@tanstack/router-plugin 1.159.11 → 1.159.14

package/dist/esm/core/code-splitter/compilers.js

@@ -2,7 +2,7 @@ import * as t from "@babel/types";
 import babel from "@babel/core";
 import * as template from "@babel/template";
 import { parseAst, findReferencedIdentifiers, deadCodeElimination, generateFromAst } from "@tanstack/router-utils";
-import { tsrSplit } from "../constants.js";
+import { tsrShared, tsrSplit } from "../constants.js";
 import { routeHmrStatement } from "../route-hmr-statement.js";
 import { createIdentifier } from "./path-ids.js";
 import { getFrameworkOptions } from "./framework-options.js";
@@ -85,6 +85,10 @@ function removeSplitSearchParamFromFilename(filename) {
   const [bareFilename] = filename.split("?");
   return bareFilename;
 }
+function addSharedSearchParamToFilename(filename) {
+  const [bareFilename] = filename.split("?");
+  return `${bareFilename}?${tsrShared}=1`;
+}
 const splittableCreateRouteFns = ["createFileRoute"];
 const unsplittableCreateRouteFns = [
   "createRootRoute",
@@ -94,6 +98,304 @@ const allCreateRouteFns = [
   ...splittableCreateRouteFns,
   ...unsplittableCreateRouteFns
 ];
+function collectIdentifiersFromNode(node) {
+  const ids = /* @__PURE__ */ new Set();
+  (function walk(n, parent, grandparent, parentKey) {
+    if (!n) return;
+    if (t.isIdentifier(n)) {
+      if (!parent || t.isReferenced(n, parent, grandparent)) {
+        ids.add(n.name);
+      }
+      return;
+    }
+    if (t.isJSXIdentifier(n)) {
+      if (parent && t.isJSXAttribute(parent) && parentKey === "name") {
+        return;
+      }
+      if (parent && t.isJSXMemberExpression(parent) && parentKey === "property") {
+        return;
+      }
+      const first = n.name[0];
+      if (first && first === first.toLowerCase()) {
+        return;
+      }
+      ids.add(n.name);
+      return;
+    }
+    for (const key of t.VISITOR_KEYS[n.type] || []) {
+      const child = n[key];
+      if (Array.isArray(child)) {
+        for (const c of child) {
+          if (c && typeof c.type === "string") {
+            walk(c, n, parent, key);
+          }
+        }
+      } else if (child && typeof child.type === "string") {
+        walk(child, n, parent, key);
+      }
+    }
+  })(node);
+  return ids;
+}
+function buildDeclarationMap(ast) {
+  const map = /* @__PURE__ */ new Map();
+  for (const stmt of ast.program.body) {
+    const decl = t.isExportNamedDeclaration(stmt) && stmt.declaration ? stmt.declaration : stmt;
+    if (t.isVariableDeclaration(decl)) {
+      for (const declarator of decl.declarations) {
+        for (const name of collectIdentifiersFromPattern(declarator.id)) {
+          map.set(name, declarator);
+        }
+      }
+    } else if (t.isFunctionDeclaration(decl) && decl.id) {
+      map.set(decl.id.name, decl);
+    } else if (t.isClassDeclaration(decl) && decl.id) {
+      map.set(decl.id.name, decl);
+    }
+  }
+  return map;
+}
+function buildDependencyGraph(declMap, localBindings) {
+  const graph = /* @__PURE__ */ new Map();
+  for (const [name, declNode] of declMap) {
+    if (!localBindings.has(name)) continue;
+    const allIds = collectIdentifiersFromNode(declNode);
+    const deps = /* @__PURE__ */ new Set();
+    for (const id of allIds) {
+      if (id !== name && localBindings.has(id)) deps.add(id);
+    }
+    graph.set(name, deps);
+  }
+  return graph;
+}
+function computeSharedBindings(opts) {
+  const ast = parseAst(opts);
+  const localModuleLevelBindings = /* @__PURE__ */ new Set();
+  for (const node of ast.program.body) {
+    collectLocalBindingsFromStatement(node, localModuleLevelBindings);
+  }
+  localModuleLevelBindings.delete("Route");
+  if (localModuleLevelBindings.size === 0) {
+    return /* @__PURE__ */ new Set();
+  }
+  function findIndexForSplitNode(str) {
+    return opts.codeSplitGroupings.findIndex(
+      (group) => group.includes(str)
+    );
+  }
+  let routeOptions;
+  babel.traverse(ast, {
+    CallExpression(path) {
+      if (!t.isIdentifier(path.node.callee)) return;
+      if (!splittableCreateRouteFns.includes(path.node.callee.name)) return;
+      if (t.isCallExpression(path.parentPath.node)) {
+        const opts2 = resolveIdentifier(path, path.parentPath.node.arguments[0]);
+        if (t.isObjectExpression(opts2)) routeOptions = opts2;
+      } else if (t.isVariableDeclarator(path.parentPath.node)) {
+        const caller = resolveIdentifier(path, path.parentPath.node.init);
+        if (t.isCallExpression(caller)) {
+          const opts2 = resolveIdentifier(path, caller.arguments[0]);
+          if (t.isObjectExpression(opts2)) routeOptions = opts2;
+        }
+      }
+    }
+  });
+  if (!routeOptions) return /* @__PURE__ */ new Set();
+  const splitGroupsPresent = /* @__PURE__ */ new Set();
+  let hasNonSplit = false;
+  for (const prop of routeOptions.properties) {
+    if (!t.isObjectProperty(prop) || !t.isIdentifier(prop.key)) continue;
+    if (prop.key.name === "codeSplitGroupings") continue;
+    if (t.isIdentifier(prop.value) && prop.value.name === "undefined") continue;
+    const groupIndex = findIndexForSplitNode(prop.key.name);
+    if (groupIndex === -1) {
+      hasNonSplit = true;
+    } else {
+      splitGroupsPresent.add(groupIndex);
+    }
+  }
+  if (!hasNonSplit && splitGroupsPresent.size < 2) return /* @__PURE__ */ new Set();
+  const declMap = buildDeclarationMap(ast);
+  const depGraph = buildDependencyGraph(declMap, localModuleLevelBindings);
+  const allLocalBindings = new Set(localModuleLevelBindings);
+  allLocalBindings.add("Route");
+  const fullDepGraph = buildDependencyGraph(declMap, allLocalBindings);
+  const refsByGroup = /* @__PURE__ */ new Map();
+  for (const prop of routeOptions.properties) {
+    if (!t.isObjectProperty(prop) || !t.isIdentifier(prop.key)) continue;
+    const key = prop.key.name;
+    if (key === "codeSplitGroupings") continue;
+    const groupIndex = findIndexForSplitNode(key);
+    const directRefs = collectModuleLevelRefsFromNode(
+      prop.value,
+      localModuleLevelBindings
+    );
+    const allRefs = new Set(directRefs);
+    expandTransitively(allRefs, depGraph);
+    for (const ref of allRefs) {
+      let groups = refsByGroup.get(ref);
+      if (!groups) {
+        groups = /* @__PURE__ */ new Set();
+        refsByGroup.set(ref, groups);
+      }
+      groups.add(groupIndex);
+    }
+  }
+  const shared = /* @__PURE__ */ new Set();
+  for (const [name, groups] of refsByGroup) {
+    if (groups.size >= 2) shared.add(name);
+  }
+  expandSharedDestructuredDeclarators(ast, refsByGroup, shared);
+  if (shared.size === 0) return shared;
+  expandDestructuredDeclarations(ast, shared);
+  removeBindingsDependingOnRoute(shared, fullDepGraph);
+  return shared;
+}
+function expandSharedDestructuredDeclarators(ast, refsByGroup, shared) {
+  for (const stmt of ast.program.body) {
+    const decl = t.isExportNamedDeclaration(stmt) && stmt.declaration ? stmt.declaration : stmt;
+    if (!t.isVariableDeclaration(decl)) continue;
+    for (const declarator of decl.declarations) {
+      if (!t.isObjectPattern(declarator.id) && !t.isArrayPattern(declarator.id))
+        continue;
+      const names = collectIdentifiersFromPattern(declarator.id);
+      const usedGroups = /* @__PURE__ */ new Set();
+      for (const name of names) {
+        const groups = refsByGroup.get(name);
+        if (!groups) continue;
+        for (const g of groups) usedGroups.add(g);
+      }
+      if (usedGroups.size >= 2) {
+        for (const name of names) {
+          shared.add(name);
+        }
+      }
+    }
+  }
+}
+function collectLocalBindingsFromStatement(node, bindings) {
+  const decl = t.isExportNamedDeclaration(node) && node.declaration ? node.declaration : node;
+  if (t.isVariableDeclaration(decl)) {
+    for (const declarator of decl.declarations) {
+      for (const name of collectIdentifiersFromPattern(declarator.id)) {
+        bindings.add(name);
+      }
+    }
+  } else if (t.isFunctionDeclaration(decl) && decl.id) {
+    bindings.add(decl.id.name);
+  } else if (t.isClassDeclaration(decl) && decl.id) {
+    bindings.add(decl.id.name);
+  }
+}
+function collectModuleLevelRefsFromNode(node, localModuleLevelBindings) {
+  const allIds = collectIdentifiersFromNode(node);
+  const refs = /* @__PURE__ */ new Set();
+  for (const name of allIds) {
+    if (localModuleLevelBindings.has(name)) refs.add(name);
+  }
+  return refs;
+}
+function expandTransitively(shared, depGraph) {
+  const queue = [...shared];
+  const visited = /* @__PURE__ */ new Set();
+  while (queue.length > 0) {
+    const name = queue.pop();
+    if (visited.has(name)) continue;
+    visited.add(name);
+    const deps = depGraph.get(name);
+    if (!deps) continue;
+    for (const dep of deps) {
+      if (!shared.has(dep)) {
+        shared.add(dep);
+        queue.push(dep);
+      }
+    }
+  }
+}
+function removeBindingsDependingOnRoute(shared, depGraph) {
+  const reverseGraph = /* @__PURE__ */ new Map();
+  for (const [name, deps] of depGraph) {
+    for (const dep of deps) {
+      let parents = reverseGraph.get(dep);
+      if (!parents) {
+        parents = /* @__PURE__ */ new Set();
+        reverseGraph.set(dep, parents);
+      }
+      parents.add(name);
+    }
+  }
+  const visited = /* @__PURE__ */ new Set();
+  const queue = ["Route"];
+  while (queue.length > 0) {
+    const cur = queue.pop();
+    if (visited.has(cur)) continue;
+    visited.add(cur);
+    const parents = reverseGraph.get(cur);
+    if (!parents) continue;
+    for (const parent of parents) {
+      if (!visited.has(parent)) queue.push(parent);
+    }
+  }
+  for (const name of [...shared]) {
+    if (visited.has(name)) {
+      shared.delete(name);
+    }
+  }
+}
+function expandDestructuredDeclarations(ast, shared) {
+  for (const stmt of ast.program.body) {
+    const decl = t.isExportNamedDeclaration(stmt) && stmt.declaration ? stmt.declaration : stmt;
+    if (!t.isVariableDeclaration(decl)) continue;
+    for (const declarator of decl.declarations) {
+      if (!t.isObjectPattern(declarator.id) && !t.isArrayPattern(declarator.id))
+        continue;
+      const names = collectIdentifiersFromPattern(declarator.id);
+      const hasShared = names.some((n) => shared.has(n));
+      if (hasShared) {
+        for (const n of names) {
+          shared.add(n);
+        }
+      }
+    }
+  }
+}
+function findExportedSharedBindings(ast, sharedBindings) {
+  const exported = /* @__PURE__ */ new Set();
+  for (const stmt of ast.program.body) {
+    if (!t.isExportNamedDeclaration(stmt) || !stmt.declaration) continue;
+    if (t.isVariableDeclaration(stmt.declaration)) {
+      for (const decl of stmt.declaration.declarations) {
+        for (const name of collectIdentifiersFromPattern(decl.id)) {
+          if (sharedBindings.has(name)) exported.add(name);
+        }
+      }
+    } else if (t.isFunctionDeclaration(stmt.declaration) && stmt.declaration.id) {
+      if (sharedBindings.has(stmt.declaration.id.name))
+        exported.add(stmt.declaration.id.name);
+    } else if (t.isClassDeclaration(stmt.declaration) && stmt.declaration.id) {
+      if (sharedBindings.has(stmt.declaration.id.name))
+        exported.add(stmt.declaration.id.name);
+    }
+  }
+  return exported;
+}
+function removeSharedDeclarations(ast, sharedBindings) {
+  ast.program.body = ast.program.body.filter((stmt) => {
+    const decl = t.isExportNamedDeclaration(stmt) && stmt.declaration ? stmt.declaration : stmt;
+    if (t.isVariableDeclaration(decl)) {
+      decl.declarations = decl.declarations.filter((declarator) => {
+        const names = collectIdentifiersFromPattern(declarator.id);
+        return !names.every((n) => sharedBindings.has(n));
+      });
+      if (decl.declarations.length === 0) return false;
+    } else if (t.isFunctionDeclaration(decl) && decl.id) {
+      if (sharedBindings.has(decl.id.name)) return false;
+    } else if (t.isClassDeclaration(decl) && decl.id) {
+      if (sharedBindings.has(decl.id.name)) return false;
+    }
+    return true;
+  });
+}
 function compileCodeSplitReferenceRoute(opts) {
   const ast = parseAst(opts);
   const refIdents = findReferencedIdentifiers(ast);
@@ -110,6 +412,7 @@ function compileCodeSplitReferenceRoute(opts) {
   let createRouteFn;
   let modified = false;
   let hmrAdded = false;
+  let sharedExportedNames;
   babel.traverse(ast, {
     Program: {
       enter(programPath) {
@@ -305,6 +608,44 @@ function compileCodeSplitReferenceRoute(opts) {
             }
           });
         }
+        if (opts.sharedBindings && opts.sharedBindings.size > 0) {
+          sharedExportedNames = findExportedSharedBindings(
+            ast,
+            opts.sharedBindings
+          );
+          removeSharedDeclarations(ast, opts.sharedBindings);
+          const sharedModuleUrl = addSharedSearchParamToFilename(opts.filename);
+          const sharedImportSpecifiers = [...opts.sharedBindings].map(
+            (name) => t.importSpecifier(t.identifier(name), t.identifier(name))
+          );
+          const [sharedImportPath] = programPath.unshiftContainer(
+            "body",
+            t.importDeclaration(
+              sharedImportSpecifiers,
+              t.stringLiteral(sharedModuleUrl)
+            )
+          );
+          sharedImportPath.traverse({
+            Identifier(identPath) {
+              if (identPath.parentPath.isImportSpecifier() && identPath.key === "local") {
+                refIdents.add(identPath);
+              }
+            }
+          });
+          if (sharedExportedNames.size > 0) {
+            const reExportSpecifiers = [...sharedExportedNames].map(
+              (name) => t.exportSpecifier(t.identifier(name), t.identifier(name))
+            );
+            programPath.pushContainer(
+              "body",
+              t.exportNamedDeclaration(
+                null,
+                reExportSpecifiers,
+                t.stringLiteral(sharedModuleUrl)
+              )
+            );
+          }
+        }
       }
     }
   });
