@prisma-next/target-mongo 0.3.0 → 0.4.0-dev.1

package/README.md

@@ -7,14 +7,18 @@ MongoDB target pack for Prisma Next.
 - **Target pack assembly**: Exports the MongoDB target pack for authoring and family composition
 - **Target metadata**: Defines the stable Mongo target identity (`kind`, `familyId`, `targetId`, `version`, `capabilities`)
 - **Codec type surface**: Exposes the base Mongo codec type map used by authoring-time type composition
+- **Migration operation factories**: Factory functions for MongoDB migration operations
 
 ## Entrypoints
 
 - `./pack`: pure target pack ref used by `@prisma-next/family-mongo` and `@prisma-next/mongo-contract-ts`
 - `./codec-types`: base Mongo codec type map
+- `./migration`: factory functions (the `Migration` base class is in `@prisma-next/family-mongo/migration`)
 
 ## Usage
 
+### Contract definition
+
 ```typescript
 import mongoFamily from '@prisma-next/family-mongo/pack';
 import { defineContract } from '@prisma-next/mongo-contract-ts/contract-builder';
@@ -25,3 +29,35 @@ const contract = defineContract({
   target: mongoTarget,
 });
 ```
+
+### Migration authoring
+
+```typescript
+import { Migration } from '@prisma-next/family-mongo/migration';
+import { createIndex, createCollection } from '@prisma-next/target-mongo/migration';
+
+export default class extends Migration {
+  plan() {
+    return [
+      createCollection("users", {
+        validator: { $jsonSchema: { required: ["email"] } },
+        validationLevel: "strict",
+      }),
+      createIndex("users", [{ field: "email", direction: 1 }], { unique: true }),
+    ]
+  }
+}
+
+Migration.run(import.meta.url)
+```
+
+Run `tsx migration.ts` to produce `ops.json` and `migration.json` (when `describe()` is implemented). Use `--dry-run` to preview without writing.
+
+### Available factories
+
+- `createIndex(collection, keys, options?)` — create an index
+- `dropIndex(collection, keys)` — drop an index
+- `createCollection(collection, options?)` — create a collection
+- `dropCollection(collection)` — drop a collection
+- `setValidation(collection, schema, options?)` — set document validation on a collection
+- `validatedCollection(name, schema, indexes)` — create a collection with a JSON Schema validator and indexes

package/dist/control.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"control.d.mts","names":[],"sources":["../src/core/marker-ledger.ts"],"sourcesContent":[],"mappings":";;;;iBA+BsB,UAAA,KAAe,KAAK,QAAQ;iBAgB5B,UAAA,KAChB;EAjBgB,SAAA,WAAU,EAAA,MAAA;EAAK,SAAA,WAAA,EAAA,MAAA;CAAa,CAAA,EAmB/C,OAnB+C,CAAA,IAAA,CAAA;AAAR,iBAiCpB,YAAA,CAjCoB,EAAA,EAkCpC,EAlCoC,EAAA,YAAA,EAAA,MAAA,EAAA,WAAA,EAAA;EAAO,SAAA,WAAA,EAAA,MAAA;EAgB3B,SAAA,WAAU,EAAA,MAG7B;AAcH,CAAA,CAAA,EAIG,OAJmB,CAAA,OAAY,CAAA;AAqBZ,iBAAA,gBAAA,CAGnB,EAAA,EAFG,EAEI,EAAA,KAAA,EAAA;;;;IAAP"}
+{"version":3,"file":"control.d.mts","names":[],"sources":["../src/core/marker-ledger.ts"],"sourcesContent":[],"mappings":";;;;iBA+BsB,UAAA,KAAe,KAAK,QAAQ;iBAgB5B,UAAA,KAChB;EAjBgB,SAAA,WAAU,EAAA,MAAA;EAAK,SAAA,WAAA,EAAA,MAAA;CAAa,CAAA,EAmB/C,OAnB+C,CAAA,IAAA,CAAA;AAAR,iBAiCpB,YAAA,CAjCoB,EAAA,EAkCpC,EAlCoC,EAAA,YAAA,EAAA,MAAA,EAAA,WAAA,EAAA;EAAO,SAAA,WAAA,EAAA,MAAA;EAgB3B,SAAA,WAAU,EAAA,MAC1B;AAgBN,CAAA,CAAA,EAIG,OAJmB,CAAA,OAAY,CAAA;AAqBZ,iBAAA,gBAAA,CAGnB,EAAA,EAFG,EAEI,EAAA,KAAA,EAAA;;;;IAAP"}

