UNPKG

kysely

Version:
46 lines (45 loc) 2.97 kB
import { UpdateQueryBuilder } from '../query-builder/update-query-builder.js'; import { DeleteQueryBuilder } from '../query-builder/delete-query-builder.js'; import { InsertQueryBuilder } from '../query-builder/insert-query-builder.js'; import { QueryCreator } from '../query-creator.js'; import { Expression } from '../expression/expression.js'; import { ShallowRecord } from '../util/type-utils.js'; import { CTEBuilderCallback } from '../query-builder/cte-builder.js'; import { CommonTableExpressionNode } from '../operation-node/common-table-expression-node.js'; export type CommonTableExpression<DB, CN extends string> = (creator: QueryCreator<DB>) => CommonTableExpressionOutput<DB, CN>; export type RecursiveCommonTableExpression<DB, CN extends string> = (creator: QueryCreator<DB & { [K in ExtractTableFromCommonTableExpressionName<CN>]: ExtractRowFromCommonTableExpressionName<CN>; }>) => CommonTableExpressionOutput<DB, CN>; export type QueryCreatorWithCommonTableExpression<DB, CN extends string, CTE> = QueryCreator<DB & { [K in ExtractTableFromCommonTableExpressionName<CN>]: ExtractRowFromCommonTableExpression<CTE>; }>; type CommonTableExpressionOutput<DB, CN extends string> = Expression<ExtractRowFromCommonTableExpressionName<CN>> | InsertQueryBuilder<DB, any, ExtractRowFromCommonTableExpressionName<CN>> | UpdateQueryBuilder<DB, any, any, ExtractRowFromCommonTableExpressionName<CN>> | DeleteQueryBuilder<DB, any, ExtractRowFromCommonTableExpressionName<CN>>; /** * Given a common CommonTableExpression CTE extracts the row type from it. * * For example a CTE `(db) => db.selectFrom('person').select(['id', 'first_name'])` * would result in `Pick<Person, 'id' | 'first_name'>`. */ type ExtractRowFromCommonTableExpression<CTE> = CTE extends (creator: QueryCreator<any>) => infer Q ? Q extends Expression<infer QO> ? QO : Q extends InsertQueryBuilder<any, any, infer QO> ? QO : Q extends UpdateQueryBuilder<any, any, any, infer QO> ? QO : Q extends DeleteQueryBuilder<any, any, infer QO> ? QO : never : never; /** * Extracts 'person' from a string like 'person(id, first_name)'. */ type ExtractTableFromCommonTableExpressionName<CN extends string> = CN extends `${infer TB}(${string})` ? TB : CN; /** * Parses a string like 'person(id, first_name)' into a type: * * { * id: any, * first_name: any * } * */ type ExtractRowFromCommonTableExpressionName<CN extends string> = CN extends `${string}(${infer CL})` ? { [C in ExtractColumnNamesFromColumnList<CL>]: any; } : ShallowRecord<string, any>; /** * Parses a string like 'id, first_name' into a type 'id' | 'first_name' */ type ExtractColumnNamesFromColumnList<R extends string> = R extends `${infer C}, ${infer RS}` ? C | ExtractColumnNamesFromColumnList<RS> : R; export declare function parseCommonTableExpression(nameOrBuilderCallback: string | CTEBuilderCallback<string>, expression: CommonTableExpression<any, string>): CommonTableExpressionNode; export {};