@fictjs/vite-plugin 0.16.0 → 0.17.1

package/dist/index.js

@@ -1,6 +1,7 @@
 // src/index.ts
 import { createHash } from "crypto";
-import { promises as fs } from "fs";
+import { promises as fs, readFileSync } from "fs";
+import { createRequire } from "module";
 import path from "path";
 import { fileURLToPath, pathToFileURL } from "url";
 import { transformAsync } from "@babel/core";
@@ -11,7 +12,11 @@ import * as t from "@babel/types";
 import { createFictPlugin } from "@fictjs/compiler";
 var traverse = typeof _traverse === "function" ? _traverse : _traverse.default;
 var generate = typeof _generate === "function" ? _generate : _generate.default;
-var CACHE_VERSION = 1;
+var CACHE_VERSION = 3;
+var require2 = createRequire(import.meta.url);
+var TRANSFORM_CACHE_FINGERPRINT = hashString(
+  [getCompilerCacheFingerprint(), String(extractAndRewriteHandlers)].join("|")
+);
 var MODULE_EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".mjs", ".cjs", ".mts", ".cts"];
 var VIRTUAL_HANDLER_PREFIX = "\0fict-handler:";
 var VIRTUAL_HANDLER_RESOLVE_PREFIX = "virtual:fict-handler:";
