@querypanel/node-sdk 1.0.6 → 1.0.8

package/dist/cjs/index.test.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const index_1 = require("./index");
+const mockResponse = (status, body) => ({
+    ok: status >= 200 && status < 300,
+    status,
+    statusText: status === 200 ? "OK" : "ERROR",
+    text: async () => JSON.stringify(body),
+});
+(0, vitest_1.describe)("QueryPanelSdkAPI.ask", () => {
+    const fetchMock = vitest_1.vi.fn();
+    const sdk = new index_1.QueryPanelSdkAPI("http://api", "test-token");
+    (0, vitest_1.beforeEach)(() => {
+        fetchMock.mockReset();
+        globalThis.fetch = fetchMock;
+    });
+    (0, vitest_1.it)("performs generate-sql and generate-chart roundtrips", async () => {
+        fetchMock
+            .mockResolvedValueOnce(mockResponse(200, {
+            sql: "SELECT 1",
+            rationale: "demo rationale",
+            context: [{ kind: "ddl", content: "CREATE TABLE" }],
+        }))
+            .mockResolvedValueOnce(mockResponse(200, {
+            vegaLiteSpec: { mark: "bar" },
+            notes: null,
+        }));
+        const executeSql = vitest_1.vi.fn().mockResolvedValue({
+            fields: ["count"],
+            rows: [{ count: 1 }],
+        });
+        const result = await sdk.ask("How many rows?", {
+            tenantId: "tenant-1",
+            executeSql,
+        });
+        (0, vitest_1.expect)(result.sql).toBe("SELECT 1");
+        (0, vitest_1.expect)(result.chart.vegaLiteSpec).toMatchObject({ mark: "bar" });
+        (0, vitest_1.expect)(result.context).toHaveLength(1);
+        (0, vitest_1.expect)(result.fields).toEqual(["count"]);
+        (0, vitest_1.expect)(result.rows).toEqual([{ count: 1 }]);
+        (0, vitest_1.expect)(fetchMock).toHaveBeenCalledWith("http://api/v2/generate-sql", vitest_1.expect.objectContaining({
+            method: "POST",
+        }));
+        (0, vitest_1.expect)(fetchMock).toHaveBeenCalledWith("http://api/v2/generate-chart", vitest_1.expect.objectContaining({
+            method: "POST",
+        }));
+    });
+    (0, vitest_1.it)("passes last_error on retry when execution fails", async () => {
+        fetchMock
+            .mockResolvedValueOnce(mockResponse(200, {
+            sql: "SELECT fail",
+            rationale: "first try",
+            context: [],
+        }))
+            .mockResolvedValueOnce(mockResponse(200, {
+            sql: "SELECT success",
+            rationale: "second try",
+            context: [],
+        }))
+            .mockResolvedValueOnce(mockResponse(200, {
+            vegaLiteSpec: { mark: "line" },
+            notes: "ready",
+        }));
+        const executeSql = vitest_1.vi
+            .fn()
+            .mockRejectedValueOnce(new Error("execution failed"))
+            .mockResolvedValueOnce({
+            fields: ["value"],
+            rows: [{ value: 2 }],
+        });
+        const result = await sdk.ask("Retry please", {
+            tenantId: "tenant-2",
+            executeSql,
+            sqlMaxAttempts: 2,
+        });
+        (0, vitest_1.expect)(result.sql).toBe("SELECT success");
+        (0, vitest_1.expect)(result.chart.notes).toBe("ready");
+        const secondCall = fetchMock.mock.calls.find(([url]) => typeof url === "string" && url.endsWith("/v2/generate-sql"));
+        (0, vitest_1.expect)(secondCall).toBeTruthy();
+        const secondBody = JSON.parse(secondCall[1].body);
+        (0, vitest_1.expect)(secondBody.last_error).toBe("execution failed");
+    });
+});
+//# sourceMappingURL=index.test.js.map

