kysely 0.19.12 → 0.21.0

package/dist/esm/operation-node/operation-node-transformer.d.ts

@@ -68,6 +68,8 @@ import { ValuesNode } from './values-node.js';
 import { SelectModifierNode } from './select-modifier-node.js';
 import { CreateTypeNode } from './create-type-node.js';
 import { DropTypeNode } from './drop-type-node.js';
+import { ExplainNode } from './explain-node.js';
+import { SchemableIdentifierNode } from './schemable-identifier-node.js';
 /**
  * Transforms an operation node tree into another one.
  *
@@ -100,9 +102,9 @@ import { DropTypeNode } from './drop-type-node.js';
 export declare class OperationNodeTransformer {
     #private;
     protected readonly nodeStack: OperationNode[];
-    readonly transformNode: <T extends OperationNode | undefined, U extends T | undefined = T>(node: U) => T;
-    protected transformNodeList<T extends OperationNode, U extends T = T>(list: readonly U[]): readonly T[];
-    protected transformNodeList<T extends OperationNode, U extends T = T>(list: readonly U[] | undefined): readonly T[] | undefined;
+    transformNode<T extends OperationNode | undefined>(node: T): T;
+    protected transformNodeImpl<T extends OperationNode>(node: T): T;
+    protected transformNodeList<T extends ReadonlyArray<OperationNode> | undefined>(list: T): T;
     protected transformSelectQuery(node: SelectQueryNode): SelectQueryNode;
     protected transformSelection(node: SelectionNode): SelectionNode;
     protected transformColumn(node: ColumnNode): ColumnNode;
@@ -166,6 +168,8 @@ export declare class OperationNodeTransformer {
     protected transformSelectModifier(node: SelectModifierNode): SelectModifierNode;
     protected transformCreateType(node: CreateTypeNode): CreateTypeNode;
     protected transformDropType(node: DropTypeNode): DropTypeNode;
+    protected transformExplain(node: ExplainNode): ExplainNode;
+    protected transformSchemableIdentifier(node: SchemableIdentifierNode): SchemableIdentifierNode;
     protected transformDataType(node: DataTypeNode): DataTypeNode;
     protected transformSelectAll(node: SelectAllNode): SelectAllNode;
     protected transformIdentifier(node: IdentifierNode): IdentifierNode;

package/dist/esm/operation-node/operation-node-transformer.js

@@ -36,6 +36,7 @@ export class OperationNodeTransformer {
         AliasNode: this.transformAlias.bind(this),
         ColumnNode: this.transformColumn.bind(this),
         IdentifierNode: this.transformIdentifier.bind(this),
+        SchemableIdentifierNode: this.transformSchemableIdentifier.bind(this),
         RawNode: this.transformRaw.bind(this),
         ReferenceNode: this.transformReference.bind(this),
         SelectQueryNode: this.transformSelectQuery.bind(this),
@@ -102,21 +103,25 @@ export class OperationNodeTransformer {
         SelectModifierNode: this.transformSelectModifier.bind(this),
         CreateTypeNode: this.transformCreateType.bind(this),
         DropTypeNode: this.transformDropType.bind(this),
+        ExplainNode: this.transformExplain.bind(this),
     });
-    transformNode = (node) => {
+    transformNode(node) {
         if (!node) {
-            return undefined;
+            return node;
         }
         this.nodeStack.push(node);
-        const out = this.#transformers[node.kind](node);
+        const out = this.transformNodeImpl(node);
         this.nodeStack.pop();
         return freeze(out);
-    };
+    }
+    transformNodeImpl(node) {
+        return this.#transformers[node.kind](node);
+    }
     transformNodeList(list) {
         if (!list) {
             return list;
         }
-        return freeze(list.map(this.transformNode));
+        return freeze(list.map((node) => this.transformNode(node)));
     }
     transformSelectQuery(node) {
         return requireAllProps({
@@ -135,6 +140,7 @@ export class OperationNodeTransformer {
             with: this.transformNode(node.with),
             having: this.transformNode(node.having),
             union: this.transformNodeList(node.union),
+            explain: this.transformNode(node.explain),
         });
     }
     transformSelection(node) {
@@ -159,7 +165,6 @@ export class OperationNodeTransformer {
     transformTable(node) {
         return requireAllProps({
             kind: 'TableNode',
-            schema: this.transformNode(node.schema),
             table: this.transformNode(node.table),
         });
     }
@@ -243,6 +248,7 @@ export class OperationNodeTransformer {
             with: this.transformNode(node.with),
             ignore: node.ignore,
             replace: node.replace,
+            explain: this.transformNode(node.explain),
         });
     }
     transformValues(node) {
@@ -261,6 +267,7 @@ export class OperationNodeTransformer {
             with: this.transformNode(node.with),
             orderBy: this.transformNode(node.orderBy),
             limit: this.transformNode(node.limit),
+            explain: this.transformNode(node.explain),
         });
     }
     transformReturning(node) {
@@ -278,6 +285,8 @@ export class OperationNodeTransformer {
             temporary: node.temporary,
             ifNotExists: node.ifNotExists,
             onCommit: node.onCommit,
+            frontModifiers: this.transformNodeList(node.frontModifiers),
+            endModifiers: this.transformNodeList(node.endModifiers),
         });
     }
     transformColumnDefinition(node) {
@@ -344,6 +353,7 @@ export class OperationNodeTransformer {
             updates: this.transformNodeList(node.updates),
             returning: this.transformNode(node.returning),
             with: this.transformNode(node.with),
+            explain: this.transformNode(node.explain),
         });
     }
     transformColumnUpdate(node) {
@@ -618,6 +628,20 @@ export class OperationNodeTransformer {
             ifExists: node.ifExists,
         });
     }
+    transformExplain(node) {
+        return requireAllProps({
+            kind: 'ExplainNode',
+            format: node.format,
+            options: this.transformNode(node.options),
+        });
+    }
+    transformSchemableIdentifier(node) {
+        return requireAllProps({
+            kind: 'SchemableIdentifierNode',
+            schema: this.transformNode(node.schema),
+            identifier: this.transformNode(node.identifier),
+        });
+    }
     transformDataType(node) {
         // An Object.freezed leaf node. No need to clone.
         return node;

package/dist/esm/operation-node/operation-node-visitor.d.ts

@@ -68,6 +68,8 @@ import { ValuesNode } from './values-node.js';
 import { SelectModifierNode } from './select-modifier-node.js';
 import { CreateTypeNode } from './create-type-node.js';
 import { DropTypeNode } from './drop-type-node.js';
+import { ExplainNode } from './explain-node.js';
+import { SchemableIdentifierNode } from './schemable-identifier-node.js';
 export declare abstract class OperationNodeVisitor {
     #private;
     protected readonly nodeStack: OperationNode[];
@@ -128,10 +130,11 @@ export declare abstract class OperationNodeVisitor {
     protected abstract visitForeignKeyConstraint(node: ForeignKeyConstraintNode): void;
     protected abstract visitUnion(node: UnionNode): void;
     protected abstract visitDataType(node: DataTypeNode): void;
-    protected abstract visitSelectAll(_: SelectAllNode): void;
-    protected abstract visitIdentifier(_: IdentifierNode): void;
-    protected abstract visitValue(_: ValueNode): void;
-    protected abstract visitPrimitiveValueList(_: PrimitiveValueListNode): void;
+    protected abstract visitSelectAll(node: SelectAllNode): void;
+    protected abstract visitIdentifier(node: IdentifierNode): void;
+    protected abstract visitSchemableIdentifier(node: SchemableIdentifierNode): void;
+    protected abstract visitValue(node: ValueNode): void;
+    protected abstract visitPrimitiveValueList(node: PrimitiveValueListNode): void;
     protected abstract visitOperator(node: OperatorNode): void;
     protected abstract visitCreateView(node: CreateViewNode): void;
     protected abstract visitDropView(node: DropViewNode): void;
@@ -142,4 +145,5 @@ export declare abstract class OperationNodeVisitor {
     protected abstract visitSelectModifier(node: SelectModifierNode): void;
     protected abstract visitCreateType(node: CreateTypeNode): void;
     protected abstract visitDropType(node: DropTypeNode): void;
+    protected abstract visitExplain(node: ExplainNode): void;
 }

package/dist/esm/operation-node/operation-node-visitor.js

@@ -9,6 +9,7 @@ export class OperationNodeVisitor {
         AliasNode: this.visitAlias.bind(this),
         ColumnNode: this.visitColumn.bind(this),
         IdentifierNode: this.visitIdentifier.bind(this),
+        SchemableIdentifierNode: this.visitSchemableIdentifier.bind(this),
         RawNode: this.visitRaw.bind(this),
         ReferenceNode: this.visitReference.bind(this),
         SelectQueryNode: this.visitSelectQuery.bind(this),
@@ -75,6 +76,7 @@ export class OperationNodeVisitor {
         SelectModifierNode: this.visitSelectModifier.bind(this),
         CreateTypeNode: this.visitCreateType.bind(this),
         DropTypeNode: this.visitDropType.bind(this),
+        ExplainNode: this.visitExplain.bind(this),
     });
     visitNode = (node) => {
         this.nodeStack.push(node);

package/dist/esm/operation-node/operation-node.d.ts

@@ -1,4 +1,4 @@
-export declare type OperationNodeKind = 'IdentifierNode' | 'RawNode' | 'SelectQueryNode' | 'SelectionNode' | 'ReferenceNode' | 'ColumnNode' | 'TableNode' | 'AliasNode' | 'FromNode' | 'SelectAllNode' | 'FilterNode' | 'AndNode' | 'OrNode' | 'ParensNode' | 'ValueNode' | 'ValueListNode' | 'PrimitiveValueListNode' | 'JoinNode' | 'OperatorNode' | 'WhereNode' | 'InsertQueryNode' | 'DeleteQueryNode' | 'ReturningNode' | 'CreateTableNode' | 'ColumnDefinitionNode' | 'AddColumnNode' | 'DropTableNode' | 'DataTypeNode' | 'OrderByNode' | 'OrderByItemNode' | 'GroupByNode' | 'GroupByItemNode' | 'UpdateQueryNode' | 'ColumnUpdateNode' | 'LimitNode' | 'OffsetNode' | 'OnConflictNode' | 'OnDuplicateKeyNode' | 'CreateIndexNode' | 'DropIndexNode' | 'ListNode' | 'ReferencesNode' | 'PrimaryKeyConstraintNode' | 'UniqueConstraintNode' | 'CheckConstraintNode' | 'ForeignKeyConstraintNode' | 'WithNode' | 'CommonTableExpressionNode' | 'HavingNode' | 'CreateSchemaNode' | 'DropSchemaNode' | 'AlterTableNode' | 'ModifyColumnNode' | 'DropColumnNode' | 'RenameColumnNode' | 'AlterColumnNode' | 'AddConstraintNode' | 'DropConstraintNode' | 'UnionNode' | 'CreateViewNode' | 'DropViewNode' | 'GeneratedNode' | 'DefaultValueNode' | 'OnNode' | 'ValuesNode' | 'CommonTableExpressionNameNode' | 'SelectModifierNode' | 'CreateTypeNode' | 'DropTypeNode';
+export declare type OperationNodeKind = 'IdentifierNode' | 'SchemableIdentifierNode' | 'RawNode' | 'SelectQueryNode' | 'SelectionNode' | 'ReferenceNode' | 'ColumnNode' | 'TableNode' | 'AliasNode' | 'FromNode' | 'SelectAllNode' | 'FilterNode' | 'AndNode' | 'OrNode' | 'ParensNode' | 'ValueNode' | 'ValueListNode' | 'PrimitiveValueListNode' | 'JoinNode' | 'OperatorNode' | 'WhereNode' | 'InsertQueryNode' | 'DeleteQueryNode' | 'ReturningNode' | 'CreateTableNode' | 'ColumnDefinitionNode' | 'AddColumnNode' | 'DropTableNode' | 'DataTypeNode' | 'OrderByNode' | 'OrderByItemNode' | 'GroupByNode' | 'GroupByItemNode' | 'UpdateQueryNode' | 'ColumnUpdateNode' | 'LimitNode' | 'OffsetNode' | 'OnConflictNode' | 'OnDuplicateKeyNode' | 'CreateIndexNode' | 'DropIndexNode' | 'ListNode' | 'ReferencesNode' | 'PrimaryKeyConstraintNode' | 'UniqueConstraintNode' | 'CheckConstraintNode' | 'ForeignKeyConstraintNode' | 'WithNode' | 'CommonTableExpressionNode' | 'HavingNode' | 'CreateSchemaNode' | 'DropSchemaNode' | 'AlterTableNode' | 'ModifyColumnNode' | 'DropColumnNode' | 'RenameColumnNode' | 'AlterColumnNode' | 'AddConstraintNode' | 'DropConstraintNode' | 'UnionNode' | 'CreateViewNode' | 'DropViewNode' | 'GeneratedNode' | 'DefaultValueNode' | 'OnNode' | 'ValuesNode' | 'CommonTableExpressionNameNode' | 'SelectModifierNode' | 'CreateTypeNode' | 'DropTypeNode' | 'ExplainNode';
 export interface OperationNode {
     readonly kind: OperationNodeKind;
 }

package/dist/esm/operation-node/reference-node.d.ts

@@ -13,5 +13,5 @@ export interface ReferenceNode extends OperationNode {
 export declare const ReferenceNode: Readonly<{
     is(node: OperationNode): node is ReferenceNode;
     create(table: TableNode, column: ColumnNode): ReferenceNode;
-    createSelectAll(table: string): ReferenceNode;
+    createSelectAll(table: TableNode): ReferenceNode;
 }>;

package/dist/esm/operation-node/reference-node.js

@@ -1,5 +1,4 @@
 /// <reference types="./reference-node.d.ts" />
-import { TableNode } from './table-node.js';
 import { SelectAllNode } from './select-all-node.js';
 import { freeze } from '../util/object-utils.js';
 /**
@@ -19,7 +18,7 @@ export const ReferenceNode = freeze({
     createSelectAll(table) {
         return freeze({
             kind: 'ReferenceNode',
-            table: TableNode.create(table),
+            table,
             column: SelectAllNode.create(),
         });
     },

package/dist/esm/operation-node/schemable-identifier-node.d.ts

@@ -0,0 +1,15 @@
+import { IdentifierNode } from './identifier-node.js';
+import { OperationNode } from './operation-node.js';
+export interface SchemableIdentifierNode extends OperationNode {
+    readonly kind: 'SchemableIdentifierNode';
+    readonly schema?: IdentifierNode;
+    readonly identifier: IdentifierNode;
+}
+/**
+ * @internal
+ */
+export declare const SchemableIdentifierNode: Readonly<{
+    is(node: OperationNode): node is SchemableIdentifierNode;
+    create(identifier: string): SchemableIdentifierNode;
+    createWithSchema(schema: string, identifier: string): SchemableIdentifierNode;
+}>;

