@prisma-next/sql-contract-psl 0.0.1 → 0.3.0-dev.113

package/src/provider.ts

@@ -1,33 +1,32 @@
 import { readFile } from 'node:fs/promises';
-import type { ContractConfig } from '@prisma-next/config/config-types';
+import type { ContractConfig, ContractSourceContext } from '@prisma-next/config/config-types';
 import type { TargetPackRef } from '@prisma-next/contract/framework-components';
 import { parsePslDocument } from '@prisma-next/psl-parser';
 import { ifDefined } from '@prisma-next/utils/defined';
-import { notOk } from '@prisma-next/utils/result';
+import { notOk, ok } from '@prisma-next/utils/result';
 import { resolve } from 'pathe';
+import type { ControlMutationDefaults } from './default-function-registry';
 import { interpretPslDocumentToSqlContractIR } from './interpreter';
 
 export interface PrismaContractOptions {
   readonly output?: string;
-  readonly target?: TargetPackRef<'sql', 'postgres'>;
-  /**
-   * Milestone-local namespace availability hook.
-   *
-   * This currently models composed extension packs by id only (for example `["pgvector"]`),
-   * and is sufficient for namespace presence checks in the PSL interpreter.
-   *
-   * Future milestones can evolve this to richer composed pack metadata/manifests when
-   * attribute-level schema/argument validation needs to move beyond namespace existence.
-   */
+  readonly target: TargetPackRef<'sql', 'postgres'>;
+  readonly scalarTypeDescriptors: ReadonlyMap<
+    string,
+    {
+      readonly codecId: string;
+      readonly nativeType: string;
+      readonly typeRef?: string;
+      readonly typeParams?: Record<string, unknown>;
+    }
+  >;
+  readonly controlMutationDefaults?: ControlMutationDefaults;
   readonly composedExtensionPacks?: readonly string[];
 }
 
-export function prismaContract(
-  schemaPath: string,
-  options?: PrismaContractOptions,
-): ContractConfig {
+export function prismaContract(schemaPath: string, options: PrismaContractOptions): ContractConfig {
   return {
-    source: async () => {
+    source: async (context: ContractSourceContext) => {
       const absoluteSchemaPath = resolve(schemaPath);
       let schema: string;
       try {
@@ -51,13 +50,27 @@ export function prismaContract(
         schema,
         sourceId: schemaPath,
       });
+      const composedExtensionPacks = [
+        ...(context.composedExtensionPacks ?? []),
+        ...(options.composedExtensionPacks ?? []),
+      ];
 
-      return interpretPslDocumentToSqlContractIR({
+      const interpreted = interpretPslDocumentToSqlContractIR({
         document,
-        ...ifDefined('target', options?.target),
-        ...ifDefined('composedExtensionPacks', options?.composedExtensionPacks),
+        target: options.target,
+        scalarTypeDescriptors: options.scalarTypeDescriptors,
+        ...ifDefined(
+          'composedExtensionPacks',
+          composedExtensionPacks.length > 0 ? composedExtensionPacks : undefined,
+        ),
+        ...ifDefined('controlMutationDefaults', options.controlMutationDefaults),
       });
+      if (!interpreted.ok) {
+        return interpreted;
+      }
+
+      return ok(interpreted.value);
     },
-    ...ifDefined('output', options?.output),
+    ...ifDefined('output', options.output),
   };
 }