@prisma-next/target-postgres 0.3.0-pr.99.6 → 0.4.0-dev.2

package/src/core/migrations/descriptor-planner.ts

@@ -0,0 +1,466 @@
+/**
+ * Descriptor-based migration planner.
+ *
+ * Takes schema issues (from verifySqlSchema) and emits PostgresMigrationOpDescriptor[].
+ * Migration strategies consume issues they recognize and produce specialized op
+ * sequences (e.g., NOT NULL backfill → addColumn(nullable) + dataTransform + setNotNull).
+ * Remaining issues get default descriptor mapping.
+ *
+ * This planner does NOT produce SqlMigrationPlanOperation — that's the resolver's job.
+ * The separation means the same descriptors work for both planner-generated and
+ * user-authored migrations.
+ */
+
+import type { Contract } from '@prisma-next/contract/types';
+import type { SqlPlannerConflict } from '@prisma-next/family-sql/control';
+import type { SchemaIssue } from '@prisma-next/framework-components/control';
+import type { SqlStorage } from '@prisma-next/sql-contract/types';
+import type { Result } from '@prisma-next/utils/result';
+import { notOk, ok } from '@prisma-next/utils/result';
+import {
+  addColumn,
+  addForeignKey,
+  addPrimaryKey,
+  addUnique,
+  alterColumnType,
+  createDependency,
+  createEnumType,
+  createIndex,
+  createTable,
+  dropColumn,
+  dropConstraint,
+  dropDefault,
+  dropIndex,
+  dropNotNull,
+  dropTable,
+  type PostgresMigrationOpDescriptor,
+  setDefault,
+  setNotNull,
+} from './operation-descriptors';
+import {
+  type MigrationStrategy,
+  migrationPlanStrategies,
+  type StrategyContext,
+} from './planner-strategies';
+
+export type { MigrationStrategy, StrategyContext };
+
+// ============================================================================
+// Issue kind ordering (dependency order)
+// ============================================================================
+
+const ISSUE_KIND_ORDER: Record<string, number> = {
+  // Dependencies and types first
+  dependency_missing: 1,
+  type_missing: 2,
+  type_values_mismatch: 3,
+  enum_values_changed: 3,
+
+  // Drops (reconciliation — clear the way for creates)
+  // FKs dropped first (they depend on other constraints)
+  extra_foreign_key: 10,
+  extra_unique_constraint: 11,
+  extra_primary_key: 12,
+  extra_index: 13,
+  extra_default: 14,
+  extra_column: 15,
+  extra_table: 16,
+
+  // Tables before columns
+  missing_table: 20,
+
+  // Columns before constraints
+  missing_column: 30,
+
+  // Reconciliation alters (on existing objects)
+  type_mismatch: 40,
+  nullability_mismatch: 41,
+  default_missing: 42,
+  default_mismatch: 43,
+
+  // Constraints after columns exist
+  primary_key_mismatch: 50,
+  unique_constraint_mismatch: 51,
+  index_mismatch: 52,
+  foreign_key_mismatch: 60,
+};
+
+function issueOrder(issue: SchemaIssue): number {
+  return ISSUE_KIND_ORDER[issue.kind] ?? 99;
+}
+
+// ============================================================================
+// Conflict helpers
+// ============================================================================
+
+function issueConflict(
+  kind: SqlPlannerConflict['kind'],
+  summary: string,
+  location?: SqlPlannerConflict['location'],
+): SqlPlannerConflict {
+  return {
+    kind,
+    summary,
+    why: 'Use `migration new` to author a custom migration for this change.',
+    ...(location ? { location } : {}),
+  };
+}
+
+// ============================================================================
+// Default issue-to-descriptor mapping
+// ============================================================================
+
+function isMissing(issue: SchemaIssue): boolean {
+  if (issue.kind === 'enum_values_changed') return false;
+  return issue.actual === undefined;
+}
+
+function mapIssue(
+  issue: SchemaIssue,
+  ctx: StrategyContext,
+): Result<readonly PostgresMigrationOpDescriptor[], SqlPlannerConflict> {
+  switch (issue.kind) {
+    // Additive — missing structures
+    case 'missing_table': {
+      if (!issue.table)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Missing table issue has no table name'),
+        );
+      const contractTable = ctx.toContract.storage.tables[issue.table];
+      if (!contractTable) {
+        return notOk(
+          issueConflict(
+            'unsupportedOperation',
+            `Table "${issue.table}" reported missing but not found in destination contract`,
+          ),
+        );
+      }
+      const ops: PostgresMigrationOpDescriptor[] = [createTable(issue.table)];
+      for (const index of contractTable.indexes) {
+        ops.push(createIndex(issue.table, [...index.columns]));
+      }
+      const explicitIndexColumnSets = new Set(
+        contractTable.indexes.map((idx) => idx.columns.join(',')),
+      );
+      for (const fk of contractTable.foreignKeys) {
+        if (fk.constraint) {
+          ops.push(addForeignKey(issue.table, [...fk.columns]));
+        }
+        if (fk.index && !explicitIndexColumnSets.has(fk.columns.join(','))) {
+          ops.push(createIndex(issue.table, [...fk.columns]));
+        }
+      }
+      for (const unique of contractTable.uniques) {
+        ops.push(addUnique(issue.table, [...unique.columns]));
+      }
+      return ok(ops);
+    }
+
+    case 'missing_column':
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Missing column issue has no table/column name'),
+        );
+      return ok([addColumn(issue.table, issue.column)]);
+
+    case 'default_missing':
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Default missing issue has no table/column name'),
+        );
+      return ok([setDefault(issue.table, issue.column)]);
+
+    // Destructive — extra structures
+    case 'extra_table':
+      if (!issue.table)
+        return notOk(issueConflict('unsupportedOperation', 'Extra table issue has no table name'));
+      return ok([dropTable(issue.table)]);
+
+    case 'extra_column':
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Extra column issue has no table/column name'),
+        );
+      return ok([dropColumn(issue.table, issue.column)]);
+
+    case 'extra_index':
+      if (!issue.table || !issue.indexOrConstraint)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Extra index issue has no table/index name'),
+        );
+      return ok([dropIndex(issue.table, issue.indexOrConstraint)]);
+
+    case 'extra_unique_constraint':
+    case 'extra_foreign_key':
+    case 'extra_primary_key':
+      if (!issue.table || !issue.indexOrConstraint)
+        return notOk(
+          issueConflict(
+            'unsupportedOperation',
+            'Extra constraint issue has no table/constraint name',
+          ),
+        );
+      return ok([dropConstraint(issue.table, issue.indexOrConstraint)]);
+
+    case 'extra_default':
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Extra default issue has no table/column name'),
+        );
+      return ok([dropDefault(issue.table, issue.column)]);
+
+    // Nullability changes
+    case 'nullability_mismatch': {
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('nullabilityConflict', 'Nullability mismatch has no table/column name'),
+        );
+      const column = ctx.toContract.storage.tables[issue.table]?.columns[issue.column];
+      if (!column)
+        return notOk(
+          issueConflict(
+            'nullabilityConflict',
+            `Column "${issue.table}"."${issue.column}" not found in destination contract`,
+          ),
+        );
+      return ok(
+        column.nullable
+          ? [dropNotNull(issue.table, issue.column)]
+          : [setNotNull(issue.table, issue.column)],
+      );
+    }
+
+    // Type changes
+    case 'type_mismatch':
+      if (!issue.table || !issue.column)
+        return notOk(issueConflict('typeMismatch', 'Type mismatch has no table/column name'));
+      return ok([alterColumnType(issue.table, issue.column)]);
+
+    // Default changes
+    case 'default_mismatch':
+      if (!issue.table || !issue.column)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Default mismatch has no table/column name'),
+        );
+      return ok([setDefault(issue.table, issue.column)]);
+
+    // Constraints — missing (actual undefined) vs mismatched (actual defined)
+    case 'primary_key_mismatch':
+      if (!issue.table)
+        return notOk(issueConflict('indexIncompatible', 'Primary key issue has no table name'));
+      if (isMissing(issue)) return ok([addPrimaryKey(issue.table)]);
+      return notOk(
+        issueConflict(
+          'indexIncompatible',
+          `Primary key on "${issue.table}" has different columns (expected: ${issue.expected}, actual: ${issue.actual})`,
+          { table: issue.table },
+        ),
+      );
+
+    case 'unique_constraint_mismatch':
+      if (!issue.table)
+        return notOk(
+          issueConflict('indexIncompatible', 'Unique constraint issue has no table name'),
+        );
+      if (isMissing(issue) && issue.expected) {
+        const columns = issue.expected.split(', ');
+        return ok([addUnique(issue.table, columns)]);
+      }
+      return notOk(
+        issueConflict(
+          'indexIncompatible',
+          `Unique constraint on "${issue.table}" differs (expected: ${issue.expected}, actual: ${issue.actual})`,
+          { table: issue.table },
+        ),
+      );
+
+    case 'index_mismatch':
+      if (!issue.table)
+        return notOk(issueConflict('indexIncompatible', 'Index issue has no table name'));
+      if (isMissing(issue) && issue.expected) {
+        const columns = issue.expected.split(', ');
+        return ok([createIndex(issue.table, columns)]);
+      }
+      return notOk(
+        issueConflict(
+          'indexIncompatible',
+          `Index on "${issue.table}" differs (expected: ${issue.expected}, actual: ${issue.actual})`,
+          { table: issue.table },
+        ),
+      );
+
+    case 'foreign_key_mismatch':
+      if (!issue.table)
+        return notOk(issueConflict('foreignKeyConflict', 'Foreign key issue has no table name'));
+      if (isMissing(issue) && issue.expected) {
+        const arrowIdx = issue.expected.indexOf(' -> ');
+        if (arrowIdx >= 0) {
+          const columns = issue.expected.slice(0, arrowIdx).split(', ');
+          return ok([addForeignKey(issue.table, columns)]);
+        }
+      }
+      return notOk(
+        issueConflict(
+          'foreignKeyConflict',
+          `Foreign key on "${issue.table}" differs (expected: ${issue.expected}, actual: ${issue.actual})`,
+          { table: issue.table },
+        ),
+      );
+
+    // Types
+    case 'type_missing': {
+      if (!issue.typeName)
+        return notOk(issueConflict('unsupportedOperation', 'Type missing issue has no typeName'));
+      const typeInstance = ctx.toContract.storage.types?.[issue.typeName];
+      if (!typeInstance) {
+        return notOk(
+          issueConflict(
+            'unsupportedOperation',
+            `Type "${issue.typeName}" reported missing but not found in destination contract`,
+          ),
+        );
+      }
+      // TODO: codec-specific descriptor dispatch should be driven by a registry, not hardcoded prefix checks
+      if (typeInstance.codecId.startsWith('pg/enum')) {
+        return ok([createEnumType(issue.typeName)]);
+      }
+      return notOk(
+        issueConflict(
+          'unsupportedOperation',
+          `Type "${issue.typeName}" uses codec "${typeInstance.codecId}" — only enum types are supported by the descriptor planner`,
+        ),
+      );
+    }
+
+    case 'type_values_mismatch':
+      return notOk(
+        issueConflict(
+          'unsupportedOperation',
+          `Type "${issue.typeName ?? 'unknown'}" values differ — type alteration not yet supported by descriptor planner`,
+        ),
+      );
+
+    // Dependencies
+    case 'dependency_missing':
+      if (!issue.dependencyId)
+        return notOk(
+          issueConflict('unsupportedOperation', 'Dependency missing issue has no dependencyId'),
+        );
+      return ok([createDependency(issue.dependencyId)]);
+    default:
+      return notOk(
+        issueConflict(
+          'unsupportedOperation',
+          `Unhandled issue kind: ${(issue as SchemaIssue).kind}`,
+        ),
+      );
+  }
+}
+
+// ============================================================================
+// Planner entry point
+// ============================================================================
+
+export interface DescriptorPlannerOptions {
+  readonly issues: readonly SchemaIssue[];
+  readonly toContract: Contract<SqlStorage>;
+  readonly fromContract: Contract<SqlStorage> | null;
+  readonly strategies?: readonly MigrationStrategy[];
+}
+
+export interface DescriptorPlannerValue {
+  readonly descriptors: readonly PostgresMigrationOpDescriptor[];
+  readonly needsDataMigration: boolean;
+}
+
+export function planDescriptors(
+  options: DescriptorPlannerOptions,
+): Result<DescriptorPlannerValue, readonly SqlPlannerConflict[]> {
+  const context: StrategyContext = {
+    toContract: options.toContract,
+    fromContract: options.fromContract,
+  };
+
+  const strategies = options.strategies ?? migrationPlanStrategies;
+
+  // Phase 1: Pattern matching — consume recognized issues
+  let remaining = options.issues;
+  const patternOps: PostgresMigrationOpDescriptor[] = [];
+
+  for (const strategy of strategies) {
+    const result = strategy(remaining, context);
+    if (result.kind === 'match') {
+      remaining = result.issues;
+      patternOps.push(...result.ops);
+    }
+  }
+
+  // Phase 2: Sort remaining issues by dependency order
+  const sorted = [...remaining].sort((a, b) => issueOrder(a) - issueOrder(b));
+
+  // Phase 3: Map remaining issues to descriptors, collecting conflicts
+  const defaultOps: PostgresMigrationOpDescriptor[] = [];
+  const conflicts: SqlPlannerConflict[] = [];
+
+  for (const issue of sorted) {
+    const result = mapIssue(issue, context);
+    if (result.ok) {
+      defaultOps.push(...result.value);
+    } else {
+      conflicts.push(result.failure);
+    }
+  }
+
+  if (conflicts.length > 0) {
+    return notOk(conflicts);
+  }
+
+  // Phase 4: Order descriptors by operation kind
+  const depOps = defaultOps.filter(
+    (op) =>
+      op.kind === 'createDependency' ||
+      op.kind === 'createEnumType' ||
+      op.kind === 'addEnumValues' ||
+      op.kind === 'dropEnumType' ||
+      op.kind === 'renameType',
+  );
+  const dropOps = defaultOps.filter(
+    (op) =>
+      op.kind === 'dropTable' ||
+      op.kind === 'dropColumn' ||
+      op.kind === 'dropConstraint' ||
+      op.kind === 'dropIndex' ||
+      op.kind === 'dropDefault',
+  );
+  const tableOps = defaultOps.filter((op) => op.kind === 'createTable');
+  const columnOps = defaultOps.filter((op) => op.kind === 'addColumn');
+  const alterOps = defaultOps.filter(
+    (op) =>
+      op.kind === 'alterColumnType' ||
+      op.kind === 'setNotNull' ||
+      op.kind === 'dropNotNull' ||
+      op.kind === 'setDefault',
+  );
+  const constraintOps = defaultOps.filter(
+    (op) =>
+      op.kind === 'addPrimaryKey' ||
+      op.kind === 'addUnique' ||
+      op.kind === 'createIndex' ||
+      op.kind === 'addForeignKey',
+  );
+
+  const descriptors: PostgresMigrationOpDescriptor[] = [
+    ...depOps,
+    ...dropOps,
+    ...tableOps,
+    ...columnOps,
+    ...patternOps,
+    ...alterOps,
+    ...constraintOps,
+  ];
+
+  return ok({
+    descriptors,
+    needsDataMigration: descriptors.some((op) => op.kind === 'dataTransform'),
+  });
+}