package/dist/esm/operation-node/schemable-identifier-node.js

@@ -0,0 +1,24 @@
+/// <reference types="./schemable-identifier-node.d.ts" />
+import { freeze } from '../util/object-utils.js';
+import { IdentifierNode } from './identifier-node.js';
+/**
+ * @internal
+ */
+export const SchemableIdentifierNode = freeze({
+    is(node) {
+        return node.kind === 'SchemableIdentifierNode';
+    },
+    create(identifier) {
+        return freeze({
+            kind: 'SchemableIdentifierNode',
+            identifier: IdentifierNode.create(identifier),
+        });
+    },
+    createWithSchema(schema, identifier) {
+        return freeze({
+            kind: 'SchemableIdentifierNode',
+            schema: IdentifierNode.create(schema),
+            identifier: IdentifierNode.create(identifier),
+        });
+    },
+});

package/dist/esm/operation-node/select-query-node.d.ts

@@ -14,6 +14,7 @@ import { WhereNode } from './where-node.js';
 import { WithNode } from './with-node.js';
 import { UnionNode } from './union-node.js';
 import { SelectModifierNode } from './select-modifier-node.js';
+import { ExplainNode } from './explain-node.js';
 export interface SelectQueryNode extends OperationNode {
     readonly kind: 'SelectQueryNode';
     readonly from: FromNode;
@@ -30,6 +31,7 @@ export interface SelectQueryNode extends OperationNode {
     readonly with?: WithNode;
     readonly having?: HavingNode;
     readonly union?: ReadonlyArray<UnionNode>;
+    readonly explain?: ExplainNode;
 }
 /**
  * @internal
@@ -48,4 +50,5 @@ export declare const SelectQueryNode: Readonly<{
     cloneWithHaving(selectNode: SelectQueryNode, filter: FilterExpressionNode): SelectQueryNode;
     cloneWithOrHaving(selectNode: SelectQueryNode, filter: FilterExpressionNode): SelectQueryNode;
     cloneWithUnion(selectNode: SelectQueryNode, union: UnionNode): SelectQueryNode;
+    cloneWithExplain(selectNode: SelectQueryNode, explain: ExplainNode): SelectQueryNode;
 }>;

package/dist/esm/operation-node/select-query-node.js

@@ -102,4 +102,10 @@ export const SelectQueryNode = freeze({
                 : freeze([union]),
         });
     },
+    cloneWithExplain(selectNode, explain) {
+        return freeze({
+            ...selectNode,
+            explain,
+        });
+    },
 });

package/dist/esm/operation-node/selection-node.d.ts

@@ -3,6 +3,7 @@ import { ColumnNode } from './column-node.js';
 import { OperationNode } from './operation-node.js';
 import { ReferenceNode } from './reference-node.js';
 import { SelectAllNode } from './select-all-node.js';
+import { TableNode } from './table-node.js';
 declare type SelectionNodeChild = ColumnNode | ReferenceNode | AliasNode | SelectAllNode;
 export interface SelectionNode extends OperationNode {
     readonly kind: 'SelectionNode';
@@ -15,6 +16,6 @@ export declare const SelectionNode: Readonly<{
     is(node: OperationNode): node is SelectionNode;
     create(selection: SelectionNodeChild): SelectionNode;
     createSelectAll(): SelectionNode;
-    createSelectAllFromTable(table: string): SelectionNode;
+    createSelectAllFromTable(table: TableNode): SelectionNode;
 }>;
 export {};

package/dist/esm/operation-node/table-node.d.ts

@@ -1,9 +1,8 @@
-import { IdentifierNode } from './identifier-node.js';
 import { OperationNode } from './operation-node.js';
+import { SchemableIdentifierNode } from './schemable-identifier-node.js';
 export interface TableNode extends OperationNode {
     readonly kind: 'TableNode';
-    readonly schema?: IdentifierNode;
-    readonly table: IdentifierNode;
+    readonly table: SchemableIdentifierNode;
 }
 /**
  * @internal

package/dist/esm/operation-node/table-node.js

@@ -1,6 +1,6 @@
 /// <reference types="./table-node.d.ts" />
 import { freeze } from '../util/object-utils.js';
-import { IdentifierNode } from './identifier-node.js';
+import { SchemableIdentifierNode } from './schemable-identifier-node.js';
 /**
  * @internal
  */
@@ -11,14 +11,13 @@ export const TableNode = freeze({
     create(table) {
         return freeze({
             kind: 'TableNode',
-            table: IdentifierNode.create(table),
+            table: SchemableIdentifierNode.create(table),
         });
     },
     createWithSchema(schema, table) {
         return freeze({
             kind: 'TableNode',
-            schema: IdentifierNode.create(schema),
-            table: IdentifierNode.create(table),
+            table: SchemableIdentifierNode.createWithSchema(schema, table),
         });
     },
 });

package/dist/esm/operation-node/update-query-node.d.ts

@@ -8,6 +8,7 @@ import { ValueListNode } from './value-list-node.js';
 import { WhereNode } from './where-node.js';
 import { WithNode } from './with-node.js';
 import { FromNode } from './from-node.js';
+import { ExplainNode } from './explain-node.js';
 export declare type UpdateValuesNode = ValueListNode | PrimitiveValueListNode;
 export interface UpdateQueryNode extends OperationNode {
     readonly kind: 'UpdateQueryNode';
@@ -18,6 +19,7 @@ export interface UpdateQueryNode extends OperationNode {
     readonly updates?: ReadonlyArray<ColumnUpdateNode>;
     readonly returning?: ReturningNode;
     readonly with?: WithNode;
+    readonly explain?: ExplainNode;
 }
 /**
  * @internal
@@ -27,4 +29,5 @@ export declare const UpdateQueryNode: Readonly<{
     create(table: TableExpressionNode, withNode?: WithNode): UpdateQueryNode;
     cloneWithFromItems(updateQuery: UpdateQueryNode, fromItems: ReadonlyArray<TableExpressionNode>): UpdateQueryNode;
     cloneWithUpdates(updateQuery: UpdateQueryNode, updates: ReadonlyArray<ColumnUpdateNode>): UpdateQueryNode;
+    cloneWithExplain(updateQuery: UpdateQueryNode, explain: ExplainNode): UpdateQueryNode;
 }>;

package/dist/esm/operation-node/update-query-node.js

@@ -31,4 +31,10 @@ export const UpdateQueryNode = freeze({
                 : updates,
         });
     },
+    cloneWithExplain(updateQuery, explain) {
+        return freeze({
+            ...updateQuery,
+            explain,
+        });
+    },
 });

package/dist/esm/parser/select-parser.js

@@ -4,6 +4,7 @@ import { SelectionNode } from '../operation-node/selection-node.js';
 import { parseAliasedStringReference } from './reference-parser.js';
 import { isDynamicReferenceBuilder, } from '../dynamic/dynamic-reference-builder.js';
 import { parseAliasedComplexExpression, } from './complex-expression-parser.js';
+import { parseTable } from './table-parser.js';
 export function parseSelectExpressionOrList(selection) {
     if (isReadonlyArray(selection)) {
         return selection.map((it) => parseSelectExpression(it));
@@ -36,7 +37,7 @@ export function parseSelectAll(table) {
 }
 function parseSelectAllArg(table) {
     if (isString(table)) {
-        return SelectionNode.createSelectAllFromTable(table);
+        return SelectionNode.createSelectAllFromTable(parseTable(table));
     }
     throw new Error(`invalid value selectAll expression: ${JSON.stringify(table)}`);
 }

package/dist/esm/plugin/camel-case/camel-case-transformer.js

@@ -10,7 +10,7 @@ export class SnakeCaseTransformer extends OperationNodeTransformer {
         node = super.transformIdentifier(node);
         return {
             ...node,
-            identifier: this.#snakeCase(node.identifier),
+            name: this.#snakeCase(node.name),
         };
     }
 }

package/dist/esm/plugin/with-schema/with-schema-transformer.d.ts

@@ -1,33 +1,9 @@
-import { AlterTableNode } from '../../operation-node/alter-table-node.js';
-import { CreateIndexNode } from '../../operation-node/create-index-node.js';
-import { CreateTableNode } from '../../operation-node/create-table-node.js';
-import { CreateTypeNode } from '../../operation-node/create-type-node.js';
-import { CreateViewNode } from '../../operation-node/create-view-node.js';
-import { DeleteQueryNode } from '../../operation-node/delete-query-node.js';
-import { DropIndexNode } from '../../operation-node/drop-index-node.js';
-import { DropTableNode } from '../../operation-node/drop-table-node.js';
-import { DropTypeNode } from '../../operation-node/drop-type-node.js';
-import { DropViewNode } from '../../operation-node/drop-view-node.js';
-import { InsertQueryNode } from '../../operation-node/insert-query-node.js';
 import { OperationNodeTransformer } from '../../operation-node/operation-node-transformer.js';
-import { SelectQueryNode } from '../../operation-node/select-query-node.js';
-import { TableNode } from '../../operation-node/table-node.js';
-import { UpdateQueryNode } from '../../operation-node/update-query-node.js';
+import { OperationNode } from '../../operation-node/operation-node.js';
+import { SchemableIdentifierNode } from '../../operation-node/schemable-identifier-node.js';
 export declare class WithSchemaTransformer extends OperationNodeTransformer {
     #private;
     constructor(schema: string);
-    protected transformSelectQuery(node: SelectQueryNode): SelectQueryNode;
-    protected transformInsertQuery(node: InsertQueryNode): InsertQueryNode;
-    protected transformUpdateQuery(node: UpdateQueryNode): UpdateQueryNode;
-    protected transformDeleteQuery(node: DeleteQueryNode): DeleteQueryNode;
-    protected transformCreateTable(node: CreateTableNode): CreateTableNode;
-    protected transformDropTable(node: DropTableNode): DropTableNode;
-    protected transformCreateIndex(node: CreateIndexNode): CreateIndexNode;
-    protected transformDropIndex(node: DropIndexNode): DropIndexNode;
-    protected transformCreateView(node: CreateViewNode): CreateViewNode;
-    protected transformDropView(node: DropViewNode): DropViewNode;
-    protected transformCreateType(node: CreateTypeNode): CreateTypeNode;
-    protected transformDropType(node: DropTypeNode): DropTypeNode;
-    protected transformAlterTable(node: AlterTableNode): AlterTableNode;
-    protected transformTable(node: TableNode): TableNode;
+    protected transformNodeImpl<T extends OperationNode>(node: T): T;
+    protected transformSchemableIdentifier(node: SchemableIdentifierNode): SchemableIdentifierNode;
 }

package/dist/esm/plugin/with-schema/with-schema-transformer.js

@@ -1,119 +1,113 @@
 /// <reference types="./with-schema-transformer.d.ts" />
 import { AliasNode } from '../../operation-node/alias-node.js';
 import { OperationNodeTransformer } from '../../operation-node/operation-node-transformer.js';
+import { SchemableIdentifierNode } from '../../operation-node/schemable-identifier-node.js';
 import { TableNode } from '../../operation-node/table-node.js';
+import { freeze } from '../../util/object-utils.js';
+// This object exist only so that we get a type error when a new RootOperationNode
+// is added. If you get a type error here, make sure to add the new root node and
+// handle it correctly in the transformer.
+//
+// DO NOT REFACTOR THIS EVEN IF IT SEEMS USELESS TO YOU!
+const ROOT_OPERATION_NODES = freeze({
+    AlterTableNode: true,
+    CreateIndexNode: true,
+    CreateSchemaNode: true,
+    CreateTableNode: true,
+    CreateTypeNode: true,
+    CreateViewNode: true,
+    DeleteQueryNode: true,
+    DropIndexNode: true,
+    DropSchemaNode: true,
+    DropTableNode: true,
+    DropTypeNode: true,
+    DropViewNode: true,
+    InsertQueryNode: true,
+    RawNode: true,
+    SelectQueryNode: true,
+    UpdateQueryNode: true,
+});
 export class WithSchemaTransformer extends OperationNodeTransformer {
     #schema;
-    #tables = new Set();
+    #schemableIds = new Set();
     constructor(schema) {
         super();
         this.#schema = schema;
     }
-    transformSelectQuery(node) {
-        return this.#transformRoot(node, (node) => super.transformSelectQuery(node));
-    }
-    transformInsertQuery(node) {
-        return this.#transformRoot(node, (node) => super.transformInsertQuery(node));
-    }
-    transformUpdateQuery(node) {
-        return this.#transformRoot(node, (node) => super.transformUpdateQuery(node));
-    }
-    transformDeleteQuery(node) {
-        return this.#transformRoot(node, (node) => super.transformDeleteQuery(node));
-    }
-    transformCreateTable(node) {
-        return this.#transformRoot(node, (node) => super.transformCreateTable(node));
-    }
-    transformDropTable(node) {
-        return this.#transformRoot(node, (node) => super.transformDropTable(node));
-    }
-    transformCreateIndex(node) {
-        return this.#transformRoot(node, (node) => super.transformCreateIndex(node));
-    }
-    transformDropIndex(node) {
-        return this.#transformRoot(node, (node) => super.transformDropIndex(node));
-    }
-    transformCreateView(node) {
-        return this.#transformRoot(node, (node) => super.transformCreateView(node));
-    }
-    transformDropView(node) {
-        return this.#transformRoot(node, (node) => super.transformDropView(node));
-    }
-    transformCreateType(node) {
-        return this.#transformRoot(node, (node) => super.transformCreateType(node));
-    }
-    transformDropType(node) {
-        return this.#transformRoot(node, (node) => super.transformDropType(node));
-    }
-    transformAlterTable(node) {
-        return this.#transformRoot(node, (node) => super.transformAlterTable(node));
+    transformNodeImpl(node) {
+        if (!this.#isRootOperationNode(node)) {
+            return super.transformNodeImpl(node);
+        }
+        const tables = this.#collectSchemableIds(node);
+        for (const table of tables) {
+            this.#schemableIds.add(table);
+        }
+        const transformed = super.transformNodeImpl(node);
+        for (const table of tables) {
+            this.#schemableIds.delete(table);
+        }
+        return transformed;
     }
-    transformTable(node) {
-        const transformed = super.transformTable(node);
-        if (transformed.schema || !this.#tables.has(node.table.identifier)) {
+    transformSchemableIdentifier(node) {
+        const transformed = super.transformSchemableIdentifier(node);
+        if (transformed.schema || !this.#schemableIds.has(node.identifier.name)) {
             return transformed;
         }
         return {
             ...transformed,
-            schema: {
+            schema: freeze({
                 kind: 'IdentifierNode',
-                identifier: this.#schema,
-            },
+                name: this.#schema,
+            }),
         };
     }
-    #transformRoot(node, transform) {
-        const tables = this.#collectTables(node);
-        for (const table of tables) {
-            this.#tables.add(table);
-        }
-        const transformed = transform(node);
-        for (const table of tables) {
-            this.#tables.delete(table);
-        }
-        return transformed;
+    #isRootOperationNode(node) {
+        return node.kind in ROOT_OPERATION_NODES;
     }
-    #collectTables(node) {
-        const tables = new Set();
+    #collectSchemableIds(node) {
+        const schemableIds = new Set();
+        if ('name' in node && node.name && SchemableIdentifierNode.is(node.name)) {
+            this.#collectSchemableId(node.name, schemableIds);
+        }
         if ('from' in node && node.from) {
-            this.#collectTablesFromTableExpressionNodes(node.from.froms, tables);
+            for (const from of node.from.froms) {
+                this.#collectSchemableIdsFromTableExpr(from, schemableIds);
+            }
         }
         if ('into' in node && node.into) {
-            this.#collectTablesFromTableExpressionNode(node.into, tables);
+            this.#collectSchemableIdsFromTableExpr(node.into, schemableIds);
         }
         if ('table' in node && node.table) {
-            this.#collectTablesFromTableExpressionNode(node.table, tables);
+            this.#collectSchemableIdsFromTableExpr(node.table, schemableIds);
         }
         if ('joins' in node && node.joins) {
-            this.#collectTablesFromJoins(node.joins, tables);
+            for (const join of node.joins) {
+                this.#collectSchemableIdsFromTableExpr(join.table, schemableIds);
+            }
         }
         if ('with' in node && node.with) {
-            this.#removeCommonTableExpressionTables(node.with, tables);
-        }
-        return tables;
-    }
-    #collectTablesFromTableExpressionNodes(nodes, tables) {
-        for (const node of nodes) {
-            this.#collectTablesFromTableExpressionNode(node, tables);
+            this.#removeCommonTableExpressionTables(node.with, schemableIds);
         }
+        return schemableIds;
     }
-    #collectTablesFromJoins(nodes, tables) {
-        for (const node of nodes) {
-            this.#collectTablesFromTableExpressionNode(node.table, tables);
-        }
-    }
-    #collectTablesFromTableExpressionNode(node, tables) {
+    #collectSchemableIdsFromTableExpr(node, schemableIds) {
         const table = TableNode.is(node)
             ? node
             : AliasNode.is(node) && TableNode.is(node.node)
                 ? node.node
                 : null;
-        if (table && !this.#tables.has(table.table.identifier)) {
-            tables.add(table.table.identifier);
+        if (table) {
+            this.#collectSchemableId(table.table, schemableIds);
+        }
+    }
+    #collectSchemableId(node, schemableIds) {
+        if (!this.#schemableIds.has(node.identifier.name)) {
+            schemableIds.add(node.identifier.name);
         }
     }
-    #removeCommonTableExpressionTables(node, tables) {
+    #removeCommonTableExpressionTables(node, schemableIds) {
         for (const expr of node.expressions) {
-            tables.delete(expr.name.table.table.identifier);
+            schemableIds.delete(expr.name.table.table.identifier.name);
         }
     }
 }