zero-com 1.10.2 → 1.12.0

package/lib/common.d.ts

@@ -8,6 +8,7 @@ export type Replacement = {
 export type Options = {
     development?: boolean;
     target?: 'client' | 'server';
+    contextDir?: string;
 };
 export type ServerFuncInfo = {
     funcId: string;
@@ -32,7 +33,9 @@ export declare const getReplacements: (compilationId: string) => {
 export declare const isFromLibrary: (callExpr: CallExpression, libraryName: string) => boolean;
 export declare const resolveFilePath: (basePath: string) => string;
 export declare const createProject: () => Project;
-export declare const buildRegistry: (contextDir: string, registry: ServerFuncRegistry) => void;
+export declare const mightNeedTransform: (content: string, filePath: string, registry: ServerFuncRegistry) => boolean;
+export declare const buildRegistry: (contextDir: string, registry: ServerFuncRegistry, project?: Project) => void;
+export declare const updateRegistryForFile: (filePath: string, contextDir: string, registry: ServerFuncRegistry, project?: Project) => void;
 export declare const getImportedServerFunctions: (sourceFile: SourceFile, registry: ServerFuncRegistry) => Map<string, ServerFuncInfo>;
 export declare const collectCallSiteReplacements: (sourceFile: SourceFile, importedFuncs: Map<string, ServerFuncInfo>) => Replacement[];
 export declare const collectHandleCallReplacements: (sourceFile: SourceFile) => Replacement[];
@@ -53,6 +56,6 @@ export type TransformOptions = {
     development?: boolean;
     target?: 'client' | 'server';
 };
-export declare const transformSourceFile: (filePath: string, content: string, registry: ServerFuncRegistry, options?: TransformOptions) => TransformResult;
+export declare const transformSourceFile: (filePath: string, content: string, registry: ServerFuncRegistry, options?: TransformOptions, project?: Project) => TransformResult;
 export declare const emitToJs: (filePath: string, content: string) => string;
 export declare const applyReplacements: (code: string, compilationId: string) => 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.collectCallSiteReplacements = exports.getImportedServerFunctions = exports.buildRegistry = 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.collectHandleCallReplacements = 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;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const magic_string_1 = __importDefault(require("magic-string"));
@@ -81,8 +81,32 @@ const getCalleeFullName = (callExpr) => {
         return expr.getText();
     return '';
 };
+// Fast pre-filter: check if a file might need transformation before doing expensive ts-morph parsing
+const mightNeedTransform = (content, filePath, registry) => {
+    var _a;
+    if (content.includes(exports.LIBRARY_NAME))
+        return true;
+    if (registry.has(filePath))
+        return true;
+    if (registry.size === 0)
+        return false;
+    // Quick import scan: check if any imported path matches a registered server function file
+    const importPattern = /from\s+['"](\.[^'"]+)['"]/g;
+    let match;
+    while ((match = importPattern.exec(content)) !== null) {
+        const importedSegment = (_a = match[1].split('/').pop()) !== null && _a !== void 0 ? _a : '';
+        if (!importedSegment)
+            continue;
+        for (const registeredPath of registry.keys()) {
+            if (registeredPath.includes(importedSegment))
+                return true;
+        }
+    }
+    return false;
+};
+exports.mightNeedTransform = mightNeedTransform;
 // Registry building
-const buildRegistry = (contextDir, registry) => {
+const buildRegistry = (contextDir, registry, project) => {
     registry.clear();
     const scanDirectory = (dir) => {
         for (const entry of fs_1.default.readdirSync(dir, { withFileTypes: true })) {
@@ -91,15 +115,30 @@ const buildRegistry = (contextDir, registry) => {
                 scanDirectory(fullPath);
             }
             else if (entry.isFile() && exports.FILE_EXTENSIONS.slice(1).includes(path_1.default.extname(entry.name))) {
-                scanFileForServerFunctions(fullPath, contextDir, registry);
+                scanFileForServerFunctions(fullPath, contextDir, registry, project);
             }
         }
     };
     scanDirectory(contextDir);
 };
 exports.buildRegistry = buildRegistry;
-const scanFileForServerFunctions = (filePath, contextDir, registry) => {
-    const sourceFile = (0, exports.createProject)().createSourceFile(filePath, fs_1.default.readFileSync(filePath, 'utf8'), { overwrite: true });
+const updateRegistryForFile = (filePath, contextDir, registry, project) => {
+    if (!fs_1.default.existsSync(filePath)) {
+        registry.delete(filePath);
+        return;
+    }
+    const ext = path_1.default.extname(filePath);
+    if (!exports.FILE_EXTENSIONS.slice(1).includes(ext))
+        return;
+    scanFileForServerFunctions(filePath, contextDir, registry, project);
+};
+exports.updateRegistryForFile = updateRegistryForFile;
+const scanFileForServerFunctions = (filePath, contextDir, registry, project) => {
+    const content = fs_1.default.readFileSync(filePath, 'utf8');
+    if (!content.includes(exports.LIBRARY_NAME))
+        return;
+    const proj = project !== null && project !== void 0 ? project : (0, exports.createProject)();
+    const sourceFile = proj.createSourceFile(filePath, content, { overwrite: true });
     const fileRegistry = new Map();
     for (const decl of sourceFile.getVariableDeclarations()) {
         if (!decl.isExported())
@@ -274,10 +313,10 @@ const generateClientStubs = (fileRegistry) => {
     return stubs;
 };
 exports.generateClientStubs = generateClientStubs;
-const transformSourceFile = (filePath, content, registry, options = {}) => {
+const transformSourceFile = (filePath, content, registry, options = {}, project) => {
     const { development = true, target } = options;
-    const project = (0, exports.createProject)();
-    const sourceFile = project.createSourceFile(filePath, content, { overwrite: true });
+    const proj = project !== null && project !== void 0 ? project : (0, exports.createProject)();
+    const sourceFile = proj.createSourceFile(filePath, content, { overwrite: true });
     const fileRegistry = registry.get(filePath);
     const isServerFunctionFile = fileRegistry && fileRegistry.size > 0;
     if (target === 'client' && isServerFunctionFile) {
@@ -308,7 +347,7 @@ const transformSourceFile = (filePath, content, registry, options = {}) => {
         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 });
+            const transformedSourceFile = proj.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 };

package/lib/rollup.js

@@ -17,10 +17,12 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const common_1 = require("./common");
 function zeroComRollupPlugin(options = {}) {
-    const { development = true, target } = options;
+    const { development = true, target, contextDir: optContextDir } = options;
     const compilationId = (0, common_1.generateCompilationId)();
     const registry = new Map();
+    const project = (0, common_1.createProject)();
     let isVite = false;
+    let scanDir = optContextDir !== null && optContextDir !== void 0 ? optContextDir : process.cwd();
     return {
         name: 'zero-com-rollup-plugin',
         // Vite-only hook — detect that we're running inside Vite
@@ -28,13 +30,20 @@ function zeroComRollupPlugin(options = {}) {
             isVite = true;
         },
         buildStart() {
-            (0, common_1.buildRegistry)(process.cwd(), registry);
+            if (!optContextDir)
+                scanDir = process.cwd();
+            (0, common_1.buildRegistry)(scanDir, registry, project);
             for (const fileRegistry of registry.values()) {
                 for (const info of fileRegistry.values()) {
                     console.log(`[ZeroComRollupPlugin] ${info.funcId}`);
                 }
             }
         },
+        watchChange(id) {
+            if (!common_1.FILE_EXTENSIONS.slice(1).includes(path_1.default.extname(id)))
+                return;
+            (0, common_1.updateRegistryForFile)(id, scanDir, registry, project);
+        },
         // Rollup path: resolveId marks files, load transforms them.
         // Skipped in Vite because meta doesn't propagate from resolveId to load.
         resolveId(source, importer, opts) {
@@ -61,7 +70,9 @@ function zeroComRollupPlugin(options = {}) {
             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, { development, target });
+            if (!(0, common_1.mightNeedTransform)(content, id, registry))
+                return null;
+            const result = (0, common_1.transformSourceFile)(id, content, registry, { development, target }, project);
             if (!result.transformed)
                 return null;
             console.log(`[ZeroComRollupPlugin] Transformed: ${path_1.default.relative(process.cwd(), id)}`);
@@ -76,12 +87,14 @@ function zeroComRollupPlugin(options = {}) {
                 return null;
             if (id[0] === '\0' || id.includes('node_modules') || !/\.(ts|tsx|js|jsx|mjs)$/.test(id))
                 return null;
+            if (!(0, common_1.mightNeedTransform)(code, id, registry))
+                return null;
             // Derive effective target: explicit option takes precedence, otherwise infer from Vite's ssr flag
             const ssr = typeof viteOptions === 'object' && viteOptions !== null && 'ssr' in viteOptions
                 ? viteOptions.ssr
                 : undefined;
             const effectiveTarget = target !== null && target !== void 0 ? target : (ssr === false ? 'client' : ssr === true ? 'server' : undefined);
-            const result = (0, common_1.transformSourceFile)(id, code, registry, { development, target: effectiveTarget });
+            const result = (0, common_1.transformSourceFile)(id, code, registry, { development, target: effectiveTarget }, project);
             if (!result.transformed)
                 return null;
             console.log(`[ZeroComRollupPlugin] Transformed: ${path_1.default.relative(process.cwd(), id)}`);

package/lib/turbopack-loader.js

@@ -6,9 +6,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = turbopackLoader;
 const path_1 = __importDefault(require("path"));
 const common_1 = require("./common");
-// Module-level cache: registry is built once and reused across invocations
+// Module-level cache: registry and project are built once and reused across invocations
 let cachedRegistry = null;
 let cachedRootDir = null;
+let cachedProject = null;
 function turbopackLoader(source) {
     var _a;
     const options = this.getOptions();
@@ -17,16 +18,21 @@ function turbopackLoader(source) {
     const development = (_a = options.development) !== null && _a !== void 0 ? _a : true;
     // Lazily build and cache the registry on first invocation or if rootDir changes
     if (!cachedRegistry || cachedRootDir !== rootDir) {
+        cachedProject = (0, common_1.createProject)();
         cachedRegistry = new Map();
         cachedRootDir = rootDir;
-        (0, common_1.buildRegistry)(rootDir, cachedRegistry);
+        (0, common_1.buildRegistry)(rootDir, cachedRegistry, cachedProject);
         for (const fileRegistry of cachedRegistry.values()) {
             for (const info of fileRegistry.values()) {
                 console.log(`[TurbopackLoader] ${info.funcId}`);
             }
         }
     }
-    const result = (0, common_1.transformSourceFile)(filePath, source, cachedRegistry, { development, target: options.target });
+    if (!(0, common_1.mightNeedTransform)(source, filePath, cachedRegistry)) {
+        this.callback(null, source);
+        return;
+    }
+    const result = (0, common_1.transformSourceFile)(filePath, source, cachedRegistry, { development, target: options.target }, cachedProject !== null && cachedProject !== void 0 ? cachedProject : undefined);
     if (!result.transformed) {
         this.callback(null, source);
         return;

package/lib/webpack-loader.d.ts

@@ -1,7 +1,9 @@
 import type { LoaderContext } from 'webpack';
+import { Project } from 'ts-morph';
 import { ServerFuncRegistry } from './common';
 export interface ZeroComLoaderOptions {
     registry: ServerFuncRegistry;
+    project?: Project;
     development: boolean;
     target?: 'client' | 'server';
 }

package/lib/webpack-loader.js

@@ -9,10 +9,14 @@ const common_1 = require("./common");
 function zeroComLoader(source) {
     const options = this.getOptions();
     const filePath = this.resourcePath;
+    if (!(0, common_1.mightNeedTransform)(source, filePath, options.registry)) {
+        this.callback(null, source);
+        return;
+    }
     const result = (0, common_1.transformSourceFile)(filePath, source, options.registry, {
         development: options.development,
         target: options.target
-    });
+    }, options.project);
     if (!result.transformed) {
         this.callback(null, source);
         return;

package/lib/webpack.d.ts

@@ -4,6 +4,7 @@ export declare class ZeroComWebpackPlugin {
     private options;
     private compilationId;
     private registry;
+    private project;
     constructor(options?: Options);
     apply(compiler: Compiler): void;
 }

package/lib/webpack.js

@@ -5,21 +5,29 @@ const common_1 = require("./common");
 class ZeroComWebpackPlugin {
     constructor(options = {}) {
         this.registry = new Map();
+        this.project = (0, common_1.createProject)();
         this.options = Object.assign({ development: true }, options);
         this.compilationId = (0, common_1.generateCompilationId)();
     }
     apply(compiler) {
+        var _a;
         const pluginName = ZeroComWebpackPlugin.name;
         const { webpack } = compiler;
+        const contextDir = (_a = this.options.contextDir) !== null && _a !== void 0 ? _a : compiler.context;
         // Build registry before compilation
         compiler.hooks.beforeCompile.tap(pluginName, () => {
-            (0, common_1.buildRegistry)(compiler.context, this.registry);
+            (0, common_1.buildRegistry)(contextDir, this.registry, this.project);
             for (const fileRegistry of this.registry.values()) {
                 for (const info of fileRegistry.values()) {
                     console.log(`[ZeroComWebpackPlugin] ${info.funcId}`);
                 }
             }
         });
+        // Incrementally update registry when a file changes during watch mode
+        compiler.hooks.invalid.tap(pluginName, (fileName) => {
+            if (fileName)
+                (0, common_1.updateRegistryForFile)(fileName, contextDir, this.registry, this.project);
+        });
         // Add loader rule for TypeScript/JavaScript files
         const loaderPath = require.resolve('./webpack-loader');
         const loaderRule = {
@@ -28,7 +36,7 @@ class ZeroComWebpackPlugin {
             enforce: 'pre',
             use: [{
                     loader: loaderPath,
-                    options: { registry: this.registry, development: this.options.development, target: this.options.target }
+                    options: { registry: this.registry, project: this.project, development: this.options.development, target: this.options.target }
                 }]
         };
         compiler.options.module.rules.unshift(loaderRule);

package/package.json

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