@query-doctor/core 0.8.9 → 0.10.0

package/dist/sql/indexes.d.cts

@@ -1,14 +1,30 @@
 'use client';
 
-import { IndexedTable } from "../optimizer/statistics.cjs";
+import { FullSchemaIndex } from "../schema.cjs";
 
 //#region src/sql/indexes.d.ts
-declare function isIndexSupported(index: IndexedTable): boolean;
+type IndexTypeHint = {
+  indexType: string;
+};
+type IndexDropHint = {
+  isPrimary: boolean;
+  isUnique: boolean;
+};
+declare function isIndexSupported(index: IndexTypeHint): 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;
+declare function isIndexProbablyDroppable(index: IndexDropHint): boolean;
+type DuplicateIndexGroup = {
+  kind: "exact";
+  indexes: FullSchemaIndex[];
+} | {
+  kind: "prefix";
+  indexes: FullSchemaIndex[];
+  sharedPrefix: FullSchemaIndex["keyColumns"];
+};
+declare function groupDuplicateIndexes(indexes: FullSchemaIndex[]): DuplicateIndexGroup[];
 //#endregion
-export { isIndexProbablyDroppable, isIndexSupported };
+export { DuplicateIndexGroup, groupDuplicateIndexes, isIndexProbablyDroppable, isIndexSupported };
 //# sourceMappingURL=indexes.d.cts.map

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

@@ -1 +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"}
+{"version":3,"file":"indexes.d.cts","names":[],"sources":["../../src/sql/indexes.ts"],"mappings":";;;;;KAEK,aAAA;EAAkB,SAAA;AAAA;AAAA,KAClB,aAAA;EAAkB,SAAA;EAAoB,QAAA;AAAA;AAAA,iBAE3B,gBAAA,CAAiB,KAAA,EAAO,aAAA;AAHR;;;;AAAA,iBAWhB,wBAAA,CAAyB,KAAA,EAAO,aAAA;AAAA,KAMpC,mBAAA;EACN,IAAA;EAAe,OAAA,EAAS,eAAA;AAAA;EAExB,IAAA;EACA,OAAA,EAAS,eAAA;EACT,YAAA,EAAc,eAAA;AAAA;AAAA,iBA2DJ,qBAAA,CACd,OAAA,EAAS,eAAA,KACR,mBAAA"}

package/dist/sql/indexes.d.mts

@@ -1,14 +1,30 @@
 'use client';
 
-import { IndexedTable } from "../optimizer/statistics.mjs";
+import { FullSchemaIndex } from "../schema.mjs";
 
 //#region src/sql/indexes.d.ts
-declare function isIndexSupported(index: IndexedTable): boolean;
+type IndexTypeHint = {
+  indexType: string;
+};
+type IndexDropHint = {
+  isPrimary: boolean;
+  isUnique: boolean;
+};
+declare function isIndexSupported(index: IndexTypeHint): 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;
+declare function isIndexProbablyDroppable(index: IndexDropHint): boolean;
+type DuplicateIndexGroup = {
+  kind: "exact";
+  indexes: FullSchemaIndex[];
+} | {
+  kind: "prefix";
+  indexes: FullSchemaIndex[];
+  sharedPrefix: FullSchemaIndex["keyColumns"];
+};
+declare function groupDuplicateIndexes(indexes: FullSchemaIndex[]): DuplicateIndexGroup[];
 //#endregion
-export { isIndexProbablyDroppable, isIndexSupported };
+export { DuplicateIndexGroup, groupDuplicateIndexes, isIndexProbablyDroppable, isIndexSupported };
 //# sourceMappingURL=indexes.d.mts.map

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

@@ -1 +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"}
+{"version":3,"file":"indexes.d.mts","names":[],"sources":["../../src/sql/indexes.ts"],"mappings":";;;;;KAEK,aAAA;EAAkB,SAAA;AAAA;AAAA,KAClB,aAAA;EAAkB,SAAA;EAAoB,QAAA;AAAA;AAAA,iBAE3B,gBAAA,CAAiB,KAAA,EAAO,aAAA;AAHR;;;;AAAA,iBAWhB,wBAAA,CAAyB,KAAA,EAAO,aAAA;AAAA,KAMpC,mBAAA;EACN,IAAA;EAAe,OAAA,EAAS,eAAA;AAAA;EAExB,IAAA;EACA,OAAA,EAAS,eAAA;EACT,YAAA,EAAc,eAAA;AAAA;AAAA,iBA2DJ,qBAAA,CACd,OAAA,EAAS,eAAA,KACR,mBAAA"}

package/dist/sql/indexes.mjs

@@ -1,16 +1,64 @@
 "use client";
 //#region src/sql/indexes.ts
 function isIndexSupported(index) {
-	return index.index_type === "btree" || index.index_type === "gin";
+	return index.indexType === "btree" || index.indexType === "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;
+	return !index.isPrimary && !index.isUnique;
+}
+function columnsEqual(a, b) {
+	if (a.length !== b.length) return false;
+	for (let i = 0; i < a.length; i++) if (a[i].name.toString() !== b[i].name.toString() || (a[i].order ?? null) !== (b[i].order ?? null) || (a[i].opclass ?? null) !== (b[i].opclass ?? null)) return false;
+	return true;
+}
+function indexesAreEqual(a, b) {
+	return a.schemaName.toString() === b.schemaName.toString() && a.tableName.toString() === b.tableName.toString() && a.indexType === b.indexType && (a.wherePredicate ?? null) === (b.wherePredicate ?? null) && columnsEqual(a.keyColumns, b.keyColumns);
+}
+function sharedPrefixColumns(shorter, longer) {
+	if (shorter.schemaName.toString() !== longer.schemaName.toString() || shorter.tableName.toString() !== longer.tableName.toString() || shorter.indexType !== longer.indexType || shorter.indexType !== "btree" || shorter.wherePredicate !== void 0 || longer.wherePredicate !== void 0 || shorter.keyColumns.length >= longer.keyColumns.length) return null;
+	for (let i = 0; i < shorter.keyColumns.length; i++) {
+		const s = shorter.keyColumns[i];
+		const l = longer.keyColumns[i];
+		if (s.name.toString() !== l.name.toString() || (s.order ?? null) !== (l.order ?? null) || (s.opclass ?? null) !== (l.opclass ?? null)) return null;
+	}
+	return shorter.keyColumns;
+}
+function groupDuplicateIndexes(indexes) {
+	const groups = [];
+	for (const index of indexes) {
+		const existing = groups.find((g) => g.kind === "exact" && indexesAreEqual(g.indexes[0], index));
+		if (existing) existing.indexes.push(index);
+		else groups.push({
+			kind: "exact",
+			indexes: [index]
+		});
+	}
+	const exactGroups = groups.filter((g) => g.indexes.length > 1);
+	const representatives = groups.filter((g) => g.indexes.length === 1).map((g) => g.indexes[0]);
+	const prefixGroups = [];
+	for (const shorter of representatives) {
+		const existing = prefixGroups.find((g) => g.indexes.some((idx) => sharedPrefixColumns(shorter, idx) !== null || sharedPrefixColumns(idx, shorter) !== null));
+		if (existing) {
+			if (!existing.indexes.includes(shorter)) existing.indexes.push(shorter);
+		} else {
+			const longer = representatives.find((idx) => idx !== shorter && sharedPrefixColumns(shorter, idx) !== null);
+			if (longer) {
+				const sharedPrefix = sharedPrefixColumns(shorter, longer);
+				prefixGroups.push({
+					kind: "prefix",
+					indexes: [longer, shorter],
+					sharedPrefix
+				});
+			}
+		}
+	}
+	return [...exactGroups, ...prefixGroups];
 }
 //#endregion
-export { isIndexProbablyDroppable, isIndexSupported };
+export { groupDuplicateIndexes, isIndexProbablyDroppable, isIndexSupported };
 
 //# sourceMappingURL=indexes.mjs.map

package/dist/sql/indexes.mjs.map

