@kaelio/ktx 0.4.1 → 0.5.0

package/node_modules/@ktx/connector-snowflake/dist/connector.js

@@ -1,517 +0,0 @@
-import { createPrivateKey } from 'node:crypto';
-import { readFileSync } from 'node:fs';
-import { homedir } from 'node:os';
-import { resolve } from 'node:path';
-import { assertReadOnlySql, limitSqlForExecution } from '@ktx/context/connections';
-import { createKtxConnectorCapabilities, } from '@ktx/context/scan';
-import * as snowflake from 'snowflake-sdk';
-import { KtxSnowflakeDialect } from './dialect.js';
-const DATE_TYPES = ['DATE', 'TIMESTAMP', 'TIMESTAMP_LTZ', 'TIMESTAMP_NTZ', 'TIMESTAMP_TZ', 'TIME'];
-function resolveStringReference(value, env) {
-    if (value.startsWith('env:')) {
-        return env[value.slice('env:'.length)] ?? '';
-    }
-    if (value.startsWith('file:')) {
-        const rawPath = value.slice('file:'.length);
-        const path = rawPath.startsWith('~') ? resolve(homedir(), rawPath.slice(1)) : rawPath;
-        return readFileSync(path, 'utf-8').trim();
-    }
-    return value;
-}
-function stringConfigValue(connection, key, env) {
-    const value = connection?.[key];
-    return typeof value === 'string' && value.trim().length > 0 ? resolveStringReference(value.trim(), env) : undefined;
-}
-function schemaNames(connection, env) {
-    if (Array.isArray(connection.schema_names) && connection.schema_names.length > 0) {
-        return connection.schema_names
-            .filter((schema) => schema.trim().length > 0)
-            .map((schema) => resolveStringReference(schema, env));
-    }
-    return [stringConfigValue(connection, 'schema_name', env) ?? 'PUBLIC'];
-}
-function firstNumber(value) {
-    const numberValue = Number(value);
-    return Number.isFinite(numberValue) ? numberValue : null;
-}
-function normalizeSnowflakeValue(value, columnType) {
-    if (columnType && DATE_TYPES.some((type) => columnType.toUpperCase().includes(type))) {
-        if (typeof value === 'number') {
-            return new Date(value).toISOString();
-        }
-        if (value instanceof Date) {
-            return value.toISOString();
-        }
-    }
-    if (typeof value === 'string') {
-        const trimmed = value.trim();
-        if ((trimmed.startsWith('{') && trimmed.endsWith('}')) || (trimmed.startsWith('[') && trimmed.endsWith(']'))) {
-            try {
-                return JSON.parse(trimmed);
-            }
-            catch {
-                return value;
-            }
-        }
-    }
-    return value;
-}
-function toSnowflakeBind(value) {
-    if (value === null || typeof value === 'string' || typeof value === 'number' || typeof value === 'boolean') {
-        return value;
-    }
-    if (value instanceof Date) {
-        return value.toISOString();
-    }
-    return String(value);
-}
-function toSnowflakeBinds(params) {
-    return params?.map((value) => toSnowflakeBind(value));
-}
-export function isKtxSnowflakeConnectionConfig(connection) {
-    return String(connection?.driver ?? '').toLowerCase() === 'snowflake';
-}
-export function snowflakeConnectionConfigFromConfig(input) {
-    const inputDriver = input.connection?.driver ?? 'unknown';
-    if (!isKtxSnowflakeConnectionConfig(input.connection)) {
-        throw new Error(`Native Snowflake connector cannot run driver "${inputDriver}"`);
-    }
-    const env = input.env ?? process.env;
-    const authMethod = input.connection?.authMethod ?? 'password';
-    const account = stringConfigValue(input.connection, 'account', env);
-    const warehouse = stringConfigValue(input.connection, 'warehouse', env);
-    const database = stringConfigValue(input.connection, 'database', env);
-    const username = stringConfigValue(input.connection, 'username', env);
-    if (!account) {
-        throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.account`);
-    }
-    if (!warehouse) {
-        throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.warehouse`);
-    }
-    if (!database) {
-        throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.database`);
-    }
-    if (!username) {
-        throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.username`);
-    }
-    const resolved = {
-        authMethod,
-        account,
-        warehouse,
-        database,
-        schemas: schemaNames(input.connection, env),
-        username,
-    };
-    const role = stringConfigValue(input.connection, 'role', env);
-    if (role) {
-        resolved.role = role;
-    }
-    if (authMethod === 'rsa') {
-        resolved.privateKey = stringConfigValue(input.connection, 'privateKey', env);
-        const passphrase = stringConfigValue(input.connection, 'passphrase', env);
-        if (passphrase) {
-            resolved.passphrase = passphrase;
-        }
-        if (!resolved.privateKey) {
-            throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.privateKey for RSA auth`);
-        }
-    }
-    else {
-        resolved.password = stringConfigValue(input.connection, 'password', env);
-        if (!resolved.password) {
-            throw new Error(`Native Snowflake connector requires connections.${input.connectionId}.password`);
-        }
-    }
-    return resolved;
-}
-class DefaultSnowflakeDriverFactory {
-    createDriver(input) {
-        return new SnowflakeSdkDriver(input.resolved, input.sdkOptionsProvider);
-    }
-}
-class SnowflakeSdkDriver {
-    resolved;
-    sdkOptionsProvider;
-    closeSdkOptions = [];
-    constructor(resolved, sdkOptionsProvider) {
-        this.resolved = resolved;
-        this.sdkOptionsProvider = sdkOptionsProvider;
-    }
-    async test() {
-        const timeoutMs = 60_000;
-        return Promise.race([
-            this.runTest(),
-            new Promise((resolveTest) => setTimeout(() => resolveTest({ success: false, error: `Connection test timed out after ${timeoutMs / 1000}s` }), timeoutMs)),
-        ]);
-    }
-    async query(sql, params) {
-        let connection = null;
-        try {
-            connection = await this.createConnection();
-            const binds = Array.isArray(params) ? toSnowflakeBinds(params) : undefined;
-            const result = await this.executeSnowflakeQuery(connection, sql, binds);
-            return { ...result, totalRows: result.rows.length, rowCount: result.rows.length };
-        }
-        catch {
-            return { headers: [], rows: [], totalRows: 0, rowCount: 0 };
-        }
-        finally {
-            if (connection) {
-                await this.destroyConnection(connection);
-            }
-        }
-    }
-    async getSchemaMetadata(schemaName = this.resolved.schemas[0] ?? 'PUBLIC') {
-        const tablesResult = await this.query(`
-        SELECT TABLE_NAME, TABLE_TYPE, COMMENT, ROW_COUNT
-        FROM INFORMATION_SCHEMA.TABLES
-        WHERE TABLE_SCHEMA = ? AND TABLE_CATALOG = ?
-        ORDER BY TABLE_NAME
-      `, [schemaName, this.resolved.database]);
-        const columnsResult = await this.query(`
-        SELECT TABLE_NAME, COLUMN_NAME, DATA_TYPE, IS_NULLABLE, COMMENT, ORDINAL_POSITION
-        FROM INFORMATION_SCHEMA.COLUMNS
-        WHERE TABLE_SCHEMA = ? AND TABLE_CATALOG = ?
-        ORDER BY TABLE_NAME, ORDINAL_POSITION
-      `, [schemaName, this.resolved.database]);
-        const columnsByTable = new Map();
-        for (const row of columnsResult.rows) {
-            const tableName = String(row[0]);
-            const columns = columnsByTable.get(tableName) ?? [];
-            columns.push({
-                name: String(row[1]),
-                type: String(row[2]),
-                nullable: row[3] === 'YES',
-                comment: row[4] ? String(row[4]) : null,
-            });
-            columnsByTable.set(tableName, columns);
-        }
-        return tablesResult.rows.map((row) => ({
-            name: String(row[0]),
-            catalog: this.resolved.database,
-            db: schemaName,
-            rowCount: firstNumber(row[3]) ?? 0,
-            comment: row[2] ? String(row[2]) : null,
-            columns: columnsByTable.get(String(row[0])) ?? [],
-        }));
-    }
-    async listSchemas() {
-        const result = await this.query(`SHOW SCHEMAS IN DATABASE "${this.resolved.database}"`);
-        return result.rows.map((row) => String(row[1])).filter((name) => name !== 'INFORMATION_SCHEMA');
-    }
-    async listTables(schemas) {
-        const filterSchemas = schemas ?? (await this.listSchemas());
-        if (filterSchemas.length === 0)
-            return [];
-        const entries = [];
-        for (const schemaName of filterSchemas) {
-            const result = await this.query(`
-        SELECT TABLE_NAME, TABLE_TYPE
-        FROM INFORMATION_SCHEMA.TABLES
-        WHERE TABLE_SCHEMA = ? AND TABLE_CATALOG = ?
-        ORDER BY TABLE_NAME
-        `, [schemaName, this.resolved.database]);
-            for (const row of result.rows) {
-                entries.push({
-                    schema: schemaName,
-                    name: String(row[0]),
-                    kind: String(row[1]) === 'VIEW' ? 'view' : 'table',
-                });
-            }
-        }
-        return entries;
-    }
-    async cleanup() {
-        const closers = this.closeSdkOptions;
-        this.closeSdkOptions = [];
-        await Promise.all(closers.map((close) => close()));
-    }
-    async runTest() {
-        let connection = null;
-        try {
-            connection = await this.createConnection();
-            await this.executeSnowflakeQuery(connection, 'SELECT 1');
-            return { success: true };
-        }
-        catch (error) {
-            return { success: false, error: error instanceof Error ? error.message : String(error) };
-        }
-        finally {
-            if (connection) {
-                await this.destroyConnection(connection);
-            }
-        }
-    }
-    async createConnection() {
-        const patch = await this.sdkOptionsProvider?.resolve({
-            account: this.resolved.account,
-            connection: { ...this.resolved, driver: 'snowflake' },
-        });
-        if (patch?.close) {
-            this.closeSdkOptions.push(patch.close);
-        }
-        const baseConfig = {
-            account: this.resolved.account,
-            username: this.resolved.username,
-            warehouse: this.resolved.warehouse,
-            database: this.resolved.database,
-            schema: this.resolved.schemas[0] ?? 'PUBLIC',
-            role: this.resolved.role,
-            ...patch?.sdkOptions,
-        };
-        const connectionConfig = this.resolved.authMethod === 'rsa'
-            ? { ...baseConfig, authenticator: 'SNOWFLAKE_JWT', privateKey: this.decryptPrivateKey() }
-            : { ...baseConfig, password: this.resolved.password };
-        const connection = snowflake.createConnection(connectionConfig);
-        return new Promise((resolveConnection, rejectConnection) => {
-            connection.connect((error, connected) => {
-                if (error) {
-                    rejectConnection(error);
-                    return;
-                }
-                const resolvedConnection = connected ?? connection;
-                this.setConnectionContext(resolvedConnection).then(() => resolveConnection(resolvedConnection), (contextError) => {
-                    resolvedConnection.destroy(() => undefined);
-                    rejectConnection(contextError);
-                });
-            });
-        });
-    }
-    async setConnectionContext(connection) {
-        if (this.resolved.role) {
-            await this.executeSnowflakeQuery(connection, `USE ROLE "${this.resolved.role}"`);
-        }
-        await this.executeSnowflakeQuery(connection, `USE WAREHOUSE "${this.resolved.warehouse}"`);
-        await this.executeSnowflakeQuery(connection, `USE DATABASE "${this.resolved.database}"`);
-        await this.executeSnowflakeQuery(connection, `USE SCHEMA "${this.resolved.schemas[0] ?? 'PUBLIC'}"`);
-    }
-    async executeSnowflakeQuery(connection, sqlText, binds) {
-        return new Promise((resolveQuery, rejectQuery) => {
-            connection.execute({
-                sqlText,
-                binds,
-                complete: (error, statement, rows) => {
-                    if (error) {
-                        rejectQuery(error);
-                        return;
-                    }
-                    const columns = statement.getColumns();
-                    const headers = columns ? columns.map((column) => column.getName()) : [];
-                    const headerTypes = columns ? columns.map((column) => column.getType()) : [];
-                    const normalizedRows = rows
-                        ? rows.map((row) => headers.map((header, index) => normalizeSnowflakeValue(row[header], headerTypes[index])))
-                        : [];
-                    resolveQuery({ headers, headerTypes, rows: normalizedRows });
-                },
-            });
-        });
-    }
-    destroyConnection(connection) {
-        return new Promise((resolveDestroy, rejectDestroy) => {
-            connection.destroy((error) => {
-                if (error) {
-                    rejectDestroy(error);
-                    return;
-                }
-                resolveDestroy();
-            });
-        });
-    }
-    decryptPrivateKey() {
-        if (!this.resolved.privateKey) {
-            throw new Error('Private key is required for RSA authentication');
-        }
-        const privateKeyObject = createPrivateKey({
-            key: this.resolved.privateKey,
-            format: 'pem',
-            ...(this.resolved.passphrase ? { passphrase: this.resolved.passphrase } : {}),
-        });
-        return privateKeyObject.export({ format: 'pem', type: 'pkcs8' });
-    }
-}
-export class KtxSnowflakeScanConnector {
-    options;
-    id;
-    driver = 'snowflake';
-    capabilities = createKtxConnectorCapabilities({
-        tableSampling: true,
-        columnSampling: true,
-        columnStats: false,
-        readOnlySql: true,
-        nestedAnalysis: true,
-        formalForeignKeys: false,
-        estimatedRowCounts: true,
-    });
-    resolved;
-    driverFactory;
-    dialect = new KtxSnowflakeDialect();
-    now;
-    driverInstance = null;
-    constructor(options) {
-        this.options = options;
-        this.resolved = snowflakeConnectionConfigFromConfig(options);
-        this.driverFactory = options.driverFactory ?? new DefaultSnowflakeDriverFactory();
-        this.now = options.now ?? (() => new Date());
-        this.id = `snowflake:${options.connectionId}`;
-    }
-    async testConnection() {
-        return this.getDriver().test();
-    }
-    async introspect(input, _ctx) {
-        this.assertConnection(input.connectionId);
-        const tables = [];
-        for (const schemaName of this.resolved.schemas) {
-            const rawTables = await this.getDriver().getSchemaMetadata(schemaName);
-            const primaryKeys = await this.primaryKeys(rawTables.map((table) => table.name), schemaName);
-            tables.push(...rawTables.map((table) => this.toSchemaTable(table, primaryKeys)));
-        }
-        return {
-            connectionId: this.options.connectionId,
-            driver: 'snowflake',
-            extractedAt: this.now().toISOString(),
-            scope: { catalogs: [this.resolved.database], schemas: this.resolved.schemas },
-            metadata: {
-                account: this.resolved.account,
-                warehouse: this.resolved.warehouse,
-                database: this.resolved.database,
-                schemas: this.resolved.schemas,
-                table_count: tables.length,
-                total_columns: tables.reduce((sum, table) => sum + table.columns.length, 0),
-            },
-            tables,
-        };
-    }
-    async sampleTable(input, _ctx) {
-        this.assertConnection(input.connectionId);
-        const result = await this.getDriver().query(this.dialect.generateSampleQuery(this.qTableName(input.table), input.limit, input.columns));
-        return { headers: result.headers, rows: result.rows, totalRows: result.totalRows };
-    }
-    async sampleColumn(input, _ctx) {
-        this.assertConnection(input.connectionId);
-        const result = await this.getDriver().query(this.dialect.generateColumnSampleQuery(this.qTableName(input.table), input.column, input.limit));
-        return {
-            values: result.rows.filter((row) => row.length > 0 && row[0] !== null).map((row) => row[0]),
-            nullCount: null,
-            distinctCount: null,
-        };
-    }
-    async columnStats(_input, _ctx) {
-        return null;
-    }
-    async executeReadOnly(input, _ctx) {
-        this.assertConnection(input.connectionId);
-        const limitedSql = limitSqlForExecution(assertReadOnlySql(input.sql), input.maxRows);
-        const prepared = this.dialect.prepareQuery(limitedSql, input.params);
-        return this.getDriver().query(prepared.sql, prepared.params);
-    }
-    async getColumnDistinctValues(table, columnName, options) {
-        const tableName = this.qTableName(table);
-        const quotedColumn = this.dialect.quoteIdentifier(columnName);
-        const cardinality = await this.singleNumber(this.dialect.generateCardinalitySampleQuery(tableName, quotedColumn, options.sampleSize ?? 10000), 'CARDINALITY');
-        if (cardinality === null) {
-            return null;
-        }
-        if (cardinality === 0) {
-            return { values: [], cardinality: 0 };
-        }
-        if (cardinality > options.maxCardinality) {
-            return { values: null, cardinality };
-        }
-        const valueRows = await this.queryRaw(this.dialect.generateDistinctValuesQuery(tableName, quotedColumn, options.limit));
-        return { values: valueRows.map((row) => String(row.VAL ?? row.val)).filter((value) => value !== 'null'), cardinality };
-    }
-    async getTableRowCount(tableName, schemaName = this.resolved.schemas[0] ?? 'PUBLIC') {
-        const tables = await this.getDriver().getSchemaMetadata(schemaName);
-        return tables.find((table) => table.name === tableName)?.rowCount ?? 0;
-    }
-    qTableName(table) {
-        return this.dialect.formatTableName(table);
-    }
-    quoteIdentifier(identifier) {
-        return this.dialect.quoteIdentifier(identifier);
-    }
-    listSchemas() {
-        return this.getDriver().listSchemas();
-    }
-    listTables(schemas) {
-        return this.getDriver().listTables(schemas);
-    }
-    async cleanup() {
-        if (this.driverInstance) {
-            await this.driverInstance.cleanup();
-            this.driverInstance = null;
-        }
-    }
-    getDriver() {
-        if (!this.driverInstance) {
-            this.driverInstance = this.driverFactory.createDriver({
-                resolved: this.resolved,
-                sdkOptionsProvider: this.options.sdkOptionsProvider,
-            });
-        }
-        return this.driverInstance;
-    }
-    async primaryKeys(tableNames, schemaName) {
-        if (tableNames.length === 0) {
-            return new Map();
-        }
-        const result = await this.getDriver().query(`
-        SELECT tc.TABLE_NAME, kcu.COLUMN_NAME
-        FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS tc
-        JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE kcu
-          ON tc.CONSTRAINT_NAME = kcu.CONSTRAINT_NAME
-          AND tc.TABLE_SCHEMA = kcu.TABLE_SCHEMA
-          AND tc.TABLE_CATALOG = kcu.TABLE_CATALOG
-        WHERE tc.CONSTRAINT_TYPE = 'PRIMARY KEY'
-          AND tc.TABLE_SCHEMA = ?
-          AND tc.TABLE_CATALOG = ?
-        ORDER BY tc.TABLE_NAME, kcu.ORDINAL_POSITION
-      `, [schemaName, this.resolved.database]);
-        const grouped = new Map();
-        for (const tableName of tableNames) {
-            grouped.set(tableName, new Set());
-        }
-        for (const row of result.rows) {
-            const tableName = String(row[0]);
-            const columnName = String(row[1]);
-            grouped.get(tableName)?.add(columnName);
-        }
-        return grouped;
-    }
-    toSchemaTable(table, primaryKeys) {
-        return {
-            catalog: table.catalog,
-            db: table.db,
-            name: table.name,
-            kind: 'table',
-            comment: table.comment,
-            estimatedRows: table.rowCount,
-            columns: table.columns.map((column) => this.toSchemaColumn(table.name, column, primaryKeys)),
-            foreignKeys: [],
-        };
-    }
-    toSchemaColumn(tableName, column, primaryKeys) {
-        return {
-            name: column.name,
-            nativeType: column.type,
-            normalizedType: this.dialect.mapDataType(column.type),
-            dimensionType: this.dialect.mapToDimensionType(column.type),
-            nullable: column.nullable,
-            primaryKey: primaryKeys.get(tableName)?.has(column.name) ?? false,
-            comment: column.comment,
-        };
-    }
-    async queryRaw(sql, params) {
-        const result = await this.getDriver().query(sql, params);
-        return result.rows.map((row) => Object.fromEntries(result.headers.map((header, index) => [header, row[index]])));
-    }
-    async singleNumber(sql, header) {
-        const rows = await this.queryRaw(sql);
-        return firstNumber(rows[0]?.[header] ?? rows[0]?.[header.toLowerCase()]);
-    }
-    assertConnection(connectionId) {
-        if (connectionId !== this.options.connectionId) {
-            throw new Error(`Snowflake connector ${this.options.connectionId} cannot scan connection ${connectionId}`);
-        }
-    }
-}