main.js

import { clipboard, ipcRenderer, remote } from "electron";
import { v4 as uuidv4 } from "uuid";
import {
  getWindowHeight,
  searchKeyValues,
  sysFile,
  mergePlugins,
  find,
  downloadZip,
  fileLists
} from "../../assets/common/utils";
import systemMethod from "../../assets/common/system";
import fs, { stat } from "fs";
import path from "path";
import { execSync } from "child_process";

const opConfig = remote.getGlobal("opConfig");

const state = {
  selected: null,
  options: [],
  showMain: false,
  current: ["market"],
  searchValue: "",
  devPlugins: mergePlugins(sysFile.getUserPlugins() || []),
  subPlaceHolder: "",
  userInfo: { ...opConfig.get() }.userInfo,
  webviewPath: "https://so.csdn.net/so/search?q=csdn",
  pluginLoading: true,
  pluginInfo: (() => {
    try {
      return window.pluginInfo || {};
    } catch (e) {}
  })()
};

const mutations = {
  setUserInfo(state, info) {
    state.userInfo = info;
  },
  changeWebviewPath(state, path) {
    state.webviewPath = path;
  },
  commonUpdate(state, payload) {
    Object.keys(payload).forEach(key => {
      state[key] = payload[key];
      if (key === "devPlugins") {
        sysFile.savePlugins(payload[key]);
      }
    });
  },
  setSubPlaceHolder(state, payload) {
    state.subPlaceHolder = payload;
  },
  deleteDevPlugin(state, payload) {
    state.devPlugins = state.devPlugins.filter(
      plugin => plugin.name !== payload.name
    );
    sysFile.savePlugins(state.devPlugins);
  },
  deleteProdPlugin(state, payload) {
    state.devPlugins = state.devPlugins.filter(
      plugin => plugin.id !== payload.id
    );
    sysFile.savePlugins(state.devPlugins);
    // todo 删除 static 目录下的对应插件
  },
  devPluginStatusChange(state, payload) {
    state.devPlugins.forEach(plugin => {
      if (plugin.name === payload.name) {
        plugin.status = !plugin.status;
      }
    });
    state.devPlugins = [...state.devPlugins];
    sysFile.savePlugins(state.devPlugins);
  }
};

const actions = {
  showMainUI({ commit, state }, payload) {
    ipcRenderer.send("changeWindowSize-rubick", {
      height: getWindowHeight()
    });
    setTimeout(() => {
      commit("commonUpdate", {
        showMain: true,
        selected: {
          key: "market",
          name: "插件中心"
        }
      });
    }, 50);
  },
  reloadDevPlugin({ commit }, payload) {
    const config = JSON.parse(
      fs.readFileSync(path.join(payload.sourceFile, "../plugin.json"), "utf-8")
    );
    const pluginConfig = {
      ...config,
      sourceFile: path.join(payload.sourceFile, `../${config.main}`)
    };
    const devPlugins = [...state.devPlugins];
    commit("commonUpdate", {
      devPlugins: devPlugins.map(plugin => {
        if (plugin.name === payload.name) {
          return {
            ...plugin,
            ...pluginConfig
          };
        }
        return plugin;
      })
    });
  },
  /**
   * @param {Object} payload payload.filePath为配置文件的绝对路径。payload.value为搜索栏文字值。
   */
  async onSearch({ commit }, payload) {
    if (state.selected && state.selected.key !== "plugin-container") {
      commit("commonUpdate", { searchValue: "" });
      return;
    }
    const value = payload.value;
    // 在插件界面不触发其他功能
    if (
      (state.selected && state.selected.key === "plugin-container") ||
      payload.searchType === "subWindow"
    ) {
      commit("commonUpdate", { searchValue: value });
      return;
    }
    const fileUrl =
      payload.filePath ||
      clipboard.read("public.file-url").replace("file://", "");
    commit("commonUpdate", { searchValue: value });
    // 复制文件
    if (payload.filePath || (fileUrl && value === "plugin.json")) {
      const config = JSON.parse(fs.readFileSync(fileUrl, "utf-8"));

      const pluginConfig = {
        ...config,
        sourceFile: path.join(fileUrl, `../${config.main || "index.html"}`),
        id: uuidv4(),
        type: "dev",
        icon: "image://" + path.join(fileUrl, `../${config.logo}`),
        subType: (() => {
          if (config.main) {
            return "";
          }
          return "template";
        })()
      };
      commit("commonUpdate", {
        selected: {
          key: "plugin",
          name: "plugin.json"
        },
        searchValue: "",
        options: [
          {
            name: "新建rubick开发插件",
            value: "new-plugin",
            icon: "https://static.91jkys.com/activity/img/b37ff555c748489f88f3adac15b76f18.png",
            desc: "新建rubick开发插件",
            click: router => {
              commit("commonUpdate", {
                showMain: true,
                devPlugins: [pluginConfig, ...state.devPlugins],
                selected: {
                  key: "plugin",
                  name: "新建rubick开发插件"
                },
                current: ["dev"]
              });
              ipcRenderer.send("changeWindowSize-rubick", {
                height: getWindowHeight()
              });
              router.push("/home/dev");
            }
          },
          {
            name: "复制路径",
            desc: "复制路径",
            value: "copy-path",
            icon: "https://static.91jkys.com/activity/img/ac0d4df0247345b9a84c8cd7ea3dd696.png",
            click: () => {
              clipboard.writeText(fileUrl);
              commit("commonUpdate", {
                showMain: false,
                selected: null,
                options: []
              });
              ipcRenderer.send("changeWindowSize-rubick", {
                height: getWindowHeight([])
              });
              remote.Notification("Rubick 通知", { body: "复制成功" });
            }
          }
        ]
      });
      // 调整窗口大小
      ipcRenderer.send("changeWindowSize-rubick", {
        height: getWindowHeight(state.options)
      });
      return;
    }

    let options = [];

    // check 是否是插件
    if (value) {
      state.devPlugins.forEach(plugin => {
        // dev 插件未开启
        if (plugin.type === "dev" && !plugin.status) return;
        const feature = plugin.features;
        feature.forEach(fe => {
          const cmds = searchKeyValues(fe.cmds, value);
          options = [
            ...options,
            ...cmds.map(cmd => ({
              name: cmd,
              value: "plugin",
              icon: plugin.sourceFile
                ? "image://" + path.join(plugin.sourceFile, `../${plugin.logo}`)
                : plugin.logo,
              desc: fe.explain,
              type: plugin.type,
              click: router => {
                actions.openPlugin(
                  { commit },
                  { cmd, plugin, feature: fe, router }
                );
              }
            }))
          ];
        });
      });

      let descMap = new Map();
      options = [
        ...options,
        ...fileLists
          .filter(plugin => {
            if (!descMap.get(plugin)) {
              descMap.set(plugin, true);
              let has = false;
              plugin.keyWords.some(keyWord => {
                if (
                  keyWord
                    .toLocaleUpperCase()
                    .indexOf(value.toLocaleUpperCase()) >= 0
                ) {
                  has = keyWord;
                  plugin.name = keyWord;
                  return true;
                }
                return false;
              });
              return has;
            } else {
              return false;
            }
          })
          .map(plugin => {
            plugin.click = () => {
              actions.openPlugin({ commit }, { plugin });
            };
            return plugin;
          })
      ];

      descMap = null;
    }

    commit("commonUpdate", {
      options
    });
    ipcRenderer.send("changeWindowSize-rubick", {
      height: getWindowHeight(state.options)
    });
  },
  async downloadPlugin({ commit }, payload) {
    const distUrl = await downloadZip(payload.downloadUrl, payload.name);
    const fileUrl = find(distUrl);

    // 复制文件
    const config = JSON.parse(
      fs.readFileSync(`${fileUrl}/plugin.json`, "utf-8")
    );
    const pluginConfig = {
      ...config,
      id: uuidv4(),
      sourceFile: `${fileUrl}/${config.main}`,
      type: "prod",
      icon: payload.logo,
      subType: (() => {
        if (config.main) {
          return "";
        }
        return "template";
      })()
    };
    commit("commonUpdate", {
      devPlugins: [pluginConfig, ...state.devPlugins]
    });
  },
  openPlugin({ commit }, { cmd, plugin, feature, router, payload }) {
    if (plugin.type === "app") {
      execSync(plugin.action);
      commit("commonUpdate", {
        selected: null,
        showMain: false,
        options: [],
        searchValue: ""
      });
      ipcRenderer.send("changeWindowSize-rubick", {
        height: getWindowHeight([])
      });
      return;
    }
    commit("commonUpdate", {
      selected: {
        key: "plugin-container",
        name: cmd.label ? cmd.label : cmd,
        icon: "image://" + path.join(plugin.sourceFile, `../${plugin.logo}`)
      },
      searchValue: "",
      showMain: true
    });
    ipcRenderer.send("changeWindowSize-rubick", {
      height: getWindowHeight()
    });
    if (plugin.type === "system") {
      systemMethod[plugin.tag][feature.code]();
      commit("commonUpdate", {
        selected: null,
        showMain: false,
        options: []
      });
      ipcRenderer.send("changeWindowSize-rubick", {
        height: getWindowHeight([])
      });
      router.push({
        path: "/home"
      });
      return;
    }
    commit("commonUpdate", {
      pluginInfo: {
        cmd,
        ...plugin,
        detail: feature,
        payload
      }
    });

    router.push({
      path: "/plugin",
      query: {
        ...plugin,
        _modify: Date.now(),
        detail: JSON.stringify(feature)
      }
    });
  }
};

export default {
  namespaced: true,
  state,
  mutations,
  actions
};