@@ -127,6 +132,9 @@ function fict(options = {}) {
         "fict",
         "fict/plus",
         "fict/advanced",
+        "fict/internal",
+        "fict/internal/list",
+        "fict/loader",
         "fict/slim",
         "fict/jsx-runtime",
         "fict/jsx-dev-runtime",
@@ -147,7 +155,12 @@ function fict(options = {}) {
         include2.delete(dep);
         exclude2.add(dep);
       }
-      const dedupePackages = ["fict", "@fictjs/runtime", "@fictjs/runtime/internal"];
+      const dedupePackages = [
+        "fict",
+        "fict/internal",
+        "@fictjs/runtime",
+        "@fictjs/runtime/internal"
+      ];
       for (const dep of dedupePackages) {
         dedupe.add(dep);
       }
@@ -266,16 +279,34 @@ function fict(options = {}) {
       }
       const cacheStore = ensureCache();
       const cacheKey = cacheStore.enabled ? buildCacheKey(filename, code, fictOptions, tsProject2) : null;
+      const shouldSplit = options.functionSplitting ?? (config?.command === "build" && (compilerOptions.resumable || !config?.build?.ssr));
       if (cacheKey) {
         const cached = await cacheStore.get(cacheKey);
         if (cached) {
-          return cached;
+          if (shouldSplit && cached.extractedHandlers?.length) {
+            for (const handler of cached.extractedHandlers) {
+              const handlerId = createHandlerId(handler.sourceModule, handler.exportName);
+              extractedHandlers.set(handlerId, handler);
+              if (config?.command === "build" && !config?.build?.ssr) {
+                this.emitFile({
+                  type: "chunk",
+                  id: `${VIRTUAL_HANDLER_RESOLVE_PREFIX}${handlerId}`,
+                  name: `handler-${handler.exportName}`
+                });
+              }
+            }
+          }
+          return {
+            code: cached.code,
+            map: cached.map
+          };
         }
       }
       try {
         const precompiledInput = isPrecompiledFictModule(code);
         let finalCode;
         let finalMap;
+        let splitResult = null;
         if (precompiledInput) {
           finalCode = code;
           finalMap = null;
@@ -297,7 +328,6 @@ function fict(options = {}) {
           finalCode = result.code;
           finalMap = result.map;
         }
-        const shouldSplit = options.functionSplitting ?? (config?.command === "build" && (compilerOptions.resumable || !config?.build?.ssr));
         debugLog("Function split decision", {
           shouldSplit,
           ssr: config?.build?.ssr,
@@ -305,7 +335,6 @@ function fict(options = {}) {
           file: filename
         });
         if (shouldSplit) {
-          let splitResult = null;
           try {
             splitResult = extractAndRewriteHandlers(finalCode, filename, extractedHandlers);
           } catch (error) {
@@ -339,7 +368,14 @@ function fict(options = {}) {
           map: finalMap
         };
         if (cacheKey) {
-          await cacheStore.set(cacheKey, transformed);
+          const cachedTransform = {
+            code: finalCode,
+            map: finalMap
+          };
+          if (shouldSplit && splitResult?.handlers.length) {
+            cachedTransform.extractedHandlers = splitResult.handlers.map((handlerName) => extractedHandlers.get(createHandlerId(filename, handlerName))).filter((handler) => !!handler);
+          }
+          await cacheStore.set(cacheKey, cachedTransform);
         }
         return transformed;
       } catch (error) {
@@ -504,6 +540,14 @@ function applyAlias(source, aliases) {
 function hashString(value) {
   return createHash("sha256").update(value).digest("hex");
 }
+function getCompilerCacheFingerprint() {
+  try {
+    const compilerEntry = require2.resolve("@fictjs/compiler");
+    return hashString(readFileSync(compilerEntry, "utf8"));
+  } catch {
+    return hashString(String(createFictPlugin));
+  }
+}
 function stableStringify(value) {
   if (value === null || typeof value !== "object") {
     return JSON.stringify(value);
@@ -535,7 +579,9 @@ function buildCacheKey(filename, code, options, tsProject) {
   const codeHash = hashString(code);
   const optionsHash = hashString(stableStringify(normalizeOptionsForCache(options)));
   const tsKey = tsProject ? `${tsProject.configHash}:${tsProject.projectVersion}` : "";
-  return hashString([CACHE_VERSION, filename, codeHash, optionsHash, tsKey].join("|"));
+  return hashString(
+    [CACHE_VERSION, TRANSFORM_CACHE_FINGERPRINT, filename, codeHash, optionsHash, tsKey].join("|")
+  );
 }
 var TransformCache = class {
   constructor(options) {
@@ -684,6 +730,30 @@ async function createTypeScriptProject(ts, rootDir, configPath) {
 function createHandlerId(sourceModule, exportName) {
   return `${sourceModule}$$${exportName}`;
 }
+function detectRuntimeImportFamilyFromCode(body) {
+  let sawFictFamily = false;
+  let sawStandaloneRuntimeFamily = false;
+  for (const stmt of body) {
+    const source = stmt && typeof stmt === "object" && "source" in stmt ? stmt.source?.value : void 0;
+    if (typeof source !== "string") continue;
+    if (source === "fict" || source === "fict/advanced" || source === "fict/internal" || source === "fict/internal/list" || source === "fict/jsx-runtime" || source === "fict/jsx-dev-runtime" || source === "fict/loader" || source === "fict/plus" || source === "fict/slim") {
+      sawFictFamily = true;
+      continue;
+    }
+    if (source === "@fictjs/runtime" || source === "@fictjs/runtime/advanced" || source === "@fictjs/runtime/internal" || source === "@fictjs/runtime/internal/list" || source === "@fictjs/runtime/jsx-runtime" || source === "@fictjs/runtime/jsx-dev-runtime" || source === "@fictjs/runtime/loader") {
+      sawStandaloneRuntimeFamily = true;
+    }
+  }
+  if (sawFictFamily) return "fict";
+  if (sawStandaloneRuntimeFamily) return "runtime";
+  return "fict";
+}
+function getRuntimeHelperModule(helperName, family) {
+  if (helperName === "keyedList") {
+    return family === "runtime" ? "@fictjs/runtime/internal/list" : "fict/internal/list";
+  }
+  return family === "runtime" ? "@fictjs/runtime/internal" : "fict/internal";
+}
 function generateHandlerModule(handler) {
   if (!handler.code) {
     return `export { ${handler.exportName} as default } from '${handler.sourceModule}';
@@ -693,11 +763,12 @@ function generateHandlerModule(handler) {
   for (const helperName of handler.helpersUsed) {
     const helper = RUNTIME_HELPERS[helperName];
     if (!helper) continue;
-    const existing = importsByModule.get(helper.from) ?? [];
+    const moduleSource = getRuntimeHelperModule(helperName, handler.runtimeImportFamily);
+    const existing = importsByModule.get(moduleSource) ?? [];
     if (!existing.includes(helper.import)) {
       existing.push(helper.import);
     }
-    importsByModule.set(helper.from, existing);
+    importsByModule.set(moduleSource, existing);
   }
   const imports = [];
   for (const [module, names] of importsByModule) {
@@ -718,20 +789,21 @@ function registerExtractedHandler(sourceModule, exportName, helpersUsed, code, l
     exportName,
     helpersUsed,
     localDeps,
-    code
+    code,
+    runtimeImportFamily: "fict"
   });
   return `${VIRTUAL_HANDLER_RESOLVE_PREFIX}${handlerId}`;
 }
 var RUNTIME_HELPERS = {
-  __fictUseLexicalScope: { import: "__fictUseLexicalScope", from: "@fictjs/runtime/internal" },
-  __fictGetScopeProps: { import: "__fictGetScopeProps", from: "@fictjs/runtime/internal" },
-  __fictGetSSRScope: { import: "__fictGetSSRScope", from: "@fictjs/runtime/internal" },
-  __fictEnsureScope: { import: "__fictEnsureScope", from: "@fictjs/runtime/internal" },
-  __fictPrepareContext: { import: "__fictPrepareContext", from: "@fictjs/runtime/internal" },
-  __fictPushContext: { import: "__fictPushContext", from: "@fictjs/runtime/internal" },
-  __fictPopContext: { import: "__fictPopContext", from: "@fictjs/runtime/internal" },
-  hydrateComponent: { import: "hydrateComponent", from: "@fictjs/runtime/internal" },
-  __fictQrl: { import: "__fictQrl", from: "@fictjs/runtime/internal" }
+  __fictUseLexicalScope: { import: "__fictUseLexicalScope", from: "fict/internal" },
+  __fictGetScopeProps: { import: "__fictGetScopeProps", from: "fict/internal" },
+  __fictGetSSRScope: { import: "__fictGetSSRScope", from: "fict/internal" },
+  __fictEnsureScope: { import: "__fictEnsureScope", from: "fict/internal" },
+  __fictPrepareContext: { import: "__fictPrepareContext", from: "fict/internal" },
+  __fictPushContext: { import: "__fictPushContext", from: "fict/internal" },
+  __fictPopContext: { import: "__fictPopContext", from: "fict/internal" },
+  hydrateComponent: { import: "hydrateComponent", from: "fict/internal" },
+  __fictQrl: { import: "__fictQrl", from: "fict/internal" }
 };
 var GLOBAL_IDENTIFIERS = /* @__PURE__ */ new Set([
   // JavaScript globals
@@ -1004,6 +1076,7 @@ function extractAndRewriteHandlers(code, sourceModule, handlerRegistry) {
   const handlerNames = [];
   const nodesToRemove = /* @__PURE__ */ new Set();
   const allLocalDeps = /* @__PURE__ */ new Set();
+  const runtimeImportFamily = detectRuntimeImportFamilyFromCode(ast.program.body);
   traverse(ast, {
     ExportNamedDeclaration(path2) {
       const declaration = path2.node.declaration;
@@ -1036,7 +1109,8 @@ function extractAndRewriteHandlers(code, sourceModule, handlerRegistry) {
             exportName: name,
             helpersUsed,
             localDeps,
-            code: handlerCode
+            code: handlerCode,
+            runtimeImportFamily
           });
           nodesToRemove.add(path2.node);
         }
@@ -1071,7 +1145,8 @@ function extractAndRewriteHandlers(code, sourceModule, handlerRegistry) {
           exportName: name,
           helpersUsed,
           localDeps,
-          code: handlerCode
+          code: handlerCode,
+          runtimeImportFamily
         });
         nodesToRemove.add(path2.node);
       }