package/dist/cjs/index.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/index.test.ts"],"names":[],"mappings":";;AAAA,mCAA8D;AAC9D,mCAAiE;AAEjE,MAAM,YAAY,GAAG,CAAC,MAAc,EAAE,IAAa,EAAE,EAAE,CAAC,CAAC;IACvD,EAAE,EAAE,MAAM,IAAI,GAAG,IAAI,MAAM,GAAG,GAAG;IACjC,MAAM;IACN,UAAU,EAAE,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO;IAC3C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;CACvC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,MAAM,SAAS,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;IAC1B,MAAM,GAAG,GAAG,IAAI,wBAAgB,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAE7D,IAAA,mBAAU,EAAC,GAAG,EAAE;QACd,SAAS,CAAC,SAAS,EAAE,CAAC;QACtB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,SAAS;aACN,qBAAqB,CACpB,YAAY,CAAC,GAAG,EAAE;YAChB,GAAG,EAAE,UAAU;YACf,SAAS,EAAE,gBAAgB;YAC3B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;SACpD,CAAC,CACH;aACA,qBAAqB,CACpB,YAAY,CAAC,GAAG,EAAE;YAChB,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;YAC7B,KAAK,EAAE,IAAI;SACZ,CAAC,CACH,CAAC;QAEJ,MAAM,UAAU,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC3C,MAAM,EAAE,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;SACrB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAoB,MAAM,GAAG,CAAC,GAAG,CAAC,gBAAgB,EAAE;YAC9D,QAAQ,EAAE,UAAU;YACpB,UAAU;SACX,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpC,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,IAAA,eAAM,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,IAAA,eAAM,EAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QACzC,IAAA,eAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5C,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,4BAA4B,EAC5B,eAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;SACf,CAAC,CACH,CAAC;QACF,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,8BAA8B,EAC9B,eAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;SACf,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,SAAS;aACN,qBAAqB,CACpB,YAAY,CAAC,GAAG,EAAE;YAChB,GAAG,EAAE,aAAa;YAClB,SAAS,EAAE,WAAW;YACtB,OAAO,EAAE,EAAE;SACZ,CAAC,CACH;aACA,qBAAqB,CACpB,YAAY,CAAC,GAAG,EAAE;YAChB,GAAG,EAAE,gBAAgB;YACrB,SAAS,EAAE,YAAY;YACvB,OAAO,EAAE,EAAE;SACZ,CAAC,CACH;aACA,qBAAqB,CACpB,YAAY,CAAC,GAAG,EAAE;YAChB,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;YAC9B,KAAK,EAAE,OAAO;SACf,CAAC,CACH,CAAC;QAEJ,MAAM,UAAU,GAAG,WAAE;aAClB,EAAE,EAAE;aACJ,qBAAqB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;aACpD,qBAAqB,CAAC;YACrB,MAAM,EAAE,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;SACrB,CAAC,CAAC;QAEL,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,QAAQ,EAAE,UAAU;YACpB,UAAU;YACV,cAAc,EAAE,CAAC;SAClB,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CACrD,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAC5D,CAAC;QACF,IAAA,eAAM,EAAC,UAAU,CAAC,CAAC,UAAU,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,UAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnD,IAAA,eAAM,EAAC,UAAU,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/cjs/schema/types.d.ts

