zero-com 1.6.4 → 1.6.5

package/lib/common.d.ts

@@ -1,4 +1,10 @@
+import type { SourceMap } from 'magic-string';
 import { CallExpression, Project, SourceFile } from 'ts-morph';
+export type Replacement = {
+    start: number;
+    end: number;
+    content: string;
+};
 export type Options = {
     development?: boolean;
 };
@@ -28,15 +34,23 @@ export declare const resolveFilePath: (basePath: string) => string;
 export declare const createProject: () => Project;
 export declare const buildRegistry: (contextDir: string, registry: ServerFuncRegistry) => void;
 export declare const getImportedServerFunctions: (sourceFile: SourceFile, registry: ServerFuncRegistry) => Map<string, ServerFuncInfo>;
-export declare const transformCallSites: (sourceFile: SourceFile, importedFuncs: Map<string, ServerFuncInfo>) => boolean;
-export declare const transformHandleCalls: (sourceFile: SourceFile) => boolean;
-export declare const transformSendCalls: (sourceFile: SourceFile) => boolean;
-export declare const transformFuncCalls: (sourceFile: SourceFile) => boolean;
+export declare const collectCallSiteReplacements: (sourceFile: SourceFile, importedFuncs: Map<string, ServerFuncInfo>) => Replacement[];
+export declare const collectHandleCallReplacements: (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) => {
+    code: string;
+    map: SourceMap;
+};
 export declare const appendRegistryCode: (sourceFile: SourceFile, fileRegistry: Map<string, ServerFuncInfo>) => string;
 export type TransformResult = {
     content: string;
     transformed: boolean;
+    map?: SourceMap;
+};
+export type TransformOptions = {
+    development?: boolean;
 };
-export declare const transformSourceFile: (filePath: string, content: string, registry: ServerFuncRegistry) => TransformResult;
+export declare const transformSourceFile: (filePath: string, content: string, registry: ServerFuncRegistry, options?: TransformOptions) => TransformResult;
 export declare const emitToJs: (filePath: string, content: string) => string;
 export declare const applyReplacements: (code: string, compilationId: string) => string;

package/lib/common.js

@@ -3,9 +3,10 @@ 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.appendRegistryCode = exports.transformFuncCalls = exports.transformSendCalls = exports.transformHandleCalls = exports.transformCallSites = exports.getImportedServerFunctions = exports.buildRegistry = exports.createProject = exports.resolveFilePath = exports.isFromLibrary = exports.getReplacements = exports.generateCompilationId = exports.formatFuncIdName = exports.FILE_EXTENSIONS = exports.LIBRARY_NAME = exports.CONTEXT_TYPE_NAME = exports.CALL_NAME = exports.HANDLE_NAME = exports.SERVER_FUNCTION_WRAPPER_NAME = exports.ZERO_COM_SERVER_REGISTRY = exports.ZERO_COM_CLIENT_CALL = void 0;
+exports.applyReplacements = exports.emitToJs = exports.transformSourceFile = exports.appendRegistryCode = exports.applyReplacementsWithMap = exports.collectFuncCallReplacements = exports.collectSendCallReplacements = exports.collectHandleCallReplacements = exports.collectCallSiteReplacements = exports.getImportedServerFunctions = exports.buildRegistry = exports.createProject = exports.resolveFilePath = exports.isFromLibrary = exports.getReplacements = exports.generateCompilationId = exports.formatFuncIdName = exports.FILE_EXTENSIONS = exports.LIBRARY_NAME = exports.CONTEXT_TYPE_NAME = exports.CALL_NAME = exports.HANDLE_NAME = exports.SERVER_FUNCTION_WRAPPER_NAME = 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"));
 const ts_morph_1 = require("ts-morph");
 // Constants
 exports.ZERO_COM_CLIENT_CALL = 'ZERO_COM_CLIENT_CALL';
@@ -153,11 +154,11 @@ const getImportedServerFunctions = (sourceFile, registry) => {
     return importedFuncs;
 };
 exports.getImportedServerFunctions = getImportedServerFunctions;
-// Transformations
-const transformCallSites = (sourceFile, importedFuncs) => {
+// Transformations - collect replacements instead of modifying AST
+const collectCallSiteReplacements = (sourceFile, importedFuncs) => {
     if (importedFuncs.size === 0)
-        return false;
-    let modified = false;
+        return [];
+    const replacements = [];
     sourceFile.forEachDescendant((node) => {
         if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
             return;
@@ -167,14 +168,17 @@ const transformCallSites = (sourceFile, importedFuncs) => {
         if (!funcInfo)
             return;
         const args = callExpr.getArguments().map(a => a.getText()).join(', ');
-        callExpr.replaceWithText(`globalThis.${exports.ZERO_COM_CLIENT_CALL}('${funcInfo.funcId}', [${args}])`);
-        modified = true;
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: `globalThis.${exports.ZERO_COM_CLIENT_CALL}('${funcInfo.funcId}', [${args}])`
+        });
     });
-    return modified;
+    return replacements;
 };