@@ -1 +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"}
+{"version":3,"file":"indexes.mjs","names":[],"sources":["../../src/sql/indexes.ts"],"sourcesContent":["import type { FullSchemaIndex } from \"../schema.js\";\n\ntype IndexTypeHint = { indexType: string };\ntype IndexDropHint = { isPrimary: boolean; isUnique: boolean };\n\nexport function isIndexSupported(index: IndexTypeHint) {\n  return index.indexType === \"btree\" || index.indexType === \"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: IndexDropHint) {\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.isPrimary && !index.isUnique;\n}\n\nexport type DuplicateIndexGroup =\n  | { kind: \"exact\"; indexes: FullSchemaIndex[] }\n  | {\n      kind: \"prefix\";\n      indexes: FullSchemaIndex[];\n      sharedPrefix: FullSchemaIndex[\"keyColumns\"];\n    };\n\nfunction columnsEqual(\n  a: FullSchemaIndex[\"keyColumns\"],\n  b: FullSchemaIndex[\"keyColumns\"],\n): boolean {\n  if (a.length !== b.length) return false;\n  for (let i = 0; i < a.length; i++) {\n    if (\n      a[i].name.toString() !== b[i].name.toString() ||\n      (a[i].order ?? null) !== (b[i].order ?? null) ||\n      (a[i].opclass ?? null) !== (b[i].opclass ?? null)\n    ) {\n      return false;\n    }\n  }\n  return true;\n}\n\nfunction indexesAreEqual(a: FullSchemaIndex, b: FullSchemaIndex): boolean {\n  return (\n    a.schemaName.toString() === b.schemaName.toString() &&\n    a.tableName.toString() === b.tableName.toString() &&\n    a.indexType === b.indexType &&\n    (a.wherePredicate ?? null) === (b.wherePredicate ?? null) &&\n    columnsEqual(a.keyColumns, b.keyColumns)\n  );\n}\n\nfunction sharedPrefixColumns(\n  shorter: FullSchemaIndex,\n  longer: FullSchemaIndex,\n): FullSchemaIndex[\"keyColumns\"] | null {\n  if (\n    shorter.schemaName.toString() !== longer.schemaName.toString() ||\n    shorter.tableName.toString() !== longer.tableName.toString() ||\n    shorter.indexType !== longer.indexType ||\n    shorter.indexType !== \"btree\" ||\n    shorter.wherePredicate !== undefined ||\n    longer.wherePredicate !== undefined ||\n    shorter.keyColumns.length >= longer.keyColumns.length\n  ) {\n    return null;\n  }\n  for (let i = 0; i < shorter.keyColumns.length; i++) {\n    const s = shorter.keyColumns[i];\n    const l = longer.keyColumns[i];\n    if (\n      s.name.toString() !== l.name.toString() ||\n      (s.order ?? null) !== (l.order ?? null) ||\n      (s.opclass ?? null) !== (l.opclass ?? null)\n    ) {\n      return null;\n    }\n  }\n  return shorter.keyColumns;\n}\n\nexport function groupDuplicateIndexes(\n  indexes: FullSchemaIndex[],\n): DuplicateIndexGroup[] {\n  const groups: DuplicateIndexGroup[] = [];\n\n  for (const index of indexes) {\n    const existing = groups.find(\n      (g) => g.kind === \"exact\" && indexesAreEqual(g.indexes[0], index),\n    );\n    if (existing) {\n      existing.indexes.push(index);\n    } else {\n      groups.push({ kind: \"exact\", indexes: [index] });\n    }\n  }\n\n  const exactGroups = groups.filter((g) => g.indexes.length > 1);\n\n  const representatives = groups\n    .filter((g) => g.indexes.length === 1)\n    .map((g) => g.indexes[0]);\n\n  const prefixGroups: Extract<DuplicateIndexGroup, { kind: \"prefix\" }>[] = [];\n\n  for (const shorter of representatives) {\n    const existing = prefixGroups.find((g) =>\n      g.indexes.some(\n        (idx) =>\n          sharedPrefixColumns(shorter, idx) !== null ||\n          sharedPrefixColumns(idx, shorter) !== null,\n      ),\n    );\n    if (existing) {\n      if (!existing.indexes.includes(shorter)) {\n        existing.indexes.push(shorter);\n      }\n    } else {\n      const longer = representatives.find(\n        (idx) => idx !== shorter && sharedPrefixColumns(shorter, idx) !== null,\n      );\n      if (longer) {\n        const sharedPrefix = sharedPrefixColumns(shorter, longer)!;\n        prefixGroups.push({\n          kind: \"prefix\",\n          indexes: [longer, shorter],\n          sharedPrefix,\n        });\n      }\n    }\n  }\n\n  return [...exactGroups, ...prefixGroups];\n}\n"],"mappings":";;AAKA,SAAgB,iBAAiB,OAAsB;AACrD,QAAO,MAAM,cAAc,WAAW,MAAM,cAAc;;;;;;AAO5D,SAAgB,yBAAyB,OAAsB;AAG7D,QAAO,CAAC,MAAM,aAAa,CAAC,MAAM;;AAWpC,SAAS,aACP,GACA,GACS;AACT,KAAI,EAAE,WAAW,EAAE,OAAQ,QAAO;AAClC,MAAK,IAAI,IAAI,GAAG,IAAI,EAAE,QAAQ,IAC5B,KACE,EAAE,GAAG,KAAK,UAAU,KAAK,EAAE,GAAG,KAAK,UAAU,KAC5C,EAAE,GAAG,SAAS,WAAW,EAAE,GAAG,SAAS,UACvC,EAAE,GAAG,WAAW,WAAW,EAAE,GAAG,WAAW,MAE5C,QAAO;AAGX,QAAO;;AAGT,SAAS,gBAAgB,GAAoB,GAA6B;AACxE,QACE,EAAE,WAAW,UAAU,KAAK,EAAE,WAAW,UAAU,IACnD,EAAE,UAAU,UAAU,KAAK,EAAE,UAAU,UAAU,IACjD,EAAE,cAAc,EAAE,cACjB,EAAE,kBAAkB,WAAW,EAAE,kBAAkB,SACpD,aAAa,EAAE,YAAY,EAAE,WAAW;;AAI5C,SAAS,oBACP,SACA,QACsC;AACtC,KACE,QAAQ,WAAW,UAAU,KAAK,OAAO,WAAW,UAAU,IAC9D,QAAQ,UAAU,UAAU,KAAK,OAAO,UAAU,UAAU,IAC5D,QAAQ,cAAc,OAAO,aAC7B,QAAQ,cAAc,WACtB,QAAQ,mBAAmB,KAAA,KAC3B,OAAO,mBAAmB,KAAA,KAC1B,QAAQ,WAAW,UAAU,OAAO,WAAW,OAE/C,QAAO;AAET,MAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,WAAW,QAAQ,KAAK;EAClD,MAAM,IAAI,QAAQ,WAAW;EAC7B,MAAM,IAAI,OAAO,WAAW;AAC5B,MACE,EAAE,KAAK,UAAU,KAAK,EAAE,KAAK,UAAU,KACtC,EAAE,SAAS,WAAW,EAAE,SAAS,UACjC,EAAE,WAAW,WAAW,EAAE,WAAW,MAEtC,QAAO;;AAGX,QAAO,QAAQ;;AAGjB,SAAgB,sBACd,SACuB;CACvB,MAAM,SAAgC,EAAE;AAExC,MAAK,MAAM,SAAS,SAAS;EAC3B,MAAM,WAAW,OAAO,MACrB,MAAM,EAAE,SAAS,WAAW,gBAAgB,EAAE,QAAQ,IAAI,MAAM,CAClE;AACD,MAAI,SACF,UAAS,QAAQ,KAAK,MAAM;MAE5B,QAAO,KAAK;GAAE,MAAM;GAAS,SAAS,CAAC,MAAM;GAAE,CAAC;;CAIpD,MAAM,cAAc,OAAO,QAAQ,MAAM,EAAE,QAAQ,SAAS,EAAE;CAE9D,MAAM,kBAAkB,OACrB,QAAQ,MAAM,EAAE,QAAQ,WAAW,EAAE,CACrC,KAAK,MAAM,EAAE,QAAQ,GAAG;CAE3B,MAAM,eAAmE,EAAE;AAE3E,MAAK,MAAM,WAAW,iBAAiB;EACrC,MAAM,WAAW,aAAa,MAAM,MAClC,EAAE,QAAQ,MACP,QACC,oBAAoB,SAAS,IAAI,KAAK,QACtC,oBAAoB,KAAK,QAAQ,KAAK,KACzC,CACF;AACD,MAAI;OACE,CAAC,SAAS,QAAQ,SAAS,QAAQ,CACrC,UAAS,QAAQ,KAAK,QAAQ;SAE3B;GACL,MAAM,SAAS,gBAAgB,MAC5B,QAAQ,QAAQ,WAAW,oBAAoB,SAAS,IAAI,KAAK,KACnE;AACD,OAAI,QAAQ;IACV,MAAM,eAAe,oBAAoB,SAAS,OAAO;AACzD,iBAAa,KAAK;KAChB,MAAM;KACN,SAAS,CAAC,QAAQ,QAAQ;KAC1B;KACD,CAAC;;;;AAKR,QAAO,CAAC,GAAG,aAAa,GAAG,aAAa"}