@@ -0,0 +1,73 @@
+export type DatabaseKind = 'clickhouse' | 'postgres' | string;
+export interface DatabaseIdentifier {
+    kind: DatabaseKind;
+    name: string;
+    schema?: string;
+    version?: string;
+}
+export interface ColumnStatistics {
+    compressedBytes?: number;
+    uncompressedBytes?: number;
+    distinctValues?: number;
+}
+export interface ColumnSchema {
+    name: string;
+    type: string;
+    rawType?: string;
+    nullable: boolean;
+    defaultKind?: string;
+    defaultExpression?: string;
+    comment?: string;
+    isPrimaryKey: boolean;
+    isForeignKey: boolean;
+    foreignKeyTable?: string;
+    foreignKeyColumn?: string;
+    maxLength?: number;
+    precision?: number;
+    scale?: number;
+    codec?: string;
+    ttlExpression?: string;
+    statistics?: ColumnStatistics;
+}
+export interface IndexSchema {
+    name: string;
+    columns: string[];
+    unique: boolean;
+    type: string;
+    definition?: string;
+}
+export interface ConstraintSchema {
+    name: string;
+    type: 'PRIMARY KEY' | 'FOREIGN KEY' | 'UNIQUE' | 'CHECK' | string;
+    columns: string[];
+    referencedTable?: string;
+    referencedColumns?: string[];
+    definition?: string;
+}
+export interface TableStatistics {
+    totalRows?: number;
+    totalBytes?: number;
+    uncompressedBytes?: number;
+}
+export interface TableSchema {
+    name: string;
+    schema: string;
+    type: 'table' | 'view' | string;
+    engine?: string;
+    comment?: string;
+    statistics?: TableStatistics;
+    columns: ColumnSchema[];
+    indexes: IndexSchema[];
+    constraints: ConstraintSchema[];
+}
+export interface SchemaIntrospection {
+    db: DatabaseIdentifier;
+    tables: TableSchema[];
+    introspectedAt: string;
+    metadata?: Record<string, unknown>;
+}
+export interface IntrospectOptions {
+    /** Optional allow-list of table names to introspect (schema-qualified or bare). */
+    tables?: string[];
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/cjs/schema/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/schema/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,YAAY,GAAG,YAAY,GAAG,UAAU,GAAG,MAAM,CAAC;AAE9D,MAAM,WAAW,kBAAkB;IAC/B,IAAI,EAAE,YAAY,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,gBAAgB;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,cAAc,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,OAAO,CAAC;IACtB,YAAY,EAAE,OAAO,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,gBAAgB,CAAC;CACjC;AAED,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,aAAa,GAAG,aAAa,GAAG,QAAQ,GAAG,OAAO,GAAG,MAAM,CAAC;IAClE,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,eAAe,CAAC;IAC7B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,WAAW,EAAE,gBAAgB,EAAE,CAAC;CACnC;AAED,MAAM,WAAW,mBAAmB;IAChC,EAAE,EAAE,kBAAkB,CAAC;IACvB,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,iBAAiB;IAC9B,mFAAmF;IACnF,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACrB"}

package/dist/cjs/schema/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/cjs/schema/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/schema/types.ts"],"names":[],"mappings":""}

package/dist/cjs/utils/clickhouse.d.ts

@@ -0,0 +1,9 @@
+export declare function isNullableType(type: string): boolean;
+export declare function unwrapTypeModifiers(type: string): string;
+export declare function extractPrecisionScale(type: string): {
+    precision?: number;
+    scale?: number;
+};
+export declare function extractFixedStringLength(type: string): number | undefined;
+export declare function parseKeyExpression(expression?: string | null): string[];
+//# sourceMappingURL=clickhouse.d.ts.map

package/dist/cjs/utils/clickhouse.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"clickhouse.d.ts","sourceRoot":"","sources":["../../../src/utils/clickhouse.ts"],"names":[],"mappings":"AAEA,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAEpD;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAYxD;AAED,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG;IACjD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB,CAWA;AAED,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAOzE;AAED,wBAAgB,kBAAkB,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,MAAM,EAAE,CAiCvE"}

package/dist/cjs/utils/clickhouse.js

