@trafficbyintent/kysely-bigquery 2.0.0

package/dist/BigQueryCompiler.js

@@ -0,0 +1,244 @@
+"use strict";
+var _BigQueryCompiler_defaultProject;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BigQueryCompiler = void 0;
+const tslib_1 = require("tslib");
+const kysely_1 = require("kysely");
+/**
+ * Query compiler for BigQuery dialect.
+ *
+ * Extends MysqlQueryCompiler and overrides methods to generate
+ * BigQuery-compatible SQL.
+ */
+class BigQueryCompiler extends kysely_1.MysqlQueryCompiler {
+    constructor(defaultProject) {
+        super();
+        _BigQueryCompiler_defaultProject.set(this, void 0);
+        tslib_1.__classPrivateFieldSet(this, _BigQueryCompiler_defaultProject, defaultProject, "f");
+    }
+    visitSetOperation(node) {
+        if (node.operator === 'union' && !node.all) {
+            this.append('union distinct ');
+            this.visitNode(node.expression);
+        }
+        else {
+            super.visitSetOperation(node);
+        }
+    }
+    visitFunction(node) {
+        const funcName = node.func.toUpperCase();
+        switch (funcName) {
+            case 'NOW':
+                this.append('CURRENT_TIMESTAMP');
+                this.visitFunctionArgumentList(node.arguments);
+                return;
+            case 'LENGTH':
+                /* BigQuery uses CHAR_LENGTH for character count */
+                this.append('CHAR_LENGTH');
+                this.visitFunctionArgumentList(node.arguments);
+                return;
+            case 'DATE_FORMAT':
+                if (node.arguments.length === 2 &&
+                    node.arguments[0] !== null &&
+                    node.arguments[0] !== undefined &&
+                    node.arguments[1] !== null &&
+                    node.arguments[1] !== undefined) {
+                    this.append('FORMAT_TIMESTAMP');
+                    this.append('(');
+                    this.visitNode(node.arguments[1]);
+                    this.append(', ');
+                    this.visitNode(node.arguments[0]);
+                    this.append(')');
+                    return;
+                }
+                break;
+            case 'DATE_ADD':
+                /*
+                 * Pass through DATE_ADD as BigQuery supports it
+                 * Note: BigQuery syntax is DATE_ADD(date, INTERVAL n unit)
+                 */
+                break;
+            default:
+                /* Use default handler for all other functions */
+                break;
+        }
+        super.visitFunction(node);
+    }
+    visitUpdateQuery(node) {
+        if (node.where === null || node.where === undefined) {
+            /*
+             * BigQuery requires WHERE clause for UPDATE statements
+             * Add WHERE TRUE to allow the query to execute
+             */
+            super.visitUpdateQuery(node);
+            this.append(' where true');
+            return;
+        }
+        super.visitUpdateQuery(node);
+    }
+    visitDeleteQuery(node) {
+        if (node.where === null || node.where === undefined) {
+            /*
+             * BigQuery requires WHERE clause for DELETE statements
+             * Add WHERE TRUE to allow the query to execute
+             */
+            super.visitDeleteQuery(node);
+            this.append(' where true');
+            return;
+        }
+        super.visitDeleteQuery(node);
+    }
+    visitTable(node) {
+        if (tslib_1.__classPrivateFieldGet(this, _BigQueryCompiler_defaultProject, "f") && node.table.kind === 'SchemableIdentifierNode') {
+            const schemableNode = node.table;
+            if (schemableNode.schema && schemableNode.schema.kind === 'IdentifierNode') {
+                this.visitIdentifier({
+                    kind: 'IdentifierNode',
+                    name: tslib_1.__classPrivateFieldGet(this, _BigQueryCompiler_defaultProject, "f"),
+                });
+                this.append('.');
+                this.visitIdentifier(schemableNode.schema);
+                this.append('.');
+                this.visitIdentifier(schemableNode.identifier);
+                return;
+            }
+        }
+        super.visitTable(node);
+    }
+    visitRaw(node) {
+        const translatedFragments = [...node.sqlFragments];
+        for (let i = 0; i < translatedFragments.length; i++) {
+            let fragment = translatedFragments[i];
+            if (fragment === null || fragment === undefined || fragment === '') {
+                continue;
+            }
+            fragment = fragment.replace(/\bNOW\s*\(\s*\)/gi, 'CURRENT_TIMESTAMP()');
+            translatedFragments[i] = fragment;
+        }
+        if (translatedFragments.some((f) => f.match(/\bDATE_FORMAT\s*\(/i))) {
+            for (let i = 0; i < translatedFragments.length; i++) {
+                let fragment = translatedFragments[i];
+                /* c8 ignore start - defensive: fragments already filtered above */
+                if (fragment === null || fragment === undefined || fragment === '') {
+                    continue;
+                }
+                /* c8 ignore stop */
+                const fullMatch = fragment.match(/\bDATE_FORMAT\s*\(\s*([^,]+)\s*,\s*('[^']*')\s*\)/gi);
+                if (fullMatch) {
+                    fragment = fragment.replace(/\bDATE_FORMAT\s*\(\s*([^,]+)\s*,\s*('[^']*')\s*\)/gi, 'FORMAT_TIMESTAMP($2, $1)');
+                    translatedFragments[i] = fragment;
+                }
+                else {
+                    translatedFragments[i] = fragment.replace(/\bDATE_FORMAT\s*\(/gi, 'FORMAT_TIMESTAMP(');
+                }
+            }
+            if (translatedFragments.length === 2 && node.parameters.length === 1) {
+                const secondFragment = translatedFragments[1];
+                if (secondFragment) {
+                    const formatMatch = secondFragment.match(/,\s*'([^']*)'(.*)$/);
+                    if (formatMatch && translatedFragments[0] && node.parameters[0]) {
+                        this.append(translatedFragments[0]);
+                        this.append(`'${formatMatch[1]}'`);
+                        this.append(', ');
+                        this.visitNode(node.parameters[0]);
+                        this.append(formatMatch[2] || '');
+                        return;
+                    }
+                }
+            }
+            this.appendFragmentsWithParams(translatedFragments, node.parameters);
+        }
+        else {
+            this.appendFragmentsWithParams(translatedFragments, node.parameters);
+        }
+    }
+    appendFragmentsWithParams(fragments, params) {
+        for (let i = 0; i < fragments.length; i++) {
+            if (i > 0) {
+                const param = params[i - 1];
+                if (param) {
+                    this.visitNode(param);
+                }
+            }
+            const fragment = fragments[i];
+            if (fragment) {
+                this.append(fragment);
+            }
+        }
+    }
+    visitFunctionArgumentList(args) {
+        this.append('(');
+        const lastNode = args[args.length - 1];
+        for (const node of args) {
+            this.visitNode(node);
+            /* c8 ignore start - defensive: currently only called with 0-1 args */
+            if (node !== lastNode) {
+                this.append(', ');
+            }
+            /* c8 ignore stop */
+        }
+        this.append(')');
+    }
+    visitColumnDefinition(node) {
+        /* Call parent implementation first */
+        super.visitColumnDefinition(node);
+        /* If column has inline constraints, append NOT ENFORCED */
+        if (node.primaryKey || node.unique || node.references) {
+            this.append(' not enforced');
+        }
+    }
+    visitPrimaryKeyConstraint(node) {
+        if (node.name) {
+            this.append('constraint ');
+            this.visitNode(node.name);
+            this.append(' ');
+        }
+        this.append('primary key');
+        if (node.columns) {
+            this.append(' (');
+            this.compileList(node.columns);
+            this.append(')');
+        }
+        /* BigQuery requires NOT ENFORCED for all constraints */
+        this.append(' not enforced');
+    }
+    visitUniqueConstraint(node) {
+        if (node.name) {
+            this.append('constraint ');
+            this.visitNode(node.name);
+            this.append(' ');
+        }
+        this.append('unique');
+        if (node.columns) {
+            this.append(' (');
+            this.compileList(node.columns);
+            this.append(')');
+        }
+        /* BigQuery requires NOT ENFORCED for all constraints */
+        this.append(' not enforced');
+    }
+    visitForeignKeyConstraint(node) {
+        if (node.name) {
+            this.append('constraint ');
+            this.visitNode(node.name);
+            this.append(' ');
+        }
+        this.append('foreign key (');
+        this.compileList(node.columns);
+        this.append(') ');
+        this.visitNode(node.references);
+        if (node.onDelete) {
+            this.append(' on delete ');
+            this.append(node.onDelete);
+        }
+        if (node.onUpdate) {
+            this.append(' on update ');
+            this.append(node.onUpdate);
+        }
+        /* BigQuery requires NOT ENFORCED for all constraints */
+        this.append(' not enforced');
+    }
+}
+exports.BigQueryCompiler = BigQueryCompiler;
+_BigQueryCompiler_defaultProject = new WeakMap();
+//# sourceMappingURL=BigQueryCompiler.js.map

package/dist/BigQueryCompiler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BigQueryCompiler.js","sourceRoot":"","sources":["../src/BigQueryCompiler.ts"],"names":[],"mappings":";;;;;AAAA,mCAcgB;AAEhB;;;;;GAKG;AACH,MAAa,gBAAiB,SAAQ,2BAAkB;IAGtD,YAAY,cAAuB;QACjC,KAAK,EAAE,CAAC;QAHD,mDAAoC;QAI3C,+BAAA,IAAI,oCAAmB,cAAc,MAAA,CAAC;IACxC,CAAC;IAEkB,iBAAiB,CAAC,IAAsB;QACzD,IAAI,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3C,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEkB,aAAa,CAAC,IAAkB;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAEzC,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,KAAK;gBACR,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;gBACjC,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC/C,OAAO;YAET,KAAK,QAAQ;gBACX,mDAAmD;gBACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBAC3B,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC/C,OAAO;YAET,KAAK,aAAa;gBAChB,IACE,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC;oBAC3B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,IAAI;oBAC1B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,SAAS;oBAC/B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,IAAI;oBAC1B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,SAAS,EAC/B,CAAC;oBACD,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;oBAChC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBACjB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBACjB,OAAO;gBACT,CAAC;gBACD,MAAM;YAER,KAAK,UAAU;gBACb;;;mBAGG;gBACH,MAAM;YAER;gBACE,iDAAiD;gBACjD,MAAM;QACV,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEkB,gBAAgB,CAAC,IAAqB;QACvD,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACpD;;;eAGG;YACH,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO;QACT,CAAC;QACD,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAEkB,gBAAgB,CAAC,IAAqB;QACvD,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACpD;;;eAGG;YACH,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO;QACT,CAAC;QACD,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAEkB,UAAU,CAAC,IAAe;QAC3C,IAAI,+BAAA,IAAI,wCAAgB,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;YAC1E,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;YAEjC,IAAI,aAAa,CAAC,MAAM,IAAI,aAAa,CAAC,MAAM,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBAC3E,IAAI,CAAC,eAAe,CAAC;oBACnB,IAAI,EAAE,gBAAgB;oBACtB,IAAI,EAAE,+BAAA,IAAI,wCAAgB;iBACT,CAAC,CAAC;gBACrB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjB,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;gBAC3C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjB,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBAC/C,OAAO;YACT,CAAC;QACH,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAEkB,QAAQ,CAAC,IAAa;QACvC,MAAM,mBAAmB,GAAG,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;QAEnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpD,IAAI,QAAQ,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;YACtC,IAAI,QAAQ,KAAK,IAAI,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,EAAE,EAAE,CAAC;gBACnE,SAAS;YACX,CAAC;YAED,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,mBAAmB,EAAE,qBAAqB,CAAC,CAAC;YAExE,mBAAmB,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC;QACpC,CAAC;QAED,IAAI,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,EAAE,CAAC;YACpE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,IAAI,QAAQ,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;gBACtC,mEAAmE;gBACnE,IAAI,QAAQ,KAAK,IAAI,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,EAAE,EAAE,CAAC;oBACnE,SAAS;gBACX,CAAC;gBACD,oBAAoB;gBAEpB,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;gBACxF,IAAI,SAAS,EAAE,CAAC;oBACd,QAAQ,GAAG,QAAQ,CAAC,OAAO,CACzB,qDAAqD,EACrD,0BAA0B,CAC3B,CAAC;oBACF,mBAAmB,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,mBAAmB,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC,sBAAsB,EAAE,mBAAmB,CAAC,CAAC;gBACzF,CAAC;YACH,CAAC;YAED,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrE,MAAM,cAAc,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;gBAC9C,IAAI,cAAc,EAAE,CAAC;oBACnB,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;oBAC/D,IAAI,WAAW,IAAI,mBAAmB,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;wBAChE,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;wBACpC,IAAI,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;wBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;wBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;wBACnC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,CAAC,yBAAyB,CAAC,mBAAmB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACvE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,yBAAyB,CAAC,mBAAmB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAEO,yBAAyB,CAC/B,SAAmB,EACnB,MAAoC;QAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC5B,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YAC9B,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IAES,yBAAyB,CAAC,IAAkC;QACpE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAErB,sEAAsE;YACtE,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;YACD,oBAAoB;QACtB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAEkB,qBAAqB,CAAC,IAA0B;QACjE,sCAAsC;QACtC,KAAK,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAElC,2DAA2D;QAC3D,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACtD,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEkB,yBAAyB,CAAC,IAA8B;QACzE,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC3B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,wDAAwD;QACxD,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC/B,CAAC;IAEkB,qBAAqB,CAAC,IAA0B;QACjE,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,wDAAwD;QACxD,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC/B,CAAC;IAEkB,yBAAyB,CAAC,IAA8B;QACzE,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAC7B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;QACD,wDAAwD;QACxD,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC/B,CAAC;CACF;AAxQD,4CAwQC"}

package/dist/BigQueryConnection.d.ts

@@ -0,0 +1,41 @@
+import { type CompiledQuery, type DatabaseConnection, type QueryResult } from 'kysely';
+import { type BigQueryDialectConfig } from './BigQueryDialect';
+/**
+ * BigQuery database connection implementation for Kysely.
+ *
+ * Handles query execution and streaming for BigQuery.
+ */
+export declare class BigQueryConnection implements DatabaseConnection {
+    #private;
+    constructor(config: BigQueryDialectConfig);
+    /**
+     * Executes a compiled query against BigQuery.
+     * @param compiledQuery - The compiled query with SQL and parameters
+     * @returns A promise that resolves to the query results
+     * @throws Error if the query fails or if null parameters are not properly typed
+     */
+    executeQuery<O>(compiledQuery: CompiledQuery): Promise<QueryResult<O>>;
+    /**
+     * Begins a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    beginTransaction(): Promise<void>;
+    /**
+     * Commits a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    commitTransaction(): Promise<void>;
+    /**
+     * Rolls back a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    rollbackTransaction(): Promise<void>;
+    /**
+     * Streams query results for handling large datasets.
+     * @param compiledQuery - The compiled query with SQL and parameters
+     * @param _chunkSize - Chunk size parameter (currently unused)
+     * @returns An async iterator that yields query results in batches
+     */
+    streamQuery<O>(compiledQuery: CompiledQuery, _chunkSize: number): AsyncIterableIterator<QueryResult<O>>;
+}
+//# sourceMappingURL=BigQueryConnection.d.ts.map

package/dist/BigQueryConnection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BigQueryConnection.d.ts","sourceRoot":"","sources":["../src/BigQueryConnection.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,aAAa,EAAE,KAAK,kBAAkB,EAAE,KAAK,WAAW,EAAE,MAAM,QAAQ,CAAC;AAIvF,OAAO,EAAE,KAAK,qBAAqB,EAAE,MAAM,mBAAmB,CAAC;AAE/D;;;;GAIG;AACH,qBAAa,kBAAmB,YAAW,kBAAkB;;gBAK/C,MAAM,EAAE,qBAAqB;IAczC;;;;;OAKG;IACG,YAAY,CAAC,CAAC,EAAE,aAAa,EAAE,aAAa,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IA8C5E;;;OAGG;IACH,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjC;;;OAGG;IACH,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIlC;;;OAGG;IACH,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC;;;;;OAKG;IACI,WAAW,CAAC,CAAC,EAClB,aAAa,EAAE,aAAa,EAC5B,UAAU,EAAE,MAAM,GACjB,qBAAqB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;CA2HzC"}

package/dist/BigQueryConnection.js

@@ -0,0 +1,222 @@
+"use strict";
+var _BigQueryConnection_instances, _BigQueryConnection_client, _BigQueryConnection_jsonDetector, _BigQueryConnection_jsonColumnNames, _BigQueryConnection_inferParamTypes, _BigQueryConnection_inferArrayType, _BigQueryConnection_processResultRow;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BigQueryConnection = void 0;
+const tslib_1 = require("tslib");
+const bigquery_1 = require("@google-cloud/bigquery");
+const jsonColumnDetector_1 = require("./jsonColumnDetector");
+/**
+ * BigQuery database connection implementation for Kysely.
+ *
+ * Handles query execution and streaming for BigQuery.
+ */
+class BigQueryConnection {
+    constructor(config) {
+        var _a;
+        _BigQueryConnection_instances.add(this);
+        _BigQueryConnection_client.set(this, void 0);
+        _BigQueryConnection_jsonDetector.set(this, void 0);
+        _BigQueryConnection_jsonColumnNames.set(this, void 0);
+        tslib_1.__classPrivateFieldSet(this, _BigQueryConnection_client, (_a = config.bigquery) !== null && _a !== void 0 ? _a : new bigquery_1.BigQuery(config.options), "f");
+        tslib_1.__classPrivateFieldSet(this, _BigQueryConnection_jsonDetector, new jsonColumnDetector_1.JsonColumnDetector(), "f");
+        /* Register known JSON columns if provided in config */
+        if (config.jsonColumns) {
+            for (const [tableName, columns] of Object.entries(config.jsonColumns)) {
+                tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonDetector, "f").registerJsonColumns(tableName, columns);
+            }
+        }
+        tslib_1.__classPrivateFieldSet(this, _BigQueryConnection_jsonColumnNames, tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonDetector, "f").getRegisteredJsonColumnNames(), "f");
+    }
+    /**
+     * Executes a compiled query against BigQuery.
+     * @param compiledQuery - The compiled query with SQL and parameters
+     * @returns A promise that resolves to the query results
+     * @throws Error if the query fails or if null parameters are not properly typed
+     */
+    executeQuery(compiledQuery) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            try {
+                const params = [...compiledQuery.parameters];
+                /* Process parameters to handle nulls and JSON serialization */
+                const processedParams = tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonDetector, "f").processParameters(compiledQuery, params);
+                const options = {
+                    query: compiledQuery.sql,
+                    params: processedParams,
+                    parseJSON: true,
+                };
+                /* BigQuery needs types array for ALL parameters when there are null parameters */
+                if (processedParams.some((p) => p === null)) {
+                    options.types = tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_instances, "m", _BigQueryConnection_inferParamTypes).call(this, processedParams);
+                }
+                const [rows] = yield tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_client, "f").query(options);
+                const processedRows = Array.isArray(rows)
+                    ? rows.map((row) => tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_instances, "m", _BigQueryConnection_processResultRow).call(this, row))
+                    : [];
+                return {
+                    rows: processedRows,
+                };
+            }
+            catch (error) {
+                /* Provide more helpful error messages */
+                if (error instanceof Error) {
+                    if (error.message.includes('Parameter types must be provided for null values') ||
+                        error.message.includes('Incorrect number of parameter types provided')) {
+                        throw new Error(`BigQuery query failed: ${error.message}\n` +
+                            'Hint: The BigQuery dialect now automatically handles null parameters. ' +
+                            "If you're still seeing this error, please report it as a bug.");
+                    }
+                    throw new Error(`BigQuery query failed: ${error.message}`);
+                }
+                throw error;
+            }
+        });
+    }
+    /**
+     * Begins a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    beginTransaction() {
+        return Promise.reject(new Error('Transactions are not supported.'));
+    }
+    /**
+     * Commits a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    commitTransaction() {
+        return Promise.reject(new Error('Transactions are not supported.'));
+    }
+    /**
+     * Rolls back a transaction (not supported by BigQuery).
+     * @returns A rejected promise as BigQuery doesn't support transactions
+     */
+    rollbackTransaction() {
+        return Promise.reject(new Error('Transactions are not supported.'));
+    }
+    /**
+     * Streams query results for handling large datasets.
+     * @param compiledQuery - The compiled query with SQL and parameters
+     * @param _chunkSize - Chunk size parameter (currently unused)
+     * @returns An async iterator that yields query results in batches
+     */
+    streamQuery(compiledQuery, _chunkSize) {
+        return tslib_1.__asyncGenerator(this, arguments, function* streamQuery_1() {
+            var _a, e_1, _b, _c;
+            const params = [...compiledQuery.parameters];
+            /* Process parameters to handle nulls and JSON serialization */
+            const processedParams = tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonDetector, "f").processParameters(compiledQuery, params);
+            const options = {
+                query: compiledQuery.sql,
+                params: processedParams,
+                parseJSON: true,
+            };
+            /* BigQuery needs types array for ALL parameters when there are null parameters */
+            if (processedParams.some((p) => p === null)) {
+                options.types = tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_instances, "m", _BigQueryConnection_inferParamTypes).call(this, processedParams);
+            }
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            let stream;
+            try {
+                // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+                stream = tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_client, "f").createQueryStream(options);
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    throw new Error(`BigQuery stream query failed: ${error.message}`);
+                }
+                throw error;
+            }
+            try {
+                try {
+                    for (var _d = true, stream_1 = tslib_1.__asyncValues(stream), stream_1_1; stream_1_1 = yield tslib_1.__await(stream_1.next()), _a = stream_1_1.done, !_a; _d = true) {
+                        _c = stream_1_1.value;
+                        _d = false;
+                        const row = _c;
+                        yield yield tslib_1.__await({
+                            rows: [tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_instances, "m", _BigQueryConnection_processResultRow).call(this, row)],
+                        });
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (!_d && !_a && (_b = stream_1.return)) yield tslib_1.__await(_b.call(stream_1));
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+            }
+            catch (error) {
+                /* Handle stream errors */
+                if (error instanceof Error) {
+                    throw new Error(`BigQuery stream error: ${error.message}`);
+                }
+                throw error;
+            }
+        });
+    }
+}
+exports.BigQueryConnection = BigQueryConnection;
+_BigQueryConnection_client = new WeakMap(), _BigQueryConnection_jsonDetector = new WeakMap(), _BigQueryConnection_jsonColumnNames = new WeakMap(), _BigQueryConnection_instances = new WeakSet(), _BigQueryConnection_inferParamTypes = function _BigQueryConnection_inferParamTypes(params) {
+    return params.map((param) => {
+        if (param === null) {
+            return 'STRING';
+        }
+        if (typeof param === 'number') {
+            return Number.isInteger(param) ? 'INT64' : 'FLOAT64';
+        }
+        if (typeof param === 'boolean') {
+            return 'BOOL';
+        }
+        if (param instanceof Date) {
+            return 'TIMESTAMP';
+        }
+        if (param instanceof Buffer) {
+            return 'BYTES';
+        }
+        if (Array.isArray(param)) {
+            return tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_instances, "m", _BigQueryConnection_inferArrayType).call(this, param);
+        }
+        if (typeof param === 'object') {
+            /*
+             * After processParameters, registered JSON values are already stringified
+             * to strings, so they never reach here. Remaining objects are STRUCTs or
+             * unregistered objects. BigQuery needs the full STRUCT<...> descriptor to
+             * use a provided type, which we can't infer generically. Returning 'STRING'
+             * is a safe fallback — BigQuery's SDK calls .toString() on the value.
+             */
+            return 'STRING';
+        }
+        return 'STRING';
+    });
+}, _BigQueryConnection_inferArrayType = function _BigQueryConnection_inferArrayType(param) {
+    if (param.length === 0) {
+        return 'ARRAY<STRING>';
+    }
+    const first = param[0];
+    if (typeof first === 'number') {
+        return Number.isInteger(first) ? 'ARRAY<INT64>' : 'ARRAY<FLOAT64>';
+    }
+    if (typeof first === 'boolean') {
+        return 'ARRAY<BOOL>';
+    }
+    return 'ARRAY<STRING>';
+}, _BigQueryConnection_processResultRow = function _BigQueryConnection_processResultRow(row) {
+    if (tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonColumnNames, "f").size === 0) {
+        return row;
+    }
+    const processedRow = {};
+    for (const [key, value] of Object.entries(row)) {
+        if (tslib_1.__classPrivateFieldGet(this, _BigQueryConnection_jsonColumnNames, "f").has(key) && typeof value === 'string' && value.length > 0) {
+            try {
+                processedRow[key] = JSON.parse(value);
+            }
+            catch (_a) {
+                processedRow[key] = value;
+            }
+        }
+        else {
+            processedRow[key] = value;
+        }
+    }
+    return processedRow;
+};
+//# sourceMappingURL=BigQueryConnection.js.map

