zero-com 1.13.2 → 1.13.4

package/README.md

@@ -229,6 +229,21 @@ When called outside `handle()`, the function executes directly with no context.
 Error: context() called outside of a server function
 ```
 
+Use `runWithContext` to provide a context in those cases:
+
+```typescript
+import { runWithContext } from 'zero-com'
+
+const authorize = async (credentials) => {
+  return runWithContext({}, async () => {
+    const user = await getUserByCredentials(credentials.email, credentials.password)
+    return user ?? null
+  })
+}
+```
+
+`runWithContext` accepts any object as the context value, which will be returned by `context()` inside any server function called within the callback.
+
 When called inside `handle()`, context is propagated automatically to the function and any nested `func()` calls, same as the normal RPC path.
 
 ## File boundary rule

package/lib/async-local-storage.browser.d.ts

@@ -0,0 +1 @@
+export declare const asyncLocalStorage: undefined;

package/lib/async-local-storage.browser.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.asyncLocalStorage = void 0;
+exports.asyncLocalStorage = undefined;

package/lib/async-local-storage.d.ts

@@ -0,0 +1,2 @@
+import { AsyncLocalStorage } from 'async_hooks';
+export declare const asyncLocalStorage: AsyncLocalStorage<any> | undefined;

package/lib/async-local-storage.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.asyncLocalStorage = void 0;
+const async_hooks_1 = require("async_hooks");
+exports.asyncLocalStorage = new async_hooks_1.AsyncLocalStorage();

package/lib/runtime.d.ts

@@ -3,12 +3,9 @@ declare global {
         [funcId: string]: (...args: any[]) => any;
     };
     var ZERO_COM_CLIENT_CALL: (funcId: string, args: any[]) => any;
-    var ZERO_COM_CONTEXT_STORAGE: {
-        run: <T>(ctx: any, fn: () => T) => T;
-        getStore: () => any;
-    } | undefined;
 }
 export declare function context<T = unknown>(): T;
 export declare function func<F extends (...args: any[]) => any>(fn: F): F;
 export declare const handle: (funcId: string, ctx: any, args: any[]) => any;
+export declare const runWithContext: <T>(ctx: any, fn: () => T) => T;
 export declare const call: (fn: (funcId: string, args: any[]) => any) => void;

package/lib/runtime.js

@@ -1,40 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.call = exports.handle = void 0;
+exports.call = exports.runWithContext = exports.handle = void 0;
 exports.context = context;
 exports.func = func;
-// Context storage - only available on server (Node.js)
-// Lazily initialized to avoid importing async_hooks on client
-function getContextStorage() {
-    if (!globalThis.ZERO_COM_CONTEXT_STORAGE) {
-        try {
-            // Dynamic require to avoid bundling async_hooks for browser.
-            // Falls back to process.getBuiltinModule() for ESM contexts (e.g. Vite SSR)
-            // where require() is not available.
-            const mod = typeof require === 'function'
-                ? require('async_hooks')
-                : process.getBuiltinModule('node:async_hooks');
-            globalThis.ZERO_COM_CONTEXT_STORAGE = new mod.AsyncLocalStorage();
-        }
-        catch (_a) {
-            // Browser environment - context storage not available
-            globalThis.ZERO_COM_CONTEXT_STORAGE = undefined;
-        }
-    }
-    return globalThis.ZERO_COM_CONTEXT_STORAGE;
-}
-// Get the current context - call this inside server functions
-function context() {
-    const storage = getContextStorage();
-    if (!storage) {
-        throw new Error('context() is only available on the server');
-    }
-    const ctx = storage.getStore();
-    if (ctx === undefined) {
-        throw new Error('context() called outside of a server function');
-    }
-    return ctx;
-}
+const async_local_storage_1 = require("./async-local-storage");
 // Default server-side implementation: call directly from registry
 // This enables server functions to call other server functions without transport.
 // If a request context exists (set by handle()), it is propagated automatically.
@@ -44,20 +13,30 @@ function context() {
 if (typeof globalThis.ZERO_COM_CLIENT_CALL === 'undefined') {
     globalThis.ZERO_COM_CLIENT_CALL = (funcId, args) => {
         var _a;
-        const storage = getContextStorage();
-        if (!storage) {
+        if (!async_local_storage_1.asyncLocalStorage) {
             throw new Error('Server function called on client without transport configured. Call call() first.');
         }
         const fn = (_a = globalThis.ZERO_COM_SERVER_REGISTRY) === null || _a === void 0 ? void 0 : _a[funcId];
         if (!fn)
             throw new Error(`Function not found: ${funcId}`);
-        const ctx = storage.getStore();
+        const ctx = async_local_storage_1.asyncLocalStorage.getStore();
         if (ctx !== undefined) {
-            return storage.run(ctx, () => fn(...args));
+            return async_local_storage_1.asyncLocalStorage.run(ctx, () => fn(...args));
         }
         return fn(...args);
     };
 }
+// Get the current context - call this inside server functions
+function context() {
+    if (!async_local_storage_1.asyncLocalStorage) {
+        throw new Error('context() is only available on the server');
+    }
+    const ctx = async_local_storage_1.asyncLocalStorage.getStore();
+    if (ctx === undefined) {
+        throw new Error('context() called outside of a server function');
+    }
+    return ctx;
+}
 // func() just returns the function as-is
 // In production mode: transformed by plugin to just the inner function
 function func(fn) {
@@ -71,13 +50,20 @@ const handle = (funcId, ctx, args) => {
     if (!fn) {
         throw new Error(`Function not found in registry: ${funcId}`);
     }
-    const storage = getContextStorage();
-    if (!storage) {
+    if (!async_local_storage_1.asyncLocalStorage) {
         throw new Error('handle() is only available on the server');
     }
-    return storage.run(ctx, () => fn(...args));
+    return async_local_storage_1.asyncLocalStorage.run(ctx, () => fn(...args));
 };
 exports.handle = handle;
+// Run a callback within a context, making context() available inside it.
+// Use this in server-only code that does not go through handle() (e.g. auth callbacks).
+const runWithContext = (ctx, fn) => {
+    if (!async_local_storage_1.asyncLocalStorage)
+        throw new Error('runWithContext() is only available on the server');
+    return async_local_storage_1.asyncLocalStorage.run(ctx, fn);
+};
+exports.runWithContext = runWithContext;
 // Client calls this to set up transport (overrides default server-side behavior)
 // In production mode: transformed by plugin to assignment
 const call = (fn) => {

package/package.json

@@ -1,7 +1,10 @@
 {
   "name": "zero-com",
-  "version": "1.13.2",
+  "version": "1.13.4",
   "main": "index.js",
+  "browser": {
+    "./lib/async-local-storage.js": "./lib/async-local-storage.browser.js"
+  },
   "repository": "https://github.com/yosbelms/zero-com",
   "keywords": [
     "webpack",