PowerToys/Wox.Infrastructure/Storage/BaseStorage.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.InteropServices.ComTypes;
using System.Text;
using System.Windows.Forms;
using Newtonsoft.Json;

namespace Wox.Infrastructure.Storage
{
    [Serializable]
    public abstract class BaseStorage<T> : IStorage where T : class,IStorage, new()
    {
        private string configFolder;

        private string ConfigFolder
        {
            get
            {
                if (string.IsNullOrEmpty(configFolder))
                {
                    string userProfilePath = Environment.GetEnvironmentVariable("USERPROFILE");
                    if (userProfilePath == null)
                    {
                        throw new ArgumentException("Environment variable USERPROFILE is empty");
                    }
                    configFolder = Path.Combine(Path.Combine(userProfilePath, ".Wox"), "Config");
                }
                return configFolder;
            }
        }

        protected string ConfigPath
        {
            get
            {
                return Path.Combine(ConfigFolder, ConfigName + FileSuffix);
            }
        }

        protected abstract string FileSuffix { get; }

        protected abstract string ConfigName { get; }

        private static object locker = new object();
        protected static T serializedObject;

        public event Action<T> AfterLoad;

        protected virtual void OnAfterLoad(T obj)
        {
            Action<T> handler = AfterLoad;
            if (handler != null) handler(obj);
        }

        public static T Instance
        {
            get
            {
                if (serializedObject == null)
                {
                    lock (locker)
                    {
                        if (serializedObject == null)
                        {
                            serializedObject = new T();
                            serializedObject.Load();
                        }
                    }
                }
                return serializedObject;
            }
        }

        /// <summary>
        /// if loading storage failed, we will try to load default
        /// </summary>
        /// <returns></returns>
        protected virtual T LoadDefault()
        {
            return serializedObject;
        }

        protected abstract void LoadInternal();
        protected abstract void SaveInternal();

        public void Load()
        {
            if (!File.Exists(ConfigPath))
            {
                if (!Directory.Exists(ConfigFolder))
                {
                    Directory.CreateDirectory(ConfigFolder);
                }
                File.Create(ConfigPath).Close();
            }
            LoadInternal();
            OnAfterLoad(serializedObject);
        }

        public void Save()
        {
            lock (locker)
            {
                SaveInternal();
            }
        }
    }
}