Function extractStatements

packages/lang-core/src/parser/parser.ts:89–148

* Extract typed statements from the symbol table. * State defaults are materialized to plain values (no raw AST in output).

(
  stmts: Statement[],
  ctx: MaterializeCtx,
)

 * State defaults are materialized to plain values (no raw AST in output).
 */
function extractStatements(
  stmts: Statement[],
  ctx: MaterializeCtx,
): {
  stateDeclarations: Record<string, unknown>;
  queryStatements: QueryStatementInfo[];
  mutationStatements: MutationStatementInfo[];
} {
  const stateDeclarations: Record<string, unknown> = {};
  const queryStatements: QueryStatementInfo[] = [];
  const mutationStatements: MutationStatementInfo[] = [];

  for (const stmt of stmts) {
    switch (stmt.kind) {
      case "state":
        stateDeclarations[stmt.id] = materializeValue(stmt.init, ctx);
        break;
      case "query":
        queryStatements.push({
          statementId: stmt.id,
          toolAST: stmt.call.args[0] ?? null,
          argsAST: stmt.call.args[1] ?? null,
          defaultsAST: stmt.call.args[2] ?? null,
          refreshAST: stmt.call.args[3] ?? null,
          deps: stmt.deps,
          complete: true,
        });
        break;
      case "mutation":
        mutationStatements.push({
          statementId: stmt.id,
          toolAST: stmt.call.args[0] ?? null,
          argsAST: stmt.call.args[1] ?? null,
        });
        break;
    }
  }

  // Auto-declare: any $var referenced in code but not explicitly declared → null default.
  // collectQueryDeps already walks AST for StateRef nodes — reuse it here.
  for (const stmt of stmts) {
    const nodes =
      stmt.kind === "state"
        ? [stmt.init]
        : stmt.kind === "value"
          ? [stmt.expr]
          : stmt.kind === "query" || stmt.kind === "mutation"
            ? stmt.call.args
            : [];
    for (const node of nodes) {
      for (const dep of collectQueryDeps(node)) {
        if (!(dep in stateDeclarations)) {
          stateDeclarations[dep] = null;
        }
      }
    }
  }