Dev

packages/vite-plugin-monkey/src/client/window.ts

@@ -1,4 +1,5 @@
 import type { MonkeyWindow } from './types/_context';
+import type { GmContextType } from '../client/index';
 
 const key =
   `__monkeyWindow-` +
@@ -10,5 +11,17 @@ const key =
     }
   })();
 
+const api_key =
+  `__monkeyApi-` +
+  (() => {
+    try {
+      return new URL(import.meta.url).origin;
+    } catch {
+      return location.origin;
+    }
+  })();
 // @ts-ignore
 export const monkeyWindow: MonkeyWindow = document[key] ?? window;
+
+// @ts-ignore
+export const monkeyApi: GmContextType = document[api_key] ?? Object.freeze({});

packages/vite-plugin-monkey/src/node/inject_template.ts

@@ -1,3 +1,5 @@
+import type { GmContextType } from '../client/index';
+
 export const fn2string = <T extends (...args: any[]) => any>(
   fn: T,
   ...args: Parameters<T>
@@ -29,6 +31,80 @@ export const fcToHtml = <T extends (...args: any[]) => any>(
   );
 };
 
+/**
+ * 根据meta信息生成处理@grant 注入到window
+ * 可修复GM_xxx is undefined
+ * @package entrySrc 脚本入口地址
+ * @param metaData // ==UserScript== 信息
+ */
+export const serverInjectGMApiFn = (entrySrc: string, metaData: string) => {
+  const api_key = `__monkeyApi-` + new URL(entrySrc).origin;
+
+  const metaDataSplit = metaData.split('\n');
+  /** 每一项都是@grant的兼容处理函数字符串 */
+  const grantCompatibilityProcessing: string[] = [];
+  /** 是否已添加GM.的处理 */
+  let isAddGMList = false;
+  for (let index = 0; index < metaDataSplit.length; index++) {
+    const metaDataItem = metaDataSplit[index];
+    const metaGrantValueMatch = metaDataItem.match(
+      /[\s]*\/\/[\s]*@grant[\s]+([\S]+)/i,
+    );
+    if (metaGrantValueMatch) {
+      const metaGrantValue =
+        metaGrantValueMatch[metaGrantValueMatch.length - 1].trim();
+      if (metaGrantValue.startsWith('GM.')) {
+        // GM.addElement
+        // GM.addStyle
+        // ...
+        if (isAddGMList) {
+          continue;
+        }
+        isAddGMList = true;
+        grantCompatibilityProcessing.push(`
+        if (window.GM == null && typeof GM === "object") {
+          Reflect.set(GM_Api, "GM", GM);
+          GM_repair_count++;
+        }`);
+      } else if (metaGrantValue.startsWith('window.')) {
+        // ↓不做处理
+        // window.close
+        // window.focus
+        // window.onurlchange
+      } else {
+        grantCompatibilityProcessing.push(`
+        if (typeof ${metaGrantValue} !== "undefined" && ${metaGrantValue} != null && window.${metaGrantValue} == null) {
+          Reflect.set(GM_Api, "${metaGrantValue}", ${metaGrantValue});
+          GM_repair_count++;
+        }`);
+      }
+    }
+  }
+
+  return `
+  ;(()=>{
+    let GM_Api = {};
+    let GM_repair_count = 0;
+    if (typeof unsafeWindow !== "undefined" && unsafeWindow == window) {
+      console.log("[vite-plugin-monkey] window == unsafeWindow repair GM api");
+${grantCompatibilityProcessing.join('\n')}
+    } else {
+      if(typeof unsafeWindow === "object" && unsafeWindow){
+        if (unsafeWindow.GM == null && typeof GM === "object") {
+          Reflect.set(GM_Api, "GM", GM);
+          GM_repair_count++;
+        }
+      }
+    }
+    Object.freeze(GM_Api);
+    document["${api_key}"] = GM_Api;
+    if(GM_repair_count > 0){
+      console.log("[vite-plugin-monkey] repair GM api count: " + GM_repair_count);
+    }
+})();
+  `;
+};
+
 export interface ScriptOptions {
   entrySrc: string;
 }
