prisma-sql 1.64.0 → 1.66.0

package/dist/index.d.mts

@@ -1,66 +1,34 @@
-import { ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
+import { Model as Model$1, ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
 import { DMMF } from '@prisma/generator-helper';
 
 type SqlDialect = 'postgres' | 'sqlite';
 
-interface BatchQuery {
-    model: string;
-    method: PrismaMethod;
-    args?: Record<string, unknown>;
+type Model = Model$1;
+interface ParamMapping {
+    index: number;
+    dynamicName?: string;
+    value?: unknown;
 }
-interface BatchCountQuery {
-    model: string;
-    method: 'count';
-    args?: {
-        where?: Record<string, unknown>;
-    };
-}
-interface BatchResult {
+interface SqlResult {
     sql: string;
     params: unknown[];
+    paramMappings?: ParamMapping[];
+    requiresReduction?: boolean;
+    includeSpec?: Record<string, any>;
+    supportsStreaming?: boolean;
 }
-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>, count: 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 Field {
-    name: string;
-    dbName: string;
-    type: string;
-    isRequired: boolean;
-    isRelation: boolean;
-    isId?: boolean;
-    relatedModel?: string;
-    relationName?: string;
-    foreignKey?: string | string[];
-    references?: string | string[];
-    isForeignKeyLocal?: boolean;
-}
-interface Model {
-    name: string;
-    tableName: string;
-    fields: Field[];
-}
-type PrismaMethod = 'findMany' | 'findFirst' | 'findUnique' | 'count' | 'aggregate' | 'groupBy';
+type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
     client: TClient;
     models?: Model[];
-    dmmf?: DMMF.Document;
-    dialect: SqlDialect;
+    dmmf?: any;
+    dialect: 'postgres' | 'sqlite';
     execute: (client: TClient, sql: string, params: unknown[]) => Promise<unknown[]>;
 }
-interface SqlResult {
-    sql: string;
-    params: unknown[];
-}
 interface PrismaSQLResult<TClient> {
     toSQL: (model: string, method: PrismaMethod, args?: Record<string, unknown>) => SqlResult;
     query: <T = unknown[]>(model: string, method: PrismaMethod, args?: Record<string, unknown>) => Promise<T>;
-    batchSql: (queries: Record<string, BatchQuery>) => SqlResult;
+    batchSql: (queries: Record<string, any>) => SqlResult;
     client: TClient;
 }
 
@@ -71,9 +39,35 @@ interface SQLDirective {
     dynamicKeys: string[];
     paramOrder: string;
     paramMappings: readonly ParamMap[];
+    requiresReduction: boolean;
+    includeSpec: Record<string, any>;
     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>, count: 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;
@@ -94,10 +88,110 @@ declare function createTransactionExecutor(deps: {
     postgresClient?: any;
 }): TransactionExecutor;
 
-declare function transformQueryResults(method: PrismaMethod, results: unknown[]): unknown;
+declare function transformQueryResults(method: string, results: unknown): unknown;
+
+type RelStats = {
+    avg: number;
+    p95: number;
+    p99: number;
+    max: number;
+    coverage: number;
+};
+
+interface WhereInSegment {
+    relationName: string;
+    relArgs: unknown;
+    childModelName: string;
+    fkFieldName: string;
+    parentKeyFieldName: string;
+    isList: boolean;
+}
+interface QueryPlan {
+    filteredArgs: any;
+    whereInSegments: WhereInSegment[];
+}
+type RelationStatsLike = Record<string, Record<string, RelStats>>;
+declare function planQueryStrategy(params: {
+    model: Model;
+    method: string;
+    args: any;
+    allModels: readonly Model[];
+    relationStats?: RelationStatsLike;
+    dialect: 'postgres' | 'sqlite';
+}): QueryPlan;
+
+interface ExecuteWhereInParams {
+    segments: WhereInSegment[];
+    parentRows: any[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+
+declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
+
+interface ReducerConfig {
+    parentModel: Model;
+    includedRelations: RelationMetadata[];
+    allModels: readonly Model[];
+}
+interface ScalarColSpec {
+    fieldName: string;
+    colName: string;
+    isJson: boolean;
+}
+interface RelationMetadata {
+    name: string;
+    cardinality: 'one' | 'many';
+    relatedModel: Model;
+    primaryKeyFields: string[];
+    includeAllScalars: boolean;
+    selectedScalarFields: string[];
+    nestedIncludes?: ReducerConfig | null;
+    path: string;
+    keyCols: string[];
+    scalarCols: ScalarColSpec[];
+}
+declare function buildReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[], prefix?: string, depth?: number): ReducerConfig;
+declare function reduceFlatRows(rows: any[], config: ReducerConfig): any[];
 
 declare function normalizeValue(value: unknown, seen?: WeakSet<object>, depth?: number): unknown;
 
+declare function createStreamingReducer(config: ReducerConfig): {
+    processRow(row: any): void;
+    getResults(): any[];
+    getParentMap(): Map<string, any>;
+};
+
+interface ProgressiveReducer {
+    processRow(row: any): void;
+    getCurrentParentKey(row: any): string | null;
+    getCompletedParent(parentKey: string): any | null;
+    getRemainingParents(): any[];
+}
+declare function createProgressiveReducer(config: ReducerConfig): ProgressiveReducer;
+
+interface StreamingWhereInParams extends ExecuteWhereInParams {
+    parentSql: string;
+    parentParams: unknown[];
+    batchSize?: number;
+    maxConcurrency?: number;
+}
+declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
+
+declare function transformAggregateRow(row: any): any;
+declare function extractCountValue(row: any): number | bigint;
+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 | undefined, allModels: readonly any[]): Promise<unknown[]>;
+declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): unknown[];
+declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
+
 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>;
@@ -105,4 +199,4 @@ 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 SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildSQL, createPrismaSQL, createToSQL, createTransactionExecutor, generateAllSQL, generateSQL, generateSQLByModel, normalizeValue, parseBatchCountResults, parseBatchResults, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };

package/dist/index.d.ts

@@ -1,66 +1,34 @@
-import { ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
+import { Model as Model$1, ParamMap, DirectiveProps } from '@dee-wan/schema-parser';
 import { DMMF } from '@prisma/generator-helper';
 
 type SqlDialect = 'postgres' | 'sqlite';
 
-interface BatchQuery {
-    model: string;
-    method: PrismaMethod;
-    args?: Record<string, unknown>;
+type Model = Model$1;
+interface ParamMapping {
+    index: number;
+    dynamicName?: string;
+    value?: unknown;
 }
-interface BatchCountQuery {
-    model: string;
-    method: 'count';
-    args?: {
-        where?: Record<string, unknown>;
-    };
-}
-interface BatchResult {
+interface SqlResult {
     sql: string;
     params: unknown[];
+    paramMappings?: ParamMapping[];
+    requiresReduction?: boolean;
+    includeSpec?: Record<string, any>;
+    supportsStreaming?: boolean;
 }
-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>, count: 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 Field {
-    name: string;
-    dbName: string;
-    type: string;
-    isRequired: boolean;
-    isRelation: boolean;
-    isId?: boolean;
-    relatedModel?: string;
-    relationName?: string;
-    foreignKey?: string | string[];
-    references?: string | string[];
-    isForeignKeyLocal?: boolean;
-}
-interface Model {
-    name: string;
-    tableName: string;
-    fields: Field[];
-}
-type PrismaMethod = 'findMany' | 'findFirst' | 'findUnique' | 'count' | 'aggregate' | 'groupBy';
+type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
     client: TClient;
     models?: Model[];
-    dmmf?: DMMF.Document;
-    dialect: SqlDialect;
+    dmmf?: any;
+    dialect: 'postgres' | 'sqlite';
     execute: (client: TClient, sql: string, params: unknown[]) => Promise<unknown[]>;
 }
-interface SqlResult {
-    sql: string;
-    params: unknown[];
-}
 interface PrismaSQLResult<TClient> {
     toSQL: (model: string, method: PrismaMethod, args?: Record<string, unknown>) => SqlResult;
     query: <T = unknown[]>(model: string, method: PrismaMethod, args?: Record<string, unknown>) => Promise<T>;
-    batchSql: (queries: Record<string, BatchQuery>) => SqlResult;
+    batchSql: (queries: Record<string, any>) => SqlResult;
     client: TClient;
 }
 
@@ -71,9 +39,35 @@ interface SQLDirective {
     dynamicKeys: string[];
     paramOrder: string;
     paramMappings: readonly ParamMap[];
+    requiresReduction: boolean;
+    includeSpec: Record<string, any>;
     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>, count: 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;
@@ -94,10 +88,110 @@ declare function createTransactionExecutor(deps: {
     postgresClient?: any;
 }): TransactionExecutor;
 
-declare function transformQueryResults(method: PrismaMethod, results: unknown[]): unknown;
+declare function transformQueryResults(method: string, results: unknown): unknown;
+
+type RelStats = {
+    avg: number;
+    p95: number;
+    p99: number;
+    max: number;
+    coverage: number;
+};
+
+interface WhereInSegment {
+    relationName: string;
+    relArgs: unknown;
+    childModelName: string;
+    fkFieldName: string;
+    parentKeyFieldName: string;
+    isList: boolean;
+}
+interface QueryPlan {
+    filteredArgs: any;
+    whereInSegments: WhereInSegment[];
+}
+type RelationStatsLike = Record<string, Record<string, RelStats>>;
+declare function planQueryStrategy(params: {
+    model: Model;
+    method: string;
+    args: any;
+    allModels: readonly Model[];
+    relationStats?: RelationStatsLike;
+    dialect: 'postgres' | 'sqlite';
+}): QueryPlan;
+
+interface ExecuteWhereInParams {
+    segments: WhereInSegment[];
+    parentRows: any[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+
+declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
+
+interface ReducerConfig {
+    parentModel: Model;
+    includedRelations: RelationMetadata[];
+    allModels: readonly Model[];
+}
+interface ScalarColSpec {
+    fieldName: string;
+    colName: string;
+    isJson: boolean;
+}
+interface RelationMetadata {
+    name: string;
+    cardinality: 'one' | 'many';
+    relatedModel: Model;
+    primaryKeyFields: string[];
+    includeAllScalars: boolean;
+    selectedScalarFields: string[];
+    nestedIncludes?: ReducerConfig | null;
+    path: string;
+    keyCols: string[];
+    scalarCols: ScalarColSpec[];
+}
+declare function buildReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[], prefix?: string, depth?: number): ReducerConfig;
+declare function reduceFlatRows(rows: any[], config: ReducerConfig): any[];
 
 declare function normalizeValue(value: unknown, seen?: WeakSet<object>, depth?: number): unknown;
 
+declare function createStreamingReducer(config: ReducerConfig): {
+    processRow(row: any): void;
+    getResults(): any[];
+    getParentMap(): Map<string, any>;
+};
+
+interface ProgressiveReducer {
+    processRow(row: any): void;
+    getCurrentParentKey(row: any): string | null;
+    getCompletedParent(parentKey: string): any | null;
+    getRemainingParents(): any[];
+}
+declare function createProgressiveReducer(config: ReducerConfig): ProgressiveReducer;
+
+interface StreamingWhereInParams extends ExecuteWhereInParams {
+    parentSql: string;
+    parentParams: unknown[];
+    batchSize?: number;
+    maxConcurrency?: number;
+}
+declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
+
+declare function transformAggregateRow(row: any): any;
+declare function extractCountValue(row: any): number | bigint;
+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 | undefined, allModels: readonly any[]): Promise<unknown[]>;
+declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): unknown[];
+declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
+
 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>;
@@ -105,4 +199,4 @@ 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 SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildSQL, createPrismaSQL, createToSQL, createTransactionExecutor, generateAllSQL, generateSQL, generateSQLByModel, normalizeValue, parseBatchCountResults, parseBatchResults, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };