@rocicorp/zero 1.6.0-canary.0 → 1.6.0-canary.10

package/out/packages/analyze-query/src/analyze-cli.js

@@ -0,0 +1,288 @@
+import { valita_exports } from "../../shared/src/valita.js";
+import { createBuilder } from "../../zql/src/query/create-builder.js";
+import { Zero } from "../../zero-client/src/client/zero.js";
+import "../../shared/src/dotenv.js";
+import { logLevel, logOptions } from "../../otel/src/log-options.js";
+import { colorConsole } from "../../shared/src/logging.js";
+import { parseOptions } from "../../shared/src/options.js";
+import { ZERO_ENV_VAR_PREFIX } from "../../zero-cache/src/config/zero-config.js";
+import { Console } from "node:console";
+import { styleText } from "node:util";
+import { WebSocket } from "ws";
+//#region ../analyze-query/src/analyze-cli.ts
+var options = {
+	zeroCacheURL: {
+		type: valita_exports.string().optional(),
+		desc: ["URL of the remote zero-cache to analyze against.", "Accepts http(s):// or ws(s):// (ws(s) is the transport actually used)."]
+	},
+	adminPassword: {
+		type: valita_exports.string().optional(),
+		desc: ["Admin password for zero-cache.", "Required when the server is configured with one; ignored in dev mode."]
+	},
+	authToken: {
+		type: valita_exports.string().optional(),
+		desc: ["Raw JWT forwarded to zero-cache.", "Used server-side to fill permission variables for the query."]
+	},
+	cookie: {
+		type: valita_exports.string().optional(),
+		desc: [
+			"Cookie header value sent on the WebSocket upgrade request,",
+			"e.g. `session=abc; foo=bar`. Use this when zero-cache is behind",
+			"a proxy that resolves auth via cookies. Merged with --headers-json",
+			"(--cookie wins on conflict)."
+		]
+	},
+	headersJson: {
+		type: valita_exports.string().optional(),
+		desc: [
+			"JSON object of arbitrary headers to send on the WebSocket upgrade",
+			"request, e.g. `{\"x-api-key\":\"...\"}`. Escape hatch for exotic auth",
+			"schemes; prefer --auth-token or --cookie when possible."
+		]
+	},
+	userId: {
+		type: valita_exports.string().optional(),
+		desc: ["Optional userID to report to zero-cache.", "Has no functional effect on analysis; defaults to \"analyze-cli\"."]
+	},
+	ast: {
+		type: valita_exports.string().optional(),
+		desc: ["JSON-encoded AST. Exactly one of --ast / --query / --query-name is required.", "The AST is sent to the server verbatim — provide it in server (post-mapping) form."]
+	},
+	query: {
+		type: valita_exports.string().optional(),
+		desc: ["ZQL query in chain form, e.g. `issue.related(\"comments\").limit(10)`.", "Evaluated against the schema you pass to runAnalyzeCLI."]
+	},
+	queryName: {
+		type: valita_exports.string().optional(),
+		desc: ["Name of a server-registered custom (named) query.", "The server resolves the name + args via its registered query handler."]
+	},
+	queryArgs: {
+		type: valita_exports.string().optional(),
+		desc: ["JSON-encoded array of arguments for --query-name. Defaults to `[]`."]
+	},
+	outputVendedRows: {
+		type: valita_exports.boolean().default(false),
+		desc: ["Include the rows read from the replica to execute the query.", "Each row appears once per read."]
+	},
+	outputSyncedRows: {
+		type: valita_exports.boolean().default(false),
+		desc: ["Include the rows that would be synced to the client."]
+	},
+	log: {
+		...logOptions,
+		level: logLevel.default("error")
+	}
+};
+var stderrConsole = new Console({
+	stdout: process.stderr,
+	stderr: process.stderr
+});
+var stderrLogSink = { log(level, context, ...args) {
+	const ctx = context ? Object.entries(context).map(([k, v]) => v === void 0 ? k : `${k}=${v}`) : [];
+	stderrConsole[level](...ctx, ...args);
+} };
+/**
+* Entry point for a user's `cli.ts`. Parses argv, connects to a remote
+* zero-cache by standing up an in-process Zero client (in-memory storage,
+* no subscriptions), calls the inspector's `analyze-query` RPC, and
+* renders the result. Intended to be called as:
+*
+* ```ts
+* import {schema} from './schema.ts';
+* import {runAnalyzeCLI} from '@rocicorp/zero/analyze';
+* await runAnalyzeCLI({schema});
+* ```
+*
+* Exits the process with code 1 on error.
+*/
+async function runAnalyzeCLI(opts) {
+	const config = parseOptions(options, {
+		argv: (opts.argv ?? process.argv.slice(2)).map((s) => s.replaceAll("\n", " ")),
+		envNamePrefix: ZERO_ENV_VAR_PREFIX,
+		description: [{
+			header: "analyze-query (remote)",
+			content: `Analyze a ZQL query against a remote zero-cache.
+
+  Connects to zero-cache's inspector protocol and reports the server-observed
+  row scans, SQLite query plans, and timings.`
+		}, {
+			header: "Examples",
+			content: `  tsx cli.ts --zero-cache-url=https://zero.example.com \\
+    --admin-password="$ZERO_ADMIN_PASSWORD" \\
+    --query='issue.related("comments").limit(10)'
+
+  tsx cli.ts --zero-cache-url=http://localhost:4848 \\
+    --ast='\\{"table": "issue", "limit": 5\\}'
+
+  tsx cli.ts --zero-cache-url=http://localhost:4848 \\
+    --query-name=issueList --query-args='[]'`
+		}]
+	});
+	if (!config.zeroCacheURL) {
+		colorConsole.error("--zero-cache-url is required. See --help for usage.");
+		process.exit(1);
+	}
+	const plan = buildQueryPlan(config);
+	const handshakeHeaders = resolveHandshakeHeaders(config);
+	if (Object.keys(handshakeHeaders).length > 0) installWebSocketHeaderShim(handshakeHeaders);
+	globalThis.TESTING ??= false;
+	const z = new Zero({
+		schema: opts.schema,
+		server: config.zeroCacheURL,
+		auth: config.authToken,
+		userID: config.userId ?? "analyze-cli",
+		kvStore: "mem",
+		logLevel: config.log.level,
+		logSink: stderrLogSink
+	});
+	let result;
+	try {
+		if (!await z.inspector.authenticate(config.adminPassword ?? "")) throw new Error("admin password rejected (or --admin-password is required)");
+		const rpcOptions = {
+			vendedRows: config.outputVendedRows,
+			syncedRows: config.outputSyncedRows
+		};
+		if (plan.kind === "ast") result = await z.inspector.analyzeServerAST(plan.ast, rpcOptions);
+		else if (plan.kind === "named") result = await z.inspector.analyzeNamedQuery(plan.name, plan.args, rpcOptions);
+		else {
+			const built = buildZqlQuery(plan.text, createBuilder(opts.schema));
+			result = await z.inspector.analyzeQuery(built, rpcOptions);
+		}
+	} catch (e) {
+		colorConsole.error(e instanceof Error ? e.message : String(e));
+		await z.close().catch(() => {});
+		process.exit(1);
+	}
+	renderResult(result, {
+		outputSyncedRows: config.outputSyncedRows,
+		outputVendedRows: config.outputVendedRows
+	});
+	await z.close();
+}
+function buildQueryPlan(config) {
+	const selectors = [
+		config.ast !== void 0 && "ast",
+		config.query !== void 0 && "query",
+		config.queryName !== void 0 && "queryName"
+	].filter(Boolean);
+	if (selectors.length === 0) {
+		colorConsole.error("Exactly one of --ast / --query / --query-name is required.");
+		process.exit(1);
+	}
+	if (selectors.length > 1) {
+		colorConsole.error(`Only one of --ast / --query / --query-name may be provided; got: ${selectors.join(", ")}`);
+		process.exit(1);
+	}
+	if (config.ast !== void 0) return {
+		kind: "ast",
+		ast: JSON.parse(config.ast)
+	};
+	if (config.query !== void 0) return {
+		kind: "zql",
+		text: config.query
+	};
+	const args = config.queryArgs ? JSON.parse(config.queryArgs) : [];
+	return {
+		kind: "named",
+		name: config.queryName,
+		args
+	};
+}
+function buildZqlQuery(queryString, builder) {
+	return new Function("builder", `return builder.${queryString};`)(builder);
+}
+function resolveHandshakeHeaders(config) {
+	const headers = {};
+	if (config.headersJson !== void 0) {
+		let parsed;
+		try {
+			parsed = JSON.parse(config.headersJson);
+		} catch (e) {
+			colorConsole.error(`--headers-json is not valid JSON: ${e instanceof Error ? e.message : String(e)}`);
+			process.exit(1);
+		}
+		if (parsed === null || typeof parsed !== "object" || Array.isArray(parsed)) {
+			colorConsole.error("--headers-json must be a JSON object.");
+			process.exit(1);
+		}
+		for (const [k, val] of Object.entries(parsed)) {
+			if (typeof val !== "string") {
+				colorConsole.error(`--headers-json values must be strings; got ${typeof val} for "${k}".`);
+				process.exit(1);
+			}
+			headers[k] = val;
+		}
+	}
+	if (config.cookie !== void 0) headers.cookie = config.cookie;
+	return headers;
+}
+function installWebSocketHeaderShim(headers) {
+	class HeaderInjectingWebSocket extends WebSocket {
+		constructor(url, protocols) {
+			super(url, protocols, { headers });
+		}
+	}
+	globalThis.WebSocket = HeaderInjectingWebSocket;
+}
+function renderResult(result, opts) {
+	if (opts.outputSyncedRows) {
+		colorConsole.log(styleText(["blue", "bold"], "=== Synced Rows: ===\n"));
+		for (const [table, rows] of Object.entries(result.syncedRows ?? {})) colorConsole.log(styleText("bold", table + ":"), rows);
+	}
+	colorConsole.log(styleText(["blue", "bold"], "=== Query Stats: ===\n"));
+	colorConsole.log(styleText("bold", "total synced rows:"), result.syncedRowCount);
+	const readRowCountsByQuery = result.readRowCountsByQuery ?? {};
+	let totalRowsRead = 0;
+	for (const table of Object.keys(readRowCountsByQuery).sort()) {
+		const counts = readRowCountsByQuery[table];
+		for (const n of Object.values(counts)) totalRowsRead += n;
+		colorConsole.log(styleText("bold", `${table} vended:`), counts);
+	}
+	colorConsole.log(styleText("bold", "Rows Read (into JS):"), colorRowsConsidered(totalRowsRead));
+	const duration = result.elapsed ?? result.end - result.start;
+	colorConsole.log(styleText("bold", "time:"), colorTime(duration), "ms");
+	if (opts.outputVendedRows) {
+		colorConsole.log(styleText(["blue", "bold"], "=== JS Row Scan Values: ===\n"));
+		for (const [table, rows] of Object.entries(result.readRows ?? {})) colorConsole.log(styleText("bold", `${table}:`), rows);
+	}
+	colorConsole.log(styleText(["blue", "bold"], "\n=== Rows Scanned (by SQLite): ===\n"));
+	const dbScansByQuery = result.dbScansByQuery ?? {};
+	let totalNVisit = 0;
+	for (const [table, queries] of Object.entries(dbScansByQuery)) {
+		colorConsole.log(styleText("bold", `${table}:`), queries);
+		for (const count of Object.values(queries)) totalNVisit += count;
+	}
+	colorConsole.log(styleText("bold", "total rows scanned:"), colorRowsConsidered(totalNVisit));
+	colorConsole.log(styleText(["blue", "bold"], "\n\n=== Query Plans: ===\n"));
+	const plans = result.sqlitePlans ?? {};
+	for (const [query, plan] of Object.entries(plans)) {
+		colorConsole.log(styleText("bold", "query"), query);
+		colorConsole.log(plan.map((row, i) => colorPlanRow(row, i)).join("\n"));
+		colorConsole.log("\n");
+	}
+	if (result.warnings.length > 0) {
+		colorConsole.log(styleText(["yellow", "bold"], "=== Warnings: ===\n"));
+		for (const w of result.warnings) colorConsole.log(styleText("yellow", w));
+	}
+}
+function colorTime(duration) {
+	if (duration < 100) return styleText("green", duration.toFixed(2) + "ms");
+	else if (duration < 1e3) return styleText("yellow", duration.toFixed(2) + "ms");
+	return styleText("red", duration.toFixed(2) + "ms");
+}
+function colorRowsConsidered(n) {
+	if (n < 1e3) return styleText("green", n.toString());
+	else if (n < 1e4) return styleText("yellow", n.toString());
+	return styleText("red", n.toString());
+}
+function colorPlanRow(row, i) {
+	if (row.includes("SCAN")) {
+		if (i === 0) return styleText("yellow", row);
+		return styleText("red", row);
+	}
+	return styleText("green", row);
+}
+//#endregion
+export { runAnalyzeCLI };
+
+//# sourceMappingURL=analyze-cli.js.map