package/dist/sql/walker.cjs

@@ -1,8 +1,8 @@
 "use client";
 require("../_virtual/_rolldown/runtime.cjs");
+const require_defineProperty = require("../_virtual/_@oxc-project_runtime@0.126.0/helpers/defineProperty.cjs");
 const require_ast_utils = require("./ast-utils.cjs");
 const require_nudges = require("./nudges.cjs");
-const require_defineProperty = require("../_virtual/_@oxc-project_runtime@0.126.0/helpers/defineProperty.cjs");
 let pgsql_deparser = require("pgsql-deparser");
 //#region src/sql/walker.ts
 const JSONB_EXTRACTION_OPS = new Set(["->", "->>"]);

package/dist/sql/walker.mjs

@@ -1,7 +1,7 @@
 "use client";
+import { _defineProperty } from "../_virtual/_@oxc-project_runtime@0.126.0/helpers/defineProperty.mjs";
 import { getNodeKind, is, isANode } from "./ast-utils.mjs";
 import { parseNudges } from "./nudges.mjs";
-import { _defineProperty } from "../_virtual/_@oxc-project_runtime@0.126.0/helpers/defineProperty.mjs";
 import { deparseSync } from "pgsql-deparser";
 //#region src/sql/walker.ts
 const JSONB_EXTRACTION_OPS = new Set(["->", "->>"]);