package/dist/migration.d.mts

@@ -0,0 +1,18 @@
+import { CreateCollectionOptions, CreateIndexOptions, MongoIndexKey, MongoMigrationPlanOperation } from "@prisma-next/mongo-query-ast/control";
+
+//#region src/core/migration-factories.d.ts
+declare function createIndex(collection: string, keys: ReadonlyArray<MongoIndexKey>, options?: CreateIndexOptions): MongoMigrationPlanOperation;
+declare function dropIndex(collection: string, keys: ReadonlyArray<MongoIndexKey>): MongoMigrationPlanOperation;
+declare function createCollection(collection: string, options?: CreateCollectionOptions): MongoMigrationPlanOperation;
+declare function dropCollection(collection: string): MongoMigrationPlanOperation;
+declare function setValidation(collection: string, schema: Record<string, unknown>, options?: {
+  validationLevel?: 'strict' | 'moderate';
+  validationAction?: 'error' | 'warn';
+}): MongoMigrationPlanOperation;
+declare function validatedCollection(name: string, schema: Record<string, unknown>, indexes: ReadonlyArray<{
+  keys: MongoIndexKey[];
+  unique?: boolean;
+}>): MongoMigrationPlanOperation[];
+//#endregion
+export { createCollection, createIndex, dropCollection, dropIndex, setValidation, validatedCollection };
+//# sourceMappingURL=migration.d.mts.map

package/dist/migration.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"migration.d.mts","names":[],"sources":["../src/core/migration-factories.ts"],"sourcesContent":[],"mappings":";;;iBAiCgB,WAAA,2BAER,cAAc,0BACV,qBACT;AAJa,iBA4CA,SAAA,CA5CW,UAAA,EAAA,MAAA,EAAA,IAAA,EA8CnB,aA9CmB,CA8CL,aA9CK,CAAA,CAAA,EA+CxB,2BA/CwB;AAEL,iBA8EN,gBAAA,CA9EM,UAAA,EAAA,MAAA,EAAA,OAAA,CAAA,EAgFV,uBAhFU,CAAA,EAiFnB,2BAjFmB;AAAd,iBAwGQ,cAAA,CAxGR,UAAA,EAAA,MAAA,CAAA,EAwG4C,2BAxG5C;AACI,iBAuHI,aAAA,CAvHJ,UAAA,EAAA,MAAA,EAAA,MAAA,EAyHF,MAzHE,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAyCZ,CAzCY,EAAA;EACT,eAAA,CAAA,EAAA,QAAA,GAAA,UAAA;EAA2B,gBAAA,CAAA,EAAA,OAAA,GAAA,MAAA;AAwC9B,CAAA,CAAA,EAkFG,2BAlFsB;AAEH,iBAoGN,mBAAA,CApGM,IAAA,EAAA,MAAA,EAAA,MAAA,EAsGZ,MAtGY,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,EAuGX,aAvGW,CAAA;EAAd,IAAA,EAuGyB,aAvGzB,EAAA;EACL,MAAA,CAAA,EAAA,OAAA;CAA2B,CAAA,CAAA,EAuG3B,2BAvG2B,EAAA"}

package/dist/migration.mjs

