From bde53772829e3ec2e5e09df841e853d7141d69b3 Mon Sep 17 00:00:00 2001
From: muwoo <2424880409@qq.com>
Date: Wed, 18 Oct 2023 11:02:14 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E5=A4=8D=20#264=20:sparkles:?=
 =?UTF-8?q?=20=E6=94=AF=E6=8C=81=20db.postAttachment=20=E5=92=8C=20db.getA?=
 =?UTF-8?q?ttachment=20:recycle:=20=E9=87=8D=E6=9E=84=E7=89=88=E6=9C=AC?=
 =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=9C=BA=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 feature/src/views/settings/database.vue |   8 +-
 package.json                            |   4 +-
 public/preload.js                       |   4 +
 src/core/db/index.ts                    |  34 +++++++
 src/core/db/webdav.ts                   |  11 +-
 src/core/plugin-handler/index.ts        |  18 ++--
 src/main/browsers/runner.ts             |  34 ++++---
 src/main/common/api.ts                  |  28 ++++-
 src/main/common/db.ts                   |  16 +++
 src/main/common/getLatestVersion.ts     |  27 +++++
 src/main/common/versionHandler.ts       | 129 +++++++-----------------
 src/main/index.ts                       |   2 +
 12 files changed, 195 insertions(+), 120 deletions(-)
 create mode 100644 src/main/common/getLatestVersion.ts

diff --git a/feature/src/views/settings/database.vue b/feature/src/views/settings/database.vue
index 5805771..bc031db 100644
--- a/feature/src/views/settings/database.vue
+++ b/feature/src/views/settings/database.vue
@@ -22,11 +22,11 @@
     <template #renderItem="{ item }">
       <a-list-item>
         <template #actions>
-          <a v-if="!item.plugin.isdownload && !item.plugin.isloading" key="list-loadmore-edit"
+          <a v-if="!item.plugin?.isdownload && !item.plugin?.isloading" key="list-loadmore-edit"
              @click="() => downloadPlugin(item.plugin)">
             <CloudDownloadOutlined style="font-size: 18px;"/>
           </a>
-          <a v-if="item.plugin.isloading" key="list-loadmore-edit">
+          <a v-if="item.plugin?.isloading" key="list-loadmore-edit">
             <LoadingOutlined style="font-size: 18px;"/>
           </a>
           <a key="list-loadmore-edit" @click="() => showKeys(item)">
@@ -36,11 +36,11 @@
         <a-list-item-meta :description="`${item.keys.length} 份文档`">
           <template #title>
             <div>
-              <span>{{ item.plugin.pluginName }}</span>
+              <span>{{ item.plugin?.pluginName }}</span>
             </div>
           </template>
           <template #avatar>
-            <a-avatar shape="square" :src="item.plugin.logo"/>
+            <a-avatar shape="square" :src="item.plugin?.logo"/>
           </template>
         </a-list-item-meta>
       </a-list-item>
