@whippsp/auth0-helper 1.0.3 → 1.0.4

package/Auth0Helper.js

@@ -1,24 +1,34 @@
 module.exports = function(userModule) {
-  // Use a short delay to ensure the user has defined their exports
-  setImmediate(() => {
-    const original = userModule.exports.onExecutePostLogin;
-    
-    if (typeof original === 'function') {
-      userModule.exports.onExecutePostLogin = async (event, api) => {
-        // 1. Automatic skip logic
-        if (event.transaction?.metadata?.skipremainingactions === 'true') {
-          console.log("Skipping.");
-          return;
-        }
+  // We immediately wrap the exports object
+  const originalExports = userModule.exports;
 
-        // 2. Inject your custom method directly into the api object
-        api.skipRemaining = () => {
-          api.transaction.setMetadata('skipremainingactions', 'true');
-        };
+  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;
+
+        // 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;
+          }
 
-        // 3. Call the original code
-        return await original(event, api);
-      };
+          // 2. Inject helper method
+          api.skipRemaining = () => {
+            console.log("Helper: Setting skip flag");
+            api.transaction.setMetadata('skipremainingactions', 'true');
+          };
+
+          // 3. Execute your original code
+          return await originalHandler(event, api);
+        };
+      }
+      return target[prop];
     }
   });
 };

package/package.json

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