@@ -42,60 +118,124 @@ export const serverInjectFn = ({ entrySrc }: ScriptOptions) => {
   // @ts-ignore
   document[key] = window;
   console.log(`[vite-plugin-monkey] mount monkeyWindow to document`);
+  const entryScript = document.createElement('script');
+  entryScript.type = 'module';
   // @ts-ignore
-  if (typeof GM_addElement === 'function') {
+  if (window.trustedTypes) {
+    // https://github.com/lisonge/vite-plugin-monkey/issues/205
     // @ts-ignore
-    GM_addElement(document.head, 'script', {
-      type: 'module',
-      src: entrySrc,
+    const policy = window.trustedTypes.createPolicy(key, {
+      createScriptURL: (input: unknown) => input,
     });
+    const trustedScriptURL = policy.createScriptURL(entrySrc);
+    entryScript.src = trustedScriptURL;
   } else {
-    const script = document.createElement('script');
-    script.type = 'module';
-    // @ts-ignore
-    if (window.trustedTypes) {
-      // https://github.com/lisonge/vite-plugin-monkey/issues/205
-      // @ts-ignore
-      const policy = window.trustedTypes.createPolicy(key, {
-        createScriptURL: (input: unknown) => input,
-      });
-      const trustedScriptURL = policy.createScriptURL(entrySrc);
-      script.src = trustedScriptURL;
+    entryScript.src = entrySrc;
+  }
+
+  const injectFn = function () {
+    let mountPositionStr = '';
+    if (document.head) {
+      if (document.head.firstChild) {
+        document.head.insertBefore(entryScript, document.head.firstChild);
+        mountPositionStr = 'document.head first';
+      } else {
+        document.head.appendChild(entryScript);
+        mountPositionStr = 'document.head last';
+      }
     } else {
-      script.src = entrySrc;
+      if (document.documentElement) {
+        if (document.documentElement.firstChild) {
+          document.documentElement.insertBefore(
+            entryScript,
+            document.documentElement.firstChild,
+          );
+          mountPositionStr = 'document.documentElement first';
+        } else {
+          document.documentElement.appendChild(entryScript);
+          mountPositionStr = 'document.documentElement last';
+        }
+      } else {
+        // 部分情况下documentElement未加载出来
+      }
     }
-    document.head.append(script);
+    return mountPositionStr == '' ? null : mountPositionStr;
+  };
+
+  const mountPosition = injectFn();
+  if (mountPosition == null) {
+    const intervalId = setInterval(() => {
+      const __mountPosition__ = injectFn();
+      if (__mountPosition__ != null) {
+        clearInterval(intervalId);
+        console.log(
+          `[vite-plugin-monkey] interval check mount entry module to ` +
+            __mountPosition__,
+        );
+      }
+    }, 5);
+  } else {
+    console.log(`[vite-plugin-monkey] mount entry module to ` + mountPosition);
+  }
+};
+
+export const cssInjectFn = (css: string) => {
+  const style = document.createElement('style');
+  style.dataset.source = 'vite-plugin-monkey';
+  style.textContent = css;
+  if (document.head) {
+    document.head.appendChild(style);
+  } else if (document.documentElement.childNodes.length === 0) {
+    document.documentElement.appendChild(style);
+  } else {
+    document.documentElement.insertBefore(
+      style,
+      document.documentElement.childNodes[0],
+    );
   }
-  console.log(`[vite-plugin-monkey] mount entry module to document.head`);
 };
 
 export const mountGmApiFn = (meta: ImportMeta, apiNames: string[] = []) => {
   const key = `__monkeyWindow-` + new URL(meta.url).origin;
+  const api_key = `__monkeyApi-` + new URL(meta.url).origin;
   // @ts-ignore
   const monkeyWindow: Window = document[key];
+  // @ts-ignore
+  const monkeyApi: Partial<GmContextType> = document[api_key] ?? {};
   if (!monkeyWindow) {
     console.log(`[vite-plugin-monkey] not found monkeyWindow`);
     return;
   }
 
   // @ts-ignore
-  window.unsafeWindow = window;
+  window.unsafeWindow = monkeyApi?.unsafeWindow ?? window;
   console.log(`[vite-plugin-monkey] mount unsafeWindow to unsafeWindow`);
 
-  apiNames.push('GM');
-  let mountedApiSize = 0;
+  const mountedApiNameList = [];
+  // @ts-ignore
+  const unmountedApiNameList = [];
+  // extra import api
+  apiNames.push('GM', 'unsafeWindow');
   apiNames.forEach((apiName) => {
     // @ts-ignore
-    const fn = monkeyWindow[apiName];
+    const fn = monkeyApi?.[apiName] ?? monkeyWindow[apiName];
     if (fn) {
       // @ts-ignore
-      window[apiName] = monkeyWindow[apiName];
-      mountedApiSize++;
+      window[apiName] = fn;
+      mountedApiNameList.push(apiName);
+    } else {
+      unmountedApiNameList.push(apiName);
     }
   });
   console.log(
-    `[vite-plugin-monkey] mount ${mountedApiSize}/${apiNames.length} GM api to unsafeWindow`,
+    `[vite-plugin-monkey] mount ${mountedApiNameList.length}/${apiNames.length} GM_api to unsafeWindow`,
   );
+  if (unmountedApiNameList.length) {
+    console.log(
+      // @ts-ignore
+      `[vite-plugin-monkey] unmount ${unmountedApiNameList.join('、')}`,
+    );
+  }
 };
 
 export const virtualHtmlTemplate = async (url: string) => {
@@ -105,7 +245,7 @@ export const virtualHtmlTemplate = async (url: string) => {
   u.searchParams.set('origin', u.origin);
   if (window == window.parent) {
     location.href = u.href;
-    await delay(500);
+    await delay(3500);
     window.close();
     return;
   }
@@ -185,7 +325,7 @@ export const previewTemplate = async (urls: string[]) => {
   if (window == window.parent && urls.length == 1) {
     const u = new URL(urls[0], location.origin);
     location.href = u.href;
-    await delay(500);
+    await delay(3500);
     window.close();
     return;
   } else if (urls.length == 0) {

packages/vite-plugin-monkey/src/node/option.ts

@@ -212,7 +212,7 @@ export const resolvedOption = (
   } = pluginOption.userscript ?? {};
 
   const { fileName = projectPkg.name + '.user.js' } = build;
-  let { metaFileName } = build;
+  let { metaFileName, metaLocalFileName } = build;
   if (typeof metaFileName == 'string') {
     const t = metaFileName;
     metaFileName = () => t;
@@ -222,7 +222,17 @@ export const resolvedOption = (
     metaFileName = undefined;
   }
 
+  if (typeof metaLocalFileName == 'string') {
+    const tt = metaLocalFileName;
+    metaLocalFileName = () => tt;
+  } else if (metaLocalFileName === true) {
+    metaLocalFileName = () =>
+      fileName.replace(/\.user\.js$/, '.meta.local.user.js');
+  } else if (metaLocalFileName === false) {
+    metaLocalFileName = undefined;
+  }
   const metaFileFc = metaFileName;
+  const metaLocalFileFc = metaLocalFileName;
 
   const cssSideEffects =
     build.cssSideEffects ||
@@ -234,9 +244,17 @@ export const resolvedOption = (
           GM_addStyle(e);
           return;
         }
-        const o = document.createElement('style');
-        o.textContent = e;
-        document.head.append(o);
+        function addStyle(cssText: string) {
+          let $style = document.createElement('style');
+          $style.innerHTML = cssText;
+          if (document.head) {
+            document.head.appendChild($style);
+          } else {
+            document.documentElement.appendChild($style);
+          }
+          return $style;
+        }
+        addStyle(e);
       };
     });
 
@@ -299,6 +317,9 @@ export const resolvedOption = (
     build: {
       fileName,
       metaFileName: metaFileFc ? () => metaFileFc(fileName) : undefined,
+      metaLocalFileName: metaLocalFileFc
+        ? () => metaLocalFileFc(fileName)
+        : undefined,
       autoGrant: build.autoGrant ?? true,
       externalGlobals: externalGlobals,
       externalResource: externalResource2,

packages/vite-plugin-monkey/src/node/plugins/finalBundle.ts

@@ -14,6 +14,7 @@ import {
 } from '../topLevelAwait';
 import type { FinalMonkeyOption } from '../types';
 import { finalMonkeyOptionToComment } from '../userscript';
+import path from 'path';
 
 const __entry_name = `__monkey.entry.js`;
 
@@ -31,7 +32,10 @@ export const finalBundlePlugin = (finalOption: FinalMonkeyOption): Plugin => {
     async configResolved(resolvedConfig) {
       viteConfig = resolvedConfig;
     },
-    async generateBundle(_, rawBundle) {
+    async generateBundle(outputOptions, rawBundle) {
+      // 输出的目录
+      let outputPath =
+        outputOptions.dir || path.resolve(viteConfig.build.outDir);
       const entryChunks: OutputChunk[] = [];
       const chunks: OutputChunk[] = [];
       Object.values(rawBundle).forEach((chunk) => {
@@ -257,6 +261,20 @@ export const finalBundlePlugin = (finalOption: FinalMonkeyOption): Plugin => {
           ),
         });
       }
+
+      if (finalOption.build.metaLocalFileName) {
+        let filePath = path.join(outputPath!, finalOption.build.fileName);
+        this.emitFile({
+          type: 'asset',
+          fileName: finalOption.build.metaLocalFileName(),
+          source: await finalMonkeyOptionToComment(
+            finalOption,
+            collectGrantSet,
+            'meta-local',
+            filePath,
+          ),
+        });
+      }
     },
   };
 };