package/dist/websocket-server.d.cts

@@ -0,0 +1,68 @@
+'use client';
+
+import { ExportedStats } from "./optimizer/statistics.cjs";
+import { OptimizedQuery } from "./query.cjs";
+import { RpcStub } from "capnweb";
+
+//#region src/websocket-server.d.ts
+type IndexDefinition = {
+  schemaName: string;
+  tableName: string;
+  indexName: string;
+  indexType?: string;
+  isUnique?: boolean;
+  columns: Array<{
+    name: string;
+    order?: "ASC" | "DESC";
+    opclass?: string;
+  }>;
+  wherePredicate?: string;
+};
+type ExtensionPresence = {
+  type: "present";
+  extensionName: "pg_stat_statements" | "pg_stat_monitor";
+  needsRestart: boolean;
+} | {
+  type: "missing";
+};
+type RepoConfig = {
+  minimumCost: number;
+  regressionThreshold: number;
+  ignoredQueryHashes: string[];
+  lastSeenQueryHashes: string[];
+  acknowledgedQueryHashes: string[];
+  comparisonBranch?: string;
+};
+interface ServerApi {
+  pushQuery(queries: OptimizedQuery[]): Promise<void>;
+  pushStats(stats: ExportedStats[]): Promise<void>;
+  pushSchema(schema: unknown): Promise<void>;
+  pushEncryptedStats(stats: Uint8Array): Promise<void>;
+  setExtensionPresence(presence: ExtensionPresence): Promise<void>;
+  log(line: string, process?: string): Promise<void>;
+  ping(): Promise<void>;
+  getRepoConfig(repo: string, branch: string): Promise<RepoConfig>;
+}
+interface ClientApi {
+  repull(): Promise<unknown>;
+  refreshQueries(): Promise<void>;
+  updateStatistics(stats: ExportedStats[]): Promise<void>;
+  hideIndex(indexName: string): Promise<void>;
+  addIndex(index: IndexDefinition): Promise<void>;
+  runQuery(query: string): Promise<void>;
+}
+type ConnectionMode = {
+  kind: "persistent";
+  environment?: string;
+} | {
+  kind: "ci";
+  branch: string;
+  sha: string;
+  label?: string;
+};
+interface UnauthenticatedServerApi {
+  authenticate(token: string, client: RpcStub<ClientApi>, mode: ConnectionMode): Promise<ServerApi>;
+}
+//#endregion
+export { ClientApi, ConnectionMode, ExtensionPresence, IndexDefinition, RepoConfig, ServerApi, UnauthenticatedServerApi };
+//# sourceMappingURL=websocket-server.d.cts.map

