feat: annotation preprocess compilation

Author: WindRunnerMax

packages/force-copy/package.json

@@ -1,6 +1,6 @@
 {
   "name": "force-copy",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "author": "Czy",
   "license": "MIT",
   "sideEffects": false,

packages/force-copy/rspack.config.js

@@ -80,6 +80,17 @@ module.exports = {
         ],
         type: "css",
       },
+      {
+        test: /\.(jsx?|tsx?)$/,
+        use: [
+          {
+            loader: "./script/define",
+            options: {
+              // debug: true,
+            },
+          },
+        ],
+      },
     ],
   },
   output: {

packages/force-copy/script/define.js

@@ -0,0 +1,62 @@
+const path = require("path");
+const fs = require("fs");
+
+// 条件编译: `process.env.PLATFORM`的深层次嵌套
+// #IFDEF #ENDIF: `C/C++`预处理指令 平台层面扩展性
+
+const src = path.resolve("src");
+/**
+ * @type {import('@rspack/core').LoaderContext}
+ * @param {string} source
+ * @returns {string}
+ */
+function DefineLoader(source) {
+  const debug = this.query.debug;
+  const resourcePath = this.resourcePath;
+  const platform = (process.env.PLATFORM || "").toLowerCase();
+  if (!platform || !resourcePath.startsWith(src)) return source;
+  // 迭代时控制该行是否命中预处理条件
+  let terser = false;
+  let terserIndex = -1;
+  /** @type {number[]} */
+  const stack = [];
+  const lines = source.split("\n");
+  const target = lines.map((line, index) => {
+    // 去掉首尾的空白 去掉行首注释符号与空白符(可选)
+    const code = line.trim().replace(/^\/\/\s*/, "");
+    // 检查预处理指令起始 `#IFDEF`只会置`true`
+    if (/^#IFDEF/.test(code)) {
+      stack.push(index);
+      // 如果是`true`继续即可
+      if (terser) return "";
+      const match = code.replace("#IFDEF", "").trim();
+      const group = match.split("|").map(item => item.trim().toLowerCase());
+      if (group.indexOf(platform) === -1) {
+        terser = true;
+        terserIndex = index;
+      }
+      return "";
+    }
+    // 检查预处理指令结束 `#IFDEF`只会置`false`
+    if (/^#ENDIF$/.test(code)) {
+      const index = stack.pop();
+      // 额外的`#ENDIF`忽略
+      if (index === undefined) return "";
+      if (index === terserIndex) {
+        terser = false;
+        terserIndex = -1;
+      }
+      return "";
+    }
+    // 如果命中预处理条件则擦除
+    if (terser) return "";
+    return line;
+  });
+  if (debug) {
+    // rm -rf ./**/*.log
+    fs.writeFile(resourcePath + ".log", target.join("\n"), () => null);
+  }
+  return target.join("\n");
+}
+
+module.exports = DefineLoader;

packages/force-copy/src/content/runtime/implant-script.ts

@@ -1,6 +1,7 @@
 export const implantScript = () => {
   const fn = window[process.env.INJECT_FILE as unknown as number] as unknown as () => void;
-  if (process.env.PLATFORM !== "chromium" && fn) {
+  // #IFDEF GECKO
+  if (fn) {
     const script = document.createElementNS("http://www.w3.org/1999/xhtml", "script");
     script.setAttribute("type", "text/javascript");
     script.innerText = `;(${fn.toString()})();`;
@@ -10,4 +11,5 @@ export const implantScript = () => {
     // @ts-ignore
     delete window[process.env.INJECT_FILE];
   }
+  // #ENDIF
 };

packages/force-copy/src/popup/utils/badge.ts

@@ -10,7 +10,10 @@ export const cipherBadgeNumber = (checked: boolean) => {
     })
     .then(tabId => {
       if (tabId) {
-        const action = process.env.PLATFORM === "chromium" ? cross.action : cross.browserAction;
+        let action: typeof cross.action | typeof cross.browserAction = cross.action;
+        // #IFDEF GECKO
+        action = cross.browserAction;
+        // #ENDIF
         action.getBadgeText({ tabId }).then(text => {
           const badge = Number(text) || 0;
           const next = badge + (checked ? 1 : -1);

packages/force-copy/src/utils/global.ts

@@ -1,7 +1,9 @@
 let env = chrome;
 
-if (process.env.PLATFORM === "gecko" && typeof browser !== "undefined") {
+// #IFDEF GECKO
+if (typeof browser !== "undefined") {
   env = browser;
 }
+// #ENDIF
 
 export const cross = env;

packages/force-copy/src/worker/runtime/content-message.ts

@@ -14,7 +14,10 @@ export const onContentMessage = (data: CWRequestType, sender: chrome.runtime.Mes
       const { payload } = data;
       if (payload && sender.tab && sender.tab.id) {
         const tabId = sender.tab.id;
-        const action = process.env.PLATFORM === "chromium" ? cross.action : cross.browserAction;
+        let action: typeof cross.action | typeof cross.browserAction = cross.action;
+        // #IFDEF GECKO
+        action = cross.browserAction;
+        // #ENDIF
         action.setBadgeText({ text: payload.toString(), tabId });
         action.setBadgeBackgroundColor({ color: "#4e5969", tabId });
       }

packages/force-copy/src/worker/runtime/implant-script.ts

@@ -4,47 +4,48 @@ import { logger } from "@/utils/logger";
 
 export const implantScript = () => {
   /**  RUN INJECT SCRIPT IN DOCUMENT START **/
-  if (process.env.PLATFORM === "chromium") {
-    // https://bugs.chromium.org/p/chromium/issues/detail?id=634381
-    // https://stackoverflow.com/questions/75495191/chrome-extension-manifest-v3-how-to-use-window-addeventlistener
-    if (cross.scripting && cross.scripting.registerContentScripts) {
-      logger.info("Register Inject Scripts By Scripting API");
-      // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/scripting/registerContentScripts
-      cross.scripting
-        .registerContentScripts([
-          {
-            matches: [...URL_MATCH],
-            runAt: "document_start",
-            world: "MAIN",
-            allFrames: true,
-            js: [process.env.INJECT_FILE + ".js"],
-            id: process.env.INJECT_FILE,
-          },
-        ])
-        .catch(err => {
-          logger.warning("Register Inject Scripts Failed", err);
-        });
-    } else {
-      logger.info("Register Inject Scripts By Tabs API");
-      // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/tabs/onUpdated
-      cross.tabs.onUpdated.addListener((_, changeInfo, tab) => {
-        if (changeInfo.status == "loading") {
-          const tabId = tab && tab.id;
-          const tabURL = tab && tab.url;
-          if (tabURL && !URL_MATCH.some(match => new RegExp(match).test(tabURL))) {
-            return void 0;
-          }
-          if (tabId && cross.scripting) {
-            cross.scripting.executeScript({
-              target: { tabId: tabId, allFrames: true },
-              files: [process.env.INJECT_FILE + ".js"],
-              injectImmediately: true,
-            });
-          }
-        }
+  // #IFDEF CHROMIUM
+  // https://bugs.chromium.org/p/chromium/issues/detail?id=634381
+  // https://stackoverflow.com/questions/75495191/chrome-extension-manifest-v3-how-to-use-window-addeventlistener
+  if (cross.scripting && cross.scripting.registerContentScripts) {
+    logger.info("Register Inject Scripts By Scripting API");
+    // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/scripting/registerContentScripts
+    cross.scripting
+      .registerContentScripts([
+        {
+          matches: [...URL_MATCH],
+          runAt: "document_start",
+          world: "MAIN",
+          allFrames: true,
+          js: [process.env.INJECT_FILE + ".js"],
+          id: process.env.INJECT_FILE,
+        },
+      ])
+      .catch(err => {
+        logger.warning("Register Inject Scripts Failed", err);
       });
-    }
   } else {
-    logger.info("Register Inject Scripts By Inline Code");
+    logger.info("Register Inject Scripts By Tabs API");
+    // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/tabs/onUpdated
+    cross.tabs.onUpdated.addListener((_, changeInfo, tab) => {
+      if (changeInfo.status == "loading") {
+        const tabId = tab && tab.id;
+        const tabURL = tab && tab.url;
+        if (tabURL && !URL_MATCH.some(match => new RegExp(match).test(tabURL))) {
+          return void 0;
+        }
+        if (tabId && cross.scripting) {
+          cross.scripting.executeScript({
+            target: { tabId: tabId, allFrames: true },
+            files: [process.env.INJECT_FILE + ".js"],
+            injectImmediately: true,
+          });
+        }
+      }
+    });
   }
+  // #ENDIF
+  // #IFDEF GECKO
+  logger.info("Register Inject Scripts By Content Script Inline Code");
+  // #ENDIF
 };