@whippsp/auth0-helper 1.0.2 → 1.0.4

package/Auth0Helper.js

@@ -1,34 +1,34 @@
 module.exports = function(userModule) {
-  let originalHandler = null;
-  console.log("Auth0Help0er module started");
+  // We immediately wrap the exports object
+  const originalExports = userModule.exports;
 
-  // Use a getter/setter on exports to "trap" the function definition
-  Object.defineProperty(userModule.exports, 'onExecutePostLogin', {
-    get: () => {
-      return async (event, api) => {
-        console.log("Auth0Help0er module override.");
+  userModule.exports = new Proxy(originalExports, {
+    get: (target, prop) => {
+      // We only care about intercepting the Auth0 handler
+      if (prop === 'onExecutePostLogin') {
+        const originalHandler = target[prop];
+        
+        if (typeof originalHandler !== 'function') return originalHandler;
 
-        // 1. Logic to check skip flag
-        if (event.transaction?.metadata?.skipremainingactions === 'true') {
-          console.log("Skipping");
-          return; 
-        }
+        // Return the WRAPPED version of your handler
+        return async (event, api) => {
+          // 1. Check skip flag
+          if (event.transaction?.metadata?.skipremainingactions === 'true') {
+            console.log("Helper: Skipping Action");
+            return;
+          }
 
-        // 2. Inject the helper
-        api.skipRemaining = () => {
-          api.transaction.setMetadata('skipremainingactions', 'true');
-        };
+          // 2. Inject helper method
+          api.skipRemaining = () => {
+            console.log("Helper: Setting skip flag");
+            api.transaction.setMetadata('skipremainingactions', 'true');
+          };
 
-        // 3. Run original logic if it exists
-        if (originalHandler) {
+          // 3. Execute your original code
           return await originalHandler(event, api);
-        }
-      };
-    },
-    set: (value) => {
-      originalHandler = value; // Capture the user's function
-    },
-    configurable: true,
-    enumerable: true
+        };
+      }
+      return target[prop];
+    }
   });
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@whippsp/auth0-helper",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "A helper to override Auth0 Actions handlers with skip logic.",
   "main": "Auth0Helper.js",
   "publishConfig": {