package/dist/websocket-server.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-server.d.cts","names":[],"sources":["../src/websocket-server.ts"],"mappings":";;;;;;;KAIY,eAAA;EACV,UAAA;EACA,SAAA;EACA,SAAA;EACA,SAAA;EACA,QAAA;EACA,OAAA,EAAS,KAAA;IACP,IAAA;IACA,KAAA;IACA,OAAA;EAAA;EAEF,cAAA;AAAA;AAAA,KAGU,iBAAA;EAEN,IAAA;EACA,aAAA;EACA,YAAA;AAAA;EAGA,IAAA;AAAA;AAAA,KAGM,UAAA;EACV,WAAA;EACA,mBAAA;EACA,kBAAA;EACA,mBAAA;EACA,uBAAA;EACA,gBAAA;AAAA;AAAA,UAGe,SAAA;EACf,SAAA,CAAU,OAAA,EAAS,cAAA,KAAmB,OAAA;EACtC,SAAA,CAAU,KAAA,EAAO,aAAA,KAAkB,OAAA;EACnC,UAAA,CAAW,MAAA,YAAkB,OAAA;EAC7B,kBAAA,CAAmB,KAAA,EAAO,UAAA,GAAa,OAAA;EACvC,oBAAA,CAAqB,QAAA,EAAU,iBAAA,GAAoB,OAAA;EACnD,GAAA,CAAI,IAAA,UAAc,OAAA,YAAmB,OAAA;EACrC,IAAA,IAAQ,OAAA;EACR,aAAA,CAAc,IAAA,UAAc,MAAA,WAAiB,OAAA,CAAQ,UAAA;AAAA;AAAA,UAGtC,SAAA;EACf,MAAA,IAAU,OAAA;EACV,cAAA,IAAkB,OAAA;EAClB,gBAAA,CAAiB,KAAA,EAAO,aAAA,KAAkB,OAAA;EAC1C,SAAA,CAAU,SAAA,WAAoB,OAAA;EAC9B,QAAA,CAAS,KAAA,EAAO,eAAA,GAAkB,OAAA;EAClC,QAAA,CAAS,KAAA,WAAgB,OAAA;AAAA;AAAA,KAGf,cAAA;EACN,IAAA;EAAoB,WAAA;AAAA;EACpB,IAAA;EAAY,MAAA;EAAgB,GAAA;EAAa,KAAA;AAAA;AAAA,UAE9B,wBAAA;EACf,YAAA,CACE,KAAA,UACA,MAAA,EAAQ,OAAA,CAAQ,SAAA,GAChB,IAAA,EAAM,cAAA,GACL,OAAA,CAAQ,SAAA;AAAA"}

package/dist/websocket-server.d.mts

@@ -0,0 +1,68 @@
+'use client';
+
+import { ExportedStats } from "./optimizer/statistics.mjs";
+import { OptimizedQuery } from "./query.mjs";
+import { RpcStub } from "capnweb";
+
+//#region src/websocket-server.d.ts
+type IndexDefinition = {
+  schemaName: string;
+  tableName: string;
+  indexName: string;
+  indexType?: string;
+  isUnique?: boolean;
+  columns: Array<{
+    name: string;
+    order?: "ASC" | "DESC";
+    opclass?: string;
+  }>;
+  wherePredicate?: string;
+};
+type ExtensionPresence = {
+  type: "present";
+  extensionName: "pg_stat_statements" | "pg_stat_monitor";
+  needsRestart: boolean;
+} | {
+  type: "missing";
+};
+type RepoConfig = {
+  minimumCost: number;
+  regressionThreshold: number;
+  ignoredQueryHashes: string[];
+  lastSeenQueryHashes: string[];
+  acknowledgedQueryHashes: string[];
+  comparisonBranch?: string;
+};
+interface ServerApi {
+  pushQuery(queries: OptimizedQuery[]): Promise<void>;
+  pushStats(stats: ExportedStats[]): Promise<void>;
+  pushSchema(schema: unknown): Promise<void>;
+  pushEncryptedStats(stats: Uint8Array): Promise<void>;
+  setExtensionPresence(presence: ExtensionPresence): Promise<void>;
+  log(line: string, process?: string): Promise<void>;
+  ping(): Promise<void>;
+  getRepoConfig(repo: string, branch: string): Promise<RepoConfig>;
+}
+interface ClientApi {
+  repull(): Promise<unknown>;
+  refreshQueries(): Promise<void>;
+  updateStatistics(stats: ExportedStats[]): Promise<void>;
+  hideIndex(indexName: string): Promise<void>;
+  addIndex(index: IndexDefinition): Promise<void>;
+  runQuery(query: string): Promise<void>;
+}
+type ConnectionMode = {
+  kind: "persistent";
+  environment?: string;
+} | {
+  kind: "ci";
+  branch: string;
+  sha: string;
+  label?: string;
+};
+interface UnauthenticatedServerApi {
+  authenticate(token: string, client: RpcStub<ClientApi>, mode: ConnectionMode): Promise<ServerApi>;
+}
+//#endregion
+export { ClientApi, ConnectionMode, ExtensionPresence, IndexDefinition, RepoConfig, ServerApi, UnauthenticatedServerApi };
+//# sourceMappingURL=websocket-server.d.mts.map

package/dist/websocket-server.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-server.d.mts","names":[],"sources":["../src/websocket-server.ts"],"mappings":";;;;;;;KAIY,eAAA;EACV,UAAA;EACA,SAAA;EACA,SAAA;EACA,SAAA;EACA,QAAA;EACA,OAAA,EAAS,KAAA;IACP,IAAA;IACA,KAAA;IACA,OAAA;EAAA;EAEF,cAAA;AAAA;AAAA,KAGU,iBAAA;EAEN,IAAA;EACA,aAAA;EACA,YAAA;AAAA;EAGA,IAAA;AAAA;AAAA,KAGM,UAAA;EACV,WAAA;EACA,mBAAA;EACA,kBAAA;EACA,mBAAA;EACA,uBAAA;EACA,gBAAA;AAAA;AAAA,UAGe,SAAA;EACf,SAAA,CAAU,OAAA,EAAS,cAAA,KAAmB,OAAA;EACtC,SAAA,CAAU,KAAA,EAAO,aAAA,KAAkB,OAAA;EACnC,UAAA,CAAW,MAAA,YAAkB,OAAA;EAC7B,kBAAA,CAAmB,KAAA,EAAO,UAAA,GAAa,OAAA;EACvC,oBAAA,CAAqB,QAAA,EAAU,iBAAA,GAAoB,OAAA;EACnD,GAAA,CAAI,IAAA,UAAc,OAAA,YAAmB,OAAA;EACrC,IAAA,IAAQ,OAAA;EACR,aAAA,CAAc,IAAA,UAAc,MAAA,WAAiB,OAAA,CAAQ,UAAA;AAAA;AAAA,UAGtC,SAAA;EACf,MAAA,IAAU,OAAA;EACV,cAAA,IAAkB,OAAA;EAClB,gBAAA,CAAiB,KAAA,EAAO,aAAA,KAAkB,OAAA;EAC1C,SAAA,CAAU,SAAA,WAAoB,OAAA;EAC9B,QAAA,CAAS,KAAA,EAAO,eAAA,GAAkB,OAAA;EAClC,QAAA,CAAS,KAAA,WAAgB,OAAA;AAAA;AAAA,KAGf,cAAA;EACN,IAAA;EAAoB,WAAA;AAAA;EACpB,IAAA;EAAY,MAAA;EAAgB,GAAA;EAAa,KAAA;AAAA;AAAA,UAE9B,wBAAA;EACf,YAAA,CACE,KAAA,UACA,MAAA,EAAQ,OAAA,CAAQ,SAAA,GAChB,IAAA,EAAM,cAAA,GACL,OAAA,CAAQ,SAAA;AAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@query-doctor/core",
-  "version": "0.8.9",
+  "version": "0.10.0",
   "private": false,
   "description": "Core logic for Query Doctor",
   "license": "",
@@ -22,14 +22,16 @@
   "scripts": {
     "dev": "tsdown --watch",
     "build": "tsdown",
+    "typecheck": "tsc --noEmit",
     "test": "vitest"
   },
   "dependencies": {
     "@pgsql/types": "^17.6.2",
+    "capnweb": "^0.7.0",
     "colorette": "^2.0.20",
     "dedent": "^1.7.2",
     "pgsql-deparser": "^17.17.2",
-    "zod": "^4.1.13"
+    "zod": "^4.3.6"
   },
   "devDependencies": {
     "@pgsql/parser": "^17.5.0",