@@ -334,6 +675,9 @@ function compileCodeSplitReferenceRoute(opts) {
 function compileCodeSplitVirtualRoute(opts) {
   const ast = parseAst(opts);
   const refIdents = findReferencedIdentifiers(ast);
+  if (opts.sharedBindings && opts.sharedBindings.size > 0) {
+    removeSharedDeclarations(ast, opts.sharedBindings);
+  }
   const intendedSplitNodes = new Set(opts.splitTargets);
   const knownExportedIdents = /* @__PURE__ */ new Set();
   babel.traverse(ast, {
@@ -592,10 +936,121 @@ function compileCodeSplitVirtualRoute(opts) {
             }
           }
         });
+        if (opts.sharedBindings && opts.sharedBindings.size > 0) {
+          const sharedImportSpecifiers = [...opts.sharedBindings].map(
+            (name) => t.importSpecifier(t.identifier(name), t.identifier(name))
+          );
+          const sharedModuleUrl = addSharedSearchParamToFilename(
+            removeSplitSearchParamFromFilename(opts.filename)
+          );
+          const [sharedImportPath] = programPath.unshiftContainer(
+            "body",
+            t.importDeclaration(
+              sharedImportSpecifiers,
+              t.stringLiteral(sharedModuleUrl)
+            )
+          );
+          sharedImportPath.traverse({
+            Identifier(identPath) {
+              if (identPath.parentPath.isImportSpecifier() && identPath.key === "local") {
+                refIdents.add(identPath);
+              }
+            }
+          });
+        }
+      }
+    }
+  });
+  deadCodeElimination(ast, refIdents);
+  {
+    const locallyBound = /* @__PURE__ */ new Set();
+    for (const stmt of ast.program.body) {
+      collectLocalBindingsFromStatement(stmt, locallyBound);
+    }
+    ast.program.body = ast.program.body.filter((stmt) => {
+      if (!t.isExpressionStatement(stmt)) return true;
+      const refs = collectIdentifiersFromNode(stmt);
+      return [...refs].some((name) => locallyBound.has(name));
+    });
+  }
+  if (ast.program.body.length === 0) {
+    ast.program.directives = [];
+  }
+  return generateFromAst(ast, {
+    sourceMaps: true,
+    sourceFileName: opts.filename,
+    filename: opts.filename
+  });
+}
+function compileCodeSplitSharedRoute(opts) {
+  const ast = parseAst(opts);
+  const refIdents = findReferencedIdentifiers(ast);
+  const localBindings = /* @__PURE__ */ new Set();
+  for (const node of ast.program.body) {
+    collectLocalBindingsFromStatement(node, localBindings);
+  }
+  localBindings.delete("Route");
+  const declMap = buildDeclarationMap(ast);
+  const depGraph = buildDependencyGraph(declMap, localBindings);
+  const keepBindings = new Set(opts.sharedBindings);
+  keepBindings.delete("Route");
+  expandTransitively(keepBindings, depGraph);
+  ast.program.body = ast.program.body.filter((stmt) => {
+    if (t.isImportDeclaration(stmt)) return true;
+    const decl = t.isExportNamedDeclaration(stmt) && stmt.declaration ? stmt.declaration : stmt;
+    if (t.isVariableDeclaration(decl)) {
+      decl.declarations = decl.declarations.filter((declarator) => {
+        const names = collectIdentifiersFromPattern(declarator.id);
+        return names.some((n) => keepBindings.has(n));
+      });
+      if (decl.declarations.length === 0) return false;
+      if (t.isExportNamedDeclaration(stmt) && stmt.declaration) {
+        return true;
       }
+      return true;
+    } else if (t.isFunctionDeclaration(decl) && decl.id) {
+      return keepBindings.has(decl.id.name);
+    } else if (t.isClassDeclaration(decl) && decl.id) {
+      return keepBindings.has(decl.id.name);
+    }
+    return false;
+  });
+  ast.program.body = ast.program.body.map((stmt) => {
+    if (t.isExportNamedDeclaration(stmt) && stmt.declaration) {
+      return stmt.declaration;
     }
+    return stmt;
   });
+  const exportNames = [...opts.sharedBindings].sort(
+    (a, b) => a.localeCompare(b)
+  );
+  const exportSpecifiers = exportNames.map(
+    (name) => t.exportSpecifier(t.identifier(name), t.identifier(name))
+  );
+  if (exportSpecifiers.length > 0) {
+    const exportDecl = t.exportNamedDeclaration(null, exportSpecifiers);
+    ast.program.body.push(exportDecl);
+    babel.traverse(ast, {
+      Program(programPath) {
+        const bodyPaths = programPath.get("body");
+        const last = bodyPaths[bodyPaths.length - 1];
+        if (last && last.isExportNamedDeclaration()) {
+          last.traverse({
+            Identifier(identPath) {
+              if (identPath.parentPath.isExportSpecifier() && identPath.key === "local") {
+                refIdents.add(identPath);
+              }
+            }
+          });
+        }
+        programPath.stop();
+      }
+    });
+  }
   deadCodeElimination(ast, refIdents);
+  if (ast.program.body.length === 0) {
+    ast.program.directives = [];
+  }
   return generateFromAst(ast, {
     sourceMaps: true,
     sourceFileName: opts.filename,
@@ -864,8 +1319,20 @@ function removeExports(ast, node) {
   return removed;
 }
 export {
+  addSharedSearchParamToFilename,
+  buildDeclarationMap,
+  buildDependencyGraph,
+  collectIdentifiersFromNode,
+  collectLocalBindingsFromStatement,
+  collectModuleLevelRefsFromNode,
   compileCodeSplitReferenceRoute,
+  compileCodeSplitSharedRoute,
   compileCodeSplitVirtualRoute,
-  detectCodeSplitGroupingsFromRoute
+  computeSharedBindings,
+  detectCodeSplitGroupingsFromRoute,
+  expandDestructuredDeclarations,
+  expandSharedDestructuredDeclarators,
+  expandTransitively,
+  removeBindingsDependingOnRoute
 };
 //# sourceMappingURL=compilers.js.map