@wulu007/stealth-hook 0.0.1 → 0.1.2

package/README.md

@@ -1,10 +1,15 @@
-
 # StealthHook
 
 A lightweight JavaScript hook library for intercepting and modifying function behavior in the browser. Designed for stealth modifications with automatic iframe detection and recursive hooking.
 
-[![npm version](https://img.shields.io/npm/v/stealth-hook)](https://www.npmjs.com/package/stealth-hook)
-[![MIT license](https://img.shields.io/npm/l/stealth-hook)](https://github.com/wulu007/StealthHook/blob/main/LICENSE)
+<div align="center">
+
+[![npm version](https://img.shields.io/npm/v/@wulu007/stealth-hook.svg?style=flat-square)](https://www.npmjs.com/package/@wulu007/stealth-hook)
+[![npm downloads](https://img.shields.io/npm/dm/@wulu007/stealth-hook.svg?style=flat-square)](https://www.npmjs.com/package/@wulu007/stealth-hook)
+[![License](https://img.shields.io/npm/l/@wulu007/stealth-hook.svg?style=flat-square)](https://github.com/wulu007/StealthHook/blob/main/LICENSE)
+[![Bundle Size](https://img.shields.io/bundlephobia/minzip/@wulu007/stealth-hook?style=flat-square)](https://bundlephobia.com/package/@wulu007/stealth-hook)
+
+</div>
 
 ## Features
 
@@ -29,7 +34,7 @@ pnpm add stealth-hook
 ### In Userscripts
 
 ```javascript
-// @require      https://unpkg.com/stealth-hook@latest/dist/index.js
+// @require      https://unpkg.com/@wulu007/stealth-hook@latest/dist/index.global.js
 // @grant        unsafeWindow
 // @run-at       document-start
 ```
@@ -60,7 +65,7 @@ hookScope(({ hookMethod }, win) => {
 // @description  Hook Example
 // @author       You
 // @match        *://*/*
-// @require      https://unpkg.com/stealth-hook@latest/dist/index.js
+// @require      https://unpkg.com/@wulu007/stealth-hook@latest/dist/index.global.js
 // @grant        unsafeWindow
 // @run-at       document-start
 // ==/UserScript==

package/dist/index.cjs

@@ -124,16 +124,18 @@ var createLogger = /* @__PURE__ */ (() => {
 
 // src/index.ts
 function hookScope(callback, rootWindow) {
+  const hookedProtoSet = /* @__PURE__ */ new WeakSet();
   const interceptorSetupMap = /* @__PURE__ */ new WeakSet();
-  const hookedWindows = /* @__PURE__ */ new WeakMap();
+  const windowIdMap = /* @__PURE__ */ new WeakMap();
+  const iframeListenerMap = /* @__PURE__ */ new WeakMap();
   let windowCounter = 0;
   const $reflectApply = Reflect.apply;
   const getWindowId = (win) => {
-    if (!hookedWindows.has(win)) {
+    if (!windowIdMap.has(win)) {
       const id = win === rootWindow ? "main" : `iframe-${++windowCounter}`;
-      hookedWindows.set(win, id);
+      windowIdMap.set(win, id);
     }
-    return hookedWindows.get(win);
+    return windowIdMap.get(win);
   };
   const setupToStringHook = (win) => {
     hook.method(win.Function.prototype, "toString", (target, args, thisArg) => {
@@ -143,54 +145,93 @@ function hookScope(callback, rootWindow) {
       return $reflectApply(target, thisArg, args);
     });
   };
-  function setupIframeInterceptor(win, applyCallback) {
-    if (interceptorSetupMap.has(win))
+  const handleIframeNode = (node, applyCallback) => {
+    const element = node;
+    const tryHook = () => {
+      try {
+        const childWin = element.contentWindow;
+        if (childWin) {
+          applyCallback(childWin);
+          setupObserver(childWin, childWin.document, applyCallback);
+        }
+      } catch {
+      }
+    };
+    tryHook();
+    const oldListener = iframeListenerMap.get(element);
+    if (oldListener)
+      element.removeEventListener("load", oldListener);
+    element.addEventListener("load", tryHook);
+    iframeListenerMap.set(element, tryHook);
+  };
+  function setupObserver(win, targetNode, applyCallback) {
+    if (interceptorSetupMap.has(targetNode))
       return;
-    interceptorSetupMap.add(win);
+    interceptorSetupMap.add(targetNode);
     const logger = createLogger(getWindowId(win));
     try {
-      const NodeProto = win.Node.prototype;
-      const hookInsertMethod = (methodName) => {
-        hook.method(NodeProto, methodName, (target, args, thisArg) => {
-          const result = $reflectApply(target, thisArg, args);
-          try {
-            const node = args[0];
-            if (!node || node.tagName !== "IFRAME")
-              return result;
-            const iframe = node;
-            const handleIframe = () => {
-              const childWin = iframe.contentWindow;
-              if (childWin) {
-                applyCallback(childWin);
-                setupIframeInterceptor(childWin, applyCallback);
+      const observer = new win.MutationObserver((mutations) => {
+        for (const mutation of mutations) {
+          if (mutation.type === "childList") {
+            mutation.addedNodes.forEach((node) => {
+              const tagName = node.tagName;
+              if (tagName === "IFRAME" || tagName === "FRAME" || tagName === "OBJECT") {
+                handleIframeNode(node, applyCallback);
+              } else if (node instanceof win.HTMLElement) {
+                if (node.childElementCount > 0) {
+                  const frames = node.querySelectorAll("iframe, frame, object");
+                  frames.forEach((frame) => handleIframeNode(frame, applyCallback));
+                }
               }
-            };
-            handleIframe();
-          } catch (e) {
-            logger.warn(`Error in ${String(methodName)} hook:`, e);
+            });
+          } else if (mutation.type === "attributes") {
+            if (mutation.target instanceof win.Element) {
+              logger.log(`\u{1F504} Detected attribute change on <${mutation.target.tagName}>`, mutation.attributeName);
+              handleIframeNode(mutation.target, applyCallback);
+            }
           }
-          return result;
-        });
-      };
-      hookInsertMethod("appendChild");
-      hookInsertMethod("insertBefore");
-      hookInsertMethod("replaceChild");
+        }
+      });
+      observer.observe(targetNode, {
+        childList: true,
+        subtree: true,
+        attributes: true,
+        attributeFilter: ["src", "srcdoc", "data"]
+      });
     } catch (e) {
-      logger.warn("\u274C Failed to setup iframe interceptor", e);
+      logger.warn("\u274C Failed to setup MutationObserver", e);
     }
   }
+  const setupShadowDomHook = (win, applyCallback) => {
+    try {
+      if (win.Element.prototype.attachShadow) {
+        hook.method(win.Element.prototype, "attachShadow", (original, args, scope) => {
+          const shadowRoot = $reflectApply(original, scope, args);
+          if (shadowRoot) {
+            setupObserver(win, shadowRoot, applyCallback);
+          }
+          return shadowRoot;
+        });
+      }
+    } catch {
+    }
+  };
   const applyToWindow = (win) => {
-    if (hookedWindows.has(win))
+    try {
+      const proto = win.Function.prototype;
+      if (hookedProtoSet.has(proto))
+        return;
+      hookedProtoSet.add(proto);
+    } catch {
       return;
+    }
     const logger = createLogger(getWindowId(win));
     setupToStringHook(win);
+    setupShadowDomHook(win, applyToWindow);
     const utils = {
       hookMethod: (obj, fnName, handler) => {
         hook.method(obj, fnName, handler);
       },
-      hookBindThisMethod(obj, fnName, handler) {
-        hook.bindThisMethod(obj, fnName, handler);
-      },
       getNativeFunction: hook.getNativeFunction,
       getNativeString: hook.getNativeString,
       getCurrentWindowId: () => getWindowId(win)
@@ -203,7 +244,7 @@ function hookScope(callback, rootWindow) {
     }
   };
   applyToWindow(rootWindow);
-  setupIframeInterceptor(rootWindow, applyToWindow);
+  setupObserver(rootWindow, rootWindow.document, applyToWindow);
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/index.d.cts

@@ -20,11 +20,14 @@ declare global {
     interface Window {
         Node: typeof Node;
         Function: typeof Function;
+        MutationObserver: typeof MutationObserver;
+        HTMLIFrameElement: typeof HTMLIFrameElement;
+        HTMLElement: typeof HTMLElement;
+        Element: typeof Element;
     }
 }
 interface HookUtils {
     hookMethod: <T extends object, K extends keyof T>(obj: T, fnName: K, handler: HookHandler<T[K] extends AnyFunction ? T[K] : AnyFunction>) => void;
-    hookBindThisMethod: <T extends object, K extends keyof T>(obj: T, fnName: K, handler: HookHandler<T[K] extends AnyFunction ? T[K] : AnyFunction>) => void;
     getCurrentWindowId: () => string | number;
     getNativeFunction: typeof hook.getNativeFunction;
     getNativeString: typeof hook.getNativeString;

package/dist/index.d.ts

@@ -20,11 +20,14 @@ declare global {
     interface Window {
         Node: typeof Node;
         Function: typeof Function;
+        MutationObserver: typeof MutationObserver;
+        HTMLIFrameElement: typeof HTMLIFrameElement;
+        HTMLElement: typeof HTMLElement;
+        Element: typeof Element;
     }
 }
 interface HookUtils {
     hookMethod: <T extends object, K extends keyof T>(obj: T, fnName: K, handler: HookHandler<T[K] extends AnyFunction ? T[K] : AnyFunction>) => void;
-    hookBindThisMethod: <T extends object, K extends keyof T>(obj: T, fnName: K, handler: HookHandler<T[K] extends AnyFunction ? T[K] : AnyFunction>) => void;
     getCurrentWindowId: () => string | number;
     getNativeFunction: typeof hook.getNativeFunction;
     getNativeString: typeof hook.getNativeString;

package/dist/index.global.js

@@ -124,16 +124,18 @@ var StealthHook = (() => {
 
   // src/index.ts
   function hookScope(callback, rootWindow) {
+    const hookedProtoSet = /* @__PURE__ */ new WeakSet();
     const interceptorSetupMap = /* @__PURE__ */ new WeakSet();
-    const hookedWindows = /* @__PURE__ */ new WeakMap();
+    const windowIdMap = /* @__PURE__ */ new WeakMap();
+    const iframeListenerMap = /* @__PURE__ */ new WeakMap();
     let windowCounter = 0;
     const $reflectApply = Reflect.apply;
     const getWindowId = (win) => {
-      if (!hookedWindows.has(win)) {
+      if (!windowIdMap.has(win)) {
         const id = win === rootWindow ? "main" : `iframe-${++windowCounter}`;
-        hookedWindows.set(win, id);
+        windowIdMap.set(win, id);
       }
-      return hookedWindows.get(win);
+      return windowIdMap.get(win);
     };
     const setupToStringHook = (win) => {
       hook.method(win.Function.prototype, "toString", (target, args, thisArg) => {
@@ -143,54 +145,93 @@ var StealthHook = (() => {
         return $reflectApply(target, thisArg, args);
       });
     };
-    function setupIframeInterceptor(win, applyCallback) {
-      if (interceptorSetupMap.has(win))
+    const handleIframeNode = (node, applyCallback) => {
+      const element = node;
+      const tryHook = () => {
+        try {
+          const childWin = element.contentWindow;
+          if (childWin) {
+            applyCallback(childWin);
+            setupObserver(childWin, childWin.document, applyCallback);
+          }
+        } catch {
+        }
+      };
+      tryHook();
+      const oldListener = iframeListenerMap.get(element);
+      if (oldListener)
+        element.removeEventListener("load", oldListener);
+      element.addEventListener("load", tryHook);
+      iframeListenerMap.set(element, tryHook);
+    };
+    function setupObserver(win, targetNode, applyCallback) {
+      if (interceptorSetupMap.has(targetNode))
         return;
-      interceptorSetupMap.add(win);
+      interceptorSetupMap.add(targetNode);
       const logger = createLogger(getWindowId(win));
       try {
-        const NodeProto = win.Node.prototype;
-        const hookInsertMethod = (methodName) => {
-          hook.method(NodeProto, methodName, (target, args, thisArg) => {
-            const result = $reflectApply(target, thisArg, args);
-            try {
-              const node = args[0];
-              if (!node || node.tagName !== "IFRAME")
-                return result;
-              const iframe = node;
-              const handleIframe = () => {
-                const childWin = iframe.contentWindow;
-                if (childWin) {
-                  applyCallback(childWin);
-                  setupIframeInterceptor(childWin, applyCallback);
+        const observer = new win.MutationObserver((mutations) => {
+          for (const mutation of mutations) {
+            if (mutation.type === "childList") {
+              mutation.addedNodes.forEach((node) => {
+                const tagName = node.tagName;
+                if (tagName === "IFRAME" || tagName === "FRAME" || tagName === "OBJECT") {
+                  handleIframeNode(node, applyCallback);
+                } else if (node instanceof win.HTMLElement) {
+                  if (node.childElementCount > 0) {
+                    const frames = node.querySelectorAll("iframe, frame, object");
+                    frames.forEach((frame) => handleIframeNode(frame, applyCallback));
+                  }
                 }
-              };
-              handleIframe();
-            } catch (e) {
-              logger.warn(`Error in ${String(methodName)} hook:`, e);
+              });
+            } else if (mutation.type === "attributes") {
+              if (mutation.target instanceof win.Element) {
+                logger.log(`\u{1F504} Detected attribute change on <${mutation.target.tagName}>`, mutation.attributeName);
+                handleIframeNode(mutation.target, applyCallback);
+              }
             }
-            return result;
-          });
-        };
-        hookInsertMethod("appendChild");
-        hookInsertMethod("insertBefore");
-        hookInsertMethod("replaceChild");
+          }
+        });
+        observer.observe(targetNode, {
+          childList: true,
+          subtree: true,
+          attributes: true,
+          attributeFilter: ["src", "srcdoc", "data"]
+        });
       } catch (e) {
-        logger.warn("\u274C Failed to setup iframe interceptor", e);
+        logger.warn("\u274C Failed to setup MutationObserver", e);
       }
     }
+    const setupShadowDomHook = (win, applyCallback) => {
+      try {
+        if (win.Element.prototype.attachShadow) {
+          hook.method(win.Element.prototype, "attachShadow", (original, args, scope) => {
+            const shadowRoot = $reflectApply(original, scope, args);
+            if (shadowRoot) {
+              setupObserver(win, shadowRoot, applyCallback);
+            }
+            return shadowRoot;
+          });
+        }
+      } catch {
+      }
+    };
     const applyToWindow = (win) => {
-      if (hookedWindows.has(win))
+      try {
+        const proto = win.Function.prototype;
+        if (hookedProtoSet.has(proto))
+          return;
+        hookedProtoSet.add(proto);
+      } catch {
         return;
+      }
       const logger = createLogger(getWindowId(win));
       setupToStringHook(win);
+      setupShadowDomHook(win, applyToWindow);
       const utils = {
         hookMethod: (obj, fnName, handler) => {
           hook.method(obj, fnName, handler);
         },
-        hookBindThisMethod(obj, fnName, handler) {
-          hook.bindThisMethod(obj, fnName, handler);
-        },
         getNativeFunction: hook.getNativeFunction,
         getNativeString: hook.getNativeString,
         getCurrentWindowId: () => getWindowId(win)
@@ -203,7 +244,7 @@ var StealthHook = (() => {
       }
     };
     applyToWindow(rootWindow);
-    setupIframeInterceptor(rootWindow, applyToWindow);
+    setupObserver(rootWindow, rootWindow.document, applyToWindow);
   }
   return __toCommonJS(index_exports);
 })();

package/dist/index.js

@@ -98,16 +98,18 @@ var createLogger = /* @__PURE__ */ (() => {
 
 // src/index.ts
 function hookScope(callback, rootWindow) {
+  const hookedProtoSet = /* @__PURE__ */ new WeakSet();
   const interceptorSetupMap = /* @__PURE__ */ new WeakSet();
-  const hookedWindows = /* @__PURE__ */ new WeakMap();
+  const windowIdMap = /* @__PURE__ */ new WeakMap();
+  const iframeListenerMap = /* @__PURE__ */ new WeakMap();
   let windowCounter = 0;
   const $reflectApply = Reflect.apply;
   const getWindowId = (win) => {
-    if (!hookedWindows.has(win)) {
+    if (!windowIdMap.has(win)) {
       const id = win === rootWindow ? "main" : `iframe-${++windowCounter}`;
-      hookedWindows.set(win, id);
+      windowIdMap.set(win, id);
     }
-    return hookedWindows.get(win);
+    return windowIdMap.get(win);
   };
   const setupToStringHook = (win) => {
     hook.method(win.Function.prototype, "toString", (target, args, thisArg) => {
@@ -117,54 +119,93 @@ function hookScope(callback, rootWindow) {
       return $reflectApply(target, thisArg, args);
     });
   };
-  function setupIframeInterceptor(win, applyCallback) {
-    if (interceptorSetupMap.has(win))
+  const handleIframeNode = (node, applyCallback) => {
+    const element = node;
+    const tryHook = () => {
+      try {
+        const childWin = element.contentWindow;
+        if (childWin) {
+          applyCallback(childWin);
+          setupObserver(childWin, childWin.document, applyCallback);
+        }
+      } catch {
+      }
+    };
+    tryHook();
+    const oldListener = iframeListenerMap.get(element);
+    if (oldListener)
+      element.removeEventListener("load", oldListener);
+    element.addEventListener("load", tryHook);
+    iframeListenerMap.set(element, tryHook);
+  };
+  function setupObserver(win, targetNode, applyCallback) {
+    if (interceptorSetupMap.has(targetNode))
       return;
-    interceptorSetupMap.add(win);
+    interceptorSetupMap.add(targetNode);
     const logger = createLogger(getWindowId(win));
     try {
-      const NodeProto = win.Node.prototype;
-      const hookInsertMethod = (methodName) => {
-        hook.method(NodeProto, methodName, (target, args, thisArg) => {
-          const result = $reflectApply(target, thisArg, args);
-          try {
-            const node = args[0];
-            if (!node || node.tagName !== "IFRAME")
-              return result;
-            const iframe = node;
-            const handleIframe = () => {
-              const childWin = iframe.contentWindow;
-              if (childWin) {
-                applyCallback(childWin);
-                setupIframeInterceptor(childWin, applyCallback);
+      const observer = new win.MutationObserver((mutations) => {
+        for (const mutation of mutations) {
+          if (mutation.type === "childList") {
+            mutation.addedNodes.forEach((node) => {
+              const tagName = node.tagName;
+              if (tagName === "IFRAME" || tagName === "FRAME" || tagName === "OBJECT") {
+                handleIframeNode(node, applyCallback);
+              } else if (node instanceof win.HTMLElement) {
+                if (node.childElementCount > 0) {
+                  const frames = node.querySelectorAll("iframe, frame, object");
+                  frames.forEach((frame) => handleIframeNode(frame, applyCallback));
+                }
               }
-            };
-            handleIframe();
-          } catch (e) {
-            logger.warn(`Error in ${String(methodName)} hook:`, e);
+            });
+          } else if (mutation.type === "attributes") {
+            if (mutation.target instanceof win.Element) {
+              logger.log(`\u{1F504} Detected attribute change on <${mutation.target.tagName}>`, mutation.attributeName);
+              handleIframeNode(mutation.target, applyCallback);
+            }
           }
-          return result;
-        });
-      };
-      hookInsertMethod("appendChild");
-      hookInsertMethod("insertBefore");
-      hookInsertMethod("replaceChild");
+        }
+      });
+      observer.observe(targetNode, {
+        childList: true,
+        subtree: true,
+        attributes: true,
+        attributeFilter: ["src", "srcdoc", "data"]
+      });
     } catch (e) {
-      logger.warn("\u274C Failed to setup iframe interceptor", e);
+      logger.warn("\u274C Failed to setup MutationObserver", e);
     }
   }
+  const setupShadowDomHook = (win, applyCallback) => {
+    try {
+      if (win.Element.prototype.attachShadow) {
+        hook.method(win.Element.prototype, "attachShadow", (original, args, scope) => {
+          const shadowRoot = $reflectApply(original, scope, args);
+          if (shadowRoot) {
+            setupObserver(win, shadowRoot, applyCallback);
+          }
+          return shadowRoot;
+        });
+      }
+    } catch {
+    }
+  };
   const applyToWindow = (win) => {
-    if (hookedWindows.has(win))
+    try {
+      const proto = win.Function.prototype;
+      if (hookedProtoSet.has(proto))
+        return;
+      hookedProtoSet.add(proto);
+    } catch {
       return;
+    }
     const logger = createLogger(getWindowId(win));
     setupToStringHook(win);
+    setupShadowDomHook(win, applyToWindow);
     const utils = {
       hookMethod: (obj, fnName, handler) => {
         hook.method(obj, fnName, handler);
       },
-      hookBindThisMethod(obj, fnName, handler) {
-        hook.bindThisMethod(obj, fnName, handler);
-      },
       getNativeFunction: hook.getNativeFunction,
       getNativeString: hook.getNativeString,
       getCurrentWindowId: () => getWindowId(win)
@@ -177,7 +218,7 @@ function hookScope(callback, rootWindow) {
     }
   };
   applyToWindow(rootWindow);
-  setupIframeInterceptor(rootWindow, applyToWindow);
+  setupObserver(rootWindow, rootWindow.document, applyToWindow);
 }
 export {
   hookScope

package/package.json

@@ -1,7 +1,8 @@
 {
   "name": "@wulu007/stealth-hook",
   "type": "module",
-  "version": "0.0.1",
+  "version": "0.1.2",
+  "packageManager": "pnpm@10.26.1",
   "description": "A stealth JavaScript hook library with automatic iframe detection and function toString spoofing.",
   "author": "wulu007",
   "license": "MIT",
@@ -33,10 +34,19 @@
     "README.md",
     "dist"
   ],
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint .",
+    "lint:fix": "eslint . --fix",
+    "prepare": "husky install"
+  },
   "devDependencies": {
     "@antfu/eslint-config": "^7.1.0",
     "@types/node": "^25.0.9",
     "eslint": "^9.39.2",
+    "husky": "^9.1.7",
     "tsup": "^8.5.1",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3"
@@ -45,9 +55,7 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "scripts": {
-    "build": "tsup",
-    "dev": "tsup --watch",
-    "typecheck": "tsc --noEmit"
+  "lint-staged": {
+    "*.{js,ts}": "eslint --fix"
   }
-}
+}