@lunora/codegen 1.0.0-alpha.5 → 1.0.0-alpha.51

package/dist/packem_shared/{discoverFunctions-DEgAcRuD.mjs → discoverFunctions-BJ-qR7Rg.mjs}

@@ -1,11 +1,12 @@
 import { readdirSync, lstatSync } from 'node:fs';
 import { join, extname, relative, sep } from 'node:path';
 import { Node, SyntaxKind } from 'ts-morph';
-import { a as parseObjectShape } from './parse-validator-tuQtHrsr.mjs';
+import { a as parseObjectShape } from './parse-validator-BSJo1HGP.mjs';
 import { s as sanitizeNamespace } from './paths-BRd6JHuF.mjs';
 
 const FUNCTION_KINDS = /* @__PURE__ */ new Set(["action", "mutation", "query", "stream"]);
-const ANY_TOKEN_RE = /\bany\b/u;
+const ANY_TOKEN_RE = /\bany\b(?!\s*(?:\?\s*)?:)/u;
+const STRING_LITERAL_SPAN_RE = /"(?:[^"\\]|\\.)*"|'(?:[^'\\]|\\.)*'|`(?:[^`\\]|\\.)*`/gu;
 const IDENTIFIER_RE = /^[A-Za-z_$][\w$]*$/u;
 const renderExpandedPropertyKey = (propertyName) => IDENTIFIER_RE.test(propertyName) ? propertyName : JSON.stringify(propertyName);
 const TS_EXTENSION_RE = /\.ts$/u;
