@prisma-next/sql-lane 0.3.0-dev.34 → 0.3.0-dev.37

package/src/sql/mutation-builder.ts

@@ -10,7 +10,7 @@ import {
   createUpdateAst,
 } from '@prisma-next/sql-relational-core/ast';
 import type { SqlQueryPlan } from '@prisma-next/sql-relational-core/plan';
-import type { QueryLaneContext } from '@prisma-next/sql-relational-core/query-lane-context';
+import type { ExecutionContext } from '@prisma-next/sql-relational-core/query-lane-context';
 import type {
   AnyColumnBuilder,
   BinaryBuilder,
@@ -75,7 +75,7 @@ export class InsertBuilderImpl<
 > implements InsertBuilder<TContract, CodecTypes, Row>
 {
   private readonly contract: TContract;
-  private readonly context: QueryLaneContext<TContract>;
+  private readonly context: ExecutionContext<TContract>;
   private readonly table: TableRef;
   private readonly values: Record<string, ParamPlaceholder>;
   private returningColumns: AnyColumnBuilder[] = [];
@@ -150,6 +150,31 @@ export class InsertBuilderImpl<
       values[columnName] = createParamRef(index, paramName);
     }
 
+    const appliedDefaults = this.context.applyMutationDefaults({
+      op: 'create',
+      table: this.table.name,
+      values,
+    });
+
+    for (const defaultValue of appliedDefaults) {
+      const columnMeta = contractTable.columns[defaultValue.column];
+      if (!columnMeta) {
+        errorUnknownColumn(defaultValue.column, this.table.name);
+      }
+
+      const index = paramValues.push(defaultValue.value);
+      paramCodecs[defaultValue.column] = columnMeta.codecId;
+      paramDescriptors.push({
+        name: defaultValue.column,
+        source: 'dsl',
+        refs: { table: this.table.name, column: defaultValue.column },
+        codecId: columnMeta.codecId,
+        nativeType: columnMeta.nativeType,
+        nullable: columnMeta.nullable,
+      });
+      values[defaultValue.column] = createParamRef(index, defaultValue.column);
+    }
+
     const returning: ColumnRef[] = this.returningColumns.map((col) => {
       // TypeScript can't narrow ColumnBuilder properly
       const c = col as unknown as { table: string; column: string };
@@ -204,7 +229,7 @@ export class UpdateBuilderImpl<
 > implements UpdateBuilder<TContract, CodecTypes, Row>
 {
   private readonly contract: TContract;
-  private readonly context: QueryLaneContext<TContract>;
+  private readonly context: ExecutionContext<TContract>;
   private readonly table: TableRef;
   private readonly set: Record<string, ParamPlaceholder>;
   private wherePredicate?: BinaryBuilder | UnaryBuilder;
@@ -300,6 +325,31 @@ export class UpdateBuilderImpl<
       set[columnName] = createParamRef(index, paramName);
     }
 
+    const appliedDefaults = this.context.applyMutationDefaults({
+      op: 'update',
+      table: this.table.name,
+      values: set,
+    });
+
+    for (const defaultValue of appliedDefaults) {
+      const columnMeta = contractTable.columns[defaultValue.column];
+      if (!columnMeta) {
+        errorUnknownColumn(defaultValue.column, this.table.name);
+      }
+
+      const index = paramValues.push(defaultValue.value);
+      paramCodecs[defaultValue.column] = columnMeta.codecId;
+      paramDescriptors.push({
+        name: defaultValue.column,
+        source: 'dsl',
+        refs: { table: this.table.name, column: defaultValue.column },
+        codecId: columnMeta.codecId,
+        nativeType: columnMeta.nativeType,
+        nullable: columnMeta.nullable,
+      });
+      set[defaultValue.column] = createParamRef(index, defaultValue.column);
+    }
+
     const whereResult = buildWhereExpr(
       this.contract,
       this.wherePredicate,
@@ -373,7 +423,7 @@ export class DeleteBuilderImpl<
 > implements DeleteBuilder<TContract, CodecTypes, Row>
 {
   private readonly contract: TContract;
-  private readonly context: QueryLaneContext<TContract>;
+  private readonly context: ExecutionContext<TContract>;
   private readonly table: TableRef;
   private wherePredicate?: BinaryBuilder | UnaryBuilder;
   private returningColumns: AnyColumnBuilder[] = [];

package/src/sql/plan.ts

@@ -285,7 +285,7 @@ export function buildMeta(args: MetaBuildArgs): PlanMeta {
     compact({
       target: args.contract.target,
       targetFamily: args.contract.targetFamily,
-      coreHash: args.contract.coreHash,
+      storageHash: args.contract.storageHash,
       lane: 'dsl',
       refs: {
         tables: Array.from(refsTables),

package/src/sql/select-builder.ts

@@ -17,7 +17,7 @@ import {
   createTableRef,
 } from '@prisma-next/sql-relational-core/ast';
 import type { SqlQueryPlan } from '@prisma-next/sql-relational-core/plan';
-import type { QueryLaneContext } from '@prisma-next/sql-relational-core/query-lane-context';
+import type { ExecutionContext } from '@prisma-next/sql-relational-core/query-lane-context';
 import type {
   AnyBinaryBuilder,
   AnyOrderBuilder,
@@ -64,7 +64,7 @@ export class SelectBuilderImpl<
 > {
   private readonly contract: TContract;
   private readonly codecTypes: CodecTypes;
-  private readonly context: QueryLaneContext<TContract>;
+  private readonly context: ExecutionContext<TContract>;
   private state: BuilderState = {};
 
   constructor(options: SqlBuilderOptions<TContract>, state?: BuilderState) {