@query-doctor/core 0.8.1 → 0.8.2-rc.1

package/dist/sql/builder.d.mts

@@ -0,0 +1,37 @@
+'use client';
+
+import { PgIdentifier } from "./pg-identifier.mjs";
+
+//#region src/sql/builder.d.ts
+type PostgresQueryBuilderCommand = "bitmapscan" | "indexscan" | "seqscan";
+declare class PostgresQueryBuilder {
+  private query;
+  private readonly commands;
+  private isIntrospection;
+  private explainFlags;
+  private _preamble;
+  private parameters;
+  private limitSubstitution?;
+  constructor(query: string);
+  get preamble(): number;
+  static createIndex(definition: string, name?: PgIdentifier): PostgresQueryBuilder;
+  enable(command: PostgresQueryBuilderCommand, value?: boolean): this;
+  withQuery(query: string): this;
+  introspect(): this;
+  explain(flags: string[]): this;
+  parameterize(parameters: Record<Parameter, number>): this;
+  replaceLimit(limit: number): this;
+  build(): string;
+  /** Return the "set a=b" parts of the command in the query separate from the explain select ... part */
+  buildParts(): {
+    commands: string;
+    query: string;
+  };
+  private generateSetCommands;
+  private generateExplain;
+  private substituteQuery;
+}
+type Parameter = `$${string}`;
+//#endregion
+export { Parameter, PostgresQueryBuilder, PostgresQueryBuilderCommand };
+//# sourceMappingURL=builder.d.mts.map

package/dist/sql/builder.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"builder.d.mts","names":[],"sources":["../../src/sql/builder.ts"],"mappings":";;;;;KAEY,2BAAA;AAAA,cAKC,oBAAA;EAAA,QASS,KAAA;EAAA,iBARH,QAAA;EAAA,QACT,eAAA;EAAA,QACA,YAAA;EAAA,QACA,SAAA;EAAA,QACA,UAAA;EAAA,QAEA,iBAAA;cAEY,KAAA;EAAA,IAEhB,QAAA,CAAA;EAAA,OAIG,WAAA,CAAY,UAAA,UAAoB,IAAA,GAAO,YAAA,GAAY,oBAAA;EAS1D,MAAA,CAAO,OAAA,EAAS,2BAAA,EAA6B,KAAA;EAU7C,SAAA,CAAU,KAAA;EAKV,UAAA,CAAA;EAKA,OAAA,CAAQ,KAAA;EAKR,YAAA,CAAa,UAAA,EAAY,MAAA,CAAO,SAAA;EAKhC,YAAA,CAAa,KAAA;EAKb,KAAA,CAAA;EA1DiB;EAoEjB,UAAA,CAAA;;;;UAUQ,mBAAA;EAAA,QASA,eAAA;EAAA,QAaA,eAAA;AAAA;AAAA,KAeE,SAAA"}

package/dist/sql/builder.mjs

@@ -0,0 +1,94 @@
+"use client";
+import { _defineProperty } from "../_virtual/_@oxc-project_runtime@0.122.0/helpers/defineProperty.mjs";
+//#region src/sql/builder.ts
+var PostgresQueryBuilder = class PostgresQueryBuilder {
+	constructor(query) {
+		this.query = query;
+		_defineProperty(this, "commands", {});
+		_defineProperty(this, "isIntrospection", false);
+		_defineProperty(this, "explainFlags", []);
+		_defineProperty(this, "_preamble", 0);
+		_defineProperty(this, "parameters", {});
+		_defineProperty(this, "limitSubstitution", void 0);
+	}
+	get preamble() {
+		return this._preamble;
+	}
+	static createIndex(definition, name) {
+		if (name) return new PostgresQueryBuilder(`create index "${name}" on ${definition};`);
+		return new PostgresQueryBuilder(`create index on ${definition};`);
+	}
+	enable(command, value = true) {
+		const commandString = `enable_${command}`;
+		if (value) this.commands[commandString] = "on";
+		else this.commands[commandString] = "off";
+		return this;
+	}
+	withQuery(query) {
+		this.query = query;
+		return this;
+	}
+	introspect() {
+		this.isIntrospection = true;
+		return this;
+	}
+	explain(flags) {
+		this.explainFlags = flags;
+		return this;
+	}
+	parameterize(parameters) {
+		Object.assign(this.parameters, parameters);
+		return this;
+	}
+	replaceLimit(limit) {
+		this.limitSubstitution = limit;
+		return this;
+	}
+	build() {
+		let commands = this.generateSetCommands();
+		commands += this.generateExplain().query;
+		if (this.isIntrospection) commands += " -- @qd_introspection";
+		return commands;
+	}
+	/** Return the "set a=b" parts of the command in the query separate from the explain select ... part */
+	buildParts() {
+		const commands = this.generateSetCommands();
+		const explain = this.generateExplain();
+		this._preamble = explain.preamble;
+		if (this.isIntrospection) explain.query += " -- @qd_introspection";
+		return {
+			commands,
+			query: explain.query
+		};
+	}
+	generateSetCommands() {
+		let commands = "";
+		for (const key in this.commands) {
+			const value = this.commands[key];
+			commands += `set local ${key}=${value};\n`;
+		}
+		return commands;
+	}
+	generateExplain() {
+		let finalQuery = "";
+		if (this.explainFlags.length > 0) finalQuery += `explain (${this.explainFlags.join(", ")}) `;
+		const query = this.substituteQuery();
+		const semicolon = query.endsWith(";") ? "" : ";";
+		const preamble = finalQuery.length;
+		finalQuery += `${query}${semicolon}`;
+		return {
+			query: finalQuery,
+			preamble
+		};
+	}
+	substituteQuery() {
+		let query = this.query;
+		if (this.limitSubstitution !== void 0) query = query.replace(/limit\s+\$\d+/g, `limit ${this.limitSubstitution}`);
+		for (const [key, value] of Object.entries(this.parameters)) query = query.replaceAll(`\\$${key}`, value.toString());
+		return query;
+	}
+};
+//#endregion
+export { PostgresQueryBuilder };
+
+//# sourceMappingURL=builder.mjs.map

package/dist/sql/builder.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"builder.mjs","names":[],"sources":["../../src/sql/builder.ts"],"sourcesContent":["import { PgIdentifier } from \"./pg-identifier\";\n\nexport type PostgresQueryBuilderCommand =\n  | \"bitmapscan\"\n  | \"indexscan\"\n  | \"seqscan\";\n\nexport class PostgresQueryBuilder {\n  private readonly commands: Record<string, string> = {};\n  private isIntrospection: boolean = false;\n  private explainFlags: string[] = [];\n  private _preamble = 0;\n  private parameters: Record<Parameter, number> = {};\n  // substitution for `limit $1` -> `limit 5`\n  private limitSubstitution?: number;\n\n  constructor(private query: string) {}\n\n  get preamble(): number {\n    return this._preamble;\n  }\n\n  static createIndex(definition: string, name?: PgIdentifier) {\n    if (name) {\n      return new PostgresQueryBuilder(\n        `create index \"${name}\" on ${definition};`,\n      );\n    }\n    return new PostgresQueryBuilder(`create index on ${definition};`);\n  }\n\n  enable(command: PostgresQueryBuilderCommand, value: boolean = true): this {\n    const commandString = `enable_${command}`;\n    if (value) {\n      this.commands[commandString] = \"on\";\n    } else {\n      this.commands[commandString] = \"off\";\n    }\n    return this;\n  }\n\n  withQuery(query: string): this {\n    this.query = query;\n    return this;\n  }\n\n  introspect(): this {\n    this.isIntrospection = true;\n    return this;\n  }\n\n  explain(flags: string[]): this {\n    this.explainFlags = flags;\n    return this;\n  }\n\n  parameterize(parameters: Record<Parameter, number>) {\n    Object.assign(this.parameters, parameters);\n    return this;\n  }\n\n  replaceLimit(limit: number) {\n    this.limitSubstitution = limit;\n    return this;\n  }\n\n  build(): string {\n    let commands = this.generateSetCommands();\n    commands += this.generateExplain().query;\n    if (this.isIntrospection) {\n      commands += \" -- @qd_introspection\";\n    }\n    return commands;\n  }\n\n  /** Return the \"set a=b\" parts of the command in the query separate from the explain select ... part */\n  buildParts() {\n    const commands = this.generateSetCommands();\n    const explain = this.generateExplain();\n    this._preamble = explain.preamble;\n    if (this.isIntrospection) {\n      explain.query += \" -- @qd_introspection\";\n    }\n    return { commands, query: explain.query };\n  }\n\n  private generateSetCommands() {\n    let commands = \"\";\n    for (const key in this.commands) {\n      const value = this.commands[key];\n      commands += `set local ${key}=${value};\\n`;\n    }\n    return commands;\n  }\n\n  private generateExplain() {\n    let finalQuery = \"\";\n    if (this.explainFlags.length > 0) {\n      finalQuery += `explain (${this.explainFlags.join(\", \")}) `;\n    }\n    const query = this.substituteQuery();\n    const semicolon = query.endsWith(\";\") ? \"\" : \";\";\n\n    const preamble = finalQuery.length;\n    finalQuery += `${query}${semicolon}`;\n    return { query: finalQuery, preamble };\n  }\n\n  private substituteQuery() {\n    let query = this.query;\n    if (this.limitSubstitution !== undefined) {\n      query = query.replace(\n        /limit\\s+\\$\\d+/g,\n        `limit ${this.limitSubstitution}`,\n      );\n    }\n    for (const [key, value] of Object.entries(this.parameters)) {\n      query = query.replaceAll(`\\\\$${key}`, value.toString());\n    }\n    return query;\n  }\n}\n\nexport type Parameter = `$${string}`;\n"],"mappings":";;;AAOA,IAAa,uBAAb,MAAa,qBAAqB;CAShC,YAAY,OAAuB;AAAf,OAAA,QAAA;wBARH,YAAmC,EAAE,CAAC;wBAC/C,mBAA2B,MAAM;wBACjC,gBAAyB,EAAE,CAAC;wBAC5B,aAAY,EAAE;wBACd,cAAwC,EAAE,CAAC;wBAE3C,qBAAA,KAAA,EAA2B;;CAInC,IAAI,WAAmB;AACrB,SAAO,KAAK;;CAGd,OAAO,YAAY,YAAoB,MAAqB;AAC1D,MAAI,KACF,QAAO,IAAI,qBACT,iBAAiB,KAAK,OAAO,WAAW,GACzC;AAEH,SAAO,IAAI,qBAAqB,mBAAmB,WAAW,GAAG;;CAGnE,OAAO,SAAsC,QAAiB,MAAY;EACxE,MAAM,gBAAgB,UAAU;AAChC,MAAI,MACF,MAAK,SAAS,iBAAiB;MAE/B,MAAK,SAAS,iBAAiB;AAEjC,SAAO;;CAGT,UAAU,OAAqB;AAC7B,OAAK,QAAQ;AACb,SAAO;;CAGT,aAAmB;AACjB,OAAK,kBAAkB;AACvB,SAAO;;CAGT,QAAQ,OAAuB;AAC7B,OAAK,eAAe;AACpB,SAAO;;CAGT,aAAa,YAAuC;AAClD,SAAO,OAAO,KAAK,YAAY,WAAW;AAC1C,SAAO;;CAGT,aAAa,OAAe;AAC1B,OAAK,oBAAoB;AACzB,SAAO;;CAGT,QAAgB;EACd,IAAI,WAAW,KAAK,qBAAqB;AACzC,cAAY,KAAK,iBAAiB,CAAC;AACnC,MAAI,KAAK,gBACP,aAAY;AAEd,SAAO;;;CAIT,aAAa;EACX,MAAM,WAAW,KAAK,qBAAqB;EAC3C,MAAM,UAAU,KAAK,iBAAiB;AACtC,OAAK,YAAY,QAAQ;AACzB,MAAI,KAAK,gBACP,SAAQ,SAAS;AAEnB,SAAO;GAAE;GAAU,OAAO,QAAQ;GAAO;;CAG3C,sBAA8B;EAC5B,IAAI,WAAW;AACf,OAAK,MAAM,OAAO,KAAK,UAAU;GAC/B,MAAM,QAAQ,KAAK,SAAS;AAC5B,eAAY,aAAa,IAAI,GAAG,MAAM;;AAExC,SAAO;;CAGT,kBAA0B;EACxB,IAAI,aAAa;AACjB,MAAI,KAAK,aAAa,SAAS,EAC7B,eAAc,YAAY,KAAK,aAAa,KAAK,KAAK,CAAC;EAEzD,MAAM,QAAQ,KAAK,iBAAiB;EACpC,MAAM,YAAY,MAAM,SAAS,IAAI,GAAG,KAAK;EAE7C,MAAM,WAAW,WAAW;AAC5B,gBAAc,GAAG,QAAQ;AACzB,SAAO;GAAE,OAAO;GAAY;GAAU;;CAGxC,kBAA0B;EACxB,IAAI,QAAQ,KAAK;AACjB,MAAI,KAAK,sBAAsB,KAAA,EAC7B,SAAQ,MAAM,QACZ,kBACA,SAAS,KAAK,oBACf;AAEH,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,WAAW,CACxD,SAAQ,MAAM,WAAW,MAAM,OAAO,MAAM,UAAU,CAAC;AAEzD,SAAO"}

