@prisma-next/sql-contract-psl 0.12.0-dev.6 → 0.12.0-dev.60

package/src/psl-attribute-parsing.ts

@@ -1,4 +1,5 @@
 import type { ContractSourceDiagnostic } from '@prisma-next/config/config-types';
+import type { ControlPolicy } from '@prisma-next/contract/types';
 import type { PslAttribute, PslSpan } from '@prisma-next/psl-parser';
 import { getPositionalArgument, parseQuotedStringLiteral } from '@prisma-next/psl-parser';
 
@@ -411,6 +412,71 @@ export function findDuplicateFieldName(fieldNames: readonly string[]): string |
   return undefined;
 }
 
+const CONTROL_POLICY_LITERALS = [
+  'managed',
+  'tolerated',
+  'external',
+  'observed',
+] as const satisfies readonly ControlPolicy[];
+
+const CONTROL_POLICY_LITERAL_SET = new Set<string>(CONTROL_POLICY_LITERALS);
+
+function isControlPolicyLiteral(value: string): value is ControlPolicy {
+  return CONTROL_POLICY_LITERAL_SET.has(value);
+}
+
+export function parseControlPolicyAttribute(input: {
+  readonly attribute: PslAttribute;
+  readonly sourceId: string;
+  readonly diagnostics: ContractSourceDiagnostic[];
+}): ControlPolicy | undefined {
+  const namedArgs = input.attribute.args.filter((arg) => arg.kind === 'named');
+  if (namedArgs.length > 0) {
+    input.diagnostics.push({
+      code: 'PSL_INVALID_ATTRIBUTE_ARGUMENT',
+      message:
+        '`@@control` does not accept named arguments; pass the policy positionally as `@@control(external)`.',
+      sourceId: input.sourceId,
+      span: input.attribute.span,
+    });
+    return undefined;
+  }
+
+  const positionalArgs = getPositionalArguments(input.attribute);
+  if (positionalArgs.length === 0) {
+    input.diagnostics.push({
+      code: 'PSL_INVALID_ATTRIBUTE_ARGUMENT',
+      message:
+        '`@@control` requires exactly one positional argument: `managed`, `tolerated`, `external`, or `observed`.',
+      sourceId: input.sourceId,
+      span: input.attribute.span,
+    });
+    return undefined;
+  }
+  if (positionalArgs.length > 1) {
+    input.diagnostics.push({
+      code: 'PSL_INVALID_ATTRIBUTE_ARGUMENT',
+      message: `\`@@control\` accepts exactly one positional argument; got ${positionalArgs.length}.`,
+      sourceId: input.sourceId,
+      span: input.attribute.span,
+    });
+    return undefined;
+  }
+
+  const token = unquoteStringLiteral(positionalArgs[0] ?? '').trim();
+  if (!isControlPolicyLiteral(token)) {
+    input.diagnostics.push({
+      code: 'PSL_INVALID_ATTRIBUTE_ARGUMENT',
+      message: `\`@@control\` argument \`${token}\` is not a known policy. Allowed: \`managed\`, \`tolerated\`, \`external\`, \`observed\`.`,
+      sourceId: input.sourceId,
+      span: input.attribute.span,
+    });
+    return undefined;
+  }
+
+  return token;
+}
+
 export function mapFieldNamesToColumns(input: {
   readonly modelName: string;
   readonly fieldNames: readonly string[];

package/src/psl-column-resolution.ts

@@ -25,6 +25,7 @@ import type {
   PslSpan,
   PslTypeConstructorCall,
 } from '@prisma-next/psl-parser';
+import { blindCast } from '@prisma-next/utils/casts';
 import {
   lowerDefaultFunctionWithRegistry,
   parseDefaultFunctionCall,
@@ -67,7 +68,9 @@ export function getAuthoringTypeConstructor(
     if (typeof current !== 'object' || current === null || Array.isArray(current)) {
       return undefined;
     }
-    current = (current as Record<string, unknown>)[segment];
+    current = blindCast<Record<string, unknown>, 'narrowed by preceding typeof/null/array guards'>(
+      current,
+    )[segment];
   }
 
   return isAuthoringTypeConstructorDescriptor(current) ? current : undefined;
@@ -94,7 +97,9 @@ export function getAuthoringEntity(
     if (typeof current !== 'object' || current === null || Array.isArray(current)) {
       return undefined;
     }
-    current = (current as Record<string, unknown>)[segment];
+    current = blindCast<Record<string, unknown>, 'narrowed by preceding typeof/null/array guards'>(
+      current,
+    )[segment];
   }
 
   return isAuthoringEntityTypeDescriptor(current) ? current : undefined;
@@ -115,7 +120,9 @@ export function getAuthoringFieldPreset(
     if (typeof current !== 'object' || current === null || Array.isArray(current)) {
       return undefined;
     }
-    current = (current as Record<string, unknown>)[segment];
+    current = blindCast<Record<string, unknown>, 'narrowed by preceding typeof/null/array guards'>(
+      current,
+    )[segment];
   }
 
   return isAuthoringFieldPresetDescriptor(current) ? current : undefined;

package/src/psl-field-resolution.ts

@@ -232,6 +232,12 @@ export function collectResolvedFields(input: CollectResolvedFieldsInput): Resolv
     if (isModelField && relationAttribute) {
       continue;
     }
+    // Cross-contract-space relation fields (e.g. `supabase:auth.User @relation(...)`) are not
+    // local model fields, but they carry a @relation attribute and should be skipped here.
+    // Their FK and RelationNode lowering is handled separately in the interpreter.
+    if (field.typeContractSpaceId !== undefined && relationAttribute) {
+      continue;
+    }
 
     const isValueObjectField = compositeTypeNames.has(field.typeName);
     const isListField = field.list;

package/src/psl-relation-resolution.ts

@@ -14,7 +14,7 @@ import {
 } from './psl-attribute-parsing';
 import { checkUncomposedNamespace, reportUncomposedNamespace } from './psl-column-resolution';
 
-export const REFERENTIAL_ACTION_MAP = {
+export const REFERENTIAL_ACTION_MAP: Record<string, ReferentialAction | undefined> = {
   NoAction: 'noAction',
   Restrict: 'restrict',
   Cascade: 'cascade',
@@ -25,7 +25,7 @@ export const REFERENTIAL_ACTION_MAP = {
   cascade: 'cascade',
   setNull: 'setNull',
   setDefault: 'setDefault',
-} as const;
+};
 
 export type ParsedRelationAttribute = {
   readonly relationName?: string;
@@ -71,8 +71,7 @@ export function normalizeReferentialAction(input: {
   readonly span: PslSpan;
   readonly diagnostics: ContractSourceDiagnostic[];
 }): ReferentialAction | undefined {
-  const normalized =
-    REFERENTIAL_ACTION_MAP[input.actionToken as keyof typeof REFERENTIAL_ACTION_MAP];
+  const normalized = REFERENTIAL_ACTION_MAP[input.actionToken];
   if (normalized) {
     return normalized;
   }