MaaAssistantArknights/src/MeoAsstGui/Helper/ViewStatusStorage.cs

// <copyright file="ViewStatusStorage.cs" company="MaaAssistantArknights">
// MeoAsstGui - A part of the MeoAssistantArknights project
// Copyright (C) 2021 MistEO and Contributors
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY
// </copyright>

using System;
using System.IO;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;

namespace MeoAsstGui
{
    /// <summary>
    /// 界面设置存储（读写json文件）
    /// </summary>
    public class ViewStatusStorage
    {
        private static readonly string _configFilename = Environment.CurrentDirectory + "\\gui.json";
        private static readonly string _configBakFilename = Environment.CurrentDirectory + "\\gui.json.bak";
        private static JObject _viewStatus = new JObject();

        /// <summary>
        /// Gets the value of a key with default value.
        /// </summary>
        /// <param name="key">The key.</param>
        /// <param name="default_value">The default value.</param>
        /// <returns>The value, or <paramref name="default_value"/> if <paramref name="key"/> is not found.</returns>
        public static string Get(string key, string default_value)
        {
            if (_viewStatus.ContainsKey(key))
            {
                return _viewStatus[key].ToString();
            }
            else
            {
                return default_value;
            }
        }

        /// <summary>
        /// Sets a key with a value.
        /// </summary>
        /// <param name="key">The key.</param>
        /// <param name="value">The value.</param>
        public static void Set(string key, string value)
        {
            _viewStatus[key] = value;
            Save();
        }

        /// <summary>
        /// Loads configuration.
        /// </summary>
        /// <param name="withRestore">Whether to restore with backup file.</param>
        /// <returns>Whether the operation is successful.</returns>
        public static bool Load(bool withRestore = true)
        {
            if (File.Exists(_configFilename))
            {
                try
                {
                    string jsonStr = File.ReadAllText(_configFilename);

                    if (jsonStr.Length <= 2 && File.Exists(_configBakFilename))
                    {
                        File.Copy(_configBakFilename, _configFilename, true);
                        jsonStr = File.ReadAllText(_configFilename);
                    }

                    // 文件存在但为空，会读出来一个null，感觉c#这库有bug，如果是null 就赋值一个空JObject
                    _viewStatus = (JObject)JsonConvert.DeserializeObject(jsonStr) ?? new JObject();
                }
                catch (Exception)
                {
                    if (withRestore && File.Exists(_configBakFilename))
                    {
                        File.Copy(_configBakFilename, _configFilename, true);
                        return Load(false);
                    }
                    else
                    {
                        _viewStatus = new JObject();
                        return false;
                    }
                }
            }
            else
            {
                _viewStatus = new JObject();
                return false;
            }

            return true;
        }

        /// <summary>
        /// Deletes configuration.
        /// </summary>
        /// <param name="key">The key.</param>
        /// <returns>Whether the operation is successful.</returns>
        public static bool Delete(string key)
        {
            try
            {
                _viewStatus.Remove(key);
                Save();
            }
            catch (Exception)
            {
                return false;
            }

            return true;
        }

        /// <summary>
        /// Saves configuration.
        /// </summary>
        /// <returns>Whether the operation is successful.</returns>
        public static bool Save()
        {
            try
            {
                var jsonStr = _viewStatus.ToString();
                if (jsonStr.Length > 2)
                {
                    using (StreamWriter sw = new StreamWriter(_configFilename))
                    {
                        sw.Write(jsonStr);
                    }

                    if (new FileInfo(_configFilename).Length > 2)
                    {
                        File.Copy(_configFilename, _configBakFilename, true);
                    }
                }
            }
            catch (Exception)
            {
                return false;
            }

            return true;
        }
    }
}