From 72f2669cce8d4378f1cecb10fffec7573af76a01 Mon Sep 17 00:00:00 2001
From: PiEgg <marksz@teamsz.xyz>
Date: Mon, 17 Dec 2018 23:56:11 +0800
Subject: [PATCH] Added: plugin reload state in localstorage

---
 src/main/index.js             |  2 ++
 src/renderer/pages/Plugin.vue | 26 ++++++++++++++++++++++----
 src/renderer/utils/LS.js      | 15 +++++++++++++++
 3 files changed, 39 insertions(+), 4 deletions(-)
 create mode 100644 src/renderer/utils/LS.js

diff --git a/src/main/index.js b/src/main/index.js
index b664f005..481efaee 100644
--- a/src/main/index.js
+++ b/src/main/index.js
@@ -512,6 +512,8 @@ if (process.env.XDG_CURRENT_DESKTOP && process.env.XDG_CURRENT_DESKTOP.includes(
 app.on('ready', () => {
   createWindow()
   createSettingWindow()
+  // reset localstorage state
+  settingWindow.webContents.executeJavaScript(`localStorage.setItem('plugins', '')`)
   if (process.platform === 'darwin' || process.platform === 'win32') {
     createTray()
   }
diff --git a/src/renderer/pages/Plugin.vue b/src/renderer/pages/Plugin.vue
index 81ac7559..d6752f74 100644
--- a/src/renderer/pages/Plugin.vue
+++ b/src/renderer/pages/Plugin.vue
@@ -81,6 +81,7 @@
 <script>
 import ConfigForm from '@/components/ConfigForm'
 import { debounce } from 'lodash'
+import LS from '@/utils/LS'
 export default {
   name: 'plugin',
   components: {
@@ -119,9 +120,15 @@ export default {
     }
   },
   created () {
+    console.log(this.pluginState)
     this.$electron.ipcRenderer.on('pluginList', (evt, list) => {
+      const plugins = LS.get('plugins')
       this.pluginList = list.map(item => {
-        item.reload = false
+        if (plugins && plugins[item.name] && plugins[item.name].reload) {
+          item.reload = true
+        } else {
+          item.reload = false
+        }
         return item
       })
       this.pluginNameList = list.map(item => item.name)
@@ -133,6 +140,7 @@ export default {
         if (item.name === plugin) {
           item.installing = false
           item.reload = true
+          this.handlePluginState(plugin, { reload: true })
         }
       })
     })
@@ -142,6 +150,7 @@ export default {
         if (item.name === plugin) {
           item.reload = true
           item.hasInstall = false
+          this.handlePluginState(plugin, { reload: true })
         }
       })
     })
@@ -157,7 +166,8 @@ export default {
         click () {
           _this.$db.read().set(`plugins.picgo-plugin-${plugin.name}`, true).write()
           plugin.enabled = true
-          // plugin.reload = true
+          plugin.reload = true
+          this.handlePluginState(plugin.name, { reload: true })
         }
       }, {
         label: '禁用插件',
@@ -165,7 +175,8 @@ export default {
         click () {
           _this.$db.read().set(`plugins.picgo-plugin-${plugin.name}`, false).write()
           plugin.enabled = false
-          // plugin.reload = true
+          plugin.reload = true
+          this.handlePluginState(plugin.name, { reload: true })
         }
       }, {
         label: '卸载插件',
@@ -202,6 +213,7 @@ export default {
       this.$electron.ipcRenderer.send('uninstallPlugin', val)
     },
     reloadApp () {
+      LS.set('plugins', '')
       this.$electron.remote.app.relaunch()
       this.$electron.remote.app.exit(0)
     },
@@ -247,8 +259,9 @@ export default {
         })
     },
     handleSearchResult (item) {
+      const name = item.package.name.replace(/picgo-plugin-/, '')
       return {
-        name: item.package.name.replace(/picgo-plugin-/, ''),
+        name: name,
         author: item.package.author.name,
         description: item.package.description,
         logo: `https://cdn.jsdelivr.net/npm/${item.package.name}/logo.png`,
@@ -259,6 +272,11 @@ export default {
         installing: false,
         reload: false
       }
+    },
+    handlePluginState (name, state) {
+      const plugins = LS.get('plugins')
+      plugins[name] = state
+      LS.set('plugins', plugins)
     }
   },
   beforeDestroy () {
diff --git a/src/renderer/utils/LS.js b/src/renderer/utils/LS.js
new file mode 100644
index 00000000..8fbd0c24
--- /dev/null
+++ b/src/renderer/utils/LS.js
@@ -0,0 +1,15 @@
+class LS {
+  get (name) {
+    if (localStorage.getItem(name)) {
+      return JSON.parse(localStorage.getItem(name))
+    } else {
+      return {}
+    }
+  }
+
+  set (name, value) {
+    return localStorage.setItem(name, JSON.stringify(value))
+  }
+}
+
+export default new LS()