package/dist/BigQueryConnection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BigQueryConnection.js","sourceRoot":"","sources":["../src/BigQueryConnection.ts"],"names":[],"mappings":";;;;;AAAA,qDAAwF;AAGxF,6DAA0D;AAI1D;;;;GAIG;AACH,MAAa,kBAAkB;IAK7B,YAAY,MAA6B;;;QAJhC,6CAAoC;QACpC,mDAAkC;QAClC,sDAA8B;QAGrC,+BAAA,IAAI,8BAAW,MAAA,MAAM,CAAC,QAAQ,mCAAI,IAAI,mBAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAA,CAAC;QAC/D,+BAAA,IAAI,oCAAiB,IAAI,uCAAkB,EAAE,MAAA,CAAC;QAE9C,uDAAuD;QACvD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtE,+BAAA,IAAI,wCAAc,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,+BAAA,IAAI,uCAAoB,+BAAA,IAAI,wCAAc,CAAC,4BAA4B,EAAE,MAAA,CAAC;IAC5E,CAAC;IAED;;;;;OAKG;IACG,YAAY,CAAI,aAA4B;;YAChD,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBAE7C,+DAA+D;gBAC/D,MAAM,eAAe,GAAG,+BAAA,IAAI,wCAAc,CAAC,iBAAiB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;gBAEpF,MAAM,OAAO,GAAU;oBACrB,KAAK,EAAE,aAAa,CAAC,GAAG;oBACxB,MAAM,EAAE,eAAe;oBACvB,SAAS,EAAE,IAAI;iBAChB,CAAC;gBAEF,kFAAkF;gBAClF,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC;oBAC5C,OAAO,CAAC,KAAK,GAAG,+BAAA,IAAI,0EAAiB,MAArB,IAAI,EAAkB,eAAe,CAAC,CAAC;gBACzD,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,+BAAA,IAAI,kCAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEjD,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;oBACvC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,+BAAA,IAAI,2EAAkB,MAAtB,IAAI,EAAmB,GAA8B,CAAC,CAAC;oBAC3E,CAAC,CAAC,EAAE,CAAC;gBAEP,OAAO;oBACL,IAAI,EAAE,aAAoB;iBAC3B,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,yCAAyC;gBACzC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,IACE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,kDAAkD,CAAC;wBAC1E,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,8CAA8C,CAAC,EACtE,CAAC;wBACD,MAAM,IAAI,KAAK,CACb,0BAA0B,KAAK,CAAC,OAAO,IAAI;4BACzC,wEAAwE;4BACxE,+DAA+D,CAClE,CAAC;oBACJ,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KAAA;IAED;;;OAGG;IACH,gBAAgB;QACd,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED;;;;;OAKG;IACI,WAAW,CAChB,aAA4B,EAC5B,UAAkB;;;YAElB,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;YAE7C,+DAA+D;YAC/D,MAAM,eAAe,GAAG,+BAAA,IAAI,wCAAc,CAAC,iBAAiB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YAEpF,MAAM,OAAO,GAAU;gBACrB,KAAK,EAAE,aAAa,CAAC,GAAG;gBACxB,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,IAAI;aAChB,CAAC;YAEF,kFAAkF;YAClF,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC;gBAC5C,OAAO,CAAC,KAAK,GAAG,+BAAA,IAAI,0EAAiB,MAArB,IAAI,EAAkB,eAAe,CAAC,CAAC;YACzD,CAAC;YAED,8DAA8D;YAC9D,IAAI,MAAW,CAAC;YAChB,IAAI,CAAC;gBACH,mEAAmE;gBACnE,MAAM,GAAG,+BAAA,IAAI,kCAAQ,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACnD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACpE,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,CAAC;;oBACH,KAAwB,eAAA,WAAA,sBAAA,MAAM,CAAA,YAAA,6FAAE,CAAC;wBAAT,sBAAM;wBAAN,WAAM;wBAAnB,MAAM,GAAG,KAAA,CAAA;wBAClB,4BAAM;4BACJ,IAAI,EAAE,CAAC,+BAAA,IAAI,2EAAkB,MAAtB,IAAI,EAAmB,GAA8B,CAAM,CAAC;yBACpE,CAAA,CAAC;oBACJ,CAAC;;;;;;;;;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,0BAA0B;gBAC1B,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KAAA;CAiFF;AAnOD,gDAmOC;qRA3EkB,MAA0B;IACzC,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAC1B,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACnB,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QACvD,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;YAC1B,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,IAAI,KAAK,YAAY,MAAM,EAAE,CAAC;YAC5B,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,+BAAA,IAAI,yEAAgB,MAApB,IAAI,EAAiB,KAAK,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B;;;;;;eAMG;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC,mFAMe,KAAgB;IAC9B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,eAAe,CAAC;IACzB,CAAC;IACD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACvB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,gBAAgB,CAAC;IACrE,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO,aAAa,CAAC;IACvB,CAAC;IACD,OAAO,eAAe,CAAC;AACzB,CAAC,uFAMiB,GAA4B;IAC5C,IAAI,+BAAA,IAAI,2CAAiB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,YAAY,GAA4B,EAAE,CAAC;IACjD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/C,IAAI,+BAAA,IAAI,2CAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpF,IAAI,CAAC;gBACH,YAAY,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxC,CAAC;YAAC,WAAM,CAAC;gBACP,YAAY,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC5B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAC5B,CAAC;IACH,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC"}

