zero-com 1.6.2 → 1.6.3

package/README.md

@@ -1,6 +1,6 @@
 # Zero-com
 
-A ~420 Bytes utility for transparently communicating client and server in full-stack projects through compile-time code transformation, with end-to-end static type checking.
+The ~85 Bytes utility for transparently communicating client and server in full-stack projects through compile-time code transformation, with end-to-end static type checking.
 
 ## Table of Contents
 

package/lib/common.d.ts

@@ -32,6 +32,7 @@ export declare const getImportedServerFunctions: (sourceFile: SourceFile, regist
 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 appendRegistryCode: (sourceFile: SourceFile, fileRegistry: Map<string, ServerFuncInfo>) => string;
 export type TransformResult = {
     content: 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.appendRegistryCode = 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.EXEC_FUNC_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.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.EXEC_FUNC_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 ts_morph_1 = require("ts-morph");
@@ -208,13 +208,40 @@ const transformSendCalls = (sourceFile) => {
     return modified;
 };
 exports.transformSendCalls = transformSendCalls;
+// Transform func(fn) calls 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;
+    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.SERVER_FUNCTION_WRAPPER_NAME)
+            return;
+        const args = callExpr.getArguments();
+        if (args.length !== 1)
+            return;
+        // Replace func(fn) with just fn
+        callExpr.replaceWithText(args[0].getText());
+        modified = true;
+    });
+    return modified;
+};
+exports.transformFuncCalls = transformFuncCalls;
 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.
     const registrations = Array.from(fileRegistry.values())
-        .map(info => `globalThis.${exports.ZERO_COM_SERVER_REGISTRY}['${info.funcId}'] = ${info.exportName}`)
-        .join(';\n');
+        .map(info => `globalThis.${exports.ZERO_COM_SERVER_REGISTRY}['${info.funcId}'] = ${info.exportName};\n` +
+        `${info.exportName}.requireContext = ${info.requireContext};`)
+        .join('\n');
     return `${sourceFile.getFullText()}
 if (!globalThis.${exports.ZERO_COM_SERVER_REGISTRY}) globalThis.${exports.ZERO_COM_SERVER_REGISTRY} = Object.create(null);
-${registrations};`;
+${registrations}`;
 };
 exports.appendRegistryCode = appendRegistryCode;
 // Helper to ensure a specific named import exists
@@ -262,6 +289,12 @@ const transformSourceFile = (filePath, content, registry) => {
     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.
+    let funcTransformed = false;
+    if (isServerFunctionFile) {
+        funcTransformed = (0, exports.transformFuncCalls)(sourceFile);
+    }
     if (handleTransformed) {
         ensureImport(sourceFile, exports.LIBRARY_NAME, exports.EXEC_FUNC_NAME);
     }

package/lib/runtime.js

@@ -3,9 +3,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.call = exports.handle = exports.execFunc = void 0;
 exports.func = func;
 // Implementation
-function func(fn) {
-    fn.requireContext = true;
-    return fn;
+// 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.');
 }
 // Internal implementation - receives the actual function from registry
 const execFunc = (sfn, ctx, args) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zero-com",
-  "version": "1.6.2",
+  "version": "1.6.3",
   "main": "index.js",
   "repository": "https://github.com/yosbelms/zero-com",
   "keywords": [