package/dist/sql/database.cjs

@@ -0,0 +1,35 @@
+"use client";
+require("../_virtual/_rolldown/runtime.cjs");
+let zod = require("zod");
+//#region src/sql/database.ts
+const PostgresVersion = zod.z.string().brand("PostgresVersion");
+/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */
+const PostgresExplainStageSchema = zod.z.lazy(() => zod.z.object({
+	"Node Type": zod.z.string(),
+	"Total Cost": zod.z.number(),
+	"Plan Width": zod.z.number(),
+	"Node Id": zod.z.number().optional(),
+	Plans: zod.z.array(PostgresExplainStageSchema).optional()
+}).passthrough());
+/**
+* Drops a disabled index. Rollsback if it fails for any reason
+* @returns Did dropping the index succeed?
+*/
+async function dropIndex(tx, index) {
+	try {
+		await tx.exec(`
+      savepoint idx_drop;
+      drop index if exists ${index} cascade;
+    `);
+		return true;
+	} catch {
+		await tx.exec(`rollback to idx_drop`);
+		return false;
+	}
+}
+//#endregion
+exports.PostgresExplainStageSchema = PostgresExplainStageSchema;
+exports.PostgresVersion = PostgresVersion;
+exports.dropIndex = dropIndex;
+
+//# sourceMappingURL=database.cjs.map

package/dist/sql/database.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.cjs","names":["z"],"sources":["../../src/sql/database.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { PgIdentifier } from \"./pg-identifier\";\n\nexport const PostgresVersion = z.string().brand(\"PostgresVersion\");\nexport type PostgresVersion = z.infer<typeof PostgresVersion>;\n\nexport interface PostgresTransaction {\n  /**\n   * Exec a query and return the result as an array of objects.\n   */\n  exec<T>(query: string, params?: unknown[]): Promise<T[]>;\n}\n\n/**\n * A shared interface for all postgres connections.\n * This is required to allow interop between pglite and regular postgres drivers\n */\nexport interface Postgres extends PostgresTransaction {\n  transaction<T>(callback: (tx: PostgresTransaction) => Promise<T>): Promise<T>;\n  cursor?<T>(\n    query: string,\n    params?: unknown[],\n    options?: { size?: number },\n  ): AsyncGenerator<T, void, unknown>;\n  // postgres returns versions as a string\n  serverNum(): Promise<PostgresVersion>;\n}\n\nexport type PostgresConnectionInput = {\n  url: string;\n};\n\nexport type PostgresFactory = (input: PostgresConnectionInput) => Postgres;\n\n// PostgreSQL EXPLAIN plan types\ndeclare const StageId: unique symbol;\nexport type PostgresStageId = number & { [StageId]: \"StageId\" };\n\nexport type PostgresStage =\n  | \"Seq Scan\"\n  | \"Limit\"\n  | \"Bitmap Heap Scan\"\n  | \"Bitmap Index Scan\"\n  | \"Index Only Scan\"\n  | \"Index Scan\"\n  | \"BitmapOr\";\n\nexport type PostgresExplainStageCommon = {\n  \"Node Type\": PostgresStage;\n  // available only when doing a trace explain using the\n  // query doctor postgres binary\n  \"Node Id\": PostgresStageId;\n  Plans?: PostgresExplainStage[];\n  \"Plan Width\": number;\n  \"Total Cost\": number;\n};\n\nexport type PostgresExplainStage =\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Index Scan\";\n      \"Index Name\": string;\n      \"Relation Name\": string;\n      Alias: string;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Seq Scan\";\n      \"Relation Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Bitmap Heap Scan\";\n      \"Relation Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Bitmap Index Scan\";\n      \"Index Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Index Only Scan\";\n      \"Index Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | PostgresExplainStageCommon;\n\n/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */\nexport const PostgresExplainStageSchema: z.ZodType<PostgresExplainStage> =\n  z.lazy(() =>\n    z\n      .object({\n        \"Node Type\": z.string(),\n        \"Total Cost\": z.number(),\n        \"Plan Width\": z.number(),\n        \"Node Id\": z.number().optional(),\n        Plans: z.array(PostgresExplainStageSchema).optional(),\n      })\n      .passthrough(),\n  ) as z.ZodType<PostgresExplainStage>;\n\nexport type PostgresExplainResult = {\n  \"QUERY PLAN\": {\n    Plan: PostgresExplainStage;\n  }[];\n};\n\n/**\n * Drops a disabled index. Rollsback if it fails for any reason\n * @returns Did dropping the index succeed?\n */\nexport async function dropIndex(\n  tx: PostgresTransaction,\n  index: PgIdentifier,\n): Promise<boolean> {\n  try {\n    await tx.exec(`\n      savepoint idx_drop;\n      drop index if exists ${index} cascade;\n    `);\n    return true;\n  } catch {\n    // no problem if droping the index fails. It should throw an error\n    await tx.exec(`rollback to idx_drop`);\n    return false;\n  }\n}\n"],"mappings":";;;;AAGA,MAAa,kBAAkBA,IAAAA,EAAE,QAAQ,CAAC,MAAM,kBAAkB;;AAiGlE,MAAa,6BACXA,IAAAA,EAAE,WACAA,IAAAA,EACG,OAAO;CACN,aAAaA,IAAAA,EAAE,QAAQ;CACvB,cAAcA,IAAAA,EAAE,QAAQ;CACxB,cAAcA,IAAAA,EAAE,QAAQ;CACxB,WAAWA,IAAAA,EAAE,QAAQ,CAAC,UAAU;CAChC,OAAOA,IAAAA,EAAE,MAAM,2BAA2B,CAAC,UAAU;CACtD,CAAC,CACD,aAAa,CACjB;;;;;AAYH,eAAsB,UACpB,IACA,OACkB;AAClB,KAAI;AACF,QAAM,GAAG,KAAK;;6BAEW,MAAM;MAC7B;AACF,SAAO;SACD;AAEN,QAAM,GAAG,KAAK,uBAAuB;AACrC,SAAO"}

package/dist/sql/database.d.cts

@@ -0,0 +1,91 @@
+'use client';
+
+import { PgIdentifier } from "./pg-identifier.cjs";
+import { z } from "zod";
+
+//#region src/sql/database.d.ts
+declare const PostgresVersion: z.core.$ZodBranded<z.ZodString, "PostgresVersion">;
+type PostgresVersion = z.infer<typeof PostgresVersion>;
+interface PostgresTransaction {
+  /**
+   * Exec a query and return the result as an array of objects.
+   */
+  exec<T>(query: string, params?: unknown[]): Promise<T[]>;
+}
+/**
+ * A shared interface for all postgres connections.
+ * This is required to allow interop between pglite and regular postgres drivers
+ */
+interface Postgres extends PostgresTransaction {
+  transaction<T>(callback: (tx: PostgresTransaction) => Promise<T>): Promise<T>;
+  cursor?<T>(query: string, params?: unknown[], options?: {
+    size?: number;
+  }): AsyncGenerator<T, void, unknown>;
+  serverNum(): Promise<PostgresVersion>;
+}
+type PostgresConnectionInput = {
+  url: string;
+};
+type PostgresFactory = (input: PostgresConnectionInput) => Postgres;
+declare const StageId: unique symbol;
+type PostgresStageId = number & {
+  [StageId]: "StageId";
+};
+type PostgresStage = "Seq Scan" | "Limit" | "Bitmap Heap Scan" | "Bitmap Index Scan" | "Index Only Scan" | "Index Scan" | "BitmapOr";
+type PostgresExplainStageCommon = {
+  "Node Type": PostgresStage;
+  "Node Id": PostgresStageId;
+  Plans?: PostgresExplainStage[];
+  "Plan Width": number;
+  "Total Cost": number;
+};
+type PostgresExplainStage = (PostgresExplainStageCommon & {
+  "Node Type": "Index Scan";
+  "Index Name": string;
+  "Relation Name": string;
+  Alias: string;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Seq Scan";
+  "Relation Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Bitmap Heap Scan";
+  "Relation Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Bitmap Index Scan";
+  "Index Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Index Only Scan";
+  "Index Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | PostgresExplainStageCommon;
+/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */
+declare const PostgresExplainStageSchema: z.ZodType<PostgresExplainStage>;
+type PostgresExplainResult = {
+  "QUERY PLAN": {
+    Plan: PostgresExplainStage;
+  }[];
+};
+/**
+ * Drops a disabled index. Rollsback if it fails for any reason
+ * @returns Did dropping the index succeed?
+ */
+declare function dropIndex(tx: PostgresTransaction, index: PgIdentifier): Promise<boolean>;
+//#endregion
+export { Postgres, PostgresConnectionInput, PostgresExplainResult, PostgresExplainStage, PostgresExplainStageCommon, PostgresExplainStageSchema, PostgresFactory, PostgresStage, PostgresStageId, PostgresTransaction, PostgresVersion, dropIndex };
+//# sourceMappingURL=database.d.cts.map

package/dist/sql/database.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.d.cts","names":[],"sources":["../../src/sql/database.ts"],"mappings":";;;;;;cAGa,eAAA,EAAe,CAAA,CAAA,IAAA,CAAA,WAAA,CAAA,CAAA,CAAA,SAAA;AAAA,KAChB,eAAA,GAAkB,CAAA,CAAE,KAAA,QAAa,eAAA;AAAA,UAE5B,mBAAA;EAHJ;;;EAOX,IAAA,IAAQ,KAAA,UAAe,MAAA,eAAqB,OAAA,CAAQ,CAAA;AAAA;;;;;UAOrC,QAAA,SAAiB,mBAAA;EAChC,WAAA,IAAe,QAAA,GAAW,EAAA,EAAI,mBAAA,KAAwB,OAAA,CAAQ,CAAA,IAAK,OAAA,CAAQ,CAAA;EAC3E,MAAA,KACE,KAAA,UACA,MAAA,cACA,OAAA;IAAY,IAAA;EAAA,IACX,cAAA,CAAe,CAAA;EAElB,SAAA,IAAa,OAAA,CAAQ,eAAA;AAAA;AAAA,KAGX,uBAAA;EACV,GAAA;AAAA;AAAA,KAGU,eAAA,IAAmB,KAAA,EAAO,uBAAA,KAA4B,QAAA;AAAA,cAGpD,OAAA;AAAA,KACF,eAAA;EAAA,CAA8B,OAAA;AAAA;AAAA,KAE9B,aAAA;AAAA,KASA,0BAAA;EACV,WAAA,EAAa,aAAA;EAGb,SAAA,EAAW,eAAA;EACX,KAAA,GAAQ,oBAAA;EACR,YAAA;EACA,YAAA;AAAA;AAAA,KAGU,oBAAA,IACP,0BAAA;EACC,WAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,eAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,eAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,YAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,YAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,KAEF,0BAAA;;cAGS,0BAAA,EAA4B,CAAA,CAAE,OAAA,CAAQ,oBAAA;AAAA,KAavC,qBAAA;EACV,YAAA;IACE,IAAA,EAAM,oBAAA;EAAA;AAAA;;AAnFV;;;iBA2FsB,SAAA,CACpB,EAAA,EAAI,mBAAA,EACJ,KAAA,EAAO,YAAA,GACN,OAAA"}

package/dist/sql/database.d.mts

@@ -0,0 +1,91 @@
+'use client';
+
+import { PgIdentifier } from "./pg-identifier.mjs";
+import { z } from "zod";
+
+//#region src/sql/database.d.ts
+declare const PostgresVersion: z.core.$ZodBranded<z.ZodString, "PostgresVersion">;
+type PostgresVersion = z.infer<typeof PostgresVersion>;
+interface PostgresTransaction {
+  /**
+   * Exec a query and return the result as an array of objects.
+   */
+  exec<T>(query: string, params?: unknown[]): Promise<T[]>;
+}
+/**
+ * A shared interface for all postgres connections.
+ * This is required to allow interop between pglite and regular postgres drivers
+ */
+interface Postgres extends PostgresTransaction {
+  transaction<T>(callback: (tx: PostgresTransaction) => Promise<T>): Promise<T>;
+  cursor?<T>(query: string, params?: unknown[], options?: {
+    size?: number;
+  }): AsyncGenerator<T, void, unknown>;
+  serverNum(): Promise<PostgresVersion>;
+}
+type PostgresConnectionInput = {
+  url: string;
+};
+type PostgresFactory = (input: PostgresConnectionInput) => Postgres;
+declare const StageId: unique symbol;
+type PostgresStageId = number & {
+  [StageId]: "StageId";
+};
+type PostgresStage = "Seq Scan" | "Limit" | "Bitmap Heap Scan" | "Bitmap Index Scan" | "Index Only Scan" | "Index Scan" | "BitmapOr";
+type PostgresExplainStageCommon = {
+  "Node Type": PostgresStage;
+  "Node Id": PostgresStageId;
+  Plans?: PostgresExplainStage[];
+  "Plan Width": number;
+  "Total Cost": number;
+};
+type PostgresExplainStage = (PostgresExplainStageCommon & {
+  "Node Type": "Index Scan";
+  "Index Name": string;
+  "Relation Name": string;
+  Alias: string;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Seq Scan";
+  "Relation Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Bitmap Heap Scan";
+  "Relation Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Bitmap Index Scan";
+  "Index Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | (PostgresExplainStageCommon & {
+  "Node Type": "Index Only Scan";
+  "Index Name": string;
+  Filter?: string;
+  "Actual Rows": number;
+  "Actual Loops": number;
+  "Rows Removed by Filter": number;
+}) | PostgresExplainStageCommon;
+/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */
+declare const PostgresExplainStageSchema: z.ZodType<PostgresExplainStage>;
+type PostgresExplainResult = {
+  "QUERY PLAN": {
+    Plan: PostgresExplainStage;
+  }[];
+};
+/**
+ * Drops a disabled index. Rollsback if it fails for any reason
+ * @returns Did dropping the index succeed?
+ */
+declare function dropIndex(tx: PostgresTransaction, index: PgIdentifier): Promise<boolean>;
+//#endregion
+export { Postgres, PostgresConnectionInput, PostgresExplainResult, PostgresExplainStage, PostgresExplainStageCommon, PostgresExplainStageSchema, PostgresFactory, PostgresStage, PostgresStageId, PostgresTransaction, PostgresVersion, dropIndex };
+//# sourceMappingURL=database.d.mts.map

package/dist/sql/database.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.d.mts","names":[],"sources":["../../src/sql/database.ts"],"mappings":";;;;;;cAGa,eAAA,EAAe,CAAA,CAAA,IAAA,CAAA,WAAA,CAAA,CAAA,CAAA,SAAA;AAAA,KAChB,eAAA,GAAkB,CAAA,CAAE,KAAA,QAAa,eAAA;AAAA,UAE5B,mBAAA;EAHJ;;;EAOX,IAAA,IAAQ,KAAA,UAAe,MAAA,eAAqB,OAAA,CAAQ,CAAA;AAAA;;;;;UAOrC,QAAA,SAAiB,mBAAA;EAChC,WAAA,IAAe,QAAA,GAAW,EAAA,EAAI,mBAAA,KAAwB,OAAA,CAAQ,CAAA,IAAK,OAAA,CAAQ,CAAA;EAC3E,MAAA,KACE,KAAA,UACA,MAAA,cACA,OAAA;IAAY,IAAA;EAAA,IACX,cAAA,CAAe,CAAA;EAElB,SAAA,IAAa,OAAA,CAAQ,eAAA;AAAA;AAAA,KAGX,uBAAA;EACV,GAAA;AAAA;AAAA,KAGU,eAAA,IAAmB,KAAA,EAAO,uBAAA,KAA4B,QAAA;AAAA,cAGpD,OAAA;AAAA,KACF,eAAA;EAAA,CAA8B,OAAA;AAAA;AAAA,KAE9B,aAAA;AAAA,KASA,0BAAA;EACV,WAAA,EAAa,aAAA;EAGb,SAAA,EAAW,eAAA;EACX,KAAA,GAAQ,oBAAA;EACR,YAAA;EACA,YAAA;AAAA;AAAA,KAGU,oBAAA,IACP,0BAAA;EACC,WAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,eAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,eAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,YAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,MAED,0BAAA;EACC,WAAA;EACA,YAAA;EACA,MAAA;EACA,aAAA;EACA,cAAA;EACA,wBAAA;AAAA,KAEF,0BAAA;;cAGS,0BAAA,EAA4B,CAAA,CAAE,OAAA,CAAQ,oBAAA;AAAA,KAavC,qBAAA;EACV,YAAA;IACE,IAAA,EAAM,oBAAA;EAAA;AAAA;;AAnFV;;;iBA2FsB,SAAA,CACpB,EAAA,EAAI,mBAAA,EACJ,KAAA,EAAO,YAAA,GACN,OAAA"}

package/dist/sql/database.mjs

@@ -0,0 +1,32 @@
+"use client";
+import { z } from "zod";
+//#region src/sql/database.ts
+const PostgresVersion = z.string().brand("PostgresVersion");
+/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */
+const PostgresExplainStageSchema = z.lazy(() => z.object({
+	"Node Type": z.string(),
+	"Total Cost": z.number(),
+	"Plan Width": z.number(),
+	"Node Id": z.number().optional(),
+	Plans: z.array(PostgresExplainStageSchema).optional()
+}).passthrough());
+/**
+* Drops a disabled index. Rollsback if it fails for any reason
+* @returns Did dropping the index succeed?
+*/
+async function dropIndex(tx, index) {
+	try {
+		await tx.exec(`
+      savepoint idx_drop;
+      drop index if exists ${index} cascade;
+    `);
+		return true;
+	} catch {
+		await tx.exec(`rollback to idx_drop`);
+		return false;
+	}
+}
+//#endregion
+export { PostgresExplainStageSchema, PostgresVersion, dropIndex };
+
+//# sourceMappingURL=database.mjs.map

package/dist/sql/database.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.mjs","names":[],"sources":["../../src/sql/database.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { PgIdentifier } from \"./pg-identifier\";\n\nexport const PostgresVersion = z.string().brand(\"PostgresVersion\");\nexport type PostgresVersion = z.infer<typeof PostgresVersion>;\n\nexport interface PostgresTransaction {\n  /**\n   * Exec a query and return the result as an array of objects.\n   */\n  exec<T>(query: string, params?: unknown[]): Promise<T[]>;\n}\n\n/**\n * A shared interface for all postgres connections.\n * This is required to allow interop between pglite and regular postgres drivers\n */\nexport interface Postgres extends PostgresTransaction {\n  transaction<T>(callback: (tx: PostgresTransaction) => Promise<T>): Promise<T>;\n  cursor?<T>(\n    query: string,\n    params?: unknown[],\n    options?: { size?: number },\n  ): AsyncGenerator<T, void, unknown>;\n  // postgres returns versions as a string\n  serverNum(): Promise<PostgresVersion>;\n}\n\nexport type PostgresConnectionInput = {\n  url: string;\n};\n\nexport type PostgresFactory = (input: PostgresConnectionInput) => Postgres;\n\n// PostgreSQL EXPLAIN plan types\ndeclare const StageId: unique symbol;\nexport type PostgresStageId = number & { [StageId]: \"StageId\" };\n\nexport type PostgresStage =\n  | \"Seq Scan\"\n  | \"Limit\"\n  | \"Bitmap Heap Scan\"\n  | \"Bitmap Index Scan\"\n  | \"Index Only Scan\"\n  | \"Index Scan\"\n  | \"BitmapOr\";\n\nexport type PostgresExplainStageCommon = {\n  \"Node Type\": PostgresStage;\n  // available only when doing a trace explain using the\n  // query doctor postgres binary\n  \"Node Id\": PostgresStageId;\n  Plans?: PostgresExplainStage[];\n  \"Plan Width\": number;\n  \"Total Cost\": number;\n};\n\nexport type PostgresExplainStage =\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Index Scan\";\n      \"Index Name\": string;\n      \"Relation Name\": string;\n      Alias: string;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Seq Scan\";\n      \"Relation Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Bitmap Heap Scan\";\n      \"Relation Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Bitmap Index Scan\";\n      \"Index Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | (PostgresExplainStageCommon & {\n      \"Node Type\": \"Index Only Scan\";\n      \"Index Name\": string;\n      Filter?: string;\n      \"Actual Rows\": number;\n      \"Actual Loops\": number;\n      \"Rows Removed by Filter\": number;\n    })\n  | PostgresExplainStageCommon;\n\n/** Zod schema for PostgresExplainStage — validates common fields, passes through variant-specific ones. */\nexport const PostgresExplainStageSchema: z.ZodType<PostgresExplainStage> =\n  z.lazy(() =>\n    z\n      .object({\n        \"Node Type\": z.string(),\n        \"Total Cost\": z.number(),\n        \"Plan Width\": z.number(),\n        \"Node Id\": z.number().optional(),\n        Plans: z.array(PostgresExplainStageSchema).optional(),\n      })\n      .passthrough(),\n  ) as z.ZodType<PostgresExplainStage>;\n\nexport type PostgresExplainResult = {\n  \"QUERY PLAN\": {\n    Plan: PostgresExplainStage;\n  }[];\n};\n\n/**\n * Drops a disabled index. Rollsback if it fails for any reason\n * @returns Did dropping the index succeed?\n */\nexport async function dropIndex(\n  tx: PostgresTransaction,\n  index: PgIdentifier,\n): Promise<boolean> {\n  try {\n    await tx.exec(`\n      savepoint idx_drop;\n      drop index if exists ${index} cascade;\n    `);\n    return true;\n  } catch {\n    // no problem if droping the index fails. It should throw an error\n    await tx.exec(`rollback to idx_drop`);\n    return false;\n  }\n}\n"],"mappings":";;;AAGA,MAAa,kBAAkB,EAAE,QAAQ,CAAC,MAAM,kBAAkB;;AAiGlE,MAAa,6BACX,EAAE,WACA,EACG,OAAO;CACN,aAAa,EAAE,QAAQ;CACvB,cAAc,EAAE,QAAQ;CACxB,cAAc,EAAE,QAAQ;CACxB,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,OAAO,EAAE,MAAM,2BAA2B,CAAC,UAAU;CACtD,CAAC,CACD,aAAa,CACjB;;;;;AAYH,eAAsB,UACpB,IACA,OACkB;AAClB,KAAI;AACF,QAAM,GAAG,KAAK;;6BAEW,MAAM;MAC7B;AACF,SAAO;SACD;AAEN,QAAM,GAAG,KAAK,uBAAuB;AACrC,SAAO"}

package/dist/sql/indexes.cjs

@@ -0,0 +1,17 @@
+"use client";
+//#region src/sql/indexes.ts
+function isIndexSupported(index) {
+	return index.index_type === "btree" || index.index_type === "gin";
+}
+/**
+* Doesn't necessarily decide whether the index can be dropped but can be
+* used to not even try dropping indexes that _definitely_ cannot be dropped
+*/
+function isIndexProbablyDroppable(index) {
+	return !index.is_primary && !index.is_unique;
+}
+//#endregion
+exports.isIndexProbablyDroppable = isIndexProbablyDroppable;
+exports.isIndexSupported = isIndexSupported;
+
+//# sourceMappingURL=indexes.cjs.map

package/dist/sql/indexes.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"indexes.cjs","names":[],"sources":["../../src/sql/indexes.ts"],"sourcesContent":["import type { IndexedTable } from \"../optimizer/statistics.js\";\n\nexport function isIndexSupported(index: IndexedTable) {\n  return index.index_type === \"btree\" || index.index_type === \"gin\";\n}\n\n/**\n * Doesn't necessarily decide whether the index can be dropped but can be\n * used to not even try dropping indexes that _definitely_ cannot be dropped\n */\nexport function isIndexProbablyDroppable(index: IndexedTable) {\n  /* TODO: until we have a better solution, this is the best we have */\n  /* The is_unique check is problematic only if the column is declared as unique */\n  return !index.is_primary && !index.is_unique;\n}\n"],"mappings":";;AAEA,SAAgB,iBAAiB,OAAqB;AACpD,QAAO,MAAM,eAAe,WAAW,MAAM,eAAe;;;;;;AAO9D,SAAgB,yBAAyB,OAAqB;AAG5D,QAAO,CAAC,MAAM,cAAc,CAAC,MAAM"}