-exports.transformCallSites = transformCallSites;
-const transformHandleCalls = (sourceFile) => {
-    let modified = false;
+exports.collectCallSiteReplacements = collectCallSiteReplacements;
+const collectHandleCallReplacements = (sourceFile) => {
+    const replacements = [];
     sourceFile.forEachDescendant((node) => {
         if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
             return;
@@ -184,18 +188,20 @@ const transformHandleCalls = (sourceFile) => {
         const args = callExpr.getArguments();
         if (args.length < 3)
             return;
-        // Inline the logic directly using a named function for better stack traces.
         const funcId = args[0].getText();
         const ctx = args[1].getText();
         const argsArray = args[2].getText();
-        callExpr.replaceWithText(`(function handle(__fn, __ctx, __args) { return __fn.requireContext ? __fn(__ctx, ...__args) : __fn(...__args); })(globalThis.${exports.ZERO_COM_SERVER_REGISTRY}[${funcId}], ${ctx}, ${argsArray})`);
-        modified = true;
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: `((__fn) => __fn.requireContext ? __fn(${ctx}, ...${argsArray}) : __fn(...${argsArray}))(globalThis.${exports.ZERO_COM_SERVER_REGISTRY}[${funcId}])`
+        });
     });
-    return modified;
+    return replacements;
 };
-exports.transformHandleCalls = transformHandleCalls;
-const transformSendCalls = (sourceFile) => {
-    let modified = false;
+exports.collectHandleCallReplacements = collectHandleCallReplacements;
+const collectSendCallReplacements = (sourceFile) => {
+    const replacements = [];
     sourceFile.forEachDescendant((node) => {
         if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
             return;
@@ -205,18 +211,21 @@ const transformSendCalls = (sourceFile) => {
         const args = callExpr.getArguments();
         if (args.length < 1)
             return;
-        callExpr.replaceWithText(`globalThis.${exports.ZERO_COM_CLIENT_CALL} = ${args[0].getText()}`);
-        modified = true;
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: `globalThis.${exports.ZERO_COM_CLIENT_CALL} = ${args[0].getText()}`
+        });
     });
-    return modified;
+    return replacements;
 };
-exports.transformSendCalls = transformSendCalls;
-// Transform func(fn) calls to just fn.
+exports.collectSendCallReplacements = collectSendCallReplacements;
+// Collect func(fn) replacements to just fn.
 // The func() wrapper is only needed for type-level transformation (RemoveContextParam).
 // At runtime, we just need the raw function. The registration code (appendRegistryCode)
 // will set requireContext on the function based on compile-time analysis.
