From c1061763e65807325066f4be0acf3d0bb4095e08 Mon Sep 17 00:00:00 2001
From: modos189 <git@modos189.ru>
Date: Mon, 17 Jul 2023 18:04:56 +0300
Subject: [PATCH] Possibility to import IITC Button data has been implemented

---
 package-lock.json                |  62 +++++++++++++---
 package.json                     |   2 +
 public/_locales/en/messages.json |  21 +++++-
 src/settings/App.vue             |   2 +-
 src/settings/backup/Main.vue     | 117 +++++++++++++++++++------------
 src/settings/backup/export.js    |  73 +++++++++++++++++++
 src/settings/backup/import.js    | 103 +++++++++++++++++++++++++++
 src/settings/backup/utils.js     |  92 ++++--------------------
 8 files changed, 335 insertions(+), 137 deletions(-)
 create mode 100644 src/settings/backup/export.js
 create mode 100644 src/settings/backup/import.js

diff --git a/package-lock.json b/package-lock.json
index a7bea45..0195c23 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,7 +9,9 @@
       "version": "2.0.2",
       "license": "GPLv3",
       "dependencies": {
+        "@bundled-es-modules/deepmerge": "^4.3.1",
         "@highlightjs/vue-plugin": "^1.0.2",
+        "await-timeout": "^1.1.1",
         "core-js": "^3.6.5",
         "highlight.js": "^10.7.3",
         "jszip": "^3.10.1",
@@ -1712,6 +1714,14 @@
         "node": ">=6.9.0"
       }
     },
+    "node_modules/@bundled-es-modules/deepmerge": {
+      "version": "4.3.1",
+      "resolved": "https://registry.npmjs.org/@bundled-es-modules/deepmerge/-/deepmerge-4.3.1.tgz",
+      "integrity": "sha512-Rk453EklPUPC3NRWc3VUNI/SSUjdBaFoaQvFRmNBNtMHVtOFD5AntiWg5kEE1hqcPqedYFDzxE3ZcMYPcA195w==",
+      "dependencies": {
+        "deepmerge": "^4.3.1"
+      }
+    },
     "node_modules/@hapi/address": {
       "version": "2.1.4",
       "resolved": "https://registry.npmjs.org/@hapi/address/-/address-2.1.4.tgz",
@@ -3359,6 +3369,14 @@
         "url": "https://tidelift.com/funding/github/npm/autoprefixer"
       }
     },
+    "node_modules/await-timeout": {
+      "version": "1.1.1",
+      "resolved": "https://registry.npmjs.org/await-timeout/-/await-timeout-1.1.1.tgz",
+      "integrity": "sha512-gsDXAS6XVc4Jt+7S92MPX6Noq69bdeXUPEaXd8dk3+yVr629LTDLxNt4j1ycBbrU+AStK2PhKIyNIM+xzWMVOQ==",
+      "engines": {
+        "node": ">=6"
+      }
+    },
     "node_modules/aws-sign2": {
       "version": "0.7.0",
       "resolved": "https://registry.npmjs.org/aws-sign2/-/aws-sign2-0.7.0.tgz",
@@ -5547,10 +5565,9 @@
       "dev": true
     },
     "node_modules/deepmerge": {
-      "version": "1.5.2",
-      "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-1.5.2.tgz",
-      "integrity": "sha512-95k0GDqvBjZavkuvzx/YqVLv/6YYa17fz6ILMSf7neqQITCPbnfEnQvEgMPNjH4kgobe7+WIL0yJEHku+H3qtQ==",
-      "dev": true,
+      "version": "4.3.1",
+      "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-4.3.1.tgz",
+      "integrity": "sha512-3sUqbMEc77XqpdNO7FRyRog+eW3ph+GYCbj+rK+uYyRMuwsVy0rMiVtPn+QJlKFvWP/1PYpapqYn0Me2knFn+A==",
       "engines": {
         "node": ">=0.10.0"
       }
@@ -15311,6 +15328,15 @@
         "node": ">=8"
       }
     },