package/dist/sql/indexes.d.cts

@@ -0,0 +1,14 @@
+'use client';
+
+import { IndexedTable } from "../optimizer/statistics.cjs";
+
+//#region src/sql/indexes.d.ts
+declare function isIndexSupported(index: IndexedTable): boolean;
+/**
+ * Doesn't necessarily decide whether the index can be dropped but can be
+ * used to not even try dropping indexes that _definitely_ cannot be dropped
+ */
+declare function isIndexProbablyDroppable(index: IndexedTable): boolean;
+//#endregion
+export { isIndexProbablyDroppable, isIndexSupported };
+//# sourceMappingURL=indexes.d.cts.map

package/dist/sql/indexes.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"indexes.d.cts","names":[],"sources":["../../src/sql/indexes.ts"],"mappings":";;;;;iBAEgB,gBAAA,CAAiB,KAAA,EAAO,YAAA;;;AAAxC;;iBAQgB,wBAAA,CAAyB,KAAA,EAAO,YAAA"}

package/dist/sql/indexes.d.mts

@@ -0,0 +1,14 @@
+'use client';
+
+import { IndexedTable } from "../optimizer/statistics.mjs";
+
+//#region src/sql/indexes.d.ts
+declare function isIndexSupported(index: IndexedTable): boolean;
+/**
+ * Doesn't necessarily decide whether the index can be dropped but can be
+ * used to not even try dropping indexes that _definitely_ cannot be dropped
+ */
+declare function isIndexProbablyDroppable(index: IndexedTable): boolean;
+//#endregion
+export { isIndexProbablyDroppable, isIndexSupported };
+//# sourceMappingURL=indexes.d.mts.map

