@prisma-next/target-mongo 0.4.0-dev.8 → 0.4.1

package/src/core/mongo-runner.ts

@@ -1,4 +1,5 @@
 import type { ContractMarkerRecord } from '@prisma-next/contract/types';
+import { errorRunnerFailed } from '@prisma-next/errors/execution';
 import type { TargetBoundComponentDescriptor } from '@prisma-next/framework-components/components';
 import type {
   MigrationOperationPolicy,
@@ -8,13 +9,29 @@ import type {
   MigrationRunnerFailure,
   MigrationRunnerResult,
 } from '@prisma-next/framework-components/control';
+import type { MongoAdapter, MongoDriver } from '@prisma-next/mongo-lowering';
 import type {
+  AnyMongoMigrationOperation,
+  MongoDataTransformCheck,
+  MongoDataTransformOperation,
   MongoDdlCommandVisitor,
   MongoInspectionCommandVisitor,
   MongoMigrationCheck,
   MongoMigrationPlanOperation,
 } from '@prisma-next/mongo-query-ast/control';
 import { notOk, ok } from '@prisma-next/utils/result';
+
+const READ_ONLY_CHECK_COMMAND_KINDS: ReadonlySet<string> = new Set(['aggregate', 'rawAggregate']);
+
+function hasProfileHash(value: unknown): value is { readonly profileHash: string } {
+  return (
+    typeof value === 'object' &&
+    value !== null &&
+    Object.hasOwn(value, 'profileHash') &&
+    typeof (value as { profileHash: unknown }).profileHash === 'string'
+  );
+}
+
 import { FilterEvaluator } from './filter-evaluator';
 import { deserializeMongoOps } from './mongo-ops-serializer';
 
@@ -38,6 +55,8 @@ export interface MarkerOperations {
 export interface MongoRunnerDependencies {
   readonly commandExecutor: MongoDdlCommandVisitor<Promise<void>>;
   readonly inspectionExecutor: MongoInspectionCommandVisitor<Promise<Record<string, unknown>[]>>;
+  readonly adapter: MongoAdapter;
+  readonly driver: MongoDriver;
   readonly markerOps: MarkerOperations;
 }
 
@@ -67,7 +86,7 @@ export class MongoMigrationRunner {
     readonly executionChecks?: MigrationRunnerExecutionChecks;
     readonly frameworkComponents: ReadonlyArray<TargetBoundComponentDescriptor<'mongo', 'mongo'>>;
   }): Promise<MigrationRunnerResult> {
-    const { commandExecutor, inspectionExecutor, markerOps } = this.deps;
+    const { commandExecutor, inspectionExecutor, adapter, driver, markerOps } = this.deps;
     const operations = deserializeMongoOps(options.plan.operations as readonly unknown[]);
 
     const policyCheck = this.enforcePolicyCompatibility(options.policy, operations);
@@ -90,9 +109,26 @@ export class MongoMigrationRunner {
     for (const operation of operations) {
       options.callbacks?.onOperationStart?.(operation);
       try {
+        if (operation.operationClass === 'data') {
+          const result = await this.executeDataTransform(
+            operation as MongoDataTransformOperation,
+            adapter,
+            driver,
+            filterEvaluator,
+            runIdempotency,
+            runPrechecks,
+            runPostchecks,
+          );
+          if (result.failure) return result.failure;
+          if (result.executed) operationsExecuted += 1;
+          continue;
+        }
+
+        const ddlOp = operation as MongoMigrationPlanOperation;
+
         if (runPostchecks && runIdempotency) {
           const allSatisfied = await this.allChecksSatisfied(
-            operation.postcheck,
+            ddlOp.postcheck,
             inspectionExecutor,
             filterEvaluator,
           );
@@ -101,7 +137,7 @@ export class MongoMigrationRunner {
 
         if (runPrechecks) {
           const precheckResult = await this.evaluateChecks(
-            operation.precheck,
+            ddlOp.precheck,
             inspectionExecutor,
             filterEvaluator,
           );
@@ -114,13 +150,13 @@ export class MongoMigrationRunner {
           }
         }
 
-        for (const step of operation.execute) {
+        for (const step of ddlOp.execute) {
           await step.command.accept(commandExecutor);
         }
 
         if (runPostchecks) {
           const postcheckResult = await this.evaluateChecks(
-            operation.postcheck,
+            ddlOp.postcheck,
             inspectionExecutor,
             filterEvaluator,
           );
@@ -140,8 +176,9 @@ export class MongoMigrationRunner {
     }
 
     const destination = options.plan.destination;
-    const contract = options.destinationContract as { profileHash?: string };
-    const profileHash = contract.profileHash ?? destination.storageHash;
+    const profileHash = hasProfileHash(options.destinationContract)
+      ? options.destinationContract.profileHash
+      : destination.storageHash;
 
     if (
       operationsExecuted === 0 &&
@@ -185,6 +222,105 @@ export class MongoMigrationRunner {
     return ok({ operationsPlanned: operations.length, operationsExecuted });
   }
 
+  private async executeDataTransform(
+    op: MongoDataTransformOperation,
+    adapter: MongoAdapter,
+    driver: MongoDriver,
+    filterEvaluator: FilterEvaluator,
+    runIdempotency: boolean,
+    runPrechecks: boolean,
+    runPostchecks: boolean,
+  ): Promise<{ executed: boolean; failure?: MigrationRunnerResult }> {
+    if (runPostchecks && runIdempotency && op.postcheck.length > 0) {
+      const allSatisfied = await this.evaluateDataTransformChecks(
+        op.postcheck,
+        adapter,
+        driver,
+        filterEvaluator,
+      );
+      if (allSatisfied) return { executed: false };
+    }
+
+    if (runPrechecks && op.precheck.length > 0) {
+      const passed = await this.evaluateDataTransformChecks(
+        op.precheck,
+        adapter,
+        driver,
+        filterEvaluator,
+      );
+      if (!passed) {
+        return {
+          executed: false,
+          failure: runnerFailure('PRECHECK_FAILED', `Operation ${op.id} failed during precheck`, {
+            meta: { operationId: op.id, name: op.name },
+          }),
+        };
+      }
+    }
+
+    for (const plan of op.run) {
+      const wireCommand = adapter.lower(plan);
+      for await (const _ of driver.execute(wireCommand)) {
+        /* consume */
+      }
+    }
+
+    if (runPostchecks && op.postcheck.length > 0) {
+      const passed = await this.evaluateDataTransformChecks(
+        op.postcheck,
+        adapter,
+        driver,
+        filterEvaluator,
+      );
+      if (!passed) {
+        return {
+          executed: false,
+          failure: runnerFailure('POSTCHECK_FAILED', `Operation ${op.id} failed during postcheck`, {
+            meta: { operationId: op.id, name: op.name },
+          }),
+        };
+      }
+    }
+
+    return { executed: true };
+  }
+
+  private async evaluateDataTransformChecks(
+    checks: readonly MongoDataTransformCheck[],
+    adapter: MongoAdapter,
+    driver: MongoDriver,
+    filterEvaluator: FilterEvaluator,
+  ): Promise<boolean> {
+    for (const check of checks) {
+      const commandKind = check.source.command.kind;
+      if (!READ_ONLY_CHECK_COMMAND_KINDS.has(commandKind)) {
+        throw errorRunnerFailed(
+          `Data-transform check rejected: command kind "${commandKind}" is not read-only`,
+          {
+            why: 'Data-transform checks must use aggregate or rawAggregate commands so the pre/postcheck path cannot mutate the database.',
+            fix: 'Author the check.source as an aggregate pipeline (or rawAggregate) rather than a DML write command.',
+            meta: {
+              checkDescription: check.description,
+              commandKind,
+              collection: check.source.collection,
+            },
+          },
+        );
+      }
+      const wireCommand = adapter.lower(check.source);
+      let matchFound = false;
+      for await (const row of driver.execute<Record<string, unknown>>(wireCommand)) {
+        if (filterEvaluator.evaluate(check.filter, row)) {
+          matchFound = true;
+          break;
+        }
+      }
+      const passed = check.expect === 'exists' ? matchFound : !matchFound;
+      if (!passed) return false;
+    }
+    return true;
+  }
+
   private async evaluateChecks(
     checks: readonly MongoMigrationCheck[],
     inspectionExecutor: MongoInspectionCommandVisitor<Promise<Record<string, unknown>[]>>,
@@ -212,7 +348,7 @@ export class MongoMigrationRunner {
 
   private enforcePolicyCompatibility(
     policy: MigrationOperationPolicy,
-    operations: readonly MongoMigrationPlanOperation[],
+    operations: readonly AnyMongoMigrationOperation[],
   ): MigrationRunnerResult | undefined {
     const allowedClasses = new Set(policy.allowedOperationClasses);
     for (const operation of operations) {

package/src/core/op-factory-call.ts

@@ -1,9 +1,32 @@
-import type { MigrationOperationClass } from '@prisma-next/framework-components/control';
+/**
+ * Mongo migration IR: one concrete `*Call` class per pure factory under
+ * `migration-factories.ts`, plus a shared `OpFactoryCallNode` abstract
+ * base. Every call class carries the literal arguments its backing
+ * factory would receive, computes a human-readable `label` in its
+ * constructor, and implements two polymorphic hooks:
+ *
+ * - `toOp()` — converts the IR node to a runtime
+ *   `MongoMigrationPlanOperation` by delegating to the matching pure
+ *   factory in `migration-factories.ts`.
+ * - `renderTypeScript()` / `importRequirements()` — inherited from
+ *   `TsExpression`. Used by `renderCallsToTypeScript` to emit the call
+ *   as a TypeScript expression inside the scaffolded `migration.ts`.
+ *
+ * The abstract base and all concrete classes are package-private.
+ * External consumers see only the framework-level `OpFactoryCall`
+ * interface and the `OpFactoryCall` union.
+ */
+
+import type {
+  OpFactoryCall as FrameworkOpFactoryCall,
+  MigrationOperationClass,
+} from '@prisma-next/framework-components/control';
 import type {
   CollModOptions,
   CreateCollectionOptions,
   CreateIndexOptions,
   MongoIndexKey,
+  MongoMigrationPlanOperation,
 } from '@prisma-next/mongo-query-ast/control';
 import type {
   MongoSchemaCollection,
@@ -11,6 +34,14 @@ import type {
   MongoSchemaIndex,
   MongoSchemaValidator,
 } from '@prisma-next/mongo-schema-ir';
+import { type ImportRequirement, jsonToTsSource, TsExpression } from '@prisma-next/ts-render';
+import {
+  collMod,
+  createCollection,
+  createIndex,
+  dropCollection,
+  dropIndex,
+} from './migration-factories';
 
 export interface CollModMeta {
   readonly id?: string;
@@ -18,19 +49,17 @@ export interface CollModMeta {
   readonly operationClass?: MigrationOperationClass;
 }
 
-export interface OpFactoryCallVisitor<R> {
-  createIndex(call: CreateIndexCall): R;
-  dropIndex(call: DropIndexCall): R;
-  createCollection(call: CreateCollectionCall): R;
-  dropCollection(call: DropCollectionCall): R;
-  collMod(call: CollModCall): R;
-}
+const TARGET_MIGRATION_MODULE = '@prisma-next/target-mongo/migration';
 
-abstract class OpFactoryCallNode {
-  abstract readonly factory: string;
+abstract class OpFactoryCallNode extends TsExpression implements FrameworkOpFactoryCall {
+  abstract readonly factoryName: string;
   abstract readonly operationClass: MigrationOperationClass;
   abstract readonly label: string;
-  abstract accept<R>(visitor: OpFactoryCallVisitor<R>): R;
+  abstract toOp(): MongoMigrationPlanOperation;
+
+  importRequirements(): readonly ImportRequirement[] {
+    return [{ moduleSpecifier: TARGET_MIGRATION_MODULE, symbol: this.factoryName }];
+  }
 
   protected freeze(): void {
     Object.freeze(this);
@@ -42,7 +71,7 @@ function formatKeys(keys: ReadonlyArray<MongoIndexKey>): string {
 }
 
 export class CreateIndexCall extends OpFactoryCallNode {
-  readonly factory = 'createIndex' as const;
+  readonly factoryName = 'createIndex' as const;
   readonly operationClass = 'additive' as const;
   readonly collection: string;
   readonly keys: ReadonlyArray<MongoIndexKey>;
@@ -62,13 +91,19 @@ export class CreateIndexCall extends OpFactoryCallNode {
     this.freeze();
   }
 
-  accept<R>(visitor: OpFactoryCallVisitor<R>): R {
-    return visitor.createIndex(this);
+  toOp(): MongoMigrationPlanOperation {
+    return createIndex(this.collection, this.keys, this.options);
+  }
+
+  renderTypeScript(): string {
+    return this.options
+      ? `createIndex(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.keys)}, ${jsonToTsSource(this.options)})`
+      : `createIndex(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.keys)})`;
   }
 }
 
 export class DropIndexCall extends OpFactoryCallNode {
-  readonly factory = 'dropIndex' as const;
+  readonly factoryName = 'dropIndex' as const;
   readonly operationClass = 'destructive' as const;
   readonly collection: string;
   readonly keys: ReadonlyArray<MongoIndexKey>;
@@ -82,13 +117,17 @@ export class DropIndexCall extends OpFactoryCallNode {
     this.freeze();
   }
 
-  accept<R>(visitor: OpFactoryCallVisitor<R>): R {
-    return visitor.dropIndex(this);
+  toOp(): MongoMigrationPlanOperation {
+    return dropIndex(this.collection, this.keys);
+  }
+
+  renderTypeScript(): string {
+    return `dropIndex(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.keys)})`;
   }
 }
 
 export class CreateCollectionCall extends OpFactoryCallNode {
-  readonly factory = 'createCollection' as const;
+  readonly factoryName = 'createCollection' as const;
   readonly operationClass = 'additive' as const;
   readonly collection: string;
   readonly options: CreateCollectionOptions | undefined;
@@ -102,13 +141,19 @@ export class CreateCollectionCall extends OpFactoryCallNode {
     this.freeze();
   }
 
-  accept<R>(visitor: OpFactoryCallVisitor<R>): R {
-    return visitor.createCollection(this);
+  toOp(): MongoMigrationPlanOperation {
+    return createCollection(this.collection, this.options);
+  }
+
+  renderTypeScript(): string {
+    return this.options
+      ? `createCollection(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.options)})`
+      : `createCollection(${jsonToTsSource(this.collection)})`;
   }
 }
 
 export class DropCollectionCall extends OpFactoryCallNode {
-  readonly factory = 'dropCollection' as const;
+  readonly factoryName = 'dropCollection' as const;
   readonly operationClass = 'destructive' as const;
   readonly collection: string;
   readonly label: string;
@@ -120,13 +165,17 @@ export class DropCollectionCall extends OpFactoryCallNode {
     this.freeze();
   }
 
-  accept<R>(visitor: OpFactoryCallVisitor<R>): R {
-    return visitor.dropCollection(this);
+  toOp(): MongoMigrationPlanOperation {
+    return dropCollection(this.collection);
+  }
+
+  renderTypeScript(): string {
+    return `dropCollection(${jsonToTsSource(this.collection)})`;
   }
 }
 
 export class CollModCall extends OpFactoryCallNode {
-  readonly factory = 'collMod' as const;
+  readonly factoryName = 'collMod' as const;
   readonly collection: string;
   readonly options: CollModOptions;
   readonly meta: CollModMeta | undefined;
@@ -143,8 +192,14 @@ export class CollModCall extends OpFactoryCallNode {
     this.freeze();
   }
 
-  accept<R>(visitor: OpFactoryCallVisitor<R>): R {
-    return visitor.collMod(this);
+  toOp(): MongoMigrationPlanOperation {
+    return collMod(this.collection, this.options, this.meta);
+  }
+
+  renderTypeScript(): string {
+    return this.meta
+      ? `collMod(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.options)}, ${jsonToTsSource(this.meta)})`
+      : `collMod(${jsonToTsSource(this.collection)}, ${jsonToTsSource(this.options)})`;
   }
 }
 

package/src/core/render-ops.ts

@@ -1,39 +1,6 @@
 import type { MongoMigrationPlanOperation } from '@prisma-next/mongo-query-ast/control';
-import {
-  collMod,
-  createCollection,
-  createIndex,
-  dropCollection,
-  dropIndex,
-} from './migration-factories';
-import type {
-  CollModCall,
-  CreateCollectionCall,
-  CreateIndexCall,
-  DropCollectionCall,
-  DropIndexCall,
-  OpFactoryCall,
-  OpFactoryCallVisitor,
-} from './op-factory-call';
-
-const renderVisitor: OpFactoryCallVisitor<MongoMigrationPlanOperation> = {
-  createIndex(call: CreateIndexCall) {
-    return createIndex(call.collection, call.keys, call.options);
-  },
-  dropIndex(call: DropIndexCall) {
-    return dropIndex(call.collection, call.keys);
-  },
-  createCollection(call: CreateCollectionCall) {
-    return createCollection(call.collection, call.options);
-  },
-  dropCollection(call: DropCollectionCall) {
-    return dropCollection(call.collection);
-  },
-  collMod(call: CollModCall) {
-    return collMod(call.collection, call.options, call.meta);
-  },
-};
+import type { OpFactoryCall } from './op-factory-call';
 
 export function renderOps(calls: ReadonlyArray<OpFactoryCall>): MongoMigrationPlanOperation[] {
-  return calls.map((call) => call.accept(renderVisitor));
+  return calls.map((call) => call.toOp());
 }

package/src/core/render-typescript.ts

@@ -1,13 +1,6 @@
 import { detectScaffoldRuntime, shebangLineFor } from '@prisma-next/migration-tools/migration-ts';
-import type {
-  CollModCall,
-  CreateCollectionCall,
-  CreateIndexCall,
-  DropCollectionCall,
-  DropIndexCall,
-  OpFactoryCall,
-  OpFactoryCallVisitor,
-} from './op-factory-call';
+import { type ImportRequirement, jsonToTsSource, renderImports } from '@prisma-next/ts-render';
+import type { OpFactoryCall } from './op-factory-call';
 
 export interface RenderMigrationMeta {
   readonly from: string;
@@ -16,22 +9,34 @@ export interface RenderMigrationMeta {
   readonly labels?: readonly string[];
 }
 
+const BASE_IMPORT: ImportRequirement = {
+  moduleSpecifier: '@prisma-next/family-mongo/migration',
+  symbol: 'Migration',
+};
+
 /**
- * Render a list of Mongo `OpFactoryCall`s as a class-flow `migration.ts`
+ * Render a list of Mongo `OpFactoryCall`s as a `migration.ts`
  * source string. The result is shebanged, extends the user-facing
  * `Migration` (i.e. `MongoMigration`) from `@prisma-next/family-mongo`, and
  * implements the abstract `operations` and `describe` members. `meta` is
  * always rendered — `describe()` is part of the `Migration` contract, so
  * even an empty stub must satisfy it; callers pass empty strings for a
  * migration-new scaffold.
+ *
+ * The walk is polymorphic: each call node contributes its own
+ * `renderTypeScript()` expression and declares its own
+ * `importRequirements()`. The top-level renderer aggregates imports
+ * across all nodes and emits one `import { … } from "…"` line per module.
+ * The `Migration` import from `@prisma-next/family-mongo/migration` is
+ * always emitted — it's driven by `meta` (the rendered scaffold always
+ * extends `Migration`), not by any node.
  */
 export function renderCallsToTypeScript(
   calls: ReadonlyArray<OpFactoryCall>,
   meta: RenderMigrationMeta,
 ): string {
-  const factoryNames = collectFactoryNames(calls);
-  const imports = buildImports(factoryNames);
-  const operationsBody = calls.map((c) => c.accept(renderCallVisitor)).join(',\n');
+  const imports = buildImports(calls);
+  const operationsBody = calls.map((c) => c.renderTypeScript()).join(',\n');
 
   return [
     shebangLineFor(detectScaffoldRuntime()),
@@ -52,20 +57,14 @@ export function renderCallsToTypeScript(
   ].join('\n');
 }
 
-function collectFactoryNames(calls: ReadonlyArray<OpFactoryCall>): string[] {
-  const names = new Set<string>();
+function buildImports(calls: ReadonlyArray<OpFactoryCall>): string {
+  const requirements: ImportRequirement[] = [BASE_IMPORT];
   for (const call of calls) {
-    names.add(call.factory);
+    for (const req of call.importRequirements()) {
+      requirements.push(req);
+    }
   }
-  return [...names].sort();
-}
-
-function buildImports(factoryNames: string[]): string {
-  const lines = ["import { Migration } from '@prisma-next/family-mongo/migration';"];
-  if (factoryNames.length > 0) {
-    lines.push(`import { ${factoryNames.join(', ')} } from '@prisma-next/target-mongo/migration';`);
-  }
-  return lines.join('\n');
+  return renderImports(requirements);
 }
 
 function buildDescribeMethod(meta: RenderMigrationMeta): string {
@@ -78,7 +77,7 @@ function buildDescribeMethod(meta: RenderMigrationMeta): string {
     lines.push(`      kind: ${JSON.stringify(meta.kind)},`);
   }
   if (meta.labels && meta.labels.length > 0) {
-    lines.push(`      labels: ${renderLiteral(meta.labels)},`);
+    lines.push(`      labels: ${jsonToTsSource(meta.labels)},`);
   }
   lines.push('    };');
   lines.push('  }');
@@ -86,58 +85,6 @@ function buildDescribeMethod(meta: RenderMigrationMeta): string {
   return lines.join('\n');
 }
 
-const renderCallVisitor: OpFactoryCallVisitor<string> = {
-  createIndex(call: CreateIndexCall) {
-    return call.options
-      ? `createIndex(${renderLiteral(call.collection)}, ${renderLiteral(call.keys)}, ${renderLiteral(call.options)})`
-      : `createIndex(${renderLiteral(call.collection)}, ${renderLiteral(call.keys)})`;
-  },
-  dropIndex(call: DropIndexCall) {
-    return `dropIndex(${renderLiteral(call.collection)}, ${renderLiteral(call.keys)})`;
-  },
-  createCollection(call: CreateCollectionCall) {
-    return call.options
-      ? `createCollection(${renderLiteral(call.collection)}, ${renderLiteral(call.options)})`
-      : `createCollection(${renderLiteral(call.collection)})`;
-  },
-  dropCollection(call: DropCollectionCall) {
-    return `dropCollection(${renderLiteral(call.collection)})`;
-  },
-  collMod(call: CollModCall) {
-    return call.meta
-      ? `collMod(${renderLiteral(call.collection)}, ${renderLiteral(call.options)}, ${renderLiteral(call.meta)})`
-      : `collMod(${renderLiteral(call.collection)}, ${renderLiteral(call.options)})`;
-  },
-};
-
-function renderLiteral(value: unknown): string {
-  if (value === undefined) return 'undefined';
-  if (value === null) return 'null';
-  if (typeof value === 'string') return JSON.stringify(value);
-  if (typeof value === 'number' || typeof value === 'boolean') return String(value);
-  if (Array.isArray(value)) {
-    if (value.length === 0) return '[]';
-    const items = value.map((v) => renderLiteral(v));
-    const singleLine = `[${items.join(', ')}]`;
-    if (singleLine.length <= 80) return singleLine;
-    return `[\n${items.map((i) => `  ${i}`).join(',\n')},\n]`;
-  }
-  if (typeof value === 'object') {
-    const entries = Object.entries(value).filter(([, v]) => v !== undefined);
-    if (entries.length === 0) return '{}';
-    const items = entries.map(([k, v]) => `${renderKey(k)}: ${renderLiteral(v)}`);
-    const singleLine = `{ ${items.join(', ')} }`;
-    if (singleLine.length <= 80) return singleLine;
-    return `{\n${items.map((i) => `  ${i}`).join(',\n')},\n}`;
-  }
-  return String(value);
-}
-
-function renderKey(key: string): string {
-  if (key === '__proto__') return JSON.stringify(key);
-  return /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(key) ? key : JSON.stringify(key);
-}
-
 function indent(text: string, spaces: number): string {
   const pad = ' '.repeat(spaces);
   return text

package/src/exports/control.ts

@@ -11,7 +11,7 @@ export type { PlanCallsResult } from '../core/mongo-planner';
 export { MongoMigrationPlanner } from '../core/mongo-planner';
 export type { MarkerOperations, MongoRunnerDependencies } from '../core/mongo-runner';
 export { MongoMigrationRunner } from '../core/mongo-runner';
-export type { CollModMeta, OpFactoryCall, OpFactoryCallVisitor } from '../core/op-factory-call';
+export type { CollModMeta, OpFactoryCall } from '../core/op-factory-call';
 export {
   CollModCall,
   CreateCollectionCall,

package/src/exports/migration.ts

@@ -3,6 +3,7 @@ export {
   collMod,
   createCollection,
   createIndex,
+  dataTransform,
   dropCollection,
   dropIndex,
   setValidation,

package/dist/migration-factories-Brzz-QGG.mjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"migration-factories-Brzz-QGG.mjs","names":[],"sources":["../src/core/migration-factories.ts"],"sourcesContent":["import type { MongoIndexKey } from '@prisma-next/mongo-query-ast/control';\nimport {\n  buildIndexOpId,\n  CollModCommand,\n  type CollModOptions,\n  CreateCollectionCommand,\n  type CreateCollectionOptions,\n  CreateIndexCommand,\n  type CreateIndexOptions,\n  DropCollectionCommand,\n  DropIndexCommand,\n  defaultMongoIndexName,\n  keysToKeySpec,\n  ListCollectionsCommand,\n  ListIndexesCommand,\n  MongoAndExpr,\n  MongoFieldFilter,\n  type MongoMigrationPlanOperation,\n} from '@prisma-next/mongo-query-ast/control';\nimport type { CollModMeta } from './op-factory-call';\n\nfunction formatKeys(keys: ReadonlyArray<MongoIndexKey>): string {\n  return keys.map((k) => `${k.field}:${k.direction}`).join(', ');\n}\n\nfunction isTextIndex(keys: ReadonlyArray<MongoIndexKey>): boolean {\n  return keys.some((k) => k.direction === 'text');\n}\n\nfunction keyFilter(keys: ReadonlyArray<MongoIndexKey>) {\n  return isTextIndex(keys)\n    ? MongoFieldFilter.eq('key._fts', 'text')\n    : MongoFieldFilter.eq('key', keysToKeySpec(keys));\n}\n\nexport function createIndex(\n  collection: string,\n  keys: ReadonlyArray<MongoIndexKey>,\n  options?: CreateIndexOptions,\n): MongoMigrationPlanOperation {\n  const name = defaultMongoIndexName(keys);\n  const filter = keyFilter(keys);\n  const fullFilter = options?.unique\n    ? MongoAndExpr.of([filter, MongoFieldFilter.eq('unique', true)])\n    : filter;\n\n  return {\n    id: buildIndexOpId('create', collection, keys),\n    label: `Create index on ${collection} (${formatKeys(keys)})`,\n    operationClass: 'additive',\n    precheck: [\n      {\n        description: `index does not already exist on ${collection}`,\n        source: new ListIndexesCommand(collection),\n        filter,\n        expect: 'notExists',\n      },\n    ],\n    execute: [\n      {\n        description: `create index on ${collection}`,\n        command: new CreateIndexCommand(collection, keys, {\n          ...options,\n          unique: options?.unique ?? undefined,\n          name,\n        }),\n      },\n    ],\n    postcheck: [\n      {\n        description: `index exists on ${collection}`,\n        source: new ListIndexesCommand(collection),\n        filter: fullFilter,\n        expect: 'exists',\n      },\n    ],\n  };\n}\n\nexport function dropIndex(\n  collection: string,\n  keys: ReadonlyArray<MongoIndexKey>,\n): MongoMigrationPlanOperation {\n  const indexName = defaultMongoIndexName(keys);\n  const filter = keyFilter(keys);\n\n  return {\n    id: buildIndexOpId('drop', collection, keys),\n    label: `Drop index on ${collection} (${formatKeys(keys)})`,\n    operationClass: 'destructive',\n    precheck: [\n      {\n        description: `index exists on ${collection}`,\n        source: new ListIndexesCommand(collection),\n        filter,\n        expect: 'exists',\n      },\n    ],\n    execute: [\n      {\n        description: `drop index on ${collection}`,\n        command: new DropIndexCommand(collection, indexName),\n      },\n    ],\n    postcheck: [\n      {\n        description: `index no longer exists on ${collection}`,\n        source: new ListIndexesCommand(collection),\n        filter,\n        expect: 'notExists',\n      },\n    ],\n  };\n}\n\nexport function createCollection(\n  collection: string,\n  options?: CreateCollectionOptions,\n): MongoMigrationPlanOperation {\n  return {\n    id: `collection.${collection}.create`,\n    label: `Create collection ${collection}`,\n    operationClass: 'additive',\n    precheck: [\n      {\n        description: `collection ${collection} does not exist`,\n        source: new ListCollectionsCommand(),\n        filter: MongoFieldFilter.eq('name', collection),\n        expect: 'notExists',\n      },\n    ],\n    execute: [\n      {\n        description: `create collection ${collection}`,\n        command: new CreateCollectionCommand(collection, options),\n      },\n    ],\n    postcheck: [],\n  };\n}\n\nexport function dropCollection(collection: string): MongoMigrationPlanOperation {\n  return {\n    id: `collection.${collection}.drop`,\n    label: `Drop collection ${collection}`,\n    operationClass: 'destructive',\n    precheck: [],\n    execute: [\n      {\n        description: `drop collection ${collection}`,\n        command: new DropCollectionCommand(collection),\n      },\n    ],\n    postcheck: [],\n  };\n}\n\nexport function setValidation(\n  collection: string,\n  schema: Record<string, unknown>,\n  options?: { validationLevel?: 'strict' | 'moderate'; validationAction?: 'error' | 'warn' },\n): MongoMigrationPlanOperation {\n  return {\n    id: `collection.${collection}.setValidation`,\n    label: `Set validation on ${collection}`,\n    operationClass: 'destructive',\n    precheck: [],\n    execute: [\n      {\n        description: `set validation on ${collection}`,\n        command: new CollModCommand(collection, {\n          validator: { $jsonSchema: schema },\n          validationLevel: options?.validationLevel,\n          validationAction: options?.validationAction,\n        }),\n      },\n    ],\n    postcheck: [],\n  };\n}\n\nexport function collMod(\n  collection: string,\n  options: CollModOptions,\n  meta?: CollModMeta,\n): MongoMigrationPlanOperation {\n  const hasValidator = options.validator != null && Object.keys(options.validator).length > 0;\n\n  return {\n    id: meta?.id ?? `collection.${collection}.collMod`,\n    label: meta?.label ?? `Modify collection ${collection}`,\n    operationClass: meta?.operationClass ?? 'destructive',\n    precheck:\n      options.validator != null\n        ? [\n            {\n              description: `collection ${collection} exists`,\n              source: new ListCollectionsCommand(),\n              filter: MongoFieldFilter.eq('name', collection),\n              expect: 'exists' as const,\n            },\n          ]\n        : [],\n    execute: [\n      {\n        description: `modify ${collection}`,\n        command: new CollModCommand(collection, options),\n      },\n    ],\n    postcheck: hasValidator\n      ? [\n          {\n            description: `validator applied on ${collection}`,\n            source: new ListCollectionsCommand(),\n            filter: MongoAndExpr.of([\n              MongoFieldFilter.eq('name', collection),\n              ...(options.validationLevel\n                ? [MongoFieldFilter.eq('options.validationLevel', options.validationLevel)]\n                : []),\n              ...(options.validationAction\n                ? [MongoFieldFilter.eq('options.validationAction', options.validationAction)]\n                : []),\n            ]),\n            expect: 'exists' as const,\n          },\n        ]\n      : [],\n  };\n}\n\nexport function validatedCollection(\n  name: string,\n  schema: Record<string, unknown>,\n  indexes: ReadonlyArray<{ keys: MongoIndexKey[]; unique?: boolean }>,\n): MongoMigrationPlanOperation[] {\n  return [\n    createCollection(name, {\n      validator: { $jsonSchema: schema },\n      validationLevel: 'strict',\n      validationAction: 'error',\n    }),\n    ...indexes.map((idx) => createIndex(name, idx.keys, { unique: idx.unique })),\n  ];\n}\n"],"mappings":";;;AAqBA,SAAS,WAAW,MAA4C;AAC9D,QAAO,KAAK,KAAK,MAAM,GAAG,EAAE,MAAM,GAAG,EAAE,YAAY,CAAC,KAAK,KAAK;;AAGhE,SAAS,YAAY,MAA6C;AAChE,QAAO,KAAK,MAAM,MAAM,EAAE,cAAc,OAAO;;AAGjD,SAAS,UAAU,MAAoC;AACrD,QAAO,YAAY,KAAK,GACpB,iBAAiB,GAAG,YAAY,OAAO,GACvC,iBAAiB,GAAG,OAAO,cAAc,KAAK,CAAC;;AAGrD,SAAgB,YACd,YACA,MACA,SAC6B;CAC7B,MAAM,OAAO,sBAAsB,KAAK;CACxC,MAAM,SAAS,UAAU,KAAK;CAC9B,MAAM,aAAa,SAAS,SACxB,aAAa,GAAG,CAAC,QAAQ,iBAAiB,GAAG,UAAU,KAAK,CAAC,CAAC,GAC9D;AAEJ,QAAO;EACL,IAAI,eAAe,UAAU,YAAY,KAAK;EAC9C,OAAO,mBAAmB,WAAW,IAAI,WAAW,KAAK,CAAC;EAC1D,gBAAgB;EAChB,UAAU,CACR;GACE,aAAa,mCAAmC;GAChD,QAAQ,IAAI,mBAAmB,WAAW;GAC1C;GACA,QAAQ;GACT,CACF;EACD,SAAS,CACP;GACE,aAAa,mBAAmB;GAChC,SAAS,IAAI,mBAAmB,YAAY,MAAM;IAChD,GAAG;IACH,QAAQ,SAAS,UAAU;IAC3B;IACD,CAAC;GACH,CACF;EACD,WAAW,CACT;GACE,aAAa,mBAAmB;GAChC,QAAQ,IAAI,mBAAmB,WAAW;GAC1C,QAAQ;GACR,QAAQ;GACT,CACF;EACF;;AAGH,SAAgB,UACd,YACA,MAC6B;CAC7B,MAAM,YAAY,sBAAsB,KAAK;CAC7C,MAAM,SAAS,UAAU,KAAK;AAE9B,QAAO;EACL,IAAI,eAAe,QAAQ,YAAY,KAAK;EAC5C,OAAO,iBAAiB,WAAW,IAAI,WAAW,KAAK,CAAC;EACxD,gBAAgB;EAChB,UAAU,CACR;GACE,aAAa,mBAAmB;GAChC,QAAQ,IAAI,mBAAmB,WAAW;GAC1C;GACA,QAAQ;GACT,CACF;EACD,SAAS,CACP;GACE,aAAa,iBAAiB;GAC9B,SAAS,IAAI,iBAAiB,YAAY,UAAU;GACrD,CACF;EACD,WAAW,CACT;GACE,aAAa,6BAA6B;GAC1C,QAAQ,IAAI,mBAAmB,WAAW;GAC1C;GACA,QAAQ;GACT,CACF;EACF;;AAGH,SAAgB,iBACd,YACA,SAC6B;AAC7B,QAAO;EACL,IAAI,cAAc,WAAW;EAC7B,OAAO,qBAAqB;EAC5B,gBAAgB;EAChB,UAAU,CACR;GACE,aAAa,cAAc,WAAW;GACtC,QAAQ,IAAI,wBAAwB;GACpC,QAAQ,iBAAiB,GAAG,QAAQ,WAAW;GAC/C,QAAQ;GACT,CACF;EACD,SAAS,CACP;GACE,aAAa,qBAAqB;GAClC,SAAS,IAAI,wBAAwB,YAAY,QAAQ;GAC1D,CACF;EACD,WAAW,EAAE;EACd;;AAGH,SAAgB,eAAe,YAAiD;AAC9E,QAAO;EACL,IAAI,cAAc,WAAW;EAC7B,OAAO,mBAAmB;EAC1B,gBAAgB;EAChB,UAAU,EAAE;EACZ,SAAS,CACP;GACE,aAAa,mBAAmB;GAChC,SAAS,IAAI,sBAAsB,WAAW;GAC/C,CACF;EACD,WAAW,EAAE;EACd;;AAGH,SAAgB,cACd,YACA,QACA,SAC6B;AAC7B,QAAO;EACL,IAAI,cAAc,WAAW;EAC7B,OAAO,qBAAqB;EAC5B,gBAAgB;EAChB,UAAU,EAAE;EACZ,SAAS,CACP;GACE,aAAa,qBAAqB;GAClC,SAAS,IAAI,eAAe,YAAY;IACtC,WAAW,EAAE,aAAa,QAAQ;IAClC,iBAAiB,SAAS;IAC1B,kBAAkB,SAAS;IAC5B,CAAC;GACH,CACF;EACD,WAAW,EAAE;EACd;;AAGH,SAAgB,QACd,YACA,SACA,MAC6B;CAC7B,MAAM,eAAe,QAAQ,aAAa,QAAQ,OAAO,KAAK,QAAQ,UAAU,CAAC,SAAS;AAE1F,QAAO;EACL,IAAI,MAAM,MAAM,cAAc,WAAW;EACzC,OAAO,MAAM,SAAS,qBAAqB;EAC3C,gBAAgB,MAAM,kBAAkB;EACxC,UACE,QAAQ,aAAa,OACjB,CACE;GACE,aAAa,cAAc,WAAW;GACtC,QAAQ,IAAI,wBAAwB;GACpC,QAAQ,iBAAiB,GAAG,QAAQ,WAAW;GAC/C,QAAQ;GACT,CACF,GACD,EAAE;EACR,SAAS,CACP;GACE,aAAa,UAAU;GACvB,SAAS,IAAI,eAAe,YAAY,QAAQ;GACjD,CACF;EACD,WAAW,eACP,CACE;GACE,aAAa,wBAAwB;GACrC,QAAQ,IAAI,wBAAwB;GACpC,QAAQ,aAAa,GAAG;IACtB,iBAAiB,GAAG,QAAQ,WAAW;IACvC,GAAI,QAAQ,kBACR,CAAC,iBAAiB,GAAG,2BAA2B,QAAQ,gBAAgB,CAAC,GACzE,EAAE;IACN,GAAI,QAAQ,mBACR,CAAC,iBAAiB,GAAG,4BAA4B,QAAQ,iBAAiB,CAAC,GAC3E,EAAE;IACP,CAAC;GACF,QAAQ;GACT,CACF,GACD,EAAE;EACP;;AAGH,SAAgB,oBACd,MACA,QACA,SAC+B;AAC/B,QAAO,CACL,iBAAiB,MAAM;EACrB,WAAW,EAAE,aAAa,QAAQ;EAClC,iBAAiB;EACjB,kBAAkB;EACnB,CAAC,EACF,GAAG,QAAQ,KAAK,QAAQ,YAAY,MAAM,IAAI,MAAM,EAAE,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAC7E"}

package/dist/op-factory-call-CfPGebEH.d.mts.map

@@ -1 +0,0 @@
-{"version":3,"file":"op-factory-call-CfPGebEH.d.mts","names":[],"sources":["../src/core/op-factory-call.ts"],"sourcesContent":[],"mappings":";;;;;UAciB,WAAA;;EAAA,SAAA,KAAW,CAAA,EAAA,MAAA;EAMX,SAAA,cAAoB,CAAA,EAHT,uBAGS;;AACC,UADrB,oBACqB,CAAA,CAAA,CAAA,CAAA;EACpB,WAAA,CAAA,IAAA,EADE,eACF,CAAA,EADoB,CACpB;EAAgB,SAAA,CAAA,IAAA,EAAhB,aAAgB,CAAA,EAAA,CAAA;EACT,gBAAA,CAAA,IAAA,EAAA,oBAAA,CAAA,EAAuB,CAAvB;EAAuB,cAAA,CAAA,IAAA,EACzB,kBADyB,CAAA,EACJ,CADI;EACzB,OAAA,CAAA,IAAA,EACP,WADO,CAAA,EACO,CADP;;uBAIR,iBAAA,CAHC;EAAc,kBAAA,OAAA,EAAA,MAAA;EAAC,kBAAA,cAAA,EAKK,uBALL;EAGhB,kBAAA,KAAiB,EAAA,MAAA;EAEI,SAAA,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAEN,oBAFM,CAEe,CAFf,CAAA,CAAA,EAEoB,CAFpB;EAEe,UAAA,MAAA,CAAA,CAAA,EAAA,IAAA;;AAAK,cAW3C,eAAA,SAAwB,iBAAA,CAXmB;EAAC,SAAA,OAAA,EAAA,aAAA;EAW5C,SAAA,cAAgB,EAAA,UAAA;EAIE,SAAA,UAAA,EAAA,MAAA;EAAd,SAAA,IAAA,EAAA,aAAA,CAAc,aAAd,CAAA;EACG,SAAA,OAAA,EAAA,kBAAA,GAAA,SAAA;EAKI,SAAA,KAAA,EAAA,MAAA;EAAd,WAAA,CAAA,UAAA,EAAA,MAAA,EAAA,IAAA,EAAA,aAAA,CAAc,aAAd,CAAA,EAAA,OAAA,CAAA,EACI,kBADJ;EACI,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAUO,oBAVP,CAU4B,CAV5B,CAAA,CAAA,EAUiC,CAVjC;;AAUO,cAKR,aAAA,SAAsB,iBAAA,CALd;EAA0B,SAAA,OAAA,EAAA,WAAA;EArBV,SAAA,cAAA,EAAA,aAAA;EAAiB,SAAA,UAAA,EAAA,MAAA;EA0BzC,SAAA,IAAA,EAII,aAJU,CAII,aAJJ,CAAA;EAII,SAAA,KAAA,EAAA,MAAA;EAAd,WAAA,CAAA,UAAA,EAAA,MAAA,EAAA,IAAA,EAGuB,aAHvB,CAGqC,aAHrC,CAAA;EAGqC,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAQjC,oBARiC,CAQZ,CARY,CAAA,CAAA,EAQP,CARO;;AAQZ,cAK7B,oBAAA,SAA6B,iBAAA,CALA;EAArB,SAAA,OAAA,EAAA,kBAAA;EAA0B,SAAA,cAAA,EAAA,UAAA;EAfZ,SAAA,UAAA,EAAA,MAAA;EAAiB,SAAA,OAAA,EAwBhC,uBAxBgC,GAAA,SAAA;EAoBvC,SAAA,KAAA,EAAA,MAAA;EAIO,WAAA,CAAA,UAAA,EAAA,MAAA,EAAA,OAAA,CAAA,EAGwB,uBAHxB;EAGwB,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAQvB,oBARuB,CAQF,CARE,CAAA,CAAA,EAQG,CARH;;AAQvB,cAKR,kBAAA,SAA2B,iBAAA,CALnB;EAA0B,SAAA,OAAA,EAAA,gBAAA;EAfL,SAAA,cAAA,EAAA,aAAA;EAAiB,SAAA,UAAA,EAAA,MAAA;EAoB9C,SAAA,KAAA,EAAA,MAAmB;EAaU,WAAA,CAAA,UAAA,EAAA,MAAA;EAArB,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAAA,oBAAA,CAAqB,CAArB,CAAA,CAAA,EAA0B,CAA1B;;AAbmB,cAkB3B,WAAA,SAAoB,iBAAA,CAlBO;EAAiB,SAAA,OAAA,EAAA,SAAA;EAkB5C,SAAA,UAAY,EAAA,MAAA;EAGL,SAAA,OAAA,EAAA,cAAA;EACH,SAAA,IAAA,EAAA,WAAA,GAAA,SAAA;EACU,SAAA,cAAA,EAAA,uBAAA;EAGgB,SAAA,KAAA,EAAA,MAAA;EAAuB,WAAA,CAAA,UAAA,EAAA,MAAA,EAAA,OAAA,EAAvB,cAAuB,EAAA,IAAA,CAAA,EAAA,WAAA;EAUxB,MAAA,CAAA,CAAA,CAAA,CAAA,OAAA,EAArB,oBAAqB,CAAA,CAAA,CAAA,CAAA,EAAK,CAAL;;AAAK,KAKnC,aAAA,GACR,eAN2C,GAO3C,aAP2C,GAQ3C,oBAR2C,GAS3C,kBAT2C,GAU3C,WAV2C;AAlBd,iBA8BjB,+BAAA,CA9BiB,KAAA,EA8BsB,gBA9BtB,CAAA,EA8ByC,kBA9BzC;AAAiB,iBA4ClC,yCAAA,CA5CkC,IAAA,EA6C1C,qBA7C0C,CAAA,EA8C/C,uBA9C+C,GAAA,SAAA"}