package/src/core/migrations/operation-descriptors.ts

@@ -0,0 +1,166 @@
+/**
+ * Postgres migration operation descriptors.
+ *
+ * Re-exports all structural SQL descriptors from @prisma-next/family-sql
+ * and adds data transform support with typed query builder callbacks.
+ */
+
+import type { Db, TableProxyContract } from '@prisma-next/sql-builder/types';
+import type { SqlQueryPlan } from '@prisma-next/sql-relational-core/plan';
+
+// Re-export structural descriptors from sql-family
+export {
+  type AddColumnDescriptor,
+  type AddEnumValuesDescriptor,
+  type AddForeignKeyDescriptor,
+  type AddPrimaryKeyDescriptor,
+  type AddUniqueDescriptor,
+  type AlterColumnTypeDescriptor,
+  addColumn,
+  addEnumValues,
+  addForeignKey,
+  addPrimaryKey,
+  addUnique,
+  alterColumnType,
+  type Buildable,
+  type CreateDependencyDescriptor,
+  type CreateEnumTypeDescriptor,
+  type CreateIndexDescriptor,
+  type CreateTableDescriptor,
+  createDependency,
+  createEnumType,
+  createIndex,
+  createTable,
+  type DropColumnDescriptor,
+  type DropConstraintDescriptor,
+  type DropDefaultDescriptor,
+  type DropEnumTypeDescriptor,
+  type DropIndexDescriptor,
+  type DropNotNullDescriptor,
+  type DropTableDescriptor,
+  dropColumn,
+  dropConstraint,
+  dropDefault,
+  dropEnumType,
+  dropIndex,
+  dropNotNull,
+  dropTable,
+  type RenameTypeDescriptor,
+  renameType,
+  type SetDefaultDescriptor,
+  type SetNotNullDescriptor,
+  type SqlMigrationOpDescriptor,
+  setDefault,
+  setNotNull,
+  TODO,
+  type TodoMarker,
+} from '@prisma-next/family-sql/operation-descriptors';
+
+import {
+  type Buildable,
+  type DataTransformDescriptor,
+  type SqlMigrationOpDescriptor,
+  builders as structuralBuilders,
+  TODO,
+  type TodoMarker,
+} from '@prisma-next/family-sql/operation-descriptors';
+
+export type { DataTransformDescriptor };
+
+// ============================================================================
+// Typed data transform inputs (for createBuilders<Contract>())
+// ============================================================================
+
+/**
+ * A single query plan input — callback, pre-built plan, or TODO placeholder.
+ * @template TContract - The contract type for the Db client. Defaults to any
+ *   (untyped). Use createBuilders<Contract>() to get typed callbacks.
+ */
+// biome-ignore lint/suspicious/noExplicitAny: default is untyped; createBuilders narrows this
+export type QueryPlanInput<TContract extends TableProxyContract = any> =
+  | ((db: Db<TContract>) => Buildable)
+  | SqlQueryPlan
+  | TodoMarker;
+
+/** Run input — a callback returning one or many buildables, or a pre-built plan/TODO. */
+// biome-ignore lint/suspicious/noExplicitAny: default is untyped; createBuilders narrows this
+export type RunInput<TContract extends TableProxyContract = any> =
+  | ((db: Db<TContract>) => Buildable | readonly Buildable[])
+  | SqlQueryPlan
+  | TodoMarker;
+
+// ============================================================================
+// Postgres descriptor union = SQL structural + data transforms
+// ============================================================================
+
+export type PostgresMigrationOpDescriptor = SqlMigrationOpDescriptor | DataTransformDescriptor;
+
+// ============================================================================
+// Data transform builder
+// ============================================================================
+
+function resolveInput(input: QueryPlanInput): QueryPlanInput {
+  if (typeof input === 'symbol' || typeof input === 'function') return input;
+  if ('build' in input && typeof (input as Buildable).build === 'function') {
+    return (input as Buildable).build();
+  }
+  return input;
+}
+
+// biome-ignore lint/suspicious/noExplicitAny: default is untyped; createBuilders narrows this
+export function dataTransform<TContract extends TableProxyContract = any>(
+  name: string,
+  options: {
+    check: QueryPlanInput<TContract> | Buildable | boolean;
+    run: RunInput<TContract> | Buildable;
+  },
+): DataTransformDescriptor {
+  const check =
+    typeof options.check === 'boolean'
+      ? options.check
+      : resolveInput(options.check as QueryPlanInput);
+
+  const run: (symbol | object | ((...args: never[]) => unknown))[] = [];
+  if (typeof options.run === 'function') {
+    run.push(options.run);
+  } else if (typeof options.run === 'symbol') {
+    run.push(options.run);
+  } else {
+    run.push(resolveInput(options.run as QueryPlanInput));
+  }
+  return {
+    kind: 'dataTransform' as const,
+    name,
+    source: 'migration.ts',
+    check,
+    run,
+  };
+}
+
+/**
+ * Creates typed migration builder functions parameterized by the contract type.
+ * The dataTransform callback receives a fully typed Db<TContract> client.
+ *
+ * Usage:
+ * ```typescript
+ * import type { Contract } from "../../src/prisma/contract.d"
+ * import { createBuilders } from "@prisma-next/target-postgres/migration-builders"
+ *
+ * const { addColumn, dataTransform, setNotNull } = createBuilders<Contract>()
+ * ```
+ */
+export function createBuilders<TContract extends TableProxyContract>() {
+  return {
+    ...structuralBuilders,
+    dataTransform: dataTransform<TContract>,
+    TODO,
+  };
+}
+
+/**
+ * All builder functions keyed by descriptor kind.
+ */
+export const builders = {
+  ...structuralBuilders,
+  dataTransform,
+} as const;