@@ -0,0 +1,126 @@
+import { CollModCommand, CreateCollectionCommand, CreateIndexCommand, DropCollectionCommand, DropIndexCommand, ListCollectionsCommand, ListIndexesCommand, MongoAndExpr, MongoFieldFilter, buildIndexOpId, defaultMongoIndexName, keysToKeySpec } from "@prisma-next/mongo-query-ast/control";
+
+//#region src/core/migration-factories.ts
+function formatKeys(keys) {
+	return keys.map((k) => `${k.field}:${k.direction}`).join(", ");
+}
+function isTextIndex(keys) {
+	return keys.some((k) => k.direction === "text");
+}
+function keyFilter(keys) {
+	return isTextIndex(keys) ? MongoFieldFilter.eq("key._fts", "text") : MongoFieldFilter.eq("key", keysToKeySpec(keys));
+}
+function createIndex(collection, keys, options) {
+	const name = defaultMongoIndexName(keys);
+	const filter = keyFilter(keys);
+	const fullFilter = options?.unique ? MongoAndExpr.of([filter, MongoFieldFilter.eq("unique", true)]) : filter;
+	return {
+		id: buildIndexOpId("create", collection, keys),
+		label: `Create index on ${collection} (${formatKeys(keys)})`,
+		operationClass: "additive",
+		precheck: [{
+			description: `index does not already exist on ${collection}`,
+			source: new ListIndexesCommand(collection),
+			filter,
+			expect: "notExists"
+		}],
+		execute: [{
+			description: `create index on ${collection}`,
+			command: new CreateIndexCommand(collection, keys, {
+				...options,
+				unique: options?.unique ?? void 0,
+				name
+			})
+		}],
+		postcheck: [{
+			description: `index exists on ${collection}`,
+			source: new ListIndexesCommand(collection),
+			filter: fullFilter,
+			expect: "exists"
+		}]
+	};
+}
+function dropIndex(collection, keys) {
+	const indexName = defaultMongoIndexName(keys);
+	const filter = keyFilter(keys);
+	return {
+		id: buildIndexOpId("drop", collection, keys),
+		label: `Drop index on ${collection} (${formatKeys(keys)})`,
+		operationClass: "destructive",
+		precheck: [{
+			description: `index exists on ${collection}`,
+			source: new ListIndexesCommand(collection),
+			filter,
+			expect: "exists"
+		}],
+		execute: [{
+			description: `drop index on ${collection}`,
+			command: new DropIndexCommand(collection, indexName)
+		}],
+		postcheck: [{
+			description: `index no longer exists on ${collection}`,
+			source: new ListIndexesCommand(collection),
+			filter,
+			expect: "notExists"
+		}]
+	};
+}
+function createCollection(collection, options) {
+	return {
+		id: `collection.${collection}.create`,
+		label: `Create collection ${collection}`,
+		operationClass: "additive",
+		precheck: [{
+			description: `collection ${collection} does not exist`,
+			source: new ListCollectionsCommand(),
+			filter: MongoFieldFilter.eq("name", collection),
+			expect: "notExists"
+		}],
+		execute: [{
+			description: `create collection ${collection}`,
+			command: new CreateCollectionCommand(collection, options)
+		}],
+		postcheck: []
+	};
+}
+function dropCollection(collection) {
+	return {
+		id: `collection.${collection}.drop`,
+		label: `Drop collection ${collection}`,
+		operationClass: "destructive",
+		precheck: [],
+		execute: [{
+			description: `drop collection ${collection}`,
+			command: new DropCollectionCommand(collection)
+		}],
+		postcheck: []
+	};
+}
+function setValidation(collection, schema, options) {
+	return {
+		id: `collection.${collection}.setValidation`,
+		label: `Set validation on ${collection}`,
+		operationClass: "destructive",
+		precheck: [],
+		execute: [{
+			description: `set validation on ${collection}`,
+			command: new CollModCommand(collection, {
+				validator: { $jsonSchema: schema },
+				validationLevel: options?.validationLevel,
+				validationAction: options?.validationAction
+			})
+		}],
+		postcheck: []
+	};
+}
+function validatedCollection(name, schema, indexes) {
+	return [createCollection(name, {
+		validator: { $jsonSchema: schema },
+		validationLevel: "strict",
+		validationAction: "error"
+	}), ...indexes.map((idx) => createIndex(name, idx.keys, { unique: idx.unique }))];
+}
+
+//#endregion
+export { createCollection, createIndex, dropCollection, dropIndex, setValidation, validatedCollection };
+//# sourceMappingURL=migration.mjs.map