@@ -221,7 +222,7 @@ const unwrapHandlerReturn = (handler) => {
   if (!rendered || rendered === "any" || rendered === "never") {
     return "unknown";
   }
-  if (ANY_TOKEN_RE.test(rendered)) {
+  if (ANY_TOKEN_RE.test(rendered.replaceAll(STRING_LITERAL_SPAN_RE, ""))) {
     return "unknown";
   }
   const handlerFilePath = handler.getSourceFile().getFilePath();
@@ -290,24 +291,73 @@ const classifyProcedureCall = (call) => {
   if (!Node.isIdentifier(callee)) {
     return void 0;
   }
-  const calleeName = resolveCalleeKind(callee);
-  if (!calleeName) {
+  const calleeName2 = resolveCalleeKind(callee);
+  if (!calleeName2) {
     return void 0;
   }
-  if (FUNCTION_KINDS.has(calleeName)) {
-    return { kind: calleeName, visibility: "public" };
+  if (FUNCTION_KINDS.has(calleeName2)) {
+    return { kind: calleeName2, visibility: "public" };
   }
-  const internalKind = INTERNAL_FACTORIES[calleeName];
+  const internalKind = INTERNAL_FACTORIES[calleeName2];
   if (internalKind) {
     return { kind: internalKind, visibility: "internal" };
   }
-  const lifecycle = LIFECYCLE_FACTORIES[calleeName];
+  const lifecycle = LIFECYCLE_FACTORIES[calleeName2];
   if (lifecycle) {
     return { kind: "mutation", lifecycle, visibility: "internal" };
   }
   return void 0;
 };
-const listLunoraSourceFiles = (directory, accumulator = []) => {
+const inlineHandler = (argument) => argument !== void 0 && (Node.isArrowFunction(argument) || Node.isFunctionExpression(argument)) ? argument : void 0;
+const isDatabaseAccessor = (receiver) => Node.isPropertyAccessExpression(receiver) && receiver.getName() === "db" || Node.isIdentifier(receiver) && receiver.getText() === "db";
+const procedureHandler = (initializer) => {
+  const argument = initializer.getArguments()[0];
+  const direct = inlineHandler(argument);
+  if (direct !== void 0) {
+    return direct;
+  }
+  if (argument === void 0 || !Node.isObjectLiteralExpression(argument)) {
+    return void 0;
+  }
+  const property = argument.getProperty("handler");
+  return property !== void 0 && Node.isPropertyAssignment(property) ? inlineHandler(property.getInitializer()) : void 0;
+};
+const calleeName = (callee) => {
+  if (Node.isIdentifier(callee)) {
+    return callee.getText();
+  }
+  return Node.isPropertyAccessExpression(callee) ? callee.getName() : "";
+};
+const chainHasStep = (receiver, method) => {
+  let node = receiver;
+  while (Node.isCallExpression(node)) {
+    const callee = node.getExpression();
+    if (!Node.isPropertyAccessExpression(callee)) {
+      break;
+    }
+    if (callee.getName() === method) {
+      return true;
+    }
+    node = callee.getExpression();
+  }
+  return false;
+};
+const chainUsesWrappedCall = (receiver, method, wrappedCallee) => {
+  let node = receiver;
+  while (Node.isCallExpression(node)) {
+    const callee = node.getExpression();
+    if (!Node.isPropertyAccessExpression(callee)) {
+      break;
+    }
+    const argument = node.getArguments()[0];
+    if (callee.getName() === method && argument !== void 0 && Node.isCallExpression(argument) && calleeName(argument.getExpression()) === wrappedCallee) {
+      return true;
+    }
+    node = callee.getExpression();
+  }
+  return false;
+};
+const listLunoraSourceFiles = (directory, accumulator = [], root = directory) => {
   let entries;
   try {
     entries = readdirSync(directory);
@@ -321,8 +371,11 @@ const listLunoraSourceFiles = (directory, accumulator = []) => {
       if (entry === "_generated" || entry === "node_modules") {
         continue;
       }
-      listLunoraSourceFiles(full, accumulator);
-    } else if (info.isFile() && extname(entry) === ".ts" && entry !== "schema.ts") {
+      listLunoraSourceFiles(full, accumulator, root);
+    } else if (info.isFile() && extname(entry) === ".ts") {
+      if (entry === "schema.ts" && directory === root) {
+        continue;
+      }
       accumulator.push(full);
     }
   }
@@ -344,7 +397,12 @@ const discoverFromCall = (call) => {
   if (classified.lifecycle) {
     return { args: {}, kind: classified.kind, lifecycle: classified.lifecycle, returnType: "void", visibility: classified.visibility };
   }
-  return { args: argsFromCall(call), kind: classified.kind, returnType: returnTypeFromCall(call), visibility: classified.visibility };
+  return {
+    args: argsFromCall(call),
+    kind: classified.kind,
+    returnType: returnTypeFromCall(call),
+    visibility: classified.visibility
+  };
 };
 const RE_EXPORT_RESOLVE_LIMIT = 8;
 const resolveExpressionToCall = (node, depth = 0) => {
@@ -457,4 +515,4 @@ const discoverFunctions = (project, lunoraDirectory) => {
   return functions;
 };
 
-export { classifyProcedureCall, discoverFunctions, listLunoraSourceFiles, lunoraRelativePath };
+export { chainHasStep, chainUsesWrappedCall, classifyProcedureCall, discoverFunctions, inlineHandler, isDatabaseAccessor, listLunoraSourceFiles, lunoraRelativePath, procedureHandler, unwrapHandlerReturn };

package/dist/packem_shared/{discoverHttpRoutes-C978pBiG.mjs → discoverHttpRoutes-daCzuTe8.mjs}

@@ -1,7 +1,7 @@
 import { relative, sep } from 'node:path';
 import { Node } from 'ts-morph';
-import { listLunoraSourceFiles } from './discoverFunctions-DEgAcRuD.mjs';
-import { p as parseValidator, a as parseObjectShape } from './parse-validator-tuQtHrsr.mjs';
+import { listLunoraSourceFiles, unwrapHandlerReturn } from './discoverFunctions-BJ-qR7Rg.mjs';
+import { p as parseValidator, a as parseObjectShape } from './parse-validator-BSJo1HGP.mjs';
 
 const TS_EXTENSION_RE = /\.ts$/u;
 const HTTP_VERBS = /* @__PURE__ */ new Set(["delete", "get", "head", "options", "patch", "post", "put"]);
@@ -72,6 +72,13 @@ const walkRouteChain = (terminalCall, terminalStep) => {
   state.path = root.path;
   return state;
 };
+const chunkTypeFromStreamTerminal = (call) => {
+  const handler = call.getArguments()[0];
+  if (!handler || !(Node.isArrowFunction(handler) || Node.isFunctionExpression(handler))) {
+    return "unknown";
+  }
+  return unwrapHandlerReturn(handler);
+};
 const routeFromTerminal = (call, callee, exportName, filePath) => {
   const step = callee.getName();
   if (!TERMINAL_STEPS.has(step)) {
@@ -90,6 +97,7 @@ const routeFromTerminal = (call, callee, exportName, filePath) => {
     path: state.path,
     searchParams: state.searchParams,
     stream: state.stream,
+    ...state.stream ? { chunkType: chunkTypeFromStreamTerminal(call) } : {},
     ...state.output ? { output: state.output } : {}
   };
 };

package/dist/packem_shared/{discoverInserts-CRQdXvHO.mjs → discoverInserts-DI4q5NxE.mjs}

@@ -1,6 +1,6 @@
 import { SyntaxKind, Node } from 'ts-morph';
 import { e as enclosingExportName } from './discover-ast-CT6BgBr4.mjs';
-import { listLunoraSourceFiles, lunoraRelativePath } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const isDatabaseInsertCall = (call) => {
   const callee = call.getExpression();
@@ -13,9 +13,31 @@ const isDatabaseInsertCall = (call) => {
   }
   return Node.isIdentifier(receiver) && receiver.getText() === "db";
 };
+const resolveStringConst = (identifier) => {
+  if (!Node.isIdentifier(identifier)) {
+    return void 0;
+  }
+  const symbol = identifier.getSymbol();
+  const declarations = symbol?.getAliasedSymbol()?.getDeclarations() ?? symbol?.getDeclarations() ?? [];
+  for (const declaration of declarations) {
+    if (Node.isVariableDeclaration(declaration)) {
+      const initializer = declaration.getInitializer();
+      if (initializer && Node.isStringLiteral(initializer)) {
+        return initializer.getLiteralText();
+      }
+    }
+  }
+  return void 0;
+};
 const tableOf = (call) => {
   const argument = call.getArguments()[0];
-  return argument && Node.isStringLiteral(argument) ? argument.getLiteralText() : "";
+  if (!argument) {
+    return "";
+  }
+  if (Node.isStringLiteral(argument)) {
+    return argument.getLiteralText();
+  }
+  return resolveStringConst(argument) ?? "";
 };
 const discoverInserts = (project, lunoraDirectory) => {
   const writes = [];

package/dist/packem_shared/{discoverMaskProcedures-B64zA740.mjs → discoverMaskProcedures-BcTOEKNU.mjs}

@@ -1,5 +1,5 @@
 import { Node, SyntaxKind } from 'ts-morph';
-import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const isMaskCall = (node) => {
   if (!Node.isCallExpression(node)) {
@@ -213,5 +213,61 @@ const discoverMaskMetadata = (project, lunoraDirectory) => {
   }
   return { columns: [...columnsByKey.values()] };
 };
+const extractMaskStrategyRows = (maskCall, exportName, relativePath) => {
+  const argument = maskCall.getArguments()[0];
+  if (!argument || !Node.isObjectLiteralExpression(argument)) {
+    return [];
+  }
+  const rows = [];
+  for (const tableProperty of argument.getProperties()) {
+    const table = memberName(tableProperty);
+    if (table === void 0 || !Node.isPropertyAssignment(tableProperty)) {
+      continue;
+    }
+    const initializer = tableProperty.getInitializer();
+    if (!initializer || !Node.isObjectLiteralExpression(initializer)) {
+      continue;
+    }
+    for (const columnProperty of initializer.getProperties()) {
+      const column = memberName(columnProperty);
+      if (column === void 0) {
+        continue;
+      }
+      const strategy = strategyOf(columnProperty);
+      if (strategy === "custom") {
+        continue;
+      }
+      rows.push({ column, exportName, file: relativePath, line: columnProperty.getStartLineNumber(), strategy, table });
+    }
+  }
+  return rows;
+};
+const maskStrategyRowsFromDeclaration = (declaration, relativePath) => {
+  if (!Node.isVariableDeclaration(declaration)) {
+    return [];
+  }
+  const initializer = declaration.getInitializer();
+  const classified = initializer && Node.isCallExpression(initializer) ? classifyProcedureCall(initializer) : void 0;
+  if (!classified?.receiver) {
+    return [];
+  }
+  return maskCallsInChain(classified.receiver).flatMap((maskCall) => extractMaskStrategyRows(maskCall, declaration.getName(), relativePath));
+};
+const discoverMaskStrategies = (project, lunoraDirectory) => {
+  const rows = [];
+  for (const filePath of listLunoraSourceFiles(lunoraDirectory)) {
+    const sourceFile = project.getSourceFile(filePath) ?? project.addSourceFileAtPath(filePath);
+    const relativePath = lunoraRelativePath(lunoraDirectory, filePath);
+    for (const statement of sourceFile.getVariableStatements()) {
+      if (!statement.isExported()) {
+        continue;
+      }
+      for (const declaration of statement.getDeclarations()) {
+        rows.push(...maskStrategyRowsFromDeclaration(declaration, relativePath));
+      }
+    }
+  }
+  return rows;
+};
 
-export { discoverMaskProcedures as default, discoverMaskMetadata };
+export { discoverMaskProcedures as default, discoverMaskMetadata, discoverMaskStrategies };

package/dist/packem_shared/{discoverMigrations-Doj_-BAA.mjs → discoverMigrations-VNUFvCwr.mjs}

@@ -1,6 +1,7 @@
 import { relative, sep } from 'node:path';
 import { SyntaxKind, Node } from 'ts-morph';
-import { listLunoraSourceFiles } from './discoverFunctions-DEgAcRuD.mjs';
+import { diagnosticAt } from './CodegenDiagnosticError-DyQ5FwkM.mjs';
+import { listLunoraSourceFiles } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const TS_EXTENSION_RE = /\.ts$/u;
 const isDefineMigration = (identifier) => {
@@ -43,9 +44,14 @@ const migrationFromDeclaration = (declaration, relativePath) => {
   const id = stringProperty(argument, "id");
   const exportName = declaration.getName();
   if (id === void 0 || id.trim() === "") {
-    throw Object.assign(
-      new Error(`Migration "${exportName}" in "${relativePath}" must declare \`id\` as a non-empty string literal so codegen can key the registry.`),
-      { code: "MIGRATION_ID_NOT_STATIC", name: "LunoraError", status: 500 }
+    throw diagnosticAt(
+      argument,
+      `Migration "${exportName}" in "${relativePath}" must declare \`id\` as a non-empty string literal so codegen can key the registry.`,
+      {
+        code: "MIGRATION_ID_NOT_STATIC",
+        name: "LunoraError",
+        status: 500
+      }
     );
   }
   return { exportName, filePath: relativePath, id, table: stringProperty(argument, "table") ?? "" };

package/dist/packem_shared/{discoverNondeterministicCalls-4KiPQxQU.mjs → discoverNondeterministicCalls-S0N2xLCq.mjs}

@@ -1,5 +1,5 @@
 import { Node, SyntaxKind } from 'ts-morph';
-import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const PROPERTY_CALLS = {
   "crypto.getRandomValues": "crypto.getRandomValues",

package/dist/packem_shared/{discoverQueries-BkIi0dBD.mjs → discoverQueries-CJnnnLpd.mjs}

@@ -1,5 +1,5 @@
 import { SyntaxKind, Node } from 'ts-morph';
-import { listLunoraSourceFiles, lunoraRelativePath } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const INDEX_METHODS = /* @__PURE__ */ new Set(["withIndex", "withSearchIndex"]);
 const isDatabaseQueryCall = (call) => {

package/dist/packem_shared/{discoverR2sqlCalls-BpDqvcUn.mjs → discoverR2sqlCalls-pnpicWfz.mjs}

@@ -1,5 +1,5 @@
 import { Node, SyntaxKind } from 'ts-morph';
-import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const handlerOf = (call, receiver) => {
   if (receiver) {

package/dist/packem_shared/{discoverRlsMetadata-DpRB1HMe.mjs → discoverRlsMetadata-DppniPUH.mjs}

@@ -1,5 +1,5 @@
 import { Node, SyntaxKind } from 'ts-morph';
-import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-DEgAcRuD.mjs';
+import { listLunoraSourceFiles, lunoraRelativePath, classifyProcedureCall } from './discoverFunctions-BJ-qR7Rg.mjs';
 
 const isRlsCall = (node) => {
   if (!Node.isCallExpression(node)) {

package/dist/packem_shared/discoverSandboxUsage-BUM2r90k.mjs

@@ -0,0 +1,38 @@
+import { listLunoraSourceFiles } from './discoverFunctions-BJ-qR7Rg.mjs';
+
+const SANDBOX_MODULE_SPECIFIERS = /* @__PURE__ */ new Set(["@lunora/agent", "@lunora/agent/sandbox"]);
+const scanImportDeclaration = (declaration) => {
+  const found = { usesSandboxBrowser: false, usesSandboxContainer: false };
+  if (!SANDBOX_MODULE_SPECIFIERS.has(declaration.getModuleSpecifierValue()) || declaration.isTypeOnly()) {
+    return found;
+  }
+  for (const named of declaration.getNamedImports()) {
+    if (named.isTypeOnly()) {
+      continue;
+    }
+    const name = named.getNameNode().getText();
+    if (name === "browserTool") {
+      found.usesSandboxBrowser = true;
+    } else if (name === "containerTool") {
+      found.usesSandboxContainer = true;
+    }
+  }
+  return found;
+};
+const discoverSandboxUsage = (project, lunoraDirectory) => {
+  const usage = { usesSandboxBrowser: false, usesSandboxContainer: false };
+  for (const filePath of listLunoraSourceFiles(lunoraDirectory)) {
+    const sourceFile = project.getSourceFile(filePath) ?? project.addSourceFileAtPath(filePath);
+    for (const declaration of sourceFile.getImportDeclarations()) {
+      const found = scanImportDeclaration(declaration);
+      usage.usesSandboxBrowser ||= found.usesSandboxBrowser;
+      usage.usesSandboxContainer ||= found.usesSandboxContainer;
+    }
+    if (usage.usesSandboxBrowser && usage.usesSandboxContainer) {
+      break;
+    }
+  }
+  return usage;
+};
+
+export { discoverSandboxUsage };