-const transformFuncCalls = (sourceFile) => {
-    let modified = false;
+const collectFuncCallReplacements = (sourceFile) => {
+    const replacements = [];
     sourceFile.forEachDescendant((node) => {
         if (node.getKind() !== ts_morph_1.SyntaxKind.CallExpression)
             return;
@@ -227,17 +236,34 @@ const transformFuncCalls = (sourceFile) => {
         if (args.length !== 1)
             return;
         // Replace func(fn) with just fn
-        callExpr.replaceWithText(args[0].getText());
-        modified = true;
+        replacements.push({
+            start: callExpr.getStart(),
+            end: callExpr.getEnd(),
+            content: args[0].getText()
+        });
     });
-    return modified;
+    return replacements;
 };
-exports.transformFuncCalls = transformFuncCalls;
+exports.collectFuncCallReplacements = collectFuncCallReplacements;
+// Apply collected replacements using MagicString for sourcemap support
+const applyReplacementsWithMap = (source, replacements, filePath) => {
+    const s = new magic_string_1.default(source);
+    // Sort replacements by start position descending to preserve positions
+    const sorted = [...replacements].sort((a, b) => b.start - a.start);
+    for (const { start, end, content } of sorted) {
+        s.overwrite(start, end, content);
+    }
+    return {
+        code: s.toString(),
+        map: s.generateMap({ source: filePath, includeContent: true, hires: true })
+    };
+};
+exports.applyReplacementsWithMap = applyReplacementsWithMap;
 const appendRegistryCode = (sourceFile, fileRegistry) => {
     // Generate registration code for each server function.
     // We set requireContext based on compile-time analysis of the function's first parameter type.
     // If the first param is context<T>, requireContext = true, otherwise false.
-    // This allows execFunc() at runtime to know whether to inject the context as the first argument.
+    // This allows handle() at runtime to know whether to inject the context as the first argument.
     const registrations = Array.from(fileRegistry.values())
         .map(info => `globalThis.${exports.ZERO_COM_SERVER_REGISTRY}['${info.funcId}'] = ${info.exportName};\n` +
         `${info.exportName}.requireContext = ${info.requireContext};`)
@@ -247,7 +273,8 @@ if (!globalThis.${exports.ZERO_COM_SERVER_REGISTRY}) globalThis.${exports.ZERO_C
 ${registrations}`;
 };
 exports.appendRegistryCode = appendRegistryCode;
-const transformSourceFile = (filePath, content, registry) => {
+const transformSourceFile = (filePath, content, registry, options = {}) => {
+    const { development = true } = options;
     const project = (0, exports.createProject)();
     const sourceFile = project.createSourceFile(filePath, content, { overwrite: true });
     const fileRegistry = registry.get(filePath);
@@ -257,17 +284,34 @@ const transformSourceFile = (filePath, content, registry) => {
     if (isServerFunctionFile) {
         fileRegistry.forEach((_, name) => importedFuncs.delete(name));
     }
-    const callsTransformed = (0, exports.transformCallSites)(sourceFile, importedFuncs);
-    const handleTransformed = (0, exports.transformHandleCalls)(sourceFile);
-    const sendTransformed = (0, exports.transformSendCalls)(sourceFile);
-    // Transform func() calls in files that define server functions.
-    // This strips the func() wrapper since it's only needed for types.
+    // Collect all replacements
+    const replacements = [];
+    // Always collect client call site replacements (needed in both dev and prod)
+    replacements.push(...(0, exports.collectCallSiteReplacements)(sourceFile, importedFuncs));
+    // In production mode, collect handle(), call(), and func() replacements.
+    // In development mode, these work at runtime via ZERO_COM_DEV_MODE flag.
+    if (!development) {
+        replacements.push(...(0, exports.collectHandleCallReplacements)(sourceFile));
+        replacements.push(...(0, exports.collectSendCallReplacements)(sourceFile));
+    }
+    // Handle server function files
     if (isServerFunctionFile) {
-        (0, exports.transformFuncCalls)(sourceFile);
+        // In production mode, strip func() wrappers
+        if (!development) {
+            replacements.push(...(0, exports.collectFuncCallReplacements)(sourceFile));
+        }
+        // Apply replacements and append registry code
+        if (replacements.length > 0) {
+            const { code, map } = (0, exports.applyReplacementsWithMap)(content, replacements, filePath);
+            // Create a new source file from the transformed code to append registry
+            const transformedSourceFile = project.createSourceFile(filePath + '.transformed', code, { overwrite: true });
+            return { content: (0, exports.appendRegistryCode)(transformedSourceFile, fileRegistry), transformed: true, map };
+        }
         return { content: (0, exports.appendRegistryCode)(sourceFile, fileRegistry), transformed: true };
     }
-    if (callsTransformed || handleTransformed || sendTransformed) {
-        return { content: sourceFile.getFullText(), transformed: true };
+    if (replacements.length > 0) {
+        const { code, map } = (0, exports.applyReplacementsWithMap)(content, replacements, filePath);
+        return { content: code, transformed: true, map };
     }
     return { content, transformed: false };
 };

package/lib/rollup.js

@@ -42,15 +42,16 @@ function zeroComRollupPlugin(options = {}) {
             });
         },
         load(id) {
-            var _a, _b;
+            var _a, _b, _c;
             if (!((_b = (_a = this.getModuleInfo(id)) === null || _a === void 0 ? void 0 : _a.meta) === null || _b === void 0 ? void 0 : _b.needsTransform) || !fs_1.default.existsSync(id))
                 return null;
             const content = fs_1.default.readFileSync(id, 'utf8');
-            const result = (0, common_1.transformSourceFile)(id, content, registry);
+            const result = (0, common_1.transformSourceFile)(id, content, registry, { development });
             if (!result.transformed)
                 return null;
             console.log(`[ZeroComRollupPlugin] Transformed: ${path_1.default.relative(process.cwd(), id)}`);
-            return (0, common_1.emitToJs)(id, result.content);
+            const code = (0, common_1.emitToJs)(id, result.content);
+            return { code, map: (_c = result.map) !== null && _c !== void 0 ? _c : null };
         },
         renderChunk(code) {
             if (development)

package/lib/runtime.d.ts

@@ -10,6 +10,6 @@ export type context<T = unknown> = T & {
 };
 type RemoveContextParam<F> = F extends (ctx: infer C, ...args: infer A) => infer R ? C extends context<unknown> ? (...args: A) => R : F : F;
 export declare function func<F extends (...args: any[]) => any>(fn: F): RemoveContextParam<F>;
-export declare const handle: (_funcId: string, _ctx: any, _args: any[]) => any;
-export declare const call: (_fn: (funcId: string, args: any[]) => Promise<any>) => void;
+export declare const handle: (funcId: string, ctx: any, args: any[]) => any;
+export declare const call: (fn: (funcId: string, args: any[]) => Promise<any>) => void;
 export {};

package/lib/runtime.js

@@ -3,19 +3,30 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.call = exports.handle = void 0;
 exports.func = func;
 // Implementation
-// User-facing function - transformed by plugin to just the inner function.
-// The plugin also appends code to register the function and set requireContext based on
-// compile-time analysis of whether the first parameter is context<T>.
-function func(_fn) {
-    throw new Error('func() was not transformed. Ensure the zero-com plugin is configured.');
+// In development mode: works at runtime, returns the function as-is.
+// In production mode: transformed by plugin to just the inner function.
+// The plugin appends code to register the function and set requireContext.
+function func(fn) {
+    return fn;
 }
-// User-facing function - transformed by plugin to inline code that checks requireContext and calls the function
-const handle = (_funcId, _ctx, _args) => {
-    throw new Error('handle() was not transformed. Ensure the zero-com plugin is configured.');
+// In development mode: works at runtime, looks up function and calls it.
+// In production mode: transformed by plugin to inline code.
+const handle = (funcId, ctx, args) => {
+    const fn = globalThis.ZERO_COM_SERVER_REGISTRY[funcId];
+    if (!fn) {
+        throw new Error(`Function not found in registry: ${funcId}`);
+    }
+    if (fn.requireContext) {
+        return fn(ctx, ...args);
+    }
+    else {
+        return fn(...args);
+    }
 };
 exports.handle = handle;
-// User-facing function - transformed by plugin to globalThis.ZERO_COM_CLIENT_CALL = fn
-const call = (_fn) => {
-    throw new Error('call() was not transformed. Ensure the zero-com plugin is configured.');
+// In development mode: works at runtime, sets the client call handler.
+// In production mode: transformed by plugin to assignment.
+const call = (fn) => {
+    globalThis.ZERO_COM_CLIENT_CALL = fn;
 };
 exports.call = call;

package/lib/webpack-loader.d.ts

@@ -2,5 +2,6 @@ import type { LoaderContext } from 'webpack';
 import { ServerFuncRegistry } from './common';
 export interface ZeroComLoaderOptions {
     registry: ServerFuncRegistry;
+    development: boolean;
 }
-export default function zeroComLoader(this: LoaderContext<ZeroComLoaderOptions>, source: string): string;
+export default function zeroComLoader(this: LoaderContext<ZeroComLoaderOptions>, source: string): void;

package/lib/webpack-loader.js

@@ -9,10 +9,13 @@ const common_1 = require("./common");
 function zeroComLoader(source) {
     const options = this.getOptions();
     const filePath = this.resourcePath;
-    const result = (0, common_1.transformSourceFile)(filePath, source, options.registry);
+    const result = (0, common_1.transformSourceFile)(filePath, source, options.registry, {
+        development: options.development
+    });
     if (!result.transformed) {
-        return source;
+        this.callback(null, source);
+        return;
     }
     console.log(`[ZeroComWebpackPlugin] Transformed: ${path_1.default.basename(filePath)}`);
-    return result.content;
+    this.callback(null, result.content, result.map);
 }

package/lib/webpack.js

@@ -24,7 +24,7 @@ class ZeroComWebpackPlugin {
             enforce: 'pre',
             use: [{
                     loader: loaderPath,
-                    options: { registry: this.registry }
+                    options: { registry: this.registry, development: this.options.development }
                 }]
         };
         compiler.options.module.rules.unshift(loaderRule);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zero-com",
-  "version": "1.6.4",
+  "version": "1.6.5",
   "main": "index.js",
   "repository": "https://github.com/yosbelms/zero-com",
   "keywords": [
@@ -22,6 +22,7 @@
   "license": "MIT",
   "description": "",
   "dependencies": {
+    "magic-string": "^0.30.21",
     "minimatch": "^10.0.1",
     "ts-morph": "^26.0.0"
   },