rubick/src/main/common/api.ts

import {
  BrowserWindow,
  ipcMain,
  dialog,
  app,
  Notification,
  nativeImage,
  clipboard,
  shell
} from "electron";
import { runner, detach } from "../browsers";
import fs from "fs";
import { LocalDb } from "@/core";
import plist from "plist";
import { DECODE_KEY } from "@/common/constans/main";

const runnerInstance = runner();
const detachInstance = detach();
const dbInstance = new LocalDb(app.getPath("userData"));

dbInstance.init();

export const API: any = {
  currentPlugin: null,
  DBKEY: "RUBICK_DB_DEFAULT",
  getCurrentWindow: (window, e) => {
    let originWindow = BrowserWindow.fromWebContents(e.sender);
    if (originWindow !== window) originWindow = detachInstance.getWindow();
    return originWindow;
  },
  __EscapeKeyDown: (event, input, window) => {
    if (input.type !== "keyDown") return;
    if (!(input.meta || input.control || input.shift || input.alt)) {
      if (input.key === "Escape") {
        API.removePlugin(null, window);
      }
      return;
    }
  },

  loadPlugin({ data: plugin }, window) {
    window.webContents.executeJavaScript(
      `window.loadPlugin(${JSON.stringify(plugin)})`
    );
    API.openPlugin({ data: plugin }, window);
  },

  openPlugin({ data: plugin }, window) {
    if (API.currentPlugin && API.currentPlugin.name === plugin.name) return;
    window.setSize(window.getSize()[0], 60);
    runnerInstance.removeView(window);
    runnerInstance.init(plugin, window);
    API.currentPlugin = plugin;
    window.webContents.executeJavaScript(
      `window.setCurrentPlugin(${JSON.stringify({
        currentPlugin: API.currentPlugin
      })})`
    );
    window.show();
    // 按 ESC 退出插件
    window.webContents.on("before-input-event", (event, input) =>
      API.__EscapeKeyDown(event, input, window)
    );
    runnerInstance
      .getView()
      .webContents.on("before-input-event", (event, input) =>
        API.__EscapeKeyDown(event, input, window)
      );
  },
  removePlugin(e, window) {
    API.currentPlugin = null;
    runnerInstance.removeView(window);
  },
  openPluginDevTools() {
    runnerInstance.getView().webContents.openDevTools({ mode: "detach" });
  },
  hideMainWindow(arg, window) {
    window.hide();
  },
  showMainWindow(arg, window) {
    window.show();
  },
  showOpenDialog({ data }, window) {
    dialog.showOpenDialogSync(window, data);
  },
  setExpendHeight({ data: height }, window: BrowserWindow, e) {
    const originWindow = API.getCurrentWindow(window, e);
    if (!originWindow) return;
    const targetHeight = height;
    originWindow.setSize(originWindow.getSize()[0], targetHeight);
  },
  setSubInput({ data }, window, e) {
    const originWindow = API.getCurrentWindow(window, e);
    if (!originWindow) return;
    originWindow.webContents.executeJavaScript(
      `window.setSubInput(${JSON.stringify({
        placeholder: data.placeholder
      })})`
    );
  },
  subInputBlur() {
    runnerInstance.getView().webContents.focus();
  },
  sendSubInputChangeEvent({ data }) {
    runnerInstance.executeHooks("SubInputChange", data);
  },
  removeSubInput(data, window, e) {
    const originWindow = API.getCurrentWindow(window, e);
    if (!originWindow) return;
    originWindow.webContents.executeJavaScript(`window.removeSubInput()`);
  },
  setSubInputValue({ data }, window, e) {
    const originWindow = API.getCurrentWindow(window, e);
    if (!originWindow) return;
    originWindow.webContents.executeJavaScript(
      `window.setSubInputValue(${JSON.stringify({
        value: data.text
      })})`
    );
  },
  getPath({ data }) {
    return app.getPath(data.name);
  },
  showNotification({ data: { body } }) {
    if (!Notification.isSupported()) return;
    "string" != typeof body && (body = String(body));
    const plugin = API.currentPlugin;
    if (!plugin) return;
    const notify = new Notification({
      title: plugin.pluginName,
      body,
      icon: plugin.logo
    });
    notify.show();
  },
  copyImage: ({ data }) => {
    const image = nativeImage.createFromDataURL(data.img);
    clipboard.writeImage(image);
  },
  copyText({ data }) {
    clipboard.writeText(String(data.text));
    return true;
  },
  copyFile: ({ data }) => {
    if (data.file && fs.existsSync(data.file)) {
      clipboard.writeBuffer(
        "NSFilenamesPboardType",
        Buffer.from(plist.build([data.file]))
      );
      return true;
    }
    return false;
  },
  dbPut({ data }) {
    return dbInstance.put(API.DBKEY, data.data);
  },
  dbGet({ data }) {
    return dbInstance.get(API.DBKEY, data.id);
  },
  dbRemove({ data }) {
    return dbInstance.remove(API.DBKEY, data.doc);
  },
  dbBulkDocs({ data }) {
    return dbInstance.bulkDocs(API.DBKEY, data.docs);
  },
  dbAllDocs({ data }) {
    return dbInstance.allDocs(API.DBKEY, data.key);
  },
  getFeatures() {
    return API.currentPlugin.features;
  },
  setFeature({ data }, window) {
    API.currentPlugin = {
      ...API.currentPlugin,
      features: (() => {
        let has = false;
        API.currentPlugin.features.some(feature => {
          has = feature.code === data.feature.code;
          return has;
        });
        if (!has) {
          return [...API.currentPlugin.features, data.feature];
        }
        return API.currentPlugin.features;
      })()
    };
    window.webContents.executeJavaScript(
      `window.updatePlugin(${JSON.stringify({
        currentPlugin: API.currentPlugin
      })})`
    );
    return true;
  },
  removeFeature({ data }, window) {
    API.currentPlugin = {
      ...API.currentPlugin,
      features: API.currentPlugin.features.filter(feature => {
        if (data.code.type) {
          return feature.code.type !== data.code.type;
        }
        return feature.code !== data.code;
      })
    };
    window.webContents.executeJavaScript(
      `window.updatePlugin(${JSON.stringify({
        currentPlugin: API.currentPlugin
      })})`
    );
    return true;
  },
  sendPluginSomeKeyDownEvent({ data: { modifiers, keyCode } }) {
    const code = DECODE_KEY[keyCode];
    if (!code || !runnerInstance.getView()) return;
    if (modifiers.length > 0) {
      runnerInstance.getView().webContents.sendInputEvent({
        type: "keyDown",
        modifiers,
        keyCode: code
      });
    } else {
      runnerInstance.getView().webContents.sendInputEvent({
        type: "keyDown",
        keyCode: code
      });
    }
  },

  detachPlugin(e, window) {
    if (!API.currentPlugin) return;
    const view = window.getBrowserView();
    window.setBrowserView(null);
    window.webContents
      .executeJavaScript(`window.getMainInputInfo()`)
      .then(res => {
        detachInstance.init(
          {
            ...API.currentPlugin,
            subInput: res
          },
          window.getBounds(),
          view
        );
        window.webContents.executeJavaScript(`window.initRubick()`);
        window.setSize(window.getSize()[0], 60);
        API.currentPlugin = null;
      });
  },
  detachInputChange({ data }) {
    API.sendSubInputChangeEvent({ data });
  },

  getLocalId() {
    return encodeURIComponent(app.getPath("home"));
  },

  shellShowItemInFolder({ data }) {
    shell.showItemInFolder(data.path);
    return true;
  }
};

export default (mainWindow: BrowserWindow) => {
  // 响应 preload.js 事件
  ipcMain.on("msg-trigger", async (event, arg) => {
    const window = arg.winId ? BrowserWindow.fromId(arg.winId) : mainWindow;
    const data = await API[arg.type](arg, window, event);
    event.returnValue = data;
    // event.sender.send(`msg-back-${arg.type}`, data);
  });
};