fix(unpatch): ensure unpatch does not hold a reference to Patch object

package.json

@@ -1,6 +1,6 @@
 {
   "name": "spitroast",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "A simple JavaScript function patcher.",
   "main": "dist/cjs.js",
   "module": "dist/esm/index.js",

src/getPatchFunc.ts

@@ -14,12 +14,13 @@ export default <CallbackType extends Function>(patchType: PatchType) =>
     callback: CallbackType,
     oneTime = false
   ) => {
-    if (typeof funcParent[funcName] !== "function")
+    let origFunc = funcParent[funcName];
+
+    if (typeof origFunc !== "function")
       throw new Error(
         `${funcName} is not a function in ${funcParent.constructor.name}`
       );
 
-    let origFunc = funcParent[funcName];
     let funcPatch = patchedFunctions.get(origFunc);
 
     if (!funcPatch) {
@@ -43,7 +44,11 @@ export default <CallbackType extends Function>(patchType: PatchType) =>
             : Reflect.get(target, prop, receiver),
       });
 
-      const runHook: any = (ctxt: unknown, args: unknown[], construct: boolean) => hook(replaceProxy, origFunc, args, ctxt, construct);
+      const runHook: any = (
+        ctxt: unknown,
+        args: unknown[],
+        construct: boolean
+      ) => hook(replaceProxy, origFunc, args, ctxt, construct);
 
       patchedFunctions.set(replaceProxy, funcPatch);
 
@@ -58,7 +63,8 @@ export default <CallbackType extends Function>(patchType: PatchType) =>
     }
 
     const hookId = Symbol();
-    const unpatchThisPatch = () => unpatch(funcPatch, hookId, patchType);
+    const funcPatchRef = new WeakRef(funcPatch);
+    const unpatchThisPatch = () => unpatch(funcPatchRef, hookId, patchType);
 
     if (oneTime) funcPatch.c.push(unpatchThisPatch);
     funcPatch[patchType].set(hookId, callback);

src/index.ts

@@ -1,5 +1,4 @@
 import getPatchFunc from "./getPatchFunc";
-import { unpatchAll } from "./unpatch";
 
 type BeforeCallback = (args: any[]) => void | undefined | any[];
 type InsteadCallback = (args: any[], origFunc: Function) => any;
@@ -9,4 +8,5 @@ const before = getPatchFunc<BeforeCallback>("b");
 const instead = getPatchFunc<InsteadCallback>("i");
 const after = getPatchFunc<AfterCallback>("a");
 
-export { instead, before, after, unpatchAll };
+export { instead, before, after };
+export { resetPatches as unpatchAll } from "./shared";

src/shared.ts

@@ -20,8 +20,6 @@ export type Patch = {
   i: Map<symbol, Function>;
 };
 
-export const patcherContext = { currentContext: { SHOULD_UNPATCH: false } };
-
 export let patchedFunctions: WeakMap<Function, Patch>;
 export let resetPatches = () =>
   (patchedFunctions = new WeakMap<Function, Patch>());

src/unpatch.ts

@@ -1,7 +1,9 @@
 import { PatchType, Patch, patchTypes, patchedFunctions } from "./shared";
 
-export function unpatch(patch: Patch, hookId: symbol, type: PatchType) {
-  if (!patch[type].delete(hookId)) return false;
+export function unpatch(patchRef: WeakRef<Patch>, hookId: symbol, type: PatchType) {
+  const patch = patchRef.deref();
+
+  if (!patch || !patch[type].delete(hookId)) return false;
   const funcParent = patch.p.deref();
 
   // If there are no more hooks for every type, remove the patch
@@ -24,6 +26,4 @@ export function unpatch(patch: Patch, hookId: symbol, type: PatchType) {
   }
 
   return true;
-}
-
-export { resetPatches as unpatchAll } from "./shared";
+}