From 6fe83d84c5343703bc94b2fa9f0a6301cf744b66 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=98=E9=9F=AC=20=E5=BC=A0?= <hongtao_zhang@outlook.com>
Date: Sat, 27 Mar 2021 17:26:53 +0800
Subject: [PATCH] make JsonStorge used with single instance

---
 Flow.Launcher.Plugin/IPublicAPI.cs | 20 ++++++++++++++--
 Flow.Launcher/PublicAPIInstance.cs | 37 ++++++++++++++++++++++++++++--
 2 files changed, 53 insertions(+), 4 deletions(-)

diff --git a/Flow.Launcher.Plugin/IPublicAPI.cs b/Flow.Launcher.Plugin/IPublicAPI.cs
index 07e2b270053..0c1cd7bfd53 100644
--- a/Flow.Launcher.Plugin/IPublicAPI.cs
+++ b/Flow.Launcher.Plugin/IPublicAPI.cs
@@ -165,10 +165,26 @@ public interface IPublicAPI
         T LoadJsonStorage<T>() where T : new();
 
         /// <summary>
-        /// Save JsonStorage for current plugin. This is the method used to save settings to json in Flow
+        /// Save JsonStorage for current plugin. This is the method used to save settings to json in Flow.Launcher
+        /// This method will save the original instance loaded with LoadJsonStorage.
         /// </summary>
         /// <typeparam name="T">Type for Serialization</typeparam>
         /// <returns></returns>
-        void SaveJsonStorage<T>(T setting) where T : new();
+        void SaveJsonStorage<T>() where T : new();
+
+        /// <summary>
+        /// Save JsonStorage for current plugin. This is the method used to save settings to json in Flow.Launcher
+        /// This method will override the original class instance loaded from LoadJsonStorage
+        /// </summary>
+        /// <typeparam name="T">Type for Serialization</typeparam>
+        /// <returns></returns>
+        void SaveJsonStorage<T>(T settings) where T : new();
+
+        /// <summary>
+        /// Backup the JsonStorage you loaded from LoadJsonStorage
+        /// </summary>
+        /// <typeparam name="T"></typeparam>
+        /// <param name="settings"></param>
+        void BackupJsonStorage<T>() where T : new();
     }
 }
diff --git a/Flow.Launcher/PublicAPIInstance.cs b/Flow.Launcher/PublicAPIInstance.cs
index 9b41c2c05ba..77902ecd9bc 100644
--- a/Flow.Launcher/PublicAPIInstance.cs
+++ b/Flow.Launcher/PublicAPIInstance.cs
@@ -131,9 +131,42 @@ public void OpenSettingDialog()
 
         public void LogException(string className, string message, Exception e, [CallerMemberName] string methodName = "") => Log.Exception(className, message, e, methodName);
 
-        public T LoadJsonStorage<T>() where T : new() => new PluginJsonStorage<T>().Load();
+        private readonly Dictionary<Type, dynamic> PluginJsonStorages = new Dictionary<Type, dynamic>();
 
-        public void SaveJsonStorage<T>(T setting) where T : new() => new PluginJsonStorage<T>(setting).Save();
+        public T LoadJsonStorage<T>() where T : new()
+        {
+            var type = typeof(T);
+            if (!PluginJsonStorages.ContainsKey(type))
+                PluginJsonStorages[type] = new PluginJsonStorage<T>();
+
+            return PluginJsonStorages[type].Load();
+        }
+
+        public void SaveJsonStorage<T>() where T : new()
+        {
+            var type = typeof(T);
+            if (!PluginJsonStorages.ContainsKey(type))
+                PluginJsonStorages[type] = new PluginJsonStorage<T>();
+
+            PluginJsonStorages[type].Save();
+        }
+
+        public void SaveJsonStorage<T>(T settings) where T : new()
+        {
+            var type = typeof(T);
+            PluginJsonStorages[type] = new PluginJsonStorage<T>(settings);
+
+            PluginJsonStorages[type].Save();
+        }
+
+        public void BackupJsonStorage<T>() where T : new()
+        {
+            var type = typeof(T);
+            if (!PluginJsonStorages.ContainsKey(type))
+                throw new InvalidOperationException("You haven't registered the JsonStorage for specific Type");
+
+            PluginJsonStorages[type].BackupOriginFile();
+        }
 
         public event FlowLauncherGlobalKeyboardEventHandler GlobalKeyboardEvent;