kiss-translator/src/background.js

import browser from "webextension-polyfill";
import {
  MSG_FETCH,
  MSG_FETCH_LIMIT,
  MSG_FETCH_CLEAR,
  MSG_TRANS_TOGGLE,
  MSG_OPEN_OPTIONS,
  MSG_SAVE_RULE,
  MSG_TRANS_TOGGLE_STYLE,
  MSG_TRANSLATE_SELECTED,
  MSG_OPEN_TRANBOX,
  CMD_TOGGLE_TRANSLATE,
  CMD_TOGGLE_STYLE,
  CMD_OPEN_OPTIONS,
  CMD_OPEN_TRANBOX,
  CMD_TRANSLATE_SELECTED,
} from "./config";
import { getSettingWithDefault, tryInitDefaultData } from "./libs/storage";
import { trySyncSettingAndRules } from "./libs/sync";
import { fetchData, fetchPool } from "./libs/fetch";
import { sendTabMsg } from "./libs/msg";
import { trySyncAllSubRules } from "./libs/subRules";
import { tryClearCaches } from "./libs";
import { saveRule } from "./libs/rules";

globalThis.ContextType = "BACKGROUND";

/**
 * 插件安装
 */
browser.runtime.onInstalled.addListener(() => {
  tryInitDefaultData();

  // 右键菜单
  browser.contextMenus.create({
    id: CMD_TRANSLATE_SELECTED,
    title: browser.i18n.getMessage("translate_selected"),
    contexts: ["selection"],
  });
  browser.contextMenus.create({
    id: "selection_separator",
    type: "separator",
    contexts: ["selection"],
  });
  browser.contextMenus.create({
    id: CMD_TOGGLE_TRANSLATE,
    title: browser.i18n.getMessage("toggle_translate"),
    contexts: ["all"],
  });
  browser.contextMenus.create({
    id: CMD_TOGGLE_STYLE,
    title: browser.i18n.getMessage("toggle_style"),
    contexts: ["all"],
  });
  browser.contextMenus.create({
    id: "options_separator",
    type: "separator",
    contexts: ["all"],
  });
  browser.contextMenus.create({
    id: CMD_OPEN_TRANBOX,
    title: browser.i18n.getMessage("open_tranbox"),
    contexts: ["all"],
  });
  browser.contextMenus.create({
    id: CMD_OPEN_OPTIONS,
    title: browser.i18n.getMessage("open_options"),
    contexts: ["all"],
  });
});

/**
 * 浏览器启动
 */
browser.runtime.onStartup.addListener(async () => {
  // 同步数据
  await trySyncSettingAndRules();

  // 清除缓存
  const setting = await getSettingWithDefault();
  if (setting.clearCache) {
    tryClearCaches();
  }

  // 同步订阅规则
  trySyncAllSubRules(setting);
});

/**
 * 监听消息
 */
browser.runtime.onMessage.addListener(
  ({ action, args }, sender, sendResponse) => {
    switch (action) {
      case MSG_FETCH:
        const { input, opts } = args;
        fetchData(input, opts)
          .then((data) => {
            sendResponse({ data });
          })
          .catch((error) => {
            sendResponse({ error: error.message, cause: error.cause });
          });
        break;
      case MSG_FETCH_LIMIT:
        const { interval, limit } = args;
        fetchPool.update(interval, limit);
        sendResponse({ data: "ok" });
        break;
      case MSG_FETCH_CLEAR:
        fetchPool.clear();
        sendResponse({ data: "ok" });
        break;
      case MSG_OPEN_OPTIONS:
        browser.runtime.openOptionsPage();
        break;
      case MSG_SAVE_RULE:
        saveRule(args);
        break;
      default:
        sendResponse({ error: `message action is unavailable: ${action}` });
    }
    return true;
  }
);

/**
 * 监听快捷键
 */
browser.commands.onCommand.addListener((command) => {
  // console.log(`Command: ${command}`);
  switch (command) {
    case CMD_TOGGLE_TRANSLATE:
      sendTabMsg(MSG_TRANS_TOGGLE);
      break;
    case CMD_TOGGLE_STYLE:
      sendTabMsg(MSG_TRANS_TOGGLE_STYLE);
      break;
    case CMD_OPEN_OPTIONS:
      browser.runtime.openOptionsPage();
      break;
    default:
  }
});

browser.contextMenus.onClicked.addListener(({ menuItemId }) => {
  switch (menuItemId) {
    case CMD_TOGGLE_TRANSLATE:
      sendTabMsg(MSG_TRANS_TOGGLE);
      break;
    case CMD_TOGGLE_STYLE:
      sendTabMsg(MSG_TRANS_TOGGLE_STYLE);
      break;
    case CMD_TRANSLATE_SELECTED:
      sendTabMsg(MSG_TRANSLATE_SELECTED);
      break;
    case CMD_OPEN_TRANBOX:
      sendTabMsg(MSG_OPEN_TRANBOX);
      break;
    case CMD_OPEN_OPTIONS:
      browser.runtime.openOptionsPage();
      break;
    default:
  }
});