package/out/packages/analyze-query/src/analyze-cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"analyze-cli.js","names":[],"sources":["../../../../../analyze-query/src/analyze-cli.ts"],"sourcesContent":["import '../../shared/src/dotenv.ts';\n\nimport {Console} from 'node:console';\nimport {styleText} from 'node:util';\nimport type {LogSink} from '@rocicorp/logger';\nimport {WebSocket as NodeWebSocket} from 'ws';\nimport {logLevel, logOptions} from '../../otel/src/log-options.ts';\nimport {colorConsole} from '../../shared/src/logging.ts';\nimport {parseOptions} from '../../shared/src/options.ts';\nimport * as v from '../../shared/src/valita.ts';\nimport {ZERO_ENV_VAR_PREFIX} from '../../zero-cache/src/config/zero-config.ts';\nimport {Zero} from '../../zero-client/src/client/zero.ts';\nimport type {AnalyzeQueryResult} from '../../zero-protocol/src/analyze-query-result.ts';\nimport type {AST} from '../../zero-protocol/src/ast.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport {createBuilder} from '../../zql/src/query/create-builder.ts';\nimport type {AnyQuery} from '../../zql/src/query/query.ts';\nimport type {SchemaQuery} from '../../zql/src/query/schema-query.ts';\n\nexport type AnalyzeCLIOptions = {\n  schema: Schema;\n  /** Defaults to `process.argv.slice(2)`. */\n  argv?: readonly string[] | undefined;\n};\n\nconst options = {\n  zeroCacheURL: {\n    type: v.string().optional(),\n    desc: [\n      'URL of the remote zero-cache to analyze against.',\n      'Accepts http(s):// or ws(s):// (ws(s) is the transport actually used).',\n    ],\n  },\n  adminPassword: {\n    type: v.string().optional(),\n    desc: [\n      'Admin password for zero-cache.',\n      'Required when the server is configured with one; ignored in dev mode.',\n    ],\n  },\n  authToken: {\n    type: v.string().optional(),\n    desc: [\n      'Raw JWT forwarded to zero-cache.',\n      'Used server-side to fill permission variables for the query.',\n    ],\n  },\n  cookie: {\n    type: v.string().optional(),\n    desc: [\n      'Cookie header value sent on the WebSocket upgrade request,',\n      'e.g. `session=abc; foo=bar`. Use this when zero-cache is behind',\n      'a proxy that resolves auth via cookies. Merged with --headers-json',\n      '(--cookie wins on conflict).',\n    ],\n  },\n  headersJson: {\n    type: v.string().optional(),\n    desc: [\n      'JSON object of arbitrary headers to send on the WebSocket upgrade',\n      'request, e.g. `{\"x-api-key\":\"...\"}`. Escape hatch for exotic auth',\n      'schemes; prefer --auth-token or --cookie when possible.',\n    ],\n  },\n  userId: {\n    type: v.string().optional(),\n    desc: [\n      'Optional userID to report to zero-cache.',\n      'Has no functional effect on analysis; defaults to \"analyze-cli\".',\n    ],\n  },\n  ast: {\n    type: v.string().optional(),\n    desc: [\n      'JSON-encoded AST. Exactly one of --ast / --query / --query-name is required.',\n      'The AST is sent to the server verbatim — provide it in server (post-mapping) form.',\n    ],\n  },\n  query: {\n    type: v.string().optional(),\n    desc: [\n      'ZQL query in chain form, e.g. `issue.related(\"comments\").limit(10)`.',\n      'Evaluated against the schema you pass to runAnalyzeCLI.',\n    ],\n  },\n  queryName: {\n    type: v.string().optional(),\n    desc: [\n      'Name of a server-registered custom (named) query.',\n      'The server resolves the name + args via its registered query handler.',\n    ],\n  },\n  queryArgs: {\n    type: v.string().optional(),\n    desc: [\n      'JSON-encoded array of arguments for --query-name. Defaults to `[]`.',\n    ],\n  },\n  outputVendedRows: {\n    type: v.boolean().default(false),\n    desc: [\n      'Include the rows read from the replica to execute the query.',\n      'Each row appears once per read.',\n    ],\n  },\n  outputSyncedRows: {\n    type: v.boolean().default(false),\n    desc: ['Include the rows that would be synced to the client.'],\n  },\n  log: {\n    ...logOptions,\n    level: logLevel.default('error'),\n  },\n};\n\ntype QueryPlan =\n  | {kind: 'ast'; ast: AST}\n  | {kind: 'zql'; text: string}\n  | {kind: 'named'; name: string; args: ReadonlyArray<unknown>};\n\n// Route all Zero client log output to stderr so stdout contains only the\n// analyze result. Shell redirection (`2>/dev/null`) can then cleanly silence\n// logs without affecting output.\nconst stderrConsole = new Console({\n  stdout: process.stderr,\n  stderr: process.stderr,\n});\nconst stderrLogSink: LogSink = {\n  log(level, context, ...args) {\n    const ctx = context\n      ? Object.entries(context).map(([k, v]) =>\n          v === undefined ? k : `${k}=${v}`,\n        )\n      : [];\n    stderrConsole[level](...ctx, ...args);\n  },\n};\n\n/**\n * Entry point for a user's `cli.ts`. Parses argv, connects to a remote\n * zero-cache by standing up an in-process Zero client (in-memory storage,\n * no subscriptions), calls the inspector's `analyze-query` RPC, and\n * renders the result. Intended to be called as:\n *\n * ```ts\n * import {schema} from './schema.ts';\n * import {runAnalyzeCLI} from '@rocicorp/zero/analyze';\n * await runAnalyzeCLI({schema});\n * ```\n *\n * Exits the process with code 1 on error.\n */\nexport async function runAnalyzeCLI(opts: AnalyzeCLIOptions): Promise<void> {\n  const argv = (opts.argv ?? process.argv.slice(2)).map(s =>\n    s.replaceAll('\\n', ' '),\n  );\n\n  const config = parseOptions(options, {\n    argv,\n    envNamePrefix: ZERO_ENV_VAR_PREFIX,\n    description: [\n      {\n        header: 'analyze-query (remote)',\n        content: `Analyze a ZQL query against a remote zero-cache.\n\n  Connects to zero-cache's inspector protocol and reports the server-observed\n  row scans, SQLite query plans, and timings.`,\n      },\n      {\n        header: 'Examples',\n        content: `  tsx cli.ts --zero-cache-url=https://zero.example.com \\\\\n    --admin-password=\"$ZERO_ADMIN_PASSWORD\" \\\\\n    --query='issue.related(\"comments\").limit(10)'\n\n  tsx cli.ts --zero-cache-url=http://localhost:4848 \\\\\n    --ast='\\\\{\"table\": \"issue\", \"limit\": 5\\\\}'\n\n  tsx cli.ts --zero-cache-url=http://localhost:4848 \\\\\n    --query-name=issueList --query-args='[]'`,\n      },\n    ],\n  });\n\n  if (!config.zeroCacheURL) {\n    colorConsole.error('--zero-cache-url is required. See --help for usage.');\n    process.exit(1);\n  }\n\n  const plan = buildQueryPlan(config);\n\n  const handshakeHeaders = resolveHandshakeHeaders(config);\n  if (Object.keys(handshakeHeaders).length > 0) {\n    installWebSocketHeaderShim(handshakeHeaders);\n  }\n\n  // zero-client and replicache reference a build-time `TESTING` global that\n  // bundlers replace with a boolean literal; under tsx there's no replacement,\n  // so provide a runtime default.\n  (globalThis as {TESTING?: boolean}).TESTING ??= false;\n\n  const z = new Zero({\n    schema: opts.schema,\n    server: config.zeroCacheURL,\n    auth: config.authToken,\n    userID: config.userId ?? 'analyze-cli',\n    kvStore: 'mem',\n    logLevel: config.log.level,\n    logSink: stderrLogSink,\n  });\n\n  let result: AnalyzeQueryResult;\n  try {\n    const authOk = await z.inspector.authenticate(config.adminPassword ?? '');\n    if (!authOk) {\n      throw new Error(\n        'admin password rejected (or --admin-password is required)',\n      );\n    }\n\n    const rpcOptions = {\n      vendedRows: config.outputVendedRows,\n      syncedRows: config.outputSyncedRows,\n    };\n\n    if (plan.kind === 'ast') {\n      result = await z.inspector.analyzeServerAST(plan.ast, rpcOptions);\n    } else if (plan.kind === 'named') {\n      result = await z.inspector.analyzeNamedQuery(\n        plan.name,\n        plan.args as ReadonlyArray<never>,\n        rpcOptions,\n      );\n    } else {\n      const built = buildZqlQuery(plan.text, createBuilder(opts.schema));\n      result = await z.inspector.analyzeQuery(built, rpcOptions);\n    }\n  } catch (e) {\n    colorConsole.error(e instanceof Error ? e.message : String(e));\n    await z.close().catch(() => {});\n    process.exit(1);\n  }\n\n  renderResult(result, {\n    outputSyncedRows: config.outputSyncedRows,\n    outputVendedRows: config.outputVendedRows,\n  });\n\n  await z.close();\n}\n\nfunction buildQueryPlan(config: {\n  ast?: string | undefined;\n  query?: string | undefined;\n  queryName?: string | undefined;\n  queryArgs?: string | undefined;\n}): QueryPlan {\n  const selectors = [\n    config.ast !== undefined && 'ast',\n    config.query !== undefined && 'query',\n    config.queryName !== undefined && 'queryName',\n  ].filter(Boolean) as string[];\n\n  if (selectors.length === 0) {\n    colorConsole.error(\n      'Exactly one of --ast / --query / --query-name is required.',\n    );\n    process.exit(1);\n  }\n  if (selectors.length > 1) {\n    colorConsole.error(\n      `Only one of --ast / --query / --query-name may be provided; got: ${selectors.join(', ')}`,\n    );\n    process.exit(1);\n  }\n\n  if (config.ast !== undefined) {\n    return {kind: 'ast', ast: JSON.parse(config.ast) as AST};\n  }\n  if (config.query !== undefined) {\n    return {kind: 'zql', text: config.query};\n  }\n  const args = config.queryArgs\n    ? (JSON.parse(config.queryArgs) as ReadonlyArray<unknown>)\n    : [];\n  return {kind: 'named', name: config.queryName as string, args};\n}\n\nfunction buildZqlQuery(\n  queryString: string,\n  builder: SchemaQuery<Schema>,\n): AnyQuery {\n  const f = new Function('builder', `return builder.${queryString};`);\n  return f(builder) as AnyQuery;\n}\n\nfunction resolveHandshakeHeaders(config: {\n  cookie?: string | undefined;\n  headersJson?: string | undefined;\n}): Record<string, string> {\n  const headers: Record<string, string> = {};\n  if (config.headersJson !== undefined) {\n    let parsed: unknown;\n    try {\n      parsed = JSON.parse(config.headersJson);\n    } catch (e) {\n      colorConsole.error(\n        `--headers-json is not valid JSON: ${e instanceof Error ? e.message : String(e)}`,\n      );\n      process.exit(1);\n    }\n    if (\n      parsed === null ||\n      typeof parsed !== 'object' ||\n      Array.isArray(parsed)\n    ) {\n      colorConsole.error('--headers-json must be a JSON object.');\n      process.exit(1);\n    }\n    for (const [k, val] of Object.entries(parsed)) {\n      if (typeof val !== 'string') {\n        colorConsole.error(\n          `--headers-json values must be strings; got ${typeof val} for \"${k}\".`,\n        );\n        process.exit(1);\n      }\n      headers[k] = val;\n    }\n  }\n  if (config.cookie !== undefined) {\n    headers.cookie = config.cookie;\n  }\n  return headers;\n}\n\nfunction installWebSocketHeaderShim(headers: Record<string, string>): void {\n  class HeaderInjectingWebSocket extends NodeWebSocket {\n    constructor(url: string | URL, protocols?: string | string[]) {\n      super(url, protocols, {headers});\n    }\n  }\n  (globalThis as {WebSocket?: unknown}).WebSocket = HeaderInjectingWebSocket;\n}\n\nfunction renderResult(\n  result: AnalyzeQueryResult,\n  opts: {outputSyncedRows: boolean; outputVendedRows: boolean},\n) {\n  if (opts.outputSyncedRows) {\n    colorConsole.log(styleText(['blue', 'bold'], '=== Synced Rows: ===\\n'));\n    for (const [table, rows] of Object.entries(result.syncedRows ?? {})) {\n      colorConsole.log(styleText('bold', table + ':'), rows);\n    }\n  }\n\n  colorConsole.log(styleText(['blue', 'bold'], '=== Query Stats: ===\\n'));\n  colorConsole.log(\n    styleText('bold', 'total synced rows:'),\n    result.syncedRowCount,\n  );\n\n  const readRowCountsByQuery = result.readRowCountsByQuery ?? {};\n  let totalRowsRead = 0;\n  for (const table of Object.keys(readRowCountsByQuery).sort()) {\n    const counts = readRowCountsByQuery[table];\n    for (const n of Object.values(counts)) {\n      totalRowsRead += n;\n    }\n    colorConsole.log(styleText('bold', `${table} vended:`), counts);\n  }\n  colorConsole.log(\n    styleText('bold', 'Rows Read (into JS):'),\n    colorRowsConsidered(totalRowsRead),\n  );\n  const duration = result.elapsed ?? result.end - result.start;\n  colorConsole.log(styleText('bold', 'time:'), colorTime(duration), 'ms');\n\n  if (opts.outputVendedRows) {\n    colorConsole.log(\n      styleText(['blue', 'bold'], '=== JS Row Scan Values: ===\\n'),\n    );\n    for (const [table, rows] of Object.entries(result.readRows ?? {})) {\n      colorConsole.log(styleText('bold', `${table}:`), rows);\n    }\n  }\n\n  colorConsole.log(\n    styleText(['blue', 'bold'], '\\n=== Rows Scanned (by SQLite): ===\\n'),\n  );\n  const dbScansByQuery = result.dbScansByQuery ?? {};\n  let totalNVisit = 0;\n  for (const [table, queries] of Object.entries(dbScansByQuery)) {\n    colorConsole.log(styleText('bold', `${table}:`), queries);\n    for (const count of Object.values(queries)) {\n      totalNVisit += count;\n    }\n  }\n  colorConsole.log(\n    styleText('bold', 'total rows scanned:'),\n    colorRowsConsidered(totalNVisit),\n  );\n\n  colorConsole.log(styleText(['blue', 'bold'], '\\n\\n=== Query Plans: ===\\n'));\n  const plans = result.sqlitePlans ?? {};\n  for (const [query, plan] of Object.entries(plans)) {\n    colorConsole.log(styleText('bold', 'query'), query);\n    colorConsole.log(plan.map((row, i) => colorPlanRow(row, i)).join('\\n'));\n    colorConsole.log('\\n');\n  }\n\n  if (result.warnings.length > 0) {\n    colorConsole.log(styleText(['yellow', 'bold'], '=== Warnings: ===\\n'));\n    for (const w of result.warnings) {\n      colorConsole.log(styleText('yellow', w));\n    }\n  }\n}\n\nfunction colorTime(duration: number) {\n  if (duration < 100) {\n    return styleText('green', duration.toFixed(2) + 'ms');\n  } else if (duration < 1000) {\n    return styleText('yellow', duration.toFixed(2) + 'ms');\n  }\n  return styleText('red', duration.toFixed(2) + 'ms');\n}\n\nfunction colorRowsConsidered(n: number) {\n  if (n < 1000) {\n    return styleText('green', n.toString());\n  } else if (n < 10000) {\n    return styleText('yellow', n.toString());\n  }\n  return styleText('red', n.toString());\n}\n\nfunction colorPlanRow(row: string, i: number) {\n  if (row.includes('SCAN')) {\n    if (i === 0) {\n      return styleText('yellow', row);\n    }\n    return styleText('red', row);\n  }\n  return styleText('green', row);\n}\n"],"mappings":";;;;;;;;;;;;AAyBA,IAAM,UAAU;CACd,cAAc;EACZ,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,oDACA,wEACF;CACF;CACA,eAAe;EACb,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,kCACA,uEACF;CACF;CACA,WAAW;EACT,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,oCACA,8DACF;CACF;CACA,QAAQ;EACN,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM;GACJ;GACA;GACA;GACA;EACF;CACF;CACA,aAAa;EACX,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM;GACJ;GACA;GACA;EACF;CACF;CACA,QAAQ;EACN,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,4CACA,oEACF;CACF;CACA,KAAK;EACH,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,gFACA,oFACF;CACF;CACA,OAAO;EACL,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,0EACA,yDACF;CACF;CACA,WAAW;EACT,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,qDACA,uEACF;CACF;CACA,WAAW;EACT,MAAM,eAAE,OAAO,EAAE,SAAS;EAC1B,MAAM,CACJ,qEACF;CACF;CACA,kBAAkB;EAChB,MAAM,eAAE,QAAQ,EAAE,QAAQ,KAAK;EAC/B,MAAM,CACJ,gEACA,iCACF;CACF;CACA,kBAAkB;EAChB,MAAM,eAAE,QAAQ,EAAE,QAAQ,KAAK;EAC/B,MAAM,CAAC,sDAAsD;CAC/D;CACA,KAAK;EACH,GAAG;EACH,OAAO,SAAS,QAAQ,OAAO;CACjC;AACF;AAUA,IAAM,gBAAgB,IAAI,QAAQ;CAChC,QAAQ,QAAQ;CAChB,QAAQ,QAAQ;AAClB,CAAC;AACD,IAAM,gBAAyB,EAC7B,IAAI,OAAO,SAAS,GAAG,MAAM;CAC3B,MAAM,MAAM,UACR,OAAO,QAAQ,OAAO,EAAE,KAAK,CAAC,GAAG,OAC/B,MAAM,KAAA,IAAY,IAAI,GAAG,EAAE,GAAG,GAChC,IACA,CAAC;CACL,cAAc,OAAO,GAAG,KAAK,GAAG,IAAI;AACtC,EACF;;;;;;;;;;;;;;;AAgBA,eAAsB,cAAc,MAAwC;CAK1E,MAAM,SAAS,aAAa,SAAS;EACnC,OALY,KAAK,QAAQ,QAAQ,KAAK,MAAM,CAAC,GAAG,KAAI,MACpD,EAAE,WAAW,MAAM,GAAG,CAItB;EACA,eAAe;EACf,aAAa,CACX;GACE,QAAQ;GACR,SAAS;;;;EAIX,GACA;GACE,QAAQ;GACR,SAAS;;;;;;;;;EASX,CACF;CACF,CAAC;CAED,IAAI,CAAC,OAAO,cAAc;EACxB,aAAa,MAAM,qDAAqD;EACxE,QAAQ,KAAK,CAAC;CAChB;CAEA,MAAM,OAAO,eAAe,MAAM;CAElC,MAAM,mBAAmB,wBAAwB,MAAM;CACvD,IAAI,OAAO,KAAK,gBAAgB,EAAE,SAAS,GACzC,2BAA2B,gBAAgB;CAM7C,WAAoC,YAAY;CAEhD,MAAM,IAAI,IAAI,KAAK;EACjB,QAAQ,KAAK;EACb,QAAQ,OAAO;EACf,MAAM,OAAO;EACb,QAAQ,OAAO,UAAU;EACzB,SAAS;EACT,UAAU,OAAO,IAAI;EACrB,SAAS;CACX,CAAC;CAED,IAAI;CACJ,IAAI;EAEF,IAAI,CAAC,MADgB,EAAE,UAAU,aAAa,OAAO,iBAAiB,EAAE,GAEtE,MAAM,IAAI,MACR,2DACF;EAGF,MAAM,aAAa;GACjB,YAAY,OAAO;GACnB,YAAY,OAAO;EACrB;EAEA,IAAI,KAAK,SAAS,OAChB,SAAS,MAAM,EAAE,UAAU,iBAAiB,KAAK,KAAK,UAAU;OAC3D,IAAI,KAAK,SAAS,SACvB,SAAS,MAAM,EAAE,UAAU,kBACzB,KAAK,MACL,KAAK,MACL,UACF;OACK;GACL,MAAM,QAAQ,cAAc,KAAK,MAAM,cAAc,KAAK,MAAM,CAAC;GACjE,SAAS,MAAM,EAAE,UAAU,aAAa,OAAO,UAAU;EAC3D;CACF,SAAS,GAAG;EACV,aAAa,MAAM,aAAa,QAAQ,EAAE,UAAU,OAAO,CAAC,CAAC;EAC7D,MAAM,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;EAC9B,QAAQ,KAAK,CAAC;CAChB;CAEA,aAAa,QAAQ;EACnB,kBAAkB,OAAO;EACzB,kBAAkB,OAAO;CAC3B,CAAC;CAED,MAAM,EAAE,MAAM;AAChB;AAEA,SAAS,eAAe,QAKV;CACZ,MAAM,YAAY;EAChB,OAAO,QAAQ,KAAA,KAAa;EAC5B,OAAO,UAAU,KAAA,KAAa;EAC9B,OAAO,cAAc,KAAA,KAAa;CACpC,EAAE,OAAO,OAAO;CAEhB,IAAI,UAAU,WAAW,GAAG;EAC1B,aAAa,MACX,4DACF;EACA,QAAQ,KAAK,CAAC;CAChB;CACA,IAAI,UAAU,SAAS,GAAG;EACxB,aAAa,MACX,oEAAoE,UAAU,KAAK,IAAI,GACzF;EACA,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI,OAAO,QAAQ,KAAA,GACjB,OAAO;EAAC,MAAM;EAAO,KAAK,KAAK,MAAM,OAAO,GAAG;CAAQ;CAEzD,IAAI,OAAO,UAAU,KAAA,GACnB,OAAO;EAAC,MAAM;EAAO,MAAM,OAAO;CAAK;CAEzC,MAAM,OAAO,OAAO,YACf,KAAK,MAAM,OAAO,SAAS,IAC5B,CAAC;CACL,OAAO;EAAC,MAAM;EAAS,MAAM,OAAO;EAAqB;CAAI;AAC/D;AAEA,SAAS,cACP,aACA,SACU;CAEV,OAAO,IADO,SAAS,WAAW,kBAAkB,YAAY,EACzD,EAAE,OAAO;AAClB;AAEA,SAAS,wBAAwB,QAGN;CACzB,MAAM,UAAkC,CAAC;CACzC,IAAI,OAAO,gBAAgB,KAAA,GAAW;EACpC,IAAI;EACJ,IAAI;GACF,SAAS,KAAK,MAAM,OAAO,WAAW;EACxC,SAAS,GAAG;GACV,aAAa,MACX,qCAAqC,aAAa,QAAQ,EAAE,UAAU,OAAO,CAAC,GAChF;GACA,QAAQ,KAAK,CAAC;EAChB;EACA,IACE,WAAW,QACX,OAAO,WAAW,YAClB,MAAM,QAAQ,MAAM,GACpB;GACA,aAAa,MAAM,uCAAuC;GAC1D,QAAQ,KAAK,CAAC;EAChB;EACA,KAAK,MAAM,CAAC,GAAG,QAAQ,OAAO,QAAQ,MAAM,GAAG;GAC7C,IAAI,OAAO,QAAQ,UAAU;IAC3B,aAAa,MACX,8CAA8C,OAAO,IAAI,QAAQ,EAAE,GACrE;IACA,QAAQ,KAAK,CAAC;GAChB;GACA,QAAQ,KAAK;EACf;CACF;CACA,IAAI,OAAO,WAAW,KAAA,GACpB,QAAQ,SAAS,OAAO;CAE1B,OAAO;AACT;AAEA,SAAS,2BAA2B,SAAuC;CACzE,MAAM,iCAAiC,UAAc;EACnD,YAAY,KAAmB,WAA+B;GAC5D,MAAM,KAAK,WAAW,EAAC,QAAO,CAAC;EACjC;CACF;CACA,WAAsC,YAAY;AACpD;AAEA,SAAS,aACP,QACA,MACA;CACA,IAAI,KAAK,kBAAkB;EACzB,aAAa,IAAI,UAAU,CAAC,QAAQ,MAAM,GAAG,wBAAwB,CAAC;EACtE,KAAK,MAAM,CAAC,OAAO,SAAS,OAAO,QAAQ,OAAO,cAAc,CAAC,CAAC,GAChE,aAAa,IAAI,UAAU,QAAQ,QAAQ,GAAG,GAAG,IAAI;CAEzD;CAEA,aAAa,IAAI,UAAU,CAAC,QAAQ,MAAM,GAAG,wBAAwB,CAAC;CACtE,aAAa,IACX,UAAU,QAAQ,oBAAoB,GACtC,OAAO,cACT;CAEA,MAAM,uBAAuB,OAAO,wBAAwB,CAAC;CAC7D,IAAI,gBAAgB;CACpB,KAAK,MAAM,SAAS,OAAO,KAAK,oBAAoB,EAAE,KAAK,GAAG;EAC5D,MAAM,SAAS,qBAAqB;EACpC,KAAK,MAAM,KAAK,OAAO,OAAO,MAAM,GAClC,iBAAiB;EAEnB,aAAa,IAAI,UAAU,QAAQ,GAAG,MAAM,SAAS,GAAG,MAAM;CAChE;CACA,aAAa,IACX,UAAU,QAAQ,sBAAsB,GACxC,oBAAoB,aAAa,CACnC;CACA,MAAM,WAAW,OAAO,WAAW,OAAO,MAAM,OAAO;CACvD,aAAa,IAAI,UAAU,QAAQ,OAAO,GAAG,UAAU,QAAQ,GAAG,IAAI;CAEtE,IAAI,KAAK,kBAAkB;EACzB,aAAa,IACX,UAAU,CAAC,QAAQ,MAAM,GAAG,+BAA+B,CAC7D;EACA,KAAK,MAAM,CAAC,OAAO,SAAS,OAAO,QAAQ,OAAO,YAAY,CAAC,CAAC,GAC9D,aAAa,IAAI,UAAU,QAAQ,GAAG,MAAM,EAAE,GAAG,IAAI;CAEzD;CAEA,aAAa,IACX,UAAU,CAAC,QAAQ,MAAM,GAAG,uCAAuC,CACrE;CACA,MAAM,iBAAiB,OAAO,kBAAkB,CAAC;CACjD,IAAI,cAAc;CAClB,KAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QAAQ,cAAc,GAAG;EAC7D,aAAa,IAAI,UAAU,QAAQ,GAAG,MAAM,EAAE,GAAG,OAAO;EACxD,KAAK,MAAM,SAAS,OAAO,OAAO,OAAO,GACvC,eAAe;CAEnB;CACA,aAAa,IACX,UAAU,QAAQ,qBAAqB,GACvC,oBAAoB,WAAW,CACjC;CAEA,aAAa,IAAI,UAAU,CAAC,QAAQ,MAAM,GAAG,4BAA4B,CAAC;CAC1E,MAAM,QAAQ,OAAO,eAAe,CAAC;CACrC,KAAK,MAAM,CAAC,OAAO,SAAS,OAAO,QAAQ,KAAK,GAAG;EACjD,aAAa,IAAI,UAAU,QAAQ,OAAO,GAAG,KAAK;EAClD,aAAa,IAAI,KAAK,KAAK,KAAK,MAAM,aAAa,KAAK,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC;EACtE,aAAa,IAAI,IAAI;CACvB;CAEA,IAAI,OAAO,SAAS,SAAS,GAAG;EAC9B,aAAa,IAAI,UAAU,CAAC,UAAU,MAAM,GAAG,qBAAqB,CAAC;EACrE,KAAK,MAAM,KAAK,OAAO,UACrB,aAAa,IAAI,UAAU,UAAU,CAAC,CAAC;CAE3C;AACF;AAEA,SAAS,UAAU,UAAkB;CACnC,IAAI,WAAW,KACb,OAAO,UAAU,SAAS,SAAS,QAAQ,CAAC,IAAI,IAAI;MAC/C,IAAI,WAAW,KACpB,OAAO,UAAU,UAAU,SAAS,QAAQ,CAAC,IAAI,IAAI;CAEvD,OAAO,UAAU,OAAO,SAAS,QAAQ,CAAC,IAAI,IAAI;AACpD;AAEA,SAAS,oBAAoB,GAAW;CACtC,IAAI,IAAI,KACN,OAAO,UAAU,SAAS,EAAE,SAAS,CAAC;MACjC,IAAI,IAAI,KACb,OAAO,UAAU,UAAU,EAAE,SAAS,CAAC;CAEzC,OAAO,UAAU,OAAO,EAAE,SAAS,CAAC;AACtC;AAEA,SAAS,aAAa,KAAa,GAAW;CAC5C,IAAI,IAAI,SAAS,MAAM,GAAG;EACxB,IAAI,MAAM,GACR,OAAO,UAAU,UAAU,GAAG;EAEhC,OAAO,UAAU,OAAO,GAAG;CAC7B;CACA,OAAO,UAAU,SAAS,GAAG;AAC/B"}

package/out/packages/analyze-query/src/bin-analyze.js

@@ -0,0 +1,282 @@
+import { valita_exports } from "../../shared/src/valita.js";
+import { must } from "../../shared/src/must.js";
+import "../../zero-protocol/src/ast.js";
+import { clientSchemaFrom } from "../../zero-schema/src/builder/schema-builder.js";
+import { asQueryInternals } from "../../zql/src/query/query-internals.js";
+import { newQuery } from "../../zql/src/query/query-impl.js";
+import { clientToServer } from "../../zero-schema/src/name-mapper.js";
+import "../../zql/src/query/query.js";
+import { QueryDelegateBase } from "../../zql/src/query/query-delegate-base.js";
+import "../../shared/src/dotenv.js";
+import { logLevel, logOptions } from "../../otel/src/log-options.js";
+import { colorConsole, createLogContext } from "../../shared/src/logging.js";
+import { parseOptions } from "../../shared/src/options.js";
+import { Debug, runtimeDebugFlags } from "../../zql/src/builder/debug-delegate.js";
+import { getShardID, upstreamSchema } from "../../zero-cache/src/types/shards.js";
+import { ZERO_ENV_VAR_PREFIX, appOptions, shardOptions, zeroOptions } from "../../zero-cache/src/config/zero-config.js";
+import { Database } from "../../zqlite/src/db.js";
+import { TableSource } from "../../zqlite/src/table-source.js";
+import { astToZQL } from "../../ast-to-zql/src/ast-to-zql.js";
+import { formatOutput } from "../../ast-to-zql/src/format.js";
+import { computeZqlSpecs, mustGetTableSpec } from "../../zero-cache/src/db/lite-tables.js";
+import { deployPermissionsOptions, loadSchemaAndPermissions } from "../../zero-cache/src/scripts/permissions.js";
+import { runAst } from "../../zero-cache/src/services/run-ast.js";
+import { pgClient } from "../../zero-cache/src/types/pg.js";
+import { explainQueries } from "../../zqlite/src/explain-queries.js";
+import { styleText } from "node:util";
+import fs from "node:fs";
+//#region ../analyze-query/src/bin-analyze.ts
+var cfg = parseOptions({
+	schema: deployPermissionsOptions.schema,
+	replicaFile: {
+		...zeroOptions.replica.file,
+		desc: ["File path to the SQLite replica to test queries against."]
+	},
+	ast: {
+		type: valita_exports.string().optional(),
+		desc: ["AST for the query to be analyzed.  Only one of ast/query/hash should be provided."]
+	},
+	query: {
+		type: valita_exports.string().optional(),
+		desc: ["Query to be analyzed in the form of: table.where(...).related(...).etc. ", "Only one of ast/query/hash should be provided."]
+	},
+	hash: {
+		type: valita_exports.string().optional(),
+		desc: [
+			"Hash of the query to be analyzed. This is used to look up the query in the database. ",
+			"Only one of ast/query/hash should be provided.",
+			"You should run this script from the directory containing your .env file to reduce the amount of",
+			"configuration required. The .env file should contain the connection URL to the CVR database."
+		]
+	},
+	applyPermissions: {
+		type: valita_exports.boolean().default(false),
+		desc: ["Whether to apply permissions (from your schema file) to the provided query."]
+	},
+	authData: {
+		type: valita_exports.string().optional(),
+		desc: ["JSON encoded payload of the auth data.", "This will be used to fill permission variables if the \"applyPermissions\" option is set"]
+	},
+	outputVendedRows: {
+		type: valita_exports.boolean().default(false),
+		desc: ["Whether to output the rows which were read from the replica in order to execute the analyzed query. ", "If the same row is read more than once it will be logged once for each time it was read."]
+	},
+	outputSyncedRows: {
+		type: valita_exports.boolean().default(false),
+		desc: ["Whether to output the rows which would be synced to the client for the analyzed query."]
+	},
+	cvr: { db: {
+		type: valita_exports.string().optional(),
+		desc: ["Connection URL to the CVR database. If using --hash, either this or --upstream-db", "must be specified."]
+	} },
+	upstream: {
+		db: {
+			desc: [`Connection URL to the "upstream" authoritative postgres database. If using --hash, `, "either this or --cvr-db must be specified."],
+			type: valita_exports.string().optional()
+		},
+		type: zeroOptions.upstream.type
+	},
+	app: appOptions,
+	shard: shardOptions,
+	log: {
+		...logOptions,
+		level: logLevel.default("error")
+	}
+}, {
+	argv: process.argv.slice(2).map((s) => s.replaceAll("\n", " ")),
+	envNamePrefix: ZERO_ENV_VAR_PREFIX,
+	description: [{
+		header: "analyze-query",
+		content: `Analyze a ZQL query and show information about how it runs against a SQLite replica.
+
+  analyze-query uses the same environment variables and flags as zero-cache-dev. If run from your development environment, it will pick up your ZERO_REPLICA_FILE, ZERO_SCHEMA_PATH, and other env vars automatically.
+
+  If run in another environment (e.g., production) you will have to specify these flags. In particular, you must have a copy of the appropriate Zero schema file to give to the --schema-path flag.`
+	}, {
+		header: "Examples",
+		content: `# In development
+  npx analyze-query --query='issue.related("comments").limit(10)'
+  npx analyze-query --ast='\\{"table": "artist","limit": 10\\}'
+  npx analyze-query --hash=1234567890
+
+  # In production
+  # First copy schema.ts to your production environment, then run:
+  npx analyze-query \\
+    --schema-path='./schema.ts' \\
+    --replica-file='/path/to/replica.db' \\
+    --query='issue.related("comments").limit(10)'
+
+  npx analyze-query \\
+    --schema-path='./schema.ts' \\
+    --replica-file='/path/to/replica.db' \\
+    --ast='\\{"table": "artist","limit": 10\\}'
+
+  # cvr-db is required when using the hash option.
+  # It is typically the same as your upstream db.
+  npx analyze-query \\
+    --schema-path='./schema.ts' \\
+    --replica-file='/path/to/replica.db' \\
+    --cvr-db='postgres://user:pass@host:port/db' \\
+    --hash=1234567890
+  `
+	}]
+});
+var config = {
+	...cfg,
+	cvr: {
+		...cfg.cvr,
+		db: cfg.cvr.db ?? cfg.upstream.db
+	}
+};
+runtimeDebugFlags.trackRowCountsVended = true;
+runtimeDebugFlags.trackRowsVended = config.outputVendedRows;
+var lc = createLogContext({ log: config.log });
+colorConsole.warn("DEPRECATED: `analyze-query` is deprecated. Please migrate to a project-specific analyze command built with `runAnalyzeCLI` from `@rocicorp/zero/analyze`.");
+if (!fs.existsSync(config.replicaFile)) {
+	colorConsole.error(`Replica file ${config.replicaFile} does not exist`);
+	process.exit(1);
+}
+var db = new Database(lc, config.replicaFile);
+var { schema, permissions } = await loadSchemaAndPermissions(config.schema.path);
+var clientSchema = clientSchemaFrom(schema).clientSchema;
+var sources = /* @__PURE__ */ new Map();
+var clientToServerMapper = clientToServer(schema.tables);
+var debug = new Debug();
+var tableSpecs = computeZqlSpecs(lc, db, { includeBackfillingColumns: false });
+var AnalyzeQueryDelegate = class extends QueryDelegateBase {
+	debug = debug;
+	defaultQueryComplete = true;
+	getSource(serverTableName) {
+		let source = sources.get(serverTableName);
+		if (source) return source;
+		const tableSpec = mustGetTableSpec(tableSpecs, serverTableName);
+		const { primaryKey } = tableSpec.tableSpec;
+		source = new TableSource(lc, config.log, db, serverTableName, tableSpec.zqlSpec, primaryKey);
+		sources.set(serverTableName, source);
+		return source;
+	}
+};
+var host = new AnalyzeQueryDelegate();
+var result;
+if (config.ast) result = await runAst(lc, clientSchema, JSON.parse(config.ast), true, {
+	applyPermissions: config.applyPermissions,
+	auth: config.authData ? {
+		type: "jwt",
+		raw: "",
+		decoded: JSON.parse(config.authData)
+	} : void 0,
+	clientToServerMapper,
+	permissions,
+	syncedRows: config.outputSyncedRows,
+	db,
+	tableSpecs,
+	host
+}, async () => {});
+else if (config.query) result = await runQuery(config.query);
+else if (config.hash) result = await runHash(config.hash);
+else {
+	colorConsole.error("No query or AST or hash provided");
+	process.exit(1);
+}
+function runQuery(queryString) {
+	const z = { query: Object.fromEntries(Object.entries(schema.tables).map(([name]) => [name, newQuery(schema, name)])) };
+	const ast = asQueryInternals(new Function("z", `return z.query.${queryString};`)(z)).ast;
+	return runAst(lc, clientSchema, ast, false, {
+		applyPermissions: config.applyPermissions,
+		auth: config.authData ? {
+			type: "jwt",
+			raw: "",
+			decoded: JSON.parse(config.authData)
+		} : void 0,
+		clientToServerMapper,
+		permissions,
+		syncedRows: config.outputSyncedRows,
+		db,
+		tableSpecs,
+		host
+	}, async () => {});
+}
+async function runHash(hash) {
+	const cvrDB = pgClient(lc, must(config.cvr.db, "CVR DB must be provided when using the hash option"), "analyze-query-hash-cvr");
+	const rows = await cvrDB`select "clientAST", "internal" from ${cvrDB(upstreamSchema(getShardID(config)) + "/cvr")}."queries" where "queryHash" = ${must(hash)} limit 1;`;
+	await cvrDB.end();
+	colorConsole.log("ZQL from Hash:");
+	const ast = rows[0].clientAST;
+	colorConsole.log(await formatOutput(ast.table + astToZQL(ast)));
+	return runAst(lc, clientSchema, ast, true, {
+		applyPermissions: config.applyPermissions,
+		auth: config.authData ? {
+			type: "jwt",
+			raw: "",
+			decoded: JSON.parse(config.authData)
+		} : void 0,
+		clientToServerMapper,
+		permissions,
+		syncedRows: config.outputSyncedRows,
+		db,
+		tableSpecs,
+		host
+	}, async () => {});
+}
+if (config.outputSyncedRows) {
+	colorConsole.log(styleText(["blue", "bold"], "=== Synced Rows: ===\n"));
+	for (const [table, rows] of Object.entries(result.syncedRows ?? {})) colorConsole.log(styleText("bold", table + ":"), rows);
+}
+colorConsole.log(styleText(["blue", "bold"], "=== Query Stats: ===\n"));
+colorConsole.log(styleText("bold", "total synced rows:"), result.syncedRowCount);
+showStats();
+if (config.outputVendedRows) {
+	colorConsole.log(styleText(["blue", "bold"], "=== JS Row Scan Values: ===\n"));
+	for (const source of sources.values()) colorConsole.log(styleText("bold", `${source.tableSchema.name}:`), debug.getVendedRows()?.[source.tableSchema.name] ?? {});
+}
+colorConsole.log(styleText(["blue", "bold"], "\n=== Rows Scanned (by SQLite): ===\n"));
+var nvisitCounts = debug.getNVisitCounts();
+var totalNVisit = 0;
+for (const [table, queries] of Object.entries(nvisitCounts)) {
+	colorConsole.log(styleText("bold", `${table}:`), queries);
+	for (const count of Object.values(queries)) totalNVisit += count;
+}
+colorConsole.log(styleText("bold", "total rows scanned:"), colorRowsConsidered(totalNVisit));
+colorConsole.log(styleText(["blue", "bold"], "\n\n=== Query Plans: ===\n"));
+var fallbackPlans = explainQueries(debug.getVendedRowCounts() ?? {}, db);
+var capturedPlans = debug.getSQLitePlans();
+var plans = {
+	...fallbackPlans,
+	...capturedPlans
+};
+for (const [query, plan] of Object.entries(plans)) {
+	colorConsole.log(styleText("bold", "query"), query);
+	colorConsole.log(plan.map((row, i) => colorPlanRow(row, i)).join("\n"));
+	colorConsole.log("\n");
+}
+function showStats() {
+	let totalRowsConsidered = 0;
+	for (const source of sources.values()) {
+		const values = Object.values(debug.getVendedRowCounts()?.[source.tableSchema.name] ?? {});
+		for (const v of values) totalRowsConsidered += v;
+		colorConsole.log(styleText("bold", source.tableSchema.name + " vended:"), debug.getVendedRowCounts()?.[source.tableSchema.name] ?? {});
+	}
+	colorConsole.log(styleText("bold", "Rows Read (into JS):"), colorRowsConsidered(totalRowsConsidered));
+	colorConsole.log(styleText("bold", "time:"), colorTime(result.end - result.start), "ms");
+}
+function colorTime(duration) {
+	if (duration < 100) return styleText("green", duration.toFixed(2) + "ms");
+	else if (duration < 1e3) return styleText("yellow", duration.toFixed(2) + "ms");
+	return styleText("red", duration.toFixed(2) + "ms");
+}
+function colorRowsConsidered(n) {
+	if (n < 1e3) return styleText("green", n.toString());
+	else if (n < 1e4) return styleText("yellow", n.toString());
+	return styleText("red", n.toString());
+}
+function colorPlanRow(row, i) {
+	if (row.includes("SCAN")) {
+		if (i === 0) return styleText("yellow", row);
+		return styleText("red", row);
+	}
+	return styleText("green", row);
+}
+//#endregion
+export {};
+
+//# sourceMappingURL=bin-analyze.js.map