zero-com 1.14.0 → 1.14.2

package/lib/common.d.ts

@@ -22,6 +22,8 @@ export declare const ZERO_COM_CONTEXT_STORAGE = "ZERO_COM_CONTEXT_STORAGE";
 export declare const SERVER_FUNCTION_WRAPPER_NAME = "func";
 export declare const HANDLE_NAME = "handle";
 export declare const CALL_NAME = "call";
+export declare const CONTEXT_NAME = "context";
+export declare const RUN_WITH_CONTEXT_NAME = "runWithContext";
 export declare const LIBRARY_NAME = "zero-com";
 export declare const FILE_EXTENSIONS: string[];
 export declare const formatFuncIdName: (funcName: string, filePath: string, line: number) => string;
@@ -41,6 +43,8 @@ export declare const collectCallSiteReplacements: (sourceFile: SourceFile, impor
 export declare const hasHandleCall: (sourceFile: SourceFile) => boolean;
 export declare const generateRegistryRequires: (registry: ServerFuncRegistry) => string;
 export declare const collectHandleCallReplacements: (sourceFile: SourceFile) => Replacement[];
+export declare const collectContextCallReplacements: (sourceFile: SourceFile) => Replacement[];
+export declare const collectRunWithContextCallReplacements: (sourceFile: SourceFile) => Replacement[];
 export declare const collectSendCallReplacements: (sourceFile: SourceFile) => Replacement[];
 export declare const collectFuncCallReplacements: (sourceFile: SourceFile) => Replacement[];
 export declare const applyReplacementsWithMap: (source: string, replacements: Replacement[], filePath: string) => {

package/lib/common.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.applyReplacements = exports.emitToJs = exports.transformSourceFile = exports.generateClientStubs = exports.appendRegistryCode = exports.applyReplacementsWithMap = exports.collectFuncCallReplacements = exports.collectSendCallReplacements = exports.collectHandleCallReplacements = exports.generateRegistryRequires = exports.hasHandleCall = exports.collectCallSiteReplacements = exports.getImportedServerFunctions = exports.updateRegistryForFile = exports.buildRegistry = exports.mightNeedTransform = exports.createProject = exports.resolveFilePath = exports.isFromLibrary = exports.getReplacements = exports.generateCompilationId = exports.formatFuncIdName = exports.FILE_EXTENSIONS = exports.LIBRARY_NAME = exports.CALL_NAME = exports.HANDLE_NAME = exports.SERVER_FUNCTION_WRAPPER_NAME = exports.ZERO_COM_CONTEXT_STORAGE = exports.ZERO_COM_SERVER_REGISTRY = exports.ZERO_COM_CLIENT_CALL = void 0;
+exports.applyReplacements = exports.emitToJs = exports.transformSourceFile = exports.generateClientStubs = exports.appendRegistryCode = exports.applyReplacementsWithMap = exports.collectFuncCallReplacements = exports.collectSendCallReplacements = exports.collectRunWithContextCallReplacements = exports.collectContextCallReplacements = exports.collectHandleCallReplacements = exports.generateRegistryRequires = exports.hasHandleCall = exports.collectCallSiteReplacements = exports.getImportedServerFunctions = exports.updateRegistryForFile = exports.buildRegistry = exports.mightNeedTransform = exports.createProject = exports.resolveFilePath = exports.isFromLibrary = exports.getReplacements = exports.generateCompilationId = exports.formatFuncIdName = exports.FILE_EXTENSIONS = exports.LIBRARY_NAME = exports.RUN_WITH_CONTEXT_NAME = exports.CONTEXT_NAME = exports.CALL_NAME = exports.HANDLE_NAME = exports.SERVER_FUNCTION_WRAPPER_NAME = exports.ZERO_COM_CONTEXT_STORAGE = exports.ZERO_COM_SERVER_REGISTRY = exports.ZERO_COM_CLIENT_CALL = void 0;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const magic_string_1 = __importDefault(require("magic-string"));
@@ -15,6 +15,8 @@ exports.ZERO_COM_CONTEXT_STORAGE = 'ZERO_COM_CONTEXT_STORAGE';
 exports.SERVER_FUNCTION_WRAPPER_NAME = 'func';
 exports.HANDLE_NAME = 'handle';
 exports.CALL_NAME = 'call';
+exports.CONTEXT_NAME = 'context';
+exports.RUN_WITH_CONTEXT_NAME = 'runWithContext';
 exports.LIBRARY_NAME = 'zero-com';
 exports.FILE_EXTENSIONS = ['', '.ts', '.tsx', '.js', '.jsx', '.mjs'];
 const formatFuncIdName = (funcName, filePath, line) => {
@@ -229,9 +231,15 @@ const hasHandleCall = (sourceFile) => {
 };
 exports.hasHandleCall = hasHandleCall;
 const generateRegistryRequires = (registry) => {
-    return Array.from(registry.keys())
+    const requires = Array.from(registry.keys())
         .map(fp => `require(${JSON.stringify(fp)});`)
         .join('\n');
+    // Initialize ZERO_COM_CONTEXT_STORAGE inline so it lives inside the bundle
+    // and gets mangled together with all references. Without this, the runtime
+    // module may be externalized (loaded via Node.js require) and set the
+    // un-mangled name while the bundle references the mangled name.
+    const initContextStorage = `if (!globalThis.${exports.ZERO_COM_CONTEXT_STORAGE}) { globalThis.${exports.ZERO_COM_CONTEXT_STORAGE} = new (require('async_hooks').AsyncLocalStorage)(); }`;
+    return initContextStorage + '\n' + requires;
 };
 exports.generateRegistryRequires = generateRegistryRequires;
 const collectHandleCallReplacements = (sourceFile) => {
@@ -258,6 +266,45 @@ const collectHandleCallReplacements = (sourceFile) => {
     return replacements;
 };
 exports.collectHandleCallReplacements = collectHandleCallReplacements;
+const collectContextCallReplacements = (sourceFile) => {
+    const replacements = [];
+    sourceFile.forEachDescendant((node) => {
+        if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
+            return;
+        const callExpr = node;
+        if (!(0, exports.isFromLibrary)(callExpr, exports.LIBRARY_NAME) || getCalleeName(callExpr) !== exports.CONTEXT_NAME)
+            return;
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: `globalThis.${exports.ZERO_COM_CONTEXT_STORAGE}.getStore()`
+        });
+    });
+    return replacements;
+};
+exports.collectContextCallReplacements = collectContextCallReplacements;
+const collectRunWithContextCallReplacements = (sourceFile) => {
+    const replacements = [];
+    sourceFile.forEachDescendant((node) => {
+        if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
+            return;
+        const callExpr = node;
+        if (!(0, exports.isFromLibrary)(callExpr, exports.LIBRARY_NAME) || getCalleeName(callExpr) !== exports.RUN_WITH_CONTEXT_NAME)
+            return;
+        const args = callExpr.getArguments();
+        if (args.length < 2)
+            return;
+        const ctx = args[0].getText();
+        const fn = args[1].getText();
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: `globalThis.${exports.ZERO_COM_CONTEXT_STORAGE}.run(${ctx}, ${fn})`
+        });
+    });
+    return replacements;
+};
+exports.collectRunWithContextCallReplacements = collectRunWithContextCallReplacements;
 const collectSendCallReplacements = (sourceFile) => {
     const replacements = [];
     sourceFile.forEachDescendant((node) => {
@@ -357,6 +404,8 @@ const transformSourceFile = (filePath, content, registry, options = {}, project)
     if (!development) {
         replacements.push(...(0, exports.collectHandleCallReplacements)(sourceFile));
         replacements.push(...(0, exports.collectSendCallReplacements)(sourceFile));
+        replacements.push(...(0, exports.collectContextCallReplacements)(sourceFile));
+        replacements.push(...(0, exports.collectRunWithContextCallReplacements)(sourceFile));
     }
     // Handle server function files
     if (isServerFunctionFile) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zero-com",
-  "version": "1.14.0",
+  "version": "1.14.2",
   "main": "index.js",
   "browser": {
     "./lib/async-local-storage.js": "./lib/async-local-storage.browser.js"