package/dist/sql/indexes.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"indexes.d.mts","names":[],"sources":["../../src/sql/indexes.ts"],"mappings":";;;;;iBAEgB,gBAAA,CAAiB,KAAA,EAAO,YAAA;;;AAAxC;;iBAQgB,wBAAA,CAAyB,KAAA,EAAO,YAAA"}

package/dist/sql/indexes.mjs

@@ -0,0 +1,16 @@
+"use client";
+//#region src/sql/indexes.ts
+function isIndexSupported(index) {
+	return index.index_type === "btree" || index.index_type === "gin";
+}
+/**
+* Doesn't necessarily decide whether the index can be dropped but can be
+* used to not even try dropping indexes that _definitely_ cannot be dropped
+*/
+function isIndexProbablyDroppable(index) {
+	return !index.is_primary && !index.is_unique;
+}
+//#endregion
+export { isIndexProbablyDroppable, isIndexSupported };
+
+//# sourceMappingURL=indexes.mjs.map

package/dist/sql/indexes.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"indexes.mjs","names":[],"sources":["../../src/sql/indexes.ts"],"sourcesContent":["import type { IndexedTable } from \"../optimizer/statistics.js\";\n\nexport function isIndexSupported(index: IndexedTable) {\n  return index.index_type === \"btree\" || index.index_type === \"gin\";\n}\n\n/**\n * Doesn't necessarily decide whether the index can be dropped but can be\n * used to not even try dropping indexes that _definitely_ cannot be dropped\n */\nexport function isIndexProbablyDroppable(index: IndexedTable) {\n  /* TODO: until we have a better solution, this is the best we have */\n  /* The is_unique check is problematic only if the column is declared as unique */\n  return !index.is_primary && !index.is_unique;\n}\n"],"mappings":";;AAEA,SAAgB,iBAAiB,OAAqB;AACpD,QAAO,MAAM,eAAe,WAAW,MAAM,eAAe;;;;;;AAO9D,SAAgB,yBAAyB,OAAqB;AAG5D,QAAO,CAAC,MAAM,cAAc,CAAC,MAAM"}