package/dist/BigQueryDialect.d.ts

@@ -0,0 +1,80 @@
+import { type BigQuery, type BigQueryOptions, type Dataset, type Table } from '@google-cloud/bigquery';
+import { type DatabaseIntrospector, type Dialect, type Driver, type Kysely, type QueryCompiler } from 'kysely';
+import { BigQueryAdapter } from './BigQueryAdapter';
+/**
+ * Configuration options for BigQuery dialect.
+ */
+export interface BigQueryDialectConfig {
+    /**
+     * BigQuery client options.
+     */
+    options?: BigQueryOptions;
+    /**
+     * Existing BigQuery, Dataset, or Table instance to use.
+     * If provided, this will be used instead of creating a new client.
+     */
+    bigquery?: BigQuery | Dataset | Table;
+    /**
+     * Optional mapping of table names to their JSON column names.
+     * This enables automatic JSON serialization for these columns.
+     *
+     * Example:
+     * ```
+     * {
+     *   'dataset.users': ['metadata', 'settings'],
+     *   'dataset.products': ['specifications']
+     * }
+     * ```
+     */
+    jsonColumns?: Record<string, string[]>;
+    /**
+     * Default GCP project ID to prepend to table references.
+     *
+     * BigQuery supports three-level table names: `project.dataset.table`.
+     * Kysely's parser only handles two-level names (`schema.table`), so
+     * three-part strings like `'my-project.dataset.table'` lose the table name.
+     *
+     * Set this to your project ID so you can write `selectFrom('dataset.table')`
+     * and the compiler emits `\`project\`.\`dataset\`.\`table\``.
+     *
+     * Example:
+     * ```
+     * new BigQueryDialect({
+     *   bigquery: client,
+     *   defaultProject: 'my-gcp-project',
+     * })
+     * ```
+     */
+    defaultProject?: string;
+}
+/**
+ * BigQuery dialect for Kysely.
+ *
+ * This dialect allows you to use Kysely with Google BigQuery.
+ */
+export declare class BigQueryDialect implements Dialect {
+    #private;
+    constructor(config?: BigQueryDialectConfig);
+    /**
+     * Creates a BigQuery adapter for Kysely.
+     * @returns A new BigQueryAdapter instance
+     */
+    createAdapter(): BigQueryAdapter;
+    /**
+     * Creates a BigQuery driver for database connections.
+     * @returns A new BigQueryDriver instance configured with the dialect settings
+     */
+    createDriver(): Driver;
+    /**
+     * Creates a BigQuery query compiler for SQL generation.
+     * @returns A new BigQueryCompiler instance that translates Kysely queries to BigQuery SQL
+     */
+    createQueryCompiler(): QueryCompiler;
+    /**
+     * Creates a BigQuery database introspector for schema discovery.
+     * @param db - The Kysely database instance
+     * @returns A new BigQueryIntrospector instance for examining database schema
+     */
+    createIntrospector(db: Kysely<unknown>): DatabaseIntrospector;
+}
+//# sourceMappingURL=BigQueryDialect.d.ts.map

package/dist/BigQueryDialect.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BigQueryDialect.d.ts","sourceRoot":"","sources":["../src/BigQueryDialect.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,eAAe,EACpB,KAAK,OAAO,EACZ,KAAK,KAAK,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,KAAK,oBAAoB,EACzB,KAAK,OAAO,EACZ,KAAK,MAAM,EACX,KAAK,MAAM,EACX,KAAK,aAAa,EACnB,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAKpD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC;;OAEG;IACH,OAAO,CAAC,EAAE,eAAe,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,EAAE,QAAQ,GAAG,OAAO,GAAG,KAAK,CAAC;IAEtC;;;;;;;;;;;OAWG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAEvC;;;;;;;;;;;;;;;;;OAiBG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,qBAAa,eAAgB,YAAW,OAAO;;gBAGjC,MAAM,CAAC,EAAE,qBAAqB;IAI1C;;;OAGG;IACH,aAAa;IAIb;;;OAGG;IACH,YAAY,IAAI,MAAM;IAItB;;;OAGG;IACH,mBAAmB,IAAI,aAAa;IAIpC;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,oBAAoB;CAwC9D"}