prisma-sql 1.68.0 → 1.69.0

package/dist/index.d.mts

@@ -1,8 +1,17 @@
 import { Model as Model$1, ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
-import { DMMF } from '@prisma/generator-helper';
 
 type SqlDialect = 'postgres' | 'sqlite';
 
+interface LateralRelationMeta {
+    name: string;
+    isList: boolean;
+    fieldTypes: Array<{
+        fieldName: string;
+        type: string;
+    }>;
+    nestedRelations: LateralRelationMeta[];
+}
+
 type Model = Model$1;
 interface ParamMapping {
     index: number;
@@ -15,8 +24,9 @@ interface SqlResult {
     paramMappings?: ParamMapping[];
     requiresReduction?: boolean;
     includeSpec?: Record<string, any>;
-    supportsStreaming?: boolean;
-    isArrayAgg?: boolean;
+    isLateral?: boolean;
+    lateralMeta?: LateralRelationMeta[];
+    skipWhereIn?: boolean;
 }
 type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
@@ -42,34 +52,12 @@ interface SQLDirective {
     paramMappings: readonly ParamMap[];
     requiresReduction: boolean;
     includeSpec: Record<string, any>;
-    isArrayAgg: boolean;
+    isLateral: boolean;
+    lateralMeta?: LateralRelationMeta[];
+    skipWhereIn?: boolean;
     originalDirective: DirectiveProps;
 }
 
-interface BatchQuery {
-    model: string;
-    method: PrismaMethod;
-    args?: Record<string, unknown>;
-}
-interface BatchCountQuery {
-    model: string;
-    method: 'count';
-    args?: {
-        where?: Record<string, unknown>;
-    };
-}
-interface BatchResult {
-    sql: string;
-    params: unknown[];
-}
-declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult & {
-    keys: string[];
-    aliases: string[];
-};
-declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
-declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
-declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
-
 interface TransactionQuery {
     model: string;
     method: PrismaMethod;
@@ -99,9 +87,12 @@ interface WhereInSegment {
     fkFieldName: string;
     parentKeyFieldName: string;
     isList: boolean;
+    perParentTake?: number;
+    perParentSkip?: number;
 }
 interface QueryPlan {
     filteredArgs: any;
+    originalArgs: any;
     whereInSegments: WhereInSegment[];
     injectedParentKeys: string[];
 }
@@ -111,8 +102,11 @@ declare function planQueryStrategy(params: {
     args: any;
     allModels: readonly Model[];
     dialect: 'postgres' | 'sqlite';
+    debug?: boolean;
 }): QueryPlan;
 
+type ExecuteFn = (sql: string, params: unknown[]) => Promise<any[]>;
+
 interface ExecuteWhereInParams {
     segments: WhereInSegment[];
     parentRows: any[];
@@ -120,32 +114,10 @@ interface ExecuteWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
-    originalArgs?: any;
-    method?: string;
+    execute: ExecuteFn;
 }
-
 declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
 
-interface ArrayAggRelationMeta {
-    name: string;
-    isList: boolean;
-    scalarFields: Array<{
-        fieldName: string;
-        colName: string;
-        isJson: boolean;
-    }>;
-    pkFieldName: string;
-}
-interface ArrayAggReducerConfig {
-    parentModel: Model;
-    parentScalarFields: string[];
-    parentJsonSet: ReadonlySet<string>;
-    relations: ArrayAggRelationMeta[];
-}
-declare function buildArrayAggReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[]): ArrayAggReducerConfig;
-declare function reduceArrayAggRows(rows: any[], config: ArrayAggReducerConfig): any[];
-
 interface StreamingWhereInParams {
     segments: WhereInSegment[];
     parentSql: string;
@@ -154,7 +126,9 @@ interface StreamingWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    execute: ExecuteFn;
+    stream?: (sql: string, params: unknown[], onRow: (row: any) => void) => Promise<void>;
+    parentTake?: number;
 }
 interface PreFetchedWhereInParams {
     segments: WhereInSegment[];
@@ -163,13 +137,46 @@ interface PreFetchedWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    execute: ExecuteFn;
 }
 declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
 declare function executeWithPreFetchedParents(params: PreFetchedWhereInParams): Promise<any[]>;
 
 declare function getPrimaryKeyField(model: Model): string;
 
+interface LateralReducerConfig {
+    parentScalarFields: string[];
+    parentJsonSet: ReadonlySet<string>;
+    relations: LateralRelationMeta[];
+}
+declare function buildLateralReducerConfig(parentModel: Model, lateralMeta: LateralRelationMeta[]): LateralReducerConfig;
+declare function reduceLateralRows(rows: any[], config: LateralReducerConfig): any[];
+
+interface BatchQuery {
+    model: string;
+    method: PrismaMethod;
+    args?: Record<string, unknown>;
+}
+interface BatchCountQuery {
+    model: string;
+    method: 'count';
+    args?: {
+        where?: Record<string, unknown>;
+    };
+}
+interface BatchResult {
+    sql: string;
+    params: unknown[];
+}
+declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult & {
+    keys: string[];
+    aliases: string[];
+};
+declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
+
+declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
+declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
+
 interface ReducerConfig {
     parentModel: Model;
     includedRelations: RelationMetadata[];
@@ -219,15 +226,37 @@ declare function getRowTransformer(method: string): ((row: any) => any) | null;
 declare function getOrPrepareStatement(client: any, sql: string): any;
 declare function shouldSqliteUseGet(method: string): boolean;
 declare function normalizeParams(params: unknown[]): unknown[];
-declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[], isArrayAgg?: boolean): Promise<unknown[]>;
+interface PostgresQueryOptions {
+    client: any;
+    sql: string;
+    params: unknown[];
+    method: string;
+    requiresReduction: boolean;
+    includeSpec?: Record<string, any>;
+    model: any;
+    allModels: readonly any[];
+    isLateral?: boolean;
+    lateralMeta?: LateralRelationMeta[];
+}
+declare function executePostgresQuery(opts: PostgresQueryOptions): Promise<unknown[]>;
 declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[]): unknown[];
 declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
 
+type RelStats = {
+    avg: number;
+    p95: number;
+    p99: number;
+    max: number;
+    coverage: number;
+};
+type RelationStatsMap = Record<string, Record<string, RelStats>>;
+declare function setRoundtripRowEquivalent(value: number): void;
+declare function setJsonRowFactor(value: number): void;
+declare function setRelationStats(stats: RelationStatsMap): void;
+declare function getRelationStats(): RelationStatsMap | undefined;
+declare function countIncludeDepth(includeSpec: Record<string, any>, model: Model, schemas: readonly Model[]): number;
+
 declare function buildSQL(model: Model, models: Model[], method: PrismaMethod, args: Record<string, unknown>, dialect: SqlDialect): SqlResult;
-declare function createToSQL(modelsOrDmmf: Model[] | DMMF.Document, dialect: SqlDialect): (model: string, method: PrismaMethod, args?: Record<string, unknown>) => SqlResult;
-declare function createPrismaSQL<TClient>(config: PrismaSQLConfig<TClient>): PrismaSQLResult<TClient>;
 declare function generateSQL(directive: DirectiveProps): SQLDirective;
-declare function generateAllSQL(directives: DirectiveProps[]): SQLDirective[];
-declare function generateSQLByModel(directives: DirectiveProps[]): Map<string, SQLDirective[]>;
 
-export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildArrayAggReducerConfig, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getPrimaryKeyField, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceArrayAggRows, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type LateralRelationMeta, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildLateralReducerConfig, buildReducerConfig, buildSQL, countIncludeDepth, createProgressiveReducer, createStreamingReducer, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateSQL, getOrPrepareStatement, getPrimaryKeyField, getRelationStats, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, reduceLateralRows, setJsonRowFactor, setRelationStats, setRoundtripRowEquivalent, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };

package/dist/index.d.ts

@@ -1,8 +1,17 @@
 import { Model as Model$1, ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
-import { DMMF } from '@prisma/generator-helper';
 
 type SqlDialect = 'postgres' | 'sqlite';
 
+interface LateralRelationMeta {
+    name: string;
+    isList: boolean;
+    fieldTypes: Array<{
+        fieldName: string;
+        type: string;
+    }>;
+    nestedRelations: LateralRelationMeta[];
+}
+
 type Model = Model$1;
 interface ParamMapping {
     index: number;
@@ -15,8 +24,9 @@ interface SqlResult {
     paramMappings?: ParamMapping[];
     requiresReduction?: boolean;
     includeSpec?: Record<string, any>;
-    supportsStreaming?: boolean;
-    isArrayAgg?: boolean;
+    isLateral?: boolean;
+    lateralMeta?: LateralRelationMeta[];
+    skipWhereIn?: boolean;
 }
 type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
@@ -42,34 +52,12 @@ interface SQLDirective {
     paramMappings: readonly ParamMap[];
     requiresReduction: boolean;
     includeSpec: Record<string, any>;
-    isArrayAgg: boolean;
+    isLateral: boolean;
+    lateralMeta?: LateralRelationMeta[];
+    skipWhereIn?: boolean;
     originalDirective: DirectiveProps;
 }
 
-interface BatchQuery {
-    model: string;
-    method: PrismaMethod;
-    args?: Record<string, unknown>;
-}
-interface BatchCountQuery {
-    model: string;
-    method: 'count';
-    args?: {
-        where?: Record<string, unknown>;
-    };
-}
-interface BatchResult {
-    sql: string;
-    params: unknown[];
-}
-declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult & {
-    keys: string[];
-    aliases: string[];
-};
-declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
-declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
-declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
-
 interface TransactionQuery {
     model: string;
     method: PrismaMethod;
@@ -99,9 +87,12 @@ interface WhereInSegment {
     fkFieldName: string;
     parentKeyFieldName: string;
     isList: boolean;
+    perParentTake?: number;
+    perParentSkip?: number;
 }
 interface QueryPlan {
     filteredArgs: any;
+    originalArgs: any;
     whereInSegments: WhereInSegment[];
     injectedParentKeys: string[];
 }
@@ -111,8 +102,11 @@ declare function planQueryStrategy(params: {
     args: any;
     allModels: readonly Model[];
     dialect: 'postgres' | 'sqlite';
+    debug?: boolean;
 }): QueryPlan;
 
+type ExecuteFn = (sql: string, params: unknown[]) => Promise<any[]>;
+
 interface ExecuteWhereInParams {
     segments: WhereInSegment[];
     parentRows: any[];
@@ -120,32 +114,10 @@ interface ExecuteWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
-    originalArgs?: any;
-    method?: string;
+    execute: ExecuteFn;
 }
-
 declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
 
-interface ArrayAggRelationMeta {
-    name: string;
-    isList: boolean;
-    scalarFields: Array<{
-        fieldName: string;
-        colName: string;
-        isJson: boolean;
-    }>;
-    pkFieldName: string;
-}
-interface ArrayAggReducerConfig {
-    parentModel: Model;
-    parentScalarFields: string[];
-    parentJsonSet: ReadonlySet<string>;
-    relations: ArrayAggRelationMeta[];
-}
-declare function buildArrayAggReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[]): ArrayAggReducerConfig;
-declare function reduceArrayAggRows(rows: any[], config: ArrayAggReducerConfig): any[];
-
 interface StreamingWhereInParams {
     segments: WhereInSegment[];
     parentSql: string;
@@ -154,7 +126,9 @@ interface StreamingWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    execute: ExecuteFn;
+    stream?: (sql: string, params: unknown[], onRow: (row: any) => void) => Promise<void>;
+    parentTake?: number;
 }
 interface PreFetchedWhereInParams {
     segments: WhereInSegment[];
@@ -163,13 +137,46 @@ interface PreFetchedWhereInParams {
     allModels: readonly Model[];
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    execute: ExecuteFn;
 }
 declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
 declare function executeWithPreFetchedParents(params: PreFetchedWhereInParams): Promise<any[]>;
 
 declare function getPrimaryKeyField(model: Model): string;
 
+interface LateralReducerConfig {
+    parentScalarFields: string[];
+    parentJsonSet: ReadonlySet<string>;
+    relations: LateralRelationMeta[];
+}
+declare function buildLateralReducerConfig(parentModel: Model, lateralMeta: LateralRelationMeta[]): LateralReducerConfig;
+declare function reduceLateralRows(rows: any[], config: LateralReducerConfig): any[];
+
+interface BatchQuery {
+    model: string;
+    method: PrismaMethod;
+    args?: Record<string, unknown>;
+}
+interface BatchCountQuery {
+    model: string;
+    method: 'count';
+    args?: {
+        where?: Record<string, unknown>;
+    };
+}
+interface BatchResult {
+    sql: string;
+    params: unknown[];
+}
+declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult & {
+    keys: string[];
+    aliases: string[];
+};
+declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
+
+declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
+declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
+
 interface ReducerConfig {
     parentModel: Model;
     includedRelations: RelationMetadata[];
@@ -219,15 +226,37 @@ declare function getRowTransformer(method: string): ((row: any) => any) | null;
 declare function getOrPrepareStatement(client: any, sql: string): any;
 declare function shouldSqliteUseGet(method: string): boolean;
 declare function normalizeParams(params: unknown[]): unknown[];
-declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[], isArrayAgg?: boolean): Promise<unknown[]>;
+interface PostgresQueryOptions {
+    client: any;
+    sql: string;
+    params: unknown[];
+    method: string;
+    requiresReduction: boolean;
+    includeSpec?: Record<string, any>;
+    model: any;
+    allModels: readonly any[];
+    isLateral?: boolean;
+    lateralMeta?: LateralRelationMeta[];
+}
+declare function executePostgresQuery(opts: PostgresQueryOptions): Promise<unknown[]>;
 declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[]): unknown[];
 declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
 
+type RelStats = {
+    avg: number;
+    p95: number;
+    p99: number;
+    max: number;
+    coverage: number;
+};
+type RelationStatsMap = Record<string, Record<string, RelStats>>;
+declare function setRoundtripRowEquivalent(value: number): void;
+declare function setJsonRowFactor(value: number): void;
+declare function setRelationStats(stats: RelationStatsMap): void;
+declare function getRelationStats(): RelationStatsMap | undefined;
+declare function countIncludeDepth(includeSpec: Record<string, any>, model: Model, schemas: readonly Model[]): number;
+
 declare function buildSQL(model: Model, models: Model[], method: PrismaMethod, args: Record<string, unknown>, dialect: SqlDialect): SqlResult;
-declare function createToSQL(modelsOrDmmf: Model[] | DMMF.Document, dialect: SqlDialect): (model: string, method: PrismaMethod, args?: Record<string, unknown>) => SqlResult;
-declare function createPrismaSQL<TClient>(config: PrismaSQLConfig<TClient>): PrismaSQLResult<TClient>;
 declare function generateSQL(directive: DirectiveProps): SQLDirective;
-declare function generateAllSQL(directives: DirectiveProps[]): SQLDirective[];
-declare function generateSQLByModel(directives: DirectiveProps[]): Map<string, SQLDirective[]>;
 
-export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildArrayAggReducerConfig, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getPrimaryKeyField, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceArrayAggRows, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type LateralRelationMeta, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildLateralReducerConfig, buildReducerConfig, buildSQL, countIncludeDepth, createProgressiveReducer, createStreamingReducer, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateSQL, getOrPrepareStatement, getPrimaryKeyField, getRelationStats, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, reduceLateralRows, setJsonRowFactor, setRelationStats, setRoundtripRowEquivalent, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };