@fluixi/compiler 1.0.0-alpha.82 → 1.0.0-alpha.84

package/dist/transform/source-locations.mjs

@@ -0,0 +1,307 @@
+// src/analyze/intrinsics.ts
+var REACTIVE_MODULE = "@fluixi/reactive/signal";
+var INTRINSICS = {
+  $signal: { export: "signal", module: REACTIVE_MODULE, returns: "signal-handle" },
+  $memo: { export: "memo", module: REACTIVE_MODULE, returns: "memo-handle" },
+  $effect: { export: "effect", module: REACTIVE_MODULE, returns: "effect" },
+  $store: { export: "store", module: "@fluixi/reactive/store", returns: "store-handle" },
+  $resource: { export: "resource", module: REACTIVE_MODULE, returns: "resource-handle" },
+  // No short runtime form for these yet, so the intrinsic stands for the `create*`
+  // name. Still worth having — no import to write — and when a shorter form is
+  // designed, the change is this line rather than a compiler pass.
+  $selector: { export: "createSelector", module: REACTIVE_MODULE, returns: "memo-accessor" },
+  $deferred: { export: "createDeferred", module: REACTIVE_MODULE, returns: "memo-accessor" },
+  // Not value constructors, but the compiler has to know them: they change what a read
+  // does. `untrack` suppresses dependency collection, which is exactly the kind of thing
+  // dependency analysis must not guess at.
+  $untrack: { export: "untrack", module: REACTIVE_MODULE, returns: "plain" },
+  $untrackStore: { export: "untrackStore", module: "@fluixi/reactive/store", returns: "plain" }
+};
+var RUNTIME_PRIMITIVES = {
+  signal: "signal-handle",
+  memo: "memo-handle",
+  effect: "effect",
+  store: "store-handle",
+  createSignal: ["signal-accessor", "signal-setter"],
+  createMemo: "memo-accessor",
+  createEffect: "effect",
+  createRenderEffect: "effect",
+  createStore: "reactive-object",
+  resource: "resource-handle",
+  createResource: "reactive-object",
+  createSelector: "memo-accessor",
+  createDeferred: "memo-accessor"
+};
+var PRIMITIVE_MODULES = [
+  "@fluixi/reactive",
+  "@fluixi/reactive/signal",
+  "@fluixi/reactive/store",
+  "@fluixi/core"
+];
+var INTRINSIC_CALL = new RegExp(`\\$(?:${Object.keys(INTRINSICS).map((n) => n.slice(1)).join("|")})\\s*\\(`);
+function isIntrinsicName(name) {
+  return Object.prototype.hasOwnProperty.call(INTRINSICS, name);
+}
+function isPrimitiveModule(source) {
+  return PRIMITIVE_MODULES.includes(source);
+}
+
+// src/analyze/reactive-bindings.ts
+function patternNames(node, out) {
+  if (!node) return;
+  switch (node.type) {
+    case "Identifier":
+      out.add(node.name);
+      return;
+    case "ObjectPattern":
+      for (const p of node.properties ?? []) {
+        if (p.type === "RestElement") patternNames(p.argument, out);
+        else patternNames(p.value, out);
+      }
+      return;
+    case "ArrayPattern":
+      for (const e of node.elements ?? []) patternNames(e, out);
+      return;
+    case "AssignmentPattern":
+      patternNames(node.left, out);
+      return;
+    case "RestElement":
+      patternNames(node.argument, out);
+      return;
+  }
+}
+function walk(root, visit) {
+  if (!root || typeof root !== "object" || typeof root.type !== "string") return;
+  visit(root);
+  for (const key of Object.keys(root)) {
+    if (key === "loc" || key === "leadingComments" || key === "trailingComments") continue;
+    const value = root[key];
+    if (Array.isArray(value)) for (const item of value) walk(item, visit);
+    else if (value && typeof value === "object") walk(value, visit);
+  }
+}
+function calledPrimitive(node, bindings) {
+  if (node.type !== "CallExpression") return void 0;
+  const callee = node.callee;
+  if (callee?.type !== "Identifier") return void 0;
+  const name = callee.name;
+  const imported = bindings.primitives.get(name);
+  if (imported) return imported;
+  const intrinsic = INTRINSICS[name];
+  if (intrinsic && !bindings.shadowed.has(name)) return intrinsic.returns;
+  return void 0;
+}
+function classify(id, returns, kinds) {
+  if (Array.isArray(returns)) {
+    if (id.type === "ArrayPattern") {
+      id.elements.forEach((element, index) => {
+        const kind = returns[index];
+        if (element?.type === "Identifier" && kind) kinds.set(element.name, kind);
+      });
+    }
+    return;
+  }
+  if (id.type === "Identifier") kinds.set(id.name, returns);
+}
+function resolveReactiveBindings(program) {
+  const bindings = { kinds: /* @__PURE__ */ new Map(), primitives: /* @__PURE__ */ new Map(), shadowed: /* @__PURE__ */ new Set() };
+  const root = program;
+  walk(root, (node) => {
+    const names = /* @__PURE__ */ new Set();
+    if (node.type === "VariableDeclarator") patternNames(node.id, names);
+    else if (node.type === "FunctionDeclaration" || node.type === "ClassDeclaration") patternNames(node.id, names);
+    else if (node.type === "ImportDefaultSpecifier" || node.type === "ImportNamespaceSpecifier") patternNames(node.local, names);
+    else if (node.type === "ImportSpecifier") patternNames(node.local, names);
+    else if (node.type === "FunctionExpression" || node.type === "ArrowFunctionExpression" || node.type === "FunctionDeclaration") {
+      for (const p of node.params ?? []) patternNames(p, names);
+    }
+    for (const name of names) if (INTRINSICS[name]) bindings.shadowed.add(name);
+  });
+  for (const statement of root.body ?? []) {
+    if (statement.type !== "ImportDeclaration") continue;
+    if (!isPrimitiveModule(statement.source?.value)) continue;
+    for (const specifier of statement.specifiers ?? []) {
+      if (specifier.type !== "ImportSpecifier") continue;
+      const importedNode = specifier.imported;
+      const imported = importedNode.type === "Identifier" ? importedNode.name : importedNode.value;
+      const returns = RUNTIME_PRIMITIVES[imported];
+      if (returns) bindings.primitives.set(specifier.local.name, returns);
+    }
+  }
+  walk(root, (node) => {
+    if (node.type !== "VariableDeclarator" || !node.init) return;
+    const returns = calledPrimitive(node.init, bindings);
+    if (returns) classify(node.id, returns, bindings.kinds);
+  });
+  return bindings;
+}
+
+// src/transform/source-locations.ts
+var SOURCE_MARKER = "__fx_source";
+var COMPONENT_MARKER = "__fx_component";
+var HOLE_MARKER = "__fx_hole";
+var markHole = (accessor, line, column, name) => `(globalThis.${HOLE_MARKER} ?? ((f) => f))(${accessor},${line},${column}${name === void 0 ? "" : `,${JSON.stringify(name)}`})`;
+var PROPS_MARKER = "__fx_props";
+var markProps = (object, at) => `(globalThis.${PROPS_MARKER} ?? ((p) => p))(${object},${JSON.stringify(at)})`;
+var RUNTIME_PRIMITIVES2 = /* @__PURE__ */ new Set([
+  "signal",
+  "memo",
+  "effect",
+  "store",
+  "resource",
+  "watch",
+  "createSignal",
+  "createMemo",
+  "createEffect",
+  "createStore",
+  "createResource"
+]);
+function mayHaveReactiveCall(code) {
+  for (const name of RUNTIME_PRIMITIVES2) {
+    if (code.includes(name)) return true;
+  }
+  return code.includes("$signal") || code.includes("$memo") || code.includes("$effect") || code.includes("$store");
+}
+var SINGLE_STATEMENT_PARENTS = /* @__PURE__ */ new Set([
+  "IfStatement",
+  "ForStatement",
+  "ForInStatement",
+  "ForOfStatement",
+  "WhileStatement",
+  "DoWhileStatement",
+  "LabeledStatement",
+  "WithStatement"
+]);
+function isLoneBranch(statement, parent) {
+  if (!parent || statement.type === "BlockStatement") return false;
+  return SINGLE_STATEMENT_PARENTS.has(parent.type);
+}
+function isExport(node) {
+  return node.type === "ExportNamedDeclaration" || node.type === "ExportDefaultDeclaration";
+}
+function isStatement(node) {
+  return node.type.endsWith("Statement") || node.type === "VariableDeclaration";
+}
+function walk2(node, visit, stack = []) {
+  if (!node || typeof node.type !== "string") return;
+  visit(node, stack);
+  stack.push(node);
+  for (const key of Object.keys(node)) {
+    if (key === "loc" || key === "parent") continue;
+    const value = node[key];
+    if (Array.isArray(value)) {
+      for (const child of value) walk2(child, visit, stack);
+    } else if (value && typeof value.type === "string") {
+      walk2(value, visit, stack);
+    }
+  }
+  stack.pop();
+}
+function nameFor(stack) {
+  const declarator = stack[stack.length - 1];
+  if (declarator?.type !== "VariableDeclarator") return void 0;
+  const id = declarator.id;
+  if (id?.type === "Identifier") return id.name;
+  if (id?.type === "ArrayPattern") {
+    const first = id.elements?.[0];
+    if (first?.type === "Identifier") return first.name;
+  }
+  return void 0;
+}
+function componentName(node) {
+  if (node.type === "FunctionDeclaration") {
+    const name = node.id?.name;
+    return name && name[0] === name[0]?.toUpperCase() ? name : void 0;
+  }
+  if (node.type === "VariableDeclarator") {
+    const id = node.id;
+    const init = node.init;
+    if (id?.type !== "Identifier" || !init) return void 0;
+    if (init.type !== "ArrowFunctionExpression" && init.type !== "FunctionExpression") return void 0;
+    const name = id.name;
+    return name[0] === name[0]?.toUpperCase() ? name : void 0;
+  }
+  return void 0;
+}
+function collectSourceEdits(program, filename) {
+  const edits = [];
+  const claimed = /* @__PURE__ */ new Set();
+  const { primitives, shadowed } = resolveReactiveBindings(program);
+  walk2(program, (node, stack) => {
+    const component = componentName(node);
+    if (component) {
+      const at = node.loc?.start;
+      let end2 = node.end;
+      for (let i = stack.length - 1; i >= 0; i--) {
+        const up = stack[i];
+        if (up.type !== "VariableDeclaration" && up.type !== "ExportNamedDeclaration" && up.type !== "ExportDefaultDeclaration") {
+          break;
+        }
+        end2 = up.end;
+      }
+      if (at) {
+        edits.push({
+          start: end2,
+          end: end2,
+          code: `;globalThis.${COMPONENT_MARKER}?.(${JSON.stringify(component)},${JSON.stringify(filename)},${at.line},${at.column});`
+        });
+      }
+    }
+    if (node.type !== "CallExpression") return;
+    const callee = node.callee;
+    if (callee?.type !== "Identifier") return;
+    const name = callee.name;
+    const intrinsic = isIntrinsicName(name) && !shadowed.has(name) ? INTRINSICS[name] : void 0;
+    const imported = primitives.has(name) || RUNTIME_PRIMITIVES2.has(name);
+    if (!intrinsic && !imported) return;
+    const position = node.loc?.start;
+    if (!position) return;
+    let statement;
+    let parent;
+    let index = -1;
+    for (let i = stack.length - 1; i >= 0; i--) {
+      if (isStatement(stack[i]) || isExport(stack[i])) {
+        statement = stack[i];
+        parent = stack[i - 1];
+        index = i;
+        break;
+      }
+    }
+    if (!statement) return;
+    let start = statement.start;
+    let end = statement.end;
+    for (let i = index - 1; i >= 0; i--) {
+      if (!isExport(stack[i])) break;
+      start = stack[i].start;
+      end = stack[i].end;
+      parent = stack[i - 1];
+    }
+    if (claimed.has(start)) return;
+    claimed.add(start);
+    const label = nameFor(stack);
+    const args = [
+      JSON.stringify(filename),
+      String(position.line),
+      String(position.column),
+      label ? JSON.stringify(label) : void 0
+    ].filter((a) => a !== void 0).join(",");
+    const mark = `globalThis.${SOURCE_MARKER}?.(${args});`;
+    if (isLoneBranch(statement, parent)) {
+      edits.push({ start, end: start, code: `{${mark}` });
+      edits.push({ start: end, end, code: "}" });
+      return;
+    }
+    edits.push({ start, end: start, code: mark });
+  });
+  return { edits, marked: edits.length };
+}
+export {
+  COMPONENT_MARKER,
+  HOLE_MARKER,
+  PROPS_MARKER,
+  SOURCE_MARKER,
+  collectSourceEdits,
+  markHole,
+  markProps,
+  mayHaveReactiveCall
+};