diff --git a/package.json b/package.json
index bb8a9f7..65e6d89 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "rubick",
-  "version": "4.0.4",
+  "version": "4.0.5",
   "author": "muwoo <2424880409@qq.com>",
   "private": true,
   "scripts": {
@@ -69,7 +69,7 @@
     "less-loader": "^5.0.0",
     "prettier": "^2.8.4",
     "typescript": "~4.1.5",
-    "vue-cli-plugin-electron-builder": "~2.1.1",
+    "vue-cli-plugin-electron-builder": "3.0.0-alpha.4",
     "worker-plugin": "^5.0.1"
   },
   "resolutions": {
diff --git a/public/preload.js b/public/preload.js
index 96ac19d..438ffdb 100644
--- a/public/preload.js
+++ b/public/preload.js
@@ -85,6 +85,10 @@ window.rubick = {
     remove: (doc) => ipcSendSync('dbRemove', { doc }),
     bulkDocs: (docs) => ipcSendSync('dbBulkDocs', { docs }),
     allDocs: (key) => ipcSendSync('dbAllDocs', { key }),
+    postAttachment: (docId, attachment, type) =>
+      ipcSendSync('dbPostAttachment', { docId, attachment, type }),
+    getAttachment: (docId) => ipcSendSync('dbGetAttachment', { docId }),
+    getAttachmentType: (docId) => ipcSendSync('dbGetAttachmentType', { docId }),
   },
   dbStorage: {
     setItem: (key, value) => {
diff --git a/src/core/db/index.ts b/src/core/db/index.ts
index e14ffb3..d815280 100644
--- a/src/core/db/index.ts
+++ b/src/core/db/index.ts
@@ -204,4 +204,38 @@ export default class DB {
     this.pouchDB = syncDb.pouchDB;
     await webdavClient.createReadStream(this.pouchDB);
   }
+
+  public async postAttachment(
+    name: string,
+    docId: string,
+    attachment: Buffer | Uint8Array,
+    type: string
+  ) {
+    const buffer = Buffer.from(attachment);
+    if (buffer.byteLength > this.docAttachmentMaxByteLength)
+      return this.errorInfo(
+        'exception',
+        'attachment data up to ' +
+          this.docAttachmentMaxByteLength / 1024 / 1024 +
+          'M'
+      );
+    try {
+      const result = await this.pouchDB.put({
+        _id: this.getDocId(name, docId),
+        _attachments: { 0: { data: buffer, content_type: type } },
+      });
+      result.id = this.replaceDocId(name, result.id);
+      return result;
+    } catch (e) {
+      return this.errorInfo(e.name, e.message);
+    }
+  }
+
+  async getAttachment(name: string, docId: string, len = '0') {
+    try {
+      return await this.pouchDB.getAttachment(this.getDocId(name, docId), len);
+    } catch (e) {
+      return null;
+    }
+  }
 }
diff --git a/src/core/db/webdav.ts b/src/core/db/webdav.ts
index 3d3e5f6..d9eb020 100644
--- a/src/core/db/webdav.ts
+++ b/src/core/db/webdav.ts
@@ -11,8 +11,8 @@ type WebDavOptions = {
 };
 
 type DBInstance = {
-  loadIt: (stream: unknown) => void;
-  dump: (stream: unknown) => void;
+  loadIt: (stream: unknown, options?: any) => void;
+  dump: (stream: unknown, options?: any) => void;
 };
 
 export default class WebDav {
@@ -54,7 +54,12 @@ export default class WebDav {
       }).show();
     }
     const ws = new MemoryStream();
-    dbInstance.dump(ws);
+    dbInstance.dump(ws, {
+      filter: (doc) => {
+        // attachment 文档导出有问题，
+        return !doc._attachments;
+      },
+    });
     ws.pipe(
       this.client.createWriteStream(this.cloudPath, {}, () => {
         new Notification({
diff --git a/src/core/plugin-handler/index.ts b/src/core/plugin-handler/index.ts
index 7fac504..2c387c6 100644
--- a/src/core/plugin-handler/index.ts
+++ b/src/core/plugin-handler/index.ts
@@ -157,13 +157,17 @@ class AdapterHandler {
    */
   private async execCommand(cmd: string, modules: string[]): Promise<string> {
     return new Promise((resolve: any, reject: any) => {
-      const args: string[] = [cmd]
-        .concat(
-          cmd !== 'uninstall' ? modules.map((m) => `${m}@latest`) : modules
-        )
-        .concat('--color=always')
-        .concat('--save')
-        .concat(`--registry=${this.registry}`);
+      let args: string[] = [cmd].concat(
+        cmd !== 'uninstall' && cmd !== 'link'
+          ? modules.map((m) => `${m}@latest`)
+          : modules
+      );
+      if (cmd !== 'link') {
+        args = args
+          .concat('--color=always')
+          .concat('--save')
+          .concat(`--registry=${this.registry}`);
+      }
 
       const npm = spawn('npm', args, {
         cwd: this.baseDir,
diff --git a/src/main/browsers/runner.ts b/src/main/browsers/runner.ts
index 903bc7f..bd41293 100644
--- a/src/main/browsers/runner.ts
+++ b/src/main/browsers/runner.ts
@@ -73,23 +73,31 @@ export default () => {
 
   const init = (plugin, window: BrowserWindow) => {
     if (view === null || view === undefined) {
-      if (viewInstance.getView(plugin.name) && !commonConst.dev()) {
-        view = viewInstance.getView(plugin.name).view;
-        window.setBrowserView(view);
-        view.inited = true;
-        viewReadyFn(window, plugin);
-      } else {
-        createView(plugin, window);
-        viewInstance.addView(plugin.name, view);
-      }
+      createView(plugin, window);
+      // if (viewInstance.getView(plugin.name) && !commonConst.dev()) {
+      //   view = viewInstance.getView(plugin.name).view;
+      //   window.setBrowserView(view);
+      //   view.inited = true;
+      //   viewReadyFn(window, plugin);
+      // } else {
+      //   createView(plugin, window);
+      //   viewInstance.addView(plugin.name, view);
+      // }
       // eslint-disable-next-line @typescript-eslint/no-var-requires
       require('@electron/remote/main').enable(view.webContents);
     }
   };
 
   const createView = (plugin, window: BrowserWindow) => {
-    const { tplPath, indexPath, development, name, main, pluginSetting, ext } =
-      plugin;
+    const {
+      tplPath,
+      indexPath,
+      development,
+      name,
+      main = 'index.html',
+      pluginSetting,
+      ext,
+    } = plugin;
     let pluginIndexPath = tplPath || indexPath;
     let preloadPath;
     let darkMode;
@@ -160,7 +168,7 @@ export default () => {
     window.removeBrowserView(view);
     window.setSize(800, 60);
     executeHooks('PluginOut', null);
-    window.webContents.executeJavaScript(`window.initRubick()`);
+    window.webContents?.executeJavaScript(`window.initRubick()`);
     view = undefined;
   };
 
@@ -174,7 +182,7 @@ export default () => {
           } catch(e) {} 
         }
       `;
-    view.webContents.executeJavaScript(evalJs);
+    view.webContents?.executeJavaScript(evalJs);
   };
 
   return {
diff --git a/src/main/common/api.ts b/src/main/common/api.ts
index a6971dd..792f02f 100644
--- a/src/main/common/api.ts
+++ b/src/main/common/api.ts
@@ -14,13 +14,18 @@ import { screenCapture } from '@/core';
 import plist from 'plist';
 import ks from 'node-key-sender';
 
-import { DECODE_KEY } from '@/common/constans/main';
+import {
+  DECODE_KEY,
+  PLUGIN_INSTALL_DIR as baseDir,
+} from '@/common/constans/main';
 import getCopyFiles from '@/common/utils/getCopyFiles';
+import common from '@/common/utils/commonConst';
 
 import mainInstance from '../index';
 import { runner, detach } from '../browsers';
 import DBInstance from './db';
 import getWinPosition from './getWinPosition';
+import path from 'path';
 
 const runnerInstance = runner();
 const detachInstance = detach();
@@ -77,8 +82,29 @@ class API extends DBInstance {
   }
 
   public openPlugin({ data: plugin }, window) {
+    if (plugin.platform && !plugin.platform.includes(process.platform)) {
+      return new Notification({
+        title: `插件不支持当前 ${process.platform} 系统`,
+        body: `插件仅支持 ${plugin.platform.join(',')}`,
+        icon: plugin.logo,
+      }).show();
+    }
     window.setSize(window.getSize()[0], 60);
     this.removePlugin(null, window);
+    // 模板文件
+    if (!plugin.main) {
+      plugin.tplPath = common.dev()
+        ? 'http://localhost:8083/#/'
+        : `file://${__static}/tpl/index.html`;
+    }
+    if (!plugin.indexPath) {
+      const pluginPath = path.resolve(baseDir, 'node_modules', plugin.name);
+      plugin.indexPath = `file://${path.join(
+        pluginPath,
+        './',
+        plugin.main || ''
+      )}`;
+    }
     runnerInstance.init(plugin, window);
     this.currentPlugin = plugin;
     window.webContents.executeJavaScript(
diff --git a/src/main/common/db.ts b/src/main/common/db.ts
index c5ac03a..beca348 100644
--- a/src/main/common/db.ts
+++ b/src/main/common/db.ts
@@ -54,4 +54,20 @@ export default class DBInstance {
   public dbImport({ data }) {
     return dbInstance.importDb(data.target);
   }
+
+  public dbPostAttachment({ data }) {
+    const { docId, attachment, type } = data;
+    return dbInstance.postAttachment(this.DBKEY, docId, attachment, type);
+  }
+
+  public dbGetAttachment({ data }) {
+    return dbInstance.getAttachment(this.DBKEY, data.docId);
+  }
+
+  public async dbGetAttachmentType({ data }) {
+    const res: any = await this.dbGet(data.docId);
+    if (!res || !res._attachments) return null;
+    const result = res._attachments[0];
+    return result ? result.content_type : null;
+  }
 }
diff --git a/src/main/common/getLatestVersion.ts b/src/main/common/getLatestVersion.ts
new file mode 100644
index 0000000..aa0953b
--- /dev/null
+++ b/src/main/common/getLatestVersion.ts
@@ -0,0 +1,27 @@
+// for referer policy, we can't use it in renderer
+import axios from 'axios';
+const RELEASE_URL = 'https://api.github.com/repos/rubickCenter/rubick/releases';
+
+export const getLatestVersion = async (isCheckBetaUpdate = false) => {
+  let res = '';
+  try {
+    res = await axios
+      .get(RELEASE_URL, {
+        headers: {
+          Referer: 'https://github.com',
+        },
+      })
+      .then((r) => {
+        const list = r.data;
+        if (isCheckBetaUpdate) {
+          const betaList = list.filter((item) => item.name.includes('beta'));
+          return betaList[0].name;
+        }
+        const normalList = list.filter((item) => !item.name.includes('beta'));
+        return normalList[0].name;
+      });
+  } catch (err) {
+    console.log(err);
+  }
+  return res;
+};
diff --git a/src/main/common/versionHandler.ts b/src/main/common/versionHandler.ts
index 54b6e7e..11f3f37 100644
--- a/src/main/common/versionHandler.ts
+++ b/src/main/common/versionHandler.ts
@@ -1,96 +1,45 @@
-import { dialog } from 'electron';
-import { autoUpdater } from 'electron-updater';
+import { dialog, shell } from 'electron';
 import pkg from '../../../package.json';
-import { main } from '../browsers';
+import { lt } from 'semver';
+import { getLatestVersion } from './getLatestVersion';
+const version = pkg.version;
+const downloadUrl = 'https://github.com/rubickCenter/rubick/releases/latest';
 
-class VersionHandler {
-  private lastestVersion: string;
-  private currentVersion: string;
-  private releaseNotes: string;
-  private isUpdate: boolean;
-
-  constructor() {
-    this.lastestVersion = '';
-    this.currentVersion = pkg.version;
-    this.releaseNotes = '';
-    this.isUpdate = false;
-    autoUpdater.autoDownload = false;
-    autoUpdater.autoInstallOnAppQuit = false;
+const checkVersion = async () => {
+  const res: string = await getLatestVersion();
+  if (res !== '') {
+    const latest = res;
+    const result = compareVersion2Update(version, latest);
+    if (result) {
+      dialog
+        .showMessageBox({
+          type: 'info',
+          title: 'Rubick 更新提示',
+          buttons: ['Yes', 'No'],
+          message: `发现新版本 v${latest}，是否更新？`,
+        })
+        .then((res) => {
+          if (res.response === 0) {
+            // if selected yes
+            shell.openExternal(downloadUrl);
+          }
+        });
+    }
+  } else {
+    return false;
   }
+};
 
-  checkForMacAndWindows() {
-    let sendUpdateMsg = false;
-    autoUpdater.removeAllListeners();
-    // update-available 会触发多次,限制只通知一次
-    autoUpdater.checkForUpdates();
-
-    autoUpdater.on('download-progress', ({ percent }) => {
-      console.log('下载进度', percent);
-      // if (percent < 50) {
-      // }
-      this.isUpdate = true;
-    });
-    autoUpdater.on('update-available', (info) => {
-      if (sendUpdateMsg) return;
-      const { version, releaseName = 'normal', releaseNotes } = info;
-      this.lastestVersion = version;
-
-      sendUpdateMsg = true;
-
-      autoUpdater.on('update-downloaded', () => {
-        console.log('下载完成');
-        this.isUpdate = false;
-        if (releaseName === 'major') {
-          autoUpdater.quitAndInstall(true, true);
-        }
-        const mainWindow = main().getWindow();
-        dialog
-          .showMessageBox(mainWindow, {
-            title: '版本更新',
-            message: `发现新版本${this.lastestVersion}，是否更新\n\n${releaseNotes}`,
-            type: 'info',
-            buttons: ['稍后提示', '立即更新'],
-          })
-          .then(({ response }) => {
-            console.log(response);
-            if (response === 1) {
-              this.update();
-            }
-          });
-      });
-
-      // 自动下载安装包
-      if (!this.isUpdate) {
-        autoUpdater.downloadUpdate();
-        console.log('download');
-      }
-    });
-    autoUpdater.on('update-not-available', (info) => {
-      if (sendUpdateMsg) return;
-      sendUpdateMsg = true;
-    });
-    autoUpdater.on('error', () => {
-      this.isUpdate = false;
-    });
+// if true -> update else return false
+const compareVersion2Update = (current: string, latest: string) => {
+  try {
+    if (latest.includes('beta')) {
+      return false;
+    }
+    return lt(current, latest);
+  } catch (e) {
+    return false;
   }
+};
 
-  checkUpdate(): void {
-    this.checkForMacAndWindows();
-  }
-
-  update() {
-    let sendUpdateMsg = false;
-
-    this.checkUpdate();
-
-    autoUpdater.on('update-downloaded', () => {
-      if (sendUpdateMsg) return;
-      sendUpdateMsg = true;
-      this.isUpdate = false;
-      autoUpdater.quitAndInstall(true, true);
-      // App.quit();
-    });
-  }
-}
-
-export default new VersionHandler();
+export default checkVersion;
diff --git a/src/main/index.ts b/src/main/index.ts
index 4e300c0..86dc8a4 100644
--- a/src/main/index.ts
+++ b/src/main/index.ts
@@ -22,6 +22,7 @@ import {
 import '../common/utils/localPlugin';
 
 import registerySystemPlugin from './common/registerySystemPlugin';
+import checkVersion from './common/versionHandler';
 
 class App {
   public windowCreator: { init: () => void; getWindow: () => BrowserWindow };
@@ -62,6 +63,7 @@ class App {
   }
   onReady() {
     const readyFunction = async () => {
+      checkVersion();
       await localConfig.init();
       const config = await localConfig.getConfig();
       if (!config.perf.common.guide) {