'use client';

import { ExportedStats } from "./optimizer/statistics.cjs";
import { LiveQueryOptimization, OptimizedQuery } from "./query.cjs";
import { RepoPolicyConfig } from "./ci/policy.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;
  conditionPolicies: RepoPolicyConfig;
};
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>;
  getProductionStats(): Promise<ExportedStats[] | null>;
}
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<LiveQueryOptimization>;
  resetStats(): 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