+    "node_modules/webpack-chain/node_modules/deepmerge": {
+      "version": "1.5.2",
+      "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-1.5.2.tgz",
+      "integrity": "sha512-95k0GDqvBjZavkuvzx/YqVLv/6YYa17fz6ILMSf7neqQITCPbnfEnQvEgMPNjH4kgobe7+WIL0yJEHku+H3qtQ==",
+      "dev": true,
+      "engines": {
+        "node": ">=0.10.0"
+      }
+    },
     "node_modules/webpack-dev-middleware": {
       "version": "3.7.3",
       "resolved": "https://registry.npmjs.org/webpack-dev-middleware/-/webpack-dev-middleware-3.7.3.tgz",
@@ -17457,6 +17483,14 @@
         "to-fast-properties": "^2.0.0"
       }
     },
+    "@bundled-es-modules/deepmerge": {
+      "version": "4.3.1",
+      "resolved": "https://registry.npmjs.org/@bundled-es-modules/deepmerge/-/deepmerge-4.3.1.tgz",
+      "integrity": "sha512-Rk453EklPUPC3NRWc3VUNI/SSUjdBaFoaQvFRmNBNtMHVtOFD5AntiWg5kEE1hqcPqedYFDzxE3ZcMYPcA195w==",
+      "requires": {
+        "deepmerge": "^4.3.1"
+      }
+    },
     "@hapi/address": {
       "version": "2.1.4",
       "resolved": "https://registry.npmjs.org/@hapi/address/-/address-2.1.4.tgz",
@@ -18832,6 +18866,11 @@
         "postcss-value-parser": "^4.1.0"
       }
     },
+    "await-timeout": {
+      "version": "1.1.1",
+      "resolved": "https://registry.npmjs.org/await-timeout/-/await-timeout-1.1.1.tgz",
+      "integrity": "sha512-gsDXAS6XVc4Jt+7S92MPX6Noq69bdeXUPEaXd8dk3+yVr629LTDLxNt4j1ycBbrU+AStK2PhKIyNIM+xzWMVOQ=="
+    },
     "aws-sign2": {
       "version": "0.7.0",
       "resolved": "https://registry.npmjs.org/aws-sign2/-/aws-sign2-0.7.0.tgz",
@@ -20580,10 +20619,9 @@
       "dev": true
     },
     "deepmerge": {
-      "version": "1.5.2",
-      "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-1.5.2.tgz",
-      "integrity": "sha512-95k0GDqvBjZavkuvzx/YqVLv/6YYa17fz6ILMSf7neqQITCPbnfEnQvEgMPNjH4kgobe7+WIL0yJEHku+H3qtQ==",
-      "dev": true
+      "version": "4.3.1",
+      "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-4.3.1.tgz",
+      "integrity": "sha512-3sUqbMEc77XqpdNO7FRyRog+eW3ph+GYCbj+rK+uYyRMuwsVy0rMiVtPn+QJlKFvWP/1PYpapqYn0Me2knFn+A=="
     },
     "default-gateway": {
       "version": "5.0.5",
@@ -28480,6 +28518,14 @@
       "requires": {
         "deepmerge": "^1.5.2",
         "javascript-stringify": "^2.0.1"
+      },
+      "dependencies": {
+        "deepmerge": {
+          "version": "1.5.2",
+          "resolved": "https://registry.npmjs.org/deepmerge/-/deepmerge-1.5.2.tgz",
+          "integrity": "sha512-95k0GDqvBjZavkuvzx/YqVLv/6YYa17fz6ILMSf7neqQITCPbnfEnQvEgMPNjH4kgobe7+WIL0yJEHku+H3qtQ==",
+          "dev": true
+        }
       }
     },
     "webpack-dev-middleware": {
diff --git a/package.json b/package.json
index d0acf40..b50ed32 100644
--- a/package.json
+++ b/package.json
@@ -10,7 +10,9 @@
     "lint": "vue-cli-service lint"
   },
   "dependencies": {
+    "@bundled-es-modules/deepmerge": "^4.3.1",
     "@highlightjs/vue-plugin": "^1.0.2",
+    "await-timeout": "^1.1.1",
     "core-js": "^3.6.5",
     "highlight.js": "^10.7.3",
     "jszip": "^3.10.1",
diff --git a/public/_locales/en/messages.json b/public/_locales/en/messages.json
index 879b772..8aba14d 100644
--- a/public/_locales/en/messages.json
+++ b/public/_locales/en/messages.json
@@ -200,12 +200,21 @@
     "export": {
         "message": "Export"
     },
+    "import_settings": {
+        "message": "Import IITC settings"
+    },
+    "import_data": {
+        "message": "Import plugins data"
+    },
+    "import_external": {
+        "message": "Import of external plugins"
+    },
+    "importFromZip": {
+        "message": "Import from zip"
+    },
     "export_settings": {
         "message": "Export IITC settings"
     },
-    "export_status": {
-        "message": "Export the enabled/disabled status of IITC plugins"
-    },
     "export_data": {
         "message": "Export plugins data"
     },
@@ -214,5 +223,11 @@
     },
     "exportToZip": {
         "message": "Export to zip"
+    },
+    "pleaseWait": {
+        "message": "Please wait"
+    },
+    "backupRestored": {
+        "message": "Backup restored"
     }
 }
