@prisma-next/sql-contract-psl 0.13.0 → 0.14.0-dev.2

package/src/psl-field-resolution.ts

@@ -1,14 +1,21 @@
 import type { ContractSourceDiagnostic } from '@prisma-next/config/config-types';
-import type { ColumnDefault, ExecutionMutationDefaultPhases } from '@prisma-next/contract/types';
+import type {
+  ColumnDefault,
+  ColumnDefaultLiteralInputValue,
+  ExecutionMutationDefaultPhases,
+} from '@prisma-next/contract/types';
 import type { AuthoringContributions } from '@prisma-next/framework-components/authoring';
 import type {
   ControlMutationDefaultRegistry,
   MutationDefaultGeneratorDescriptor,
 } from '@prisma-next/framework-components/control';
 import type { PslAttribute, PslField, PslModel } from '@prisma-next/psl-parser';
+import type { EnumTypeHandle } from '@prisma-next/sql-contract-ts/contract-builder';
+import { blindCast } from '@prisma-next/utils/casts';
 import { ifDefined } from '@prisma-next/utils/defined';
 import {
   getAttribute,
+  getPositionalArgumentEntry,
   lowerFirst,
   parseConstraintMapArgument,
   parseMapName,
@@ -21,6 +28,61 @@ import {
   resolveFieldTypeDescriptor,
 } from './psl-column-resolution';
 
+type LoweredFieldDefault = {
+  readonly defaultValue?: ColumnDefault;
+  readonly executionDefaults?: ExecutionMutationDefaultPhases;
+};
+
+function lowerEnumDefaultForField(input: {
+  readonly modelName: string;
+  readonly fieldName: string;
+  readonly defaultAttribute: PslAttribute;
+  readonly enumHandle: EnumTypeHandle;
+  readonly sourceId: string;
+  readonly diagnostics: ContractSourceDiagnostic[];
+}): LoweredFieldDefault {
+  const expressionEntry = getPositionalArgumentEntry(input.defaultAttribute);
+  if (!expressionEntry) {
+    return {};
+  }
+
+  const raw = expressionEntry.value.trim();
+  const isQuotedString = /^(['"]).*\1$/.test(raw);
+  const isFunctionCall = raw.includes('(') && raw.endsWith(')');
+
+  if (isQuotedString || isFunctionCall) {
+    input.diagnostics.push({
+      code: 'PSL_ENUM_DEFAULT_MUST_BE_MEMBER_NAME',
+      message: `Field "${input.modelName}.${input.fieldName}" @default on an enum field must name a member (e.g. @default(Low)), not a raw value or function.`,
+      sourceId: input.sourceId,
+      span: input.defaultAttribute.span,
+    });
+    return {};
+  }
+
+  const match = input.enumHandle.enumMembers.find((m) => m.name === raw);
+  if (!match) {
+    const validNames = input.enumHandle.enumMembers.map((m) => m.name).join(', ');
+    input.diagnostics.push({
+      code: 'PSL_ENUM_UNKNOWN_DEFAULT_MEMBER',
+      message: `Field "${input.modelName}.${input.fieldName}" @default(${raw}) does not name a member of ${input.enumHandle.enumName}. Valid members: ${validNames}.`,
+      sourceId: input.sourceId,
+      span: input.defaultAttribute.span,
+    });
+    return {};
+  }
+
+  return {
+    defaultValue: {
+      kind: 'literal',
+      value: blindCast<
+        ColumnDefaultLiteralInputValue,
+        'enum member values are codec-validated JsonValue-compatible scalars'
+      >(match.value),
+    },
+  };
+}
+
 export type ResolvedField = {
   readonly field: PslField;
   readonly columnName: string;
@@ -76,6 +138,7 @@ export interface CollectResolvedFieldsInput {
   readonly diagnostics: ContractSourceDiagnostic[];
   readonly sourceId: string;
   readonly scalarTypeDescriptors: ReadonlyMap<string, ColumnDescriptor>;
+  readonly enumHandles?: ReadonlyMap<string, EnumTypeHandle>;
 }
 
 const BUILTIN_FIELD_ATTRIBUTE_NAMES: ReadonlySet<string> = new Set([
@@ -95,7 +158,7 @@ const BUILTIN_FIELD_ATTRIBUTE_NAMES: ReadonlySet<string> = new Set([
  * migrated (so they don't get told to do what they just did).
  *
  * Pairing the suppression predicate with the hint makes each entry
- * self-contained: a future entry for, say, `@id` ↔ `id.uuidv7()` cannot
+ * self-contained: a future entry for, say, `@id` ↔ `id.uuidv7String()` cannot
  * silently inherit the wrong predicate when added.
  */
 interface RemovedAttributeRule {
@@ -225,6 +288,7 @@ export function collectResolvedFields(input: CollectResolvedFieldsInput): Resolv
     diagnostics,
     sourceId,
     scalarTypeDescriptors,
+    enumHandles,
   } = input;
   const resolvedFields: ResolvedField[] = [];
 
@@ -350,17 +414,27 @@ export function collectResolvedFields(input: CollectResolvedFieldsInput): Resolv
       });
       continue;
     }
-    const loweredDefault = defaultAttribute
-      ? lowerDefaultForField({
-          modelName: model.name,
-          fieldName: field.name,
-          defaultAttribute,
-          columnDescriptor: descriptor,
-          generatorDescriptorById,
-          sourceId,
-          defaultFunctionRegistry,
-          diagnostics,
-        })
+    const enumHandle = enumHandles?.get(field.typeName);
+    const loweredDefault: LoweredFieldDefault = defaultAttribute
+      ? enumHandle
+        ? lowerEnumDefaultForField({
+            modelName: model.name,
+            fieldName: field.name,
+            defaultAttribute,
+            enumHandle,
+            sourceId,
+            diagnostics,
+          })
+        : lowerDefaultForField({
+            modelName: model.name,
+            fieldName: field.name,
+            defaultAttribute,
+            columnDescriptor: descriptor,
+            generatorDescriptorById,
+            sourceId,
+            defaultFunctionRegistry,
+            diagnostics,
+          })
       : {};
     const loweredOnCreate = loweredDefault.executionDefaults?.onCreate;
     if (field.optional && loweredOnCreate) {
@@ -374,7 +448,9 @@ export function collectResolvedFields(input: CollectResolvedFieldsInput): Resolv
       });
       continue;
     }
-    if (loweredOnCreate) {
+    const fieldUsesNamedType =
+      field.typeRef !== undefined || namedTypeDescriptors.has(field.typeName);
+    if (loweredOnCreate && !fieldUsesNamedType) {
       const generatorDescriptor = generatorDescriptorById.get(loweredOnCreate.id);
       const generatedDescriptor = generatorDescriptor?.resolveGeneratedColumnDescriptor?.({
         generated: loweredOnCreate,