@@ -0,0 +1,98 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isNullableType = isNullableType;
+exports.unwrapTypeModifiers = unwrapTypeModifiers;
+exports.extractPrecisionScale = extractPrecisionScale;
+exports.extractFixedStringLength = extractFixedStringLength;
+exports.parseKeyExpression = parseKeyExpression;
+const WRAPPER_REGEX = /^(Nullable|LowCardinality|SimpleAggregateFunction)\((.+)\)$/i;
+function isNullableType(type) {
+    return /Nullable\s*\(/i.test(type);
+}
+function unwrapTypeModifiers(type) {
+    let current = type.trim();
+    let match = WRAPPER_REGEX.exec(current);
+    while (match) {
+        const inner = match[2];
+        if (!inner) {
+            break;
+        }
+        current = inner.trim();
+        match = WRAPPER_REGEX.exec(current);
+    }
+    return current;
+}
+function extractPrecisionScale(type) {
+    const unwrapped = unwrapTypeModifiers(type);
+    const decimalMatch = unwrapped.match(/Decimal(?:\d+)?\((\d+)\s*,\s*(\d+)\)/i);
+    if (!decimalMatch)
+        return {};
+    const precision = decimalMatch[1];
+    const scale = decimalMatch[2];
+    if (!precision || !scale)
+        return {};
+    return {
+        precision: Number.parseInt(precision, 10),
+        scale: Number.parseInt(scale, 10),
+    };
+}
+function extractFixedStringLength(type) {
+    const unwrapped = unwrapTypeModifiers(type);
+    const match = unwrapped.match(/^(?:FixedString|StringFixed)\((\d+)\)$/i);
+    if (!match)
+        return undefined;
+    const length = match[1];
+    if (!length)
+        return undefined;
+    return Number.parseInt(length, 10);
+}
+function parseKeyExpression(expression) {
+    if (!expression)
+        return [];
+    let value = expression.trim();
+    if (!value)
+        return [];
+    if (/^tuple\s*\(/i.test(value) && value.endsWith(')')) {
+        value = value.replace(/^tuple\s*\(/i, '').replace(/\)$/, '');
+    }
+    const columns = [];
+    let depth = 0;
+    let token = '';
+    for (const ch of value) {
+        if (ch === '(') {
+            depth += 1;
+            token += ch;
+            continue;
+        }
+        if (ch === ')') {
+            depth = Math.max(0, depth - 1);
+            token += ch;
+            continue;
+        }
+        if (ch === ',' && depth === 0) {
+            const col = token.trim();
+            if (col)
+                columns.push(stripWrapper(col));
+            token = '';
+            continue;
+        }
+        token += ch;
+    }
+    const last = token.trim();
+    if (last)
+        columns.push(stripWrapper(last));
+    return columns.filter(Boolean);
+}
+function stripWrapper(value) {
+    const noQuotes = stripQuotes(value);
+    const withoutTicks = noQuotes.replace(/`/g, '').trim();
+    const parts = withoutTicks.split('.');
+    return parts[parts.length - 1]?.trim() ?? '';
+}
+function stripQuotes(value) {
+    if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
+        return value.slice(1, -1);
+    }
+    return value;
+}
+//# sourceMappingURL=clickhouse.js.map

package/dist/cjs/utils/clickhouse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clickhouse.js","sourceRoot":"","sources":["../../../src/utils/clickhouse.ts"],"names":[],"mappings":";;AAEA,wCAEC;AAED,kDAYC;AAED,sDAcC;AAED,4DAOC;AAED,gDAiCC;AA9ED,MAAM,aAAa,GAAG,8DAA8D,CAAC;AAErF,SAAgB,cAAc,CAAC,IAAY;IACvC,OAAO,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACvC,CAAC;AAED,SAAgB,mBAAmB,CAAC,IAAY;IAC5C,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC1B,IAAI,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,OAAO,KAAK,EAAE,CAAC;QACX,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACvB,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,MAAM;QACV,CAAC;QACD,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QACvB,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAgB,qBAAqB,CAAC,IAAY;IAI9C,MAAM,SAAS,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,YAAY,GAAG,SAAS,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC9E,IAAI,CAAC,YAAY;QAAE,OAAO,EAAE,CAAC;IAC7B,MAAM,SAAS,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;IAC9B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACpC,OAAO;QACH,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC;QACzC,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;KACpC,CAAC;AACN,CAAC;AAED,SAAgB,wBAAwB,CAAC,IAAY;IACjD,MAAM,SAAS,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;IACzE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACxB,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACvC,CAAC;AAED,SAAgB,kBAAkB,CAAC,UAA0B;IACzD,IAAI,CAAC,UAAU;QAAE,OAAO,EAAE,CAAC;IAC3B,IAAI,KAAK,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,IAAI,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACpD,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,KAAK,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;QACrB,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YACb,KAAK,IAAI,CAAC,CAAC;YACX,KAAK,IAAI,EAAE,CAAC;YACZ,SAAS;QACb,CAAC;QACD,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YACb,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YAC/B,KAAK,IAAI,EAAE,CAAC;YACZ,SAAS;QACb,CAAC;QACD,IAAI,EAAE,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;YACzB,IAAI,GAAG;gBAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;YACzC,KAAK,GAAG,EAAE,CAAC;YACX,SAAS;QACb,CAAC;QACD,KAAK,IAAI,EAAE,CAAC;IAChB,CAAC;IACD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,IAAI,IAAI;QAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,YAAY,CAAC,KAAa;IAC/B,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AACjD,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAC9B,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QACnG,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC"}

package/dist/esm/connectors/base.d.ts

@@ -0,0 +1,14 @@
+import type { DataFormat } from '@clickhouse/client';
+import type { IntrospectOptions, SchemaIntrospection } from '../schema/types.js';
+export interface QueryOptions {
+    params?: Record<string, unknown>;
+    format?: DataFormat;
+    settings?: Record<string, unknown>;
+}
+export interface DatabaseConnector {
+    connect(): Promise<void>;
+    query<T = Record<string, unknown>>(sql: string, options?: QueryOptions): Promise<T[]>;
+    introspect(options?: IntrospectOptions): Promise<SchemaIntrospection>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/esm/connectors/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/connectors/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,KAAK,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEjF,MAAM,WAAW,YAAY;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,UAAU,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,iBAAiB;IAC9B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IACtF,UAAU,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACtE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1B"}

package/dist/esm/connectors/base.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=base.js.map

package/dist/esm/connectors/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/connectors/base.ts"],"names":[],"mappings":""}

package/dist/esm/connectors/clickhouse.d.ts

@@ -0,0 +1,35 @@
+import { type ClickHouseClient, type ClickHouseClientConfigOptions } from "@clickhouse/client";
+import type { IntrospectOptions, SchemaIntrospection } from "../schema/types.js";
+import type { DatabaseConnector, QueryOptions } from "./base.js";
+export interface ClickHouseConnectorConfig {
+    /** Fully qualified ClickHouse URL. Overrides host/port/protocol when provided. */
+    url?: string;
+    host?: string;
+    port?: number;
+    protocol?: "http" | "https";
+    database: string;
+    username?: string;
+    password?: string;
+    application?: string;
+    clickhouseSettings?: Record<string, unknown>;
+    clientOptions?: Partial<ClickHouseClientConfigOptions>;
+}
+export interface ClickHouseConnectorOverrides {
+    client?: ClickHouseClient;
+    clientFactory?: () => ClickHouseClient;
+}
+export declare class ClickHouseConnector implements DatabaseConnector {
+    private readonly config;
+    private readonly overrides;
+    private client;
+    private readonly defaultFormat;
+    constructor(config: ClickHouseConnectorConfig, overrides?: ClickHouseConnectorOverrides);
+    connect(): Promise<void>;
+    query<T = Record<string, unknown>>(sql: string, options?: QueryOptions): Promise<T[]>;
+    introspect(options?: IntrospectOptions): Promise<SchemaIntrospection>;
+    close(): Promise<void>;
+    private ensureClient;
+    private safePing;
+    private buildClientConfig;
+}
+//# sourceMappingURL=clickhouse.d.ts.map

package/dist/esm/connectors/clickhouse.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"clickhouse.d.ts","sourceRoot":"","sources":["../../../src/connectors/clickhouse.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,gBAAgB,EACrB,KAAK,6BAA6B,EAIlC,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAEX,iBAAiB,EACjB,mBAAmB,EAEnB,MAAM,oBAAoB,CAAC;AAQ5B,OAAO,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEjE,MAAM,WAAW,yBAAyB;IACzC,kFAAkF;IAClF,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC7C,aAAa,CAAC,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,4BAA4B;IAC5C,MAAM,CAAC,EAAE,gBAAgB,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,gBAAgB,CAAC;CACvC;AA4BD,qBAAa,mBAAoB,YAAW,iBAAiB;IAO3D,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAP3B,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAEC;gBAGb,MAAM,EAAE,yBAAyB,EACjC,SAAS,GAAE,4BAAiC;IAKxD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAIxB,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE,YAAY,GACpB,OAAO,CAAC,CAAC,EAAE,CAAC;IAmCT,UAAU,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAyGrE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAMd,YAAY;YAcZ,QAAQ;IAYtB,OAAO,CAAC,iBAAiB;CAkCzB"}

package/dist/esm/connectors/clickhouse.js

@@ -0,0 +1,288 @@
+import { createClient, } from "@clickhouse/client";
+import { extractFixedStringLength, extractPrecisionScale, isNullableType, parseKeyExpression, unwrapTypeModifiers, } from "../utils/clickhouse.js";
+export class ClickHouseConnector {
+    constructor(config, overrides = {}) {
+        this.config = config;
+        this.overrides = overrides;
+        this.defaultFormat = "JSONEachRow";
+        this.client = overrides.client ?? null;
+    }
+    async connect() {
+        await this.ensureClient();
+    }
+    async query(sql, options) {
+        const client = await this.ensureClient();
+        const params = {
+            query: sql,
+        };
+        const format = options?.format ?? this.defaultFormat;
+        if (format) {
+            Object.assign(params, { format });
+        }
+        if (options?.params) {
+            Object.assign(params, { query_params: options.params });
+        }
+        if (options?.settings) {
+            Object.assign(params, {
+                clickhouse_settings: options.settings,
+            });
+        }
+        const result = await client.query(params);
+        const payload = await result.json();
+        if (Array.isArray(payload)) {
+            return payload;
+        }
+        if (payload && typeof payload === "object") {
+            const maybeData = payload.data;
+            if (Array.isArray(maybeData)) {
+                return maybeData;
+            }
+        }
+        return [];
+    }
+    async introspect(options) {
+        const allowTables = normalizeTableFilter(options?.tables);
+        const hasFilter = allowTables.length > 0;
+        const queryParams = {
+            db: this.config.database,
+        };
+        if (hasFilter) {
+            queryParams.tables = allowTables;
+        }
+        const filterClause = hasFilter ? " AND name IN {tables:Array(String)}" : "";
+        const tables = await this.query(`SELECT name, engine, comment, total_rows, total_bytes, is_view, primary_key, sorting_key
+             FROM system.tables
+             WHERE database = {db:String}${filterClause}
+             ORDER BY name`, { params: queryParams });
+        const columnFilterClause = hasFilter
+            ? " AND table IN {tables:Array(String)}"
+            : "";
+        const columns = await this.query(`SELECT table, name, type, position, default_kind, default_expression, comment,
+                    codec_expression, ttl_expression, is_in_primary_key,
+                    data_compressed_bytes, data_uncompressed_bytes
+             FROM system.columns
+             WHERE database = {db:String}${columnFilterClause}
+             ORDER BY table, position`, { params: queryParams });
+        const columnsByTable = new Map();
+        for (const rawColumn of columns) {
+            const list = columnsByTable.get(rawColumn.table) ?? [];
+            list.push(transformColumnRow(rawColumn));
+            columnsByTable.set(rawColumn.table, list);
+        }
+        const tableSchemas = tables.map((table) => {
+            const tableColumns = columnsByTable.get(table.name) ?? [];
+            const primaryKeyColumns = parseKeyExpression(table.primary_key);
+            const totalRows = toNumber(table.total_rows);
+            const totalBytes = toNumber(table.total_bytes);
+            for (const column of tableColumns) {
+                column.isPrimaryKey =
+                    column.isPrimaryKey || primaryKeyColumns.includes(column.name);
+            }
+            const indexes = primaryKeyColumns.length
+                ? [
+                    {
+                        name: "primary_key",
+                        columns: primaryKeyColumns,
+                        unique: true,
+                        type: "PRIMARY KEY",
+                        ...(table.primary_key ? { definition: table.primary_key } : {}),
+                    },
+                ]
+                : [];
+            const constraints = primaryKeyColumns.length
+                ? [
+                    {
+                        name: "primary_key",
+                        type: "PRIMARY KEY",
+                        columns: primaryKeyColumns,
+                    },
+                ]
+                : [];
+            const base = {
+                name: table.name,
+                schema: this.config.database,
+                type: asTableType(table.is_view),
+                engine: table.engine,
+                columns: tableColumns,
+                indexes,
+                constraints,
+            };
+            const comment = sanitize(table.comment);
+            if (comment !== undefined) {
+                base.comment = comment;
+            }
+            const statistics = buildTableStatistics(totalRows, totalBytes);
+            if (statistics) {
+                base.statistics = statistics;
+            }
+            return base;
+        });
+        return {
+            db: {
+                kind: "clickhouse",
+                name: this.config.database,
+            },
+            tables: tableSchemas,
+            introspectedAt: new Date().toISOString(),
+        };
+    }
+    async close() {
+        if (!this.client)
+            return;
+        await this.client.close();
+        this.client = null;
+    }
+    async ensureClient() {
+        if (this.client) {
+            return this.client;
+        }
+        this.client =
+            this.overrides?.clientFactory?.() ??
+                createClient(this.buildClientConfig());
+        const pingResult = await this.safePing(this.client);
+        if (pingResult && "success" in pingResult && pingResult.success === false) {
+            throw pingResult.error ?? new Error("ClickHouse ping failed");
+        }
+        return this.client;
+    }
+    async safePing(client) {
+        try {
+            return await client.ping();
+        }
+        catch (error) {
+            await client.close();
+            this.client = null;
+            throw error;
+        }
+    }
+    buildClientConfig() {
+        const merged = {
+            ...(this.config.clientOptions ?? {}),
+        };
+        if (this.config.url) {
+            merged.url = this.config.url;
+        }
+        else if (this.config.host) {
+            const protocol = this.config.protocol ?? "https";
+            const port = this.config.port ?? (protocol === "https" ? 8443 : 8123);
+            merged.url = `${protocol}://${this.config.host}:${port}`;
+        }
+        merged.database = this.config.database;
+        if (this.config.username) {
+            merged.username = this.config.username;
+        }
+        if (this.config.password) {
+            merged.password = this.config.password;
+        }
+        if (this.config.application) {
+            merged.application = this.config.application;
+        }
+        if (this.config.clickhouseSettings) {
+            merged.clickhouse_settings = {
+                ...(this.config.clientOptions?.clickhouse_settings ?? {}),
+                ...this.config.clickhouseSettings,
+            };
+        }
+        return merged;
+    }
+}
+function normalizeTableFilter(tables) {
+    if (!tables?.length)
+        return [];
+    const seen = new Set();
+    const normalized = [];
+    for (const table of tables) {
+        if (!table)
+            continue;
+        const trimmed = table.trim();
+        if (!trimmed)
+            continue;
+        const parts = trimmed.split(".");
+        const tableName = parts[parts.length - 1];
+        if (!tableName || seen.has(tableName))
+            continue;
+        seen.add(tableName);
+        normalized.push(tableName);
+    }
+    return normalized;
+}
+function transformColumnRow(row) {
+    const nullable = isNullableType(row.type);
+    const unwrappedType = unwrapTypeModifiers(row.type);
+    const { precision, scale } = extractPrecisionScale(row.type);
+    const maxLength = extractFixedStringLength(row.type);
+    const compressedBytes = toNumber(row.data_compressed_bytes);
+    const uncompressedBytes = toNumber(row.data_uncompressed_bytes);
+    const column = {
+        name: row.name,
+        type: unwrappedType,
+        nullable,
+        isPrimaryKey: Boolean(toNumber(row.is_in_primary_key)),
+        isForeignKey: false,
+    };
+    column.rawType = row.type;
+    const defaultKind = sanitize(row.default_kind);
+    if (defaultKind !== undefined)
+        column.defaultKind = defaultKind;
+    const defaultExpression = sanitize(row.default_expression);
+    if (defaultExpression !== undefined)
+        column.defaultExpression = defaultExpression;
+    const comment = sanitize(row.comment);
+    if (comment !== undefined)
+        column.comment = comment;
+    const codec = sanitize(row.codec_expression);
+    if (codec !== undefined)
+        column.codec = codec;
+    const ttlExpression = sanitize(row.ttl_expression);
+    if (ttlExpression !== undefined)
+        column.ttlExpression = ttlExpression;
+    if (maxLength !== undefined)
+        column.maxLength = maxLength;
+    if (precision !== undefined)
+        column.precision = precision;
+    if (scale !== undefined)
+        column.scale = scale;
+    if (compressedBytes !== undefined || uncompressedBytes !== undefined) {
+        column.statistics = {};
+        if (compressedBytes !== undefined)
+            column.statistics.compressedBytes = compressedBytes;
+        if (uncompressedBytes !== undefined)
+            column.statistics.uncompressedBytes = uncompressedBytes;
+    }
+    return column;
+}
+function toNumber(value) {
+    if (value === null || value === undefined)
+        return undefined;
+    if (typeof value === "number")
+        return value;
+    const parsed = Number.parseFloat(String(value));
+    return Number.isNaN(parsed) ? undefined : parsed;
+}
+function sanitize(value) {
+    if (value === null || value === undefined)
+        return undefined;
+    const trimmed = String(value).trim();
+    return trimmed.length ? trimmed : undefined;
+}
+function asTableType(isView) {
+    if (typeof isView === "number") {
+        return isView > 0 ? "view" : "table";
+    }
+    if (typeof isView === "string") {
+        const normalized = isView.toLowerCase();
+        return normalized === "1" || normalized === "true" ? "view" : "table";
+    }
+    return "table";
+}
+function buildTableStatistics(totalRows, totalBytes) {
+    if (totalRows === undefined && totalBytes === undefined)
+        return undefined;
+    const stats = {};
+    if (totalRows !== undefined)
+        stats.totalRows = totalRows;
+    if (totalBytes !== undefined)
+        stats.totalBytes = totalBytes;
+    return stats;
+}
+//# sourceMappingURL=clickhouse.js.map