diff --git a/src/settings/App.vue b/src/settings/App.vue
index 9d0acf3..0492ef7 100644
--- a/src/settings/App.vue
+++ b/src/settings/App.vue
@@ -75,7 +75,7 @@ body {
   align-items: center;
   flex-direction: column;
   background: #2b2b2b;
-  height: 100vh;
+  height: 100%;
   width: 100%;
 }
 
diff --git a/src/settings/backup/Main.vue b/src/settings/backup/Main.vue
index 35dc888..711ddce 100644
--- a/src/settings/backup/Main.vue
+++ b/src/settings/backup/Main.vue
@@ -4,7 +4,30 @@
     <div class="parent">
       <h1>{{ _("import") }}</h1>
       <div class="card">
-        TODO
+        <label class="setting-check">
+          <input type="checkbox" v-model="import_settings" />
+          <span>{{ _("import_settings") }}</span>
+        </label>
+        <label class="setting-check">
+          <input type="checkbox" v-model="import_data" />
+          <span>{{ _("import_data") }}</span>
+        </label>
+        <label class="setting-check">
+          <input type="checkbox" v-model="import_external" />
+          <span>{{ _("import_external") }}</span>
+        </label>
+        <div class="btn disabled" v-if="is_wait">{{ _("pleaseWait") }}</div>
+        <form v-on:click="$refs.input.click()" v-if="!is_wait">
+          <div class="btn">{{ _("importFromZip") }}</div>
+          <input
+            type="file"
+            ref="input"
+            accept="zip,application/octet-stream,application/zip,application/x-zip,application/x-zip-compressed"
+            id="input"
+            style="display: none"
+            v-on:change="handleImport"
+          />
+        </form>
       </div>
       <h1>{{ _("export") }}</h1>
       <div class="card">
@@ -12,10 +35,6 @@
           <input type="checkbox" v-model="export_settings" />
           <span>{{ _("export_settings") }}</span>
         </label>
-        <label class="setting-check">
-          <input type="checkbox" v-model="export_status" />
-          <span>{{ _("export_status") }}</span>
-        </label>
         <label class="setting-check">
           <input type="checkbox" v-model="export_data" />
           <span>{{ _("export_data") }}</span>
@@ -24,7 +43,7 @@
           <input type="checkbox" v-model="export_external" />
           <span>{{ _("export_external") }}</span>
         </label>
-        <div class="btn" @click="goExport">{{ _("exportToZip") }}</div>
+        <div class="btn" @click="handleExport">{{ _("exportToZip") }}</div>
       </div>
     </div>
   </div>