package/dist/migration.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"migration.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  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';\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 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":";;;AAmBA,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,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/package.json

@@ -1,23 +1,24 @@
 {
   "name": "@prisma-next/target-mongo",
-  "version": "0.3.0",
+  "version": "0.4.0-dev.1",
   "type": "module",
   "sideEffects": false,
   "description": "MongoDB target pack for Prisma Next",
   "dependencies": {
     "mongodb": "^6.16.0",
-    "@prisma-next/contract": "0.3.0",
-    "@prisma-next/framework-components": "0.3.0",
-    "@prisma-next/mongo-query-ast": "0.3.0"
+    "@prisma-next/contract": "0.4.0-dev.1",
+    "@prisma-next/mongo-query-ast": "0.4.0-dev.1",
+    "@prisma-next/framework-components": "0.4.0-dev.1"
   },
   "devDependencies": {
     "mongodb-memory-server": "10.4.3",
+    "pathe": "^2.0.3",
     "tsdown": "0.18.4",
     "typescript": "5.9.3",
     "vitest": "4.0.17",
     "@prisma-next/test-utils": "0.0.1",
-    "@prisma-next/tsdown": "0.0.0",
-    "@prisma-next/tsconfig": "0.0.0"
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/tsdown": "0.0.0"
   },
   "files": [
     "dist",
@@ -26,6 +27,7 @@
   "exports": {
     "./codec-types": "./dist/codec-types.mjs",
     "./control": "./dist/control.mjs",
+    "./migration": "./dist/migration.mjs",
     "./pack": "./dist/pack.mjs",
     "./package.json": "./package.json"
   },

package/src/core/migration-factories.ts

@@ -0,0 +1,193 @@
+import type { MongoIndexKey } from '@prisma-next/mongo-query-ast/control';
+import {
+  buildIndexOpId,
+  CollModCommand,
+  CreateCollectionCommand,
+  type CreateCollectionOptions,
+  CreateIndexCommand,
+  type CreateIndexOptions,
+  DropCollectionCommand,
+  DropIndexCommand,
+  defaultMongoIndexName,
+  keysToKeySpec,
+  ListCollectionsCommand,
+  ListIndexesCommand,
+  MongoAndExpr,
+  MongoFieldFilter,
+  type MongoMigrationPlanOperation,
+} from '@prisma-next/mongo-query-ast/control';
+
+function formatKeys(keys: ReadonlyArray<MongoIndexKey>): string {
+  return keys.map((k) => `${k.field}:${k.direction}`).join(', ');
+}
+
+function isTextIndex(keys: ReadonlyArray<MongoIndexKey>): boolean {
+  return keys.some((k) => k.direction === 'text');
+}
+
+function keyFilter(keys: ReadonlyArray<MongoIndexKey>) {
+  return isTextIndex(keys)
+    ? MongoFieldFilter.eq('key._fts', 'text')
+    : MongoFieldFilter.eq('key', keysToKeySpec(keys));
+}
+
+export function createIndex(
+  collection: string,
+  keys: ReadonlyArray<MongoIndexKey>,
+  options?: CreateIndexOptions,
+): MongoMigrationPlanOperation {
+  const name = defaultMongoIndexName(keys);
+  const filter = keyFilter(keys);
+  const fullFilter = options?.unique
+    ? MongoAndExpr.of([filter, MongoFieldFilter.eq('unique', true)])
+    : filter;
+
+  return {
+    id: buildIndexOpId('create', collection, keys),
+    label: `Create index on ${collection} (${formatKeys(keys)})`,
+    operationClass: 'additive',
+    precheck: [
+      {
+        description: `index does not already exist on ${collection}`,
+        source: new ListIndexesCommand(collection),
+        filter,
+        expect: 'notExists',
+      },
+    ],
+    execute: [
+      {
+        description: `create index on ${collection}`,
+        command: new CreateIndexCommand(collection, keys, {
+          ...options,
+          unique: options?.unique ?? undefined,
+          name,
+        }),
+      },
+    ],
+    postcheck: [
+      {
+        description: `index exists on ${collection}`,
+        source: new ListIndexesCommand(collection),
+        filter: fullFilter,
+        expect: 'exists',
+      },
+    ],
+  };
+}
+
+export function dropIndex(
+  collection: string,
+  keys: ReadonlyArray<MongoIndexKey>,
+): MongoMigrationPlanOperation {
+  const indexName = defaultMongoIndexName(keys);
+  const filter = keyFilter(keys);
+
+  return {
+    id: buildIndexOpId('drop', collection, keys),
+    label: `Drop index on ${collection} (${formatKeys(keys)})`,
+    operationClass: 'destructive',
+    precheck: [
+      {
+        description: `index exists on ${collection}`,
+        source: new ListIndexesCommand(collection),
+        filter,
+        expect: 'exists',
+      },
+    ],
+    execute: [
+      {
+        description: `drop index on ${collection}`,
+        command: new DropIndexCommand(collection, indexName),
+      },
+    ],
+    postcheck: [
+      {
+        description: `index no longer exists on ${collection}`,
+        source: new ListIndexesCommand(collection),
+        filter,
+        expect: 'notExists',
+      },
+    ],
+  };
+}
+
+export function createCollection(
+  collection: string,
+  options?: CreateCollectionOptions,
+): MongoMigrationPlanOperation {
+  return {
+    id: `collection.${collection}.create`,
+    label: `Create collection ${collection}`,
+    operationClass: 'additive',
+    precheck: [
+      {
+        description: `collection ${collection} does not exist`,
+        source: new ListCollectionsCommand(),
+        filter: MongoFieldFilter.eq('name', collection),
+        expect: 'notExists',
+      },
+    ],
+    execute: [
+      {
+        description: `create collection ${collection}`,
+        command: new CreateCollectionCommand(collection, options),
+      },
+    ],
+    postcheck: [],
+  };
+}
+
+export function dropCollection(collection: string): MongoMigrationPlanOperation {
+  return {
+    id: `collection.${collection}.drop`,
+    label: `Drop collection ${collection}`,
+    operationClass: 'destructive',
+    precheck: [],
+    execute: [
+      {
+        description: `drop collection ${collection}`,
+        command: new DropCollectionCommand(collection),
+      },
+    ],
+    postcheck: [],
+  };
+}
+
+export function setValidation(
+  collection: string,
+  schema: Record<string, unknown>,
+  options?: { validationLevel?: 'strict' | 'moderate'; validationAction?: 'error' | 'warn' },
+): MongoMigrationPlanOperation {
+  return {
+    id: `collection.${collection}.setValidation`,
+    label: `Set validation on ${collection}`,
+    operationClass: 'destructive',
+    precheck: [],
+    execute: [
+      {
+        description: `set validation on ${collection}`,
+        command: new CollModCommand(collection, {
+          validator: { $jsonSchema: schema },
+          validationLevel: options?.validationLevel,
+          validationAction: options?.validationAction,
+        }),
+      },
+    ],
+    postcheck: [],
+  };
+}
+
+export function validatedCollection(
+  name: string,
+  schema: Record<string, unknown>,
+  indexes: ReadonlyArray<{ keys: MongoIndexKey[]; unique?: boolean }>,
+): MongoMigrationPlanOperation[] {
+  return [
+    createCollection(name, {
+      validator: { $jsonSchema: schema },
+      validationLevel: 'strict',
+      validationAction: 'error',
+    }),
+    ...indexes.map((idx) => createIndex(name, idx.keys, { unique: idx.unique })),
+  ];
+}

package/src/exports/migration.ts

@@ -0,0 +1,8 @@
+export {
+  createCollection,
+  createIndex,
+  dropCollection,
+  dropIndex,
+  setValidation,
+  validatedCollection,
+} from '../core/migration-factories';