@fluixi/compiler 1.0.0-alpha.83 → 1.0.0-alpha.85

package/dist/transform/source-locations.mjs

@@ -1 +1,307 @@
-var u="@fluixi/reactive/signal",d={$signal:{export:"signal",module:u,returns:"signal-handle"},$memo:{export:"memo",module:u,returns:"memo-handle"},$effect:{export:"effect",module:u,returns:"effect"},$store:{export:"store",module:"@fluixi/reactive/store",returns:"store-handle"},$resource:{export:"resource",module:u,returns:"resource-handle"},$selector:{export:"createSelector",module:u,returns:"memo-accessor"},$deferred:{export:"createDeferred",module:u,returns:"memo-accessor"},$untrack:{export:"untrack",module:u,returns:"plain"},$untrackStore:{export:"untrackStore",module:"@fluixi/reactive/store",returns:"plain"}},D={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"},F=["@fluixi/reactive","@fluixi/reactive/signal","@fluixi/reactive/store","@fluixi/core"],z=new RegExp(`\\$(?:${Object.keys(d).map(e=>e.slice(1)).join("|")})\\s*\\(`);function M(e){return Object.prototype.hasOwnProperty.call(d,e)}function _(e){return F.includes(e)}function c(e,n){if(e)switch(e.type){case"Identifier":n.add(e.name);return;case"ObjectPattern":for(let r of e.properties??[])r.type==="RestElement"?c(r.argument,n):c(r.value,n);return;case"ArrayPattern":for(let r of e.elements??[])c(r,n);return;case"AssignmentPattern":c(e.left,n);return;case"RestElement":c(e.argument,n);return}}function N(e,n){if(!(!e||typeof e!="object"||typeof e.type!="string")){n(e);for(let r of Object.keys(e)){if(r==="loc"||r==="leadingComments"||r==="trailingComments")continue;let t=e[r];if(Array.isArray(t))for(let i of t)N(i,n);else t&&typeof t=="object"&&N(t,n)}}}function j(e,n){if(e.type!=="CallExpression")return;let r=e.callee;if(r?.type!=="Identifier")return;let t=r.name,i=n.primitives.get(t);if(i)return i;let o=d[t];if(o&&!n.shadowed.has(t))return o.returns}function V(e,n,r){if(Array.isArray(n)){e.type==="ArrayPattern"&&e.elements.forEach((t,i)=>{let o=n[i];t?.type==="Identifier"&&o&&r.set(t.name,o)});return}e.type==="Identifier"&&r.set(e.name,n)}function A(e){let n={kinds:new Map,primitives:new Map,shadowed:new Set},r=e;N(r,t=>{let i=new Set;if(t.type==="VariableDeclarator")c(t.id,i);else if(t.type==="FunctionDeclaration"||t.type==="ClassDeclaration")c(t.id,i);else if(t.type==="ImportDefaultSpecifier"||t.type==="ImportNamespaceSpecifier")c(t.local,i);else if(t.type==="ImportSpecifier")c(t.local,i);else if(t.type==="FunctionExpression"||t.type==="ArrowFunctionExpression"||t.type==="FunctionDeclaration")for(let o of t.params??[])c(o,i);for(let o of i)d[o]&&n.shadowed.add(o)});for(let t of r.body??[])if(t.type==="ImportDeclaration"&&_(t.source?.value))for(let i of t.specifiers??[]){if(i.type!=="ImportSpecifier")continue;let o=i.imported,l=o.type==="Identifier"?o.name:o.value,a=D[l];a&&n.primitives.set(i.local.name,a)}return N(r,t=>{if(t.type!=="VariableDeclarator"||!t.init)return;let i=j(t.init,n);i&&V(t.id,i,n.kinds)}),n}var K="__fx_source",L="__fx_component",B="__fx_hole",te=(e,n,r,t)=>`(globalThis.${B} ?? ((f) => f))(${e},${n},${r}${t===void 0?"":`,${JSON.stringify(t)}`})`,U="__fx_props",ne=(e,n)=>`(globalThis.${U} ?? ((p) => p))(${e},${JSON.stringify(n)})`,C=new Set(["signal","memo","effect","store","resource","watch","createSignal","createMemo","createEffect","createStore","createResource"]);function re(e){for(let n of C)if(e.includes(n))return!0;return e.includes("$signal")||e.includes("$memo")||e.includes("$effect")||e.includes("$store")}var J=new Set(["IfStatement","ForStatement","ForInStatement","ForOfStatement","WhileStatement","DoWhileStatement","LabeledStatement","WithStatement"]);function H(e,n){return!n||e.type==="BlockStatement"?!1:J.has(n.type)}function T(e){return e.type==="ExportNamedDeclaration"||e.type==="ExportDefaultDeclaration"}function W(e){return e.type.endsWith("Statement")||e.type==="VariableDeclaration"}function h(e,n,r=[]){if(!(!e||typeof e.type!="string")){n(e,r),r.push(e);for(let t of Object.keys(e)){if(t==="loc"||t==="parent")continue;let i=e[t];if(Array.isArray(i))for(let o of i)h(o,n,r);else i&&typeof i.type=="string"&&h(i,n,r)}r.pop()}}function G(e){let n=e[e.length-1];if(n?.type!=="VariableDeclarator")return;let r=n.id;if(r?.type==="Identifier")return r.name;if(r?.type==="ArrayPattern"){let t=r.elements?.[0];if(t?.type==="Identifier")return t.name}}function q(e){if(e.type==="FunctionDeclaration"){let n=e.id?.name;return n&&n[0]===n[0]?.toUpperCase()?n:void 0}if(e.type==="VariableDeclarator"){let n=e.id,r=e.init;if(n?.type!=="Identifier"||!r||r.type!=="ArrowFunctionExpression"&&r.type!=="FunctionExpression")return;let t=n.name;return t[0]===t[0]?.toUpperCase()?t:void 0}}function ie(e,n){let r=[],t=new Set,{primitives:i,shadowed:o}=A(e);return h(e,(l,a)=>{let E=q(l);if(E){let s=l.loc?.start,I=l.end;for(let R=a.length-1;R>=0;R--){let g=a[R];if(g.type!=="VariableDeclaration"&&g.type!=="ExportNamedDeclaration"&&g.type!=="ExportDefaultDeclaration")break;I=g.end}s&&r.push({start:I,end:I,code:`;globalThis.${L}?.(${JSON.stringify(E)},${JSON.stringify(n)},${s.line},${s.column});`})}if(l.type!=="CallExpression")return;let b=l.callee;if(b?.type!=="Identifier")return;let p=b.name,O=M(p)&&!o.has(p)?d[p]:void 0,P=i.has(p)||C.has(p);if(!O&&!P)return;let y=l.loc?.start;if(!y)return;let m,S,v=-1;for(let s=a.length-1;s>=0;s--)if(W(a[s])||T(a[s])){m=a[s],S=a[s-1],v=s;break}if(!m)return;let f=m.start,x=m.end;for(let s=v-1;s>=0&&T(a[s]);s--)f=a[s].start,x=a[s].end,S=a[s-1];if(t.has(f))return;t.add(f);let $=G(a),k=[JSON.stringify(n),String(y.line),String(y.column),$?JSON.stringify($):void 0].filter(s=>s!==void 0).join(","),w=`globalThis.${K}?.(${k});`;if(H(m,S)){r.push({start:f,end:f,code:`{${w}`}),r.push({start:x,end:x,code:"}"});return}r.push({start:f,end:f,code:w})}),{edits:r,marked:r.length}}export{L as COMPONENT_MARKER,B as HOLE_MARKER,U as PROPS_MARKER,K as SOURCE_MARKER,ie as collectSourceEdits,te as markHole,ne as markProps,re as mayHaveReactiveCall};
+// 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
+};