@@ -33,66 +52,70 @@
 <script>
 import { _ } from "@/i18n";
 import {
+  createBackupZip,
   filterExternalPlugins,
   filterStorageIitcSettings,
-  filterStoragePluginsSettings,
-  filterStoragePluginsStatus,
-  formatDate
-} from "./utils";
-
-const JSZip = require("jszip");
-
-const saveAs = (blob, fileName) => {
-  const link = document.createElement("a");
-  // create a blobURI pointing to our Blob
-  link.href = URL.createObjectURL(blob);
-  link.download = fileName;
-  // some browser needs the anchor to be in the doc
-  document.body.append(link);
-  link.click();
-  link.remove();
-  // in case the Blob uses a lot of memory
-  setTimeout(() => URL.revokeObjectURL(link.href), 7000);
-};
+  filterStoragePluginsSettings
+} from "./export";
+import {
+  getBackupDataFromZip,
+  importBackupExternalPlugins,
+  importBackupIitcSettings,
+  importBackupPluginsSettings
+} from "@/settings/backup/import";
 
 export default {
   name: "PageBackup",
   data() {
     return {
+      is_wait: false,
+      import_settings: true,
+      import_status: true,
+      import_data: true,
+      import_external: true,
       export_settings: true,
-      export_status: true,
       export_data: true,
       export_external: true
     };
   },
   methods: {
     _: _,
-    async goExport() {
-      const zip = new JSZip();
-      const backup = await this.getBackupData();
+    async handleImport(e) {
+      const target = e.target;
+      const files = target.files;
+      if (files.length === 0) return;
 
-      zip.file("iitc-button.json", JSON.stringify(backup.data));
-
-      for (const channel in backup.external_plugins) {
-        for (const external_plugin_name in backup.external_plugins[channel]) {
-          zip.file(
-            `${channel}/${external_plugin_name}`,
-            backup.external_plugins[channel][external_plugin_name]
-          );
-        }
-      }
+      this.is_wait = true;
+      const backup = await getBackupDataFromZip(files[0]);
+      const default_channel = await browser.storage.local
+        .get(["channel"])
+        .then(data => data.channel);
 
-      const filename = "iitc-backup_" + formatDate(new Date()) + ".zip";
-      zip.generateAsync({ type: "blob" }).then(function(content) {
-        saveAs(content, filename);
-      });
+      if (this.import_settings)
+        await importBackupIitcSettings(
+          backup.data.iitc_settings,
+          default_channel
+        );
+      if (this.import_data)
+        await importBackupPluginsSettings(backup.data.plugins_data);
+      if (this.import_external)
+        await importBackupExternalPlugins(
+          backup.external_plugins,
+          default_channel
+        );
+      const message = _("backupRestored");
+      alert(message);
+      this.is_wait = false;
+    },
+    async handleExport() {
+      const backup = await this.getBackupData();
+      await createBackupZip(backup);
     },
     async getBackupData() {
       const backup = {
         external_plugins: {},
         data: {
           iitc_settings: {},
-          plugins_status: {},
           plugins_data: {},
           app: "IITC Button"
         }
@@ -101,8 +124,6 @@ export default {
 
       if (this.export_settings)
         backup.data.iitc_settings = filterStorageIitcSettings(all_storage);
-      if (this.export_status)
-        backup.data.plugins_status = filterStoragePluginsStatus(all_storage);
       if (this.export_data)
         backup.data.plugins_data = filterStoragePluginsSettings(all_storage);
       if (this.export_external)
@@ -154,4 +175,8 @@ h1 {
 .btn:hover {
   background: #094559;
 }
+
+.btn.disabled {
+  background: #484848;
+}
 </style>
diff --git a/src/settings/backup/export.js b/src/settings/backup/export.js
new file mode 100644
index 0000000..e480b20
--- /dev/null
+++ b/src/settings/backup/export.js
@@ -0,0 +1,73 @@
+//@license magnet:?xt=urn:btih:1f739d935676111cfff4b4693e3816e664797050&dn=gpl-3.0.txt GPL-v3
+import JSZip from "jszip";
+import { backup_json_name, formatDate, saveAs } from "@/settings/backup/utils";
+
+export const createBackupZip = async backup => {
+  const zip = new JSZip();
+
+  zip.file(backup_json_name, JSON.stringify(backup.data));
+
+  for (const channel in backup.external_plugins) {
+    for (const external_plugin_name in backup.external_plugins[channel]) {
+      zip.file(
+        `${channel}/${external_plugin_name}`,
+        backup.external_plugins[channel][external_plugin_name]
+      );
+    }
+  }
+
+  const filename = "iitc-backup_" + formatDate(new Date()) + ".zip";
+  zip.generateAsync({ type: "blob" }).then(function(content) {
+    saveAs(content, filename);
+  });
+};
+
+export const filterStorageIitcSettings = all_storage => {
+  const iitc_settings = {};
+  const storage_keys = [
+    "channel",
+    "network_host",
+    "release_update_check_interval",
+    "beta_update_check_interval",
+    "custom_update_check_interval"
+  ];
+
+  for (const key in all_storage) {
+    if (storage_keys.includes(key)) {
+      iitc_settings[key] = all_storage[key];
+    }
+  }
+  return iitc_settings;
+};
+
+export const filterStoragePluginsSettings = all_storage => {
+  const plugins_storage = {};
+  for (const key in all_storage) {
+    if (key.startsWith("VMin")) {
+      plugins_storage[key] = all_storage[key];
+    }
+  }
+  return plugins_storage;
+};
+
+export const filterExternalPlugins = all_storage => {
+  const external_plugins = {};
+  const storage_keys = [
+    "release_plugins_user",
+    "beta_plugins_user",
+    "custom_plugins_user"
+  ];
+
+  for (const key in all_storage) {
+    if (storage_keys.includes(key)) {
+      const channel = key.split("_")[0];
+      external_plugins[channel] = {};
+      for (const plugin_uid in all_storage[key]) {
+        const plugin_filename = all_storage[key][plugin_uid]["filename"];
+        external_plugins[channel][plugin_filename] =
+          all_storage[key][plugin_uid]["code"];
+      }
+    }
+  }
+  return external_plugins;
+};
diff --git a/src/settings/backup/import.js b/src/settings/backup/import.js
new file mode 100644
index 0000000..915887f
--- /dev/null
+++ b/src/settings/backup/import.js
@@ -0,0 +1,103 @@
+//@license magnet:?xt=urn:btih:1f739d935676111cfff4b4693e3816e664797050&dn=gpl-3.0.txt GPL-v3
+import JSZip from "jszip";
+import Timeout from "await-timeout";
+import deepmerge from "@bundled-es-modules/deepmerge";
+import { backup_json_name } from "@/settings/backup/utils";
+import { parseMeta } from "lib-iitc-manager";
+
+export const getBackupDataFromZip = async file => {
+  const backup = {
+    external_plugins: {},
+    data: {}
+  };
+  const zip = await JSZip.loadAsync(file);
+
+  for (const zipName in zip.files) {
+    const zipEntry = zip.files[zipName];
+    if (!zipEntry.dir) {
+      const filename = zipEntry.name;
+      const filename_split = filename.split("/");
+
+      if (filename === backup_json_name) {
+        // import iitc-button.json
+        backup.data = JSON.parse(await zip.file(filename).async("string"));
+      } else if (
+        filename_split.length > 1 &&
+        ["release", "beta", "custom"].includes(filename_split[0])
+      ) {
+        // import external plugins
+        const channel = filename_split[0];
+        const plugin_filename = filename_split[1];
+        if (!Object.hasOwn(backup.external_plugins, channel)) {
+          backup.external_plugins[channel] = {};
+        }
+        backup.external_plugins[channel][plugin_filename] = await zip
+          .file(filename)
+          .async("string");
+      }
+    }
+  }
+  return backup;
+};
+
+export const importBackupIitcSettings = async (backup, default_channel) => {
+  await browser.storage.local.set(backup);
+
+  const set_channel = backup.channel;
+  if (set_channel !== default_channel) {
+    await browser.runtime.sendMessage({
+      type: "setChannel",
+      value: set_channel
+    });
+  }
+  await Timeout.set(1000);
+};
+
+export const importBackupPluginsSettings = async backup => {
+  const all_storage = await browser.storage.local.get(null);
+
+  const vMinRecords = {};
+  Object.keys(all_storage).forEach(key => {
+    if (key.startsWith("VMin")) {
+      vMinRecords[key] = all_storage[key];
+    }
+  });
+
+  const new_storage = deepmerge(vMinRecords, backup);
+  await browser.storage.local.set(new_storage);
+};
+
+export const importBackupExternalPlugins = async (backup, default_channel) => {
+  const scripts = [];
+  let current_channel = null;
+
+  for (const channel of Object.keys(backup)) {
+    await browser.runtime.sendMessage({
+      type: "setChannel",
+      value: channel
+    });
+    current_channel = channel;
+
+    await Timeout.set(1000);
+
+    for (const [filename, code] of Object.entries(backup[channel])) {
+      const meta = parseMeta(code);
+      meta["filename"] = filename;
+      scripts.push({ meta: meta, code: code });
+    }
+
+    await browser.runtime.sendMessage({
+      type: "addUserScripts",
+      scripts: scripts
+    });
+
+    await Timeout.set(100);
+  }
+
+  if (current_channel !== default_channel) {
+    await browser.runtime.sendMessage({
+      type: "setChannel",
+      value: default_channel
+    });
+  }
+};
diff --git a/src/settings/backup/utils.js b/src/settings/backup/utils.js
index 8f32440..288b217 100644
--- a/src/settings/backup/utils.js
+++ b/src/settings/backup/utils.js
@@ -1,4 +1,6 @@
 //@license magnet:?xt=urn:btih:1f739d935676111cfff4b4693e3816e664797050&dn=gpl-3.0.txt GPL-v3
+export const backup_json_name = "iitc-button.json";
+
 export function formatDate(date) {
   const year = date.getFullYear();
   const month = padZero(date.getMonth() + 1);
@@ -19,83 +21,15 @@ function padZero(number) {
   return number;
 }
 
-function getSubstringBeforeUnderscore(str) {
-  const underscoreIndex = str.indexOf("_");
-  if (underscoreIndex !== -1) {
-    return str.substr(0, underscoreIndex);
-  }
-  return str;
-}
-
-export const filterStorageIitcSettings = all_storage => {
-  const iitc_settings = {};
-  const storage_keys = [
-    "channel",
-    "network_host",
-    "release_update_check_interval",
-    "beta_update_check_interval",
-    "custom_update_check_interval"
-  ];
-
-  for (const key in all_storage) {
-    if (storage_keys.includes(key)) {
-      iitc_settings[key] = all_storage[key];
-    }
-  }
-  return iitc_settings;
-};
-
-export const filterStoragePluginsStatus = all_storage => {
-  const plugins_status = {};
-  const storage_keys = [
-    "release_plugins_local",
-    "beta_plugins_local",
-    "custom_plugins_local",
-    "release_plugins_user",
-    "beta_plugins_user",
-    "custom_plugins_user"
-  ];
-
-  for (const key in all_storage) {
-    if (storage_keys.includes(key)) {
-      plugins_status[key] = {};
-      for (const plugin_uid in all_storage[key]) {
-        plugins_status[key][plugin_uid] =
-          all_storage[key][plugin_uid]["status"];
-      }
-    }
-  }
-  return plugins_status;
-};
-
-export const filterStoragePluginsSettings = all_storage => {
-  const plugins_storage = {};
-  for (const key in all_storage) {
-    if (key.startsWith("VMin")) {
-      plugins_storage[key] = all_storage[key];
-    }
-  }
-  return plugins_storage;
-};
-
-export const filterExternalPlugins = all_storage => {
-  const external_plugins = {};
-  const storage_keys = [
-    "release_plugins_user",
-    "beta_plugins_user",
-    "custom_plugins_user"
-  ];
-
-  for (const key in all_storage) {
-    if (storage_keys.includes(key)) {
-      const channel = getSubstringBeforeUnderscore(key);
-      external_plugins[channel] = {};
-      for (const plugin_uid in all_storage[key]) {
-        const plugin_filename = all_storage[key][plugin_uid]["filename"];
-        external_plugins[channel][plugin_filename] =
-          all_storage[key][plugin_uid]["code"];
-      }
-    }
-  }
-  return external_plugins;
+export const saveAs = (blob, fileName) => {
+  const link = document.createElement("a");
+  // create a blobURI pointing to our Blob
+  link.href = URL.createObjectURL(blob);
+  link.download = fileName;
+  // some browser needs the anchor to be in the doc
+  document.body.append(link);
+  link.click();
+  link.remove();
+  // in case the Blob uses a lot of memory
+  setTimeout(() => URL.revokeObjectURL(link.href), 7000);
 };