@atscript/mongo 0.0.17 → 0.0.19

package/dist/index.cjs

@@ -106,6 +106,15 @@ const annotations = { mongo: {
 			});
 		}
 	}),
+	autoIndexes: new __atscript_core.AnnotationSpec({
+		description: "Switch on/off the automatic index creation. Works with as-mongo moost controller.\n\nDefault: true",
+		nodeType: ["interface"],
+		argument: {
+			name: "type",
+			type: "boolean",
+			description: "On/Off the automatic index creation"
+		}
+	}),
 	index: {
 		plain: new __atscript_core.AnnotationSpec({
 			description: "Defines a **standard MongoDB index** on a field.\n\n- Improves query performance on indexed fields.\n- Can be used for **single-field** or **compound** indexes.\n\n**Example:**\n```atscript\n@mongo.index.plain \"departmentIndex\"\ndepartment: string\n```\n",
@@ -278,6 +287,28 @@ const annotations = { mongo: {
 			});
 			return errors;
 		}
+	}) },
+	array: { uniqueItems: new __atscript_core.AnnotationSpec({
+		description: "Marks an **array field** as containing *globally unique items* when handling **patch `$insert` operations**.\n\n- Forces the patcher to use **set-semantics** (`$setUnion`) instead of a plain append, so duplicates are silently skipped.\n- Has **no effect** on `$replace`, `$update`, or `$remove`.\n- If the array’s element type already defines one or more `@meta.isKey` properties, *uniqueness is implied* and this annotation is unnecessary (but harmless).\n\n**Example:**\n```atscript\n@mongo.array.uniqueItems\ntags: string[]\n\n// Later in a patch payload …\n{\n  $insert: {\n    tags: [\"mongo\", \"mongo\"] // second \"mongo\" is ignored\n  }\n}\n```\n",
+		nodeType: ["prop"],
+		multiple: false,
+		validate(token, args, doc) {
+			const field = token.parentNode;
+			const errors = [];
+			const definition = field.getDefinition();
+			if (!definition) return errors;
+			let wrongType = false;
+			if ((0, __atscript_core.isRef)(definition)) {
+				const def = doc.unwindType(definition.id, definition.chain)?.def;
+				if (!(0, __atscript_core.isArray)(def)) wrongType = true;
+			} else if (!(0, __atscript_core.isArray)(definition)) wrongType = true;
+			if (wrongType) errors.push({
+				message: `[mongo] type of array expected when using @mongo.array.uniqueItems`,
+				severity: 1,
+				range: token.range
+			});
+			return errors;
+		}
 	}) }
 } };
 
@@ -305,6 +336,304 @@ const NoopLogger = {
 	debug: () => {}
 };
 
+//#endregion
+//#region packages/mongo/src/lib/validate-plugins.ts
+const validateMongoIdPlugin = (ctx, def, value) => {
+	if (ctx.path === "_id" && def.type.tags.has("objectId")) return ctx.validateAnnotatedType(def, value instanceof mongodb.ObjectId ? value.toString() : value);
+};
+const validateMongoUniqueArrayItemsPlugin = (ctx, def, value) => {
+	if (def.metadata.has("mongo.array.uniqueItems") && def.type.kind === "array") {
+		if (Array.isArray(value)) {
+			const separator = "▼↩";
+			const seen = new Set();
+			const keyProps = CollectionPatcher.getKeyProps(def);
+			for (const item of value) {
+				let key = "";
+				if (keyProps.size) for (const prop of keyProps) key += JSON.stringify(item[prop]) + separator;
+else key = JSON.stringify(item);
+				if (seen.has(key)) {
+					ctx.error(`Duplicate items are not allowed`);
+					return false;
+				}
+				seen.add(key);
+			}
+		}
+	}
+	return undefined;
+};
+
+//#endregion
+//#region packages/mongo/src/lib/collection-patcher.ts
+function _define_property$2(obj, key, value) {
+	if (key in obj) Object.defineProperty(obj, key, {
+		value,
+		enumerable: true,
+		configurable: true,
+		writable: true
+	});
+else obj[key] = value;
+	return obj;
+}
+var CollectionPatcher = class CollectionPatcher {
+	/**
+	* Extract a set of *key properties* (annotated with `@meta.isKey`) from an
+	* array‐of‐objects type definition. These keys uniquely identify an element
+	* inside the array and are later used for `$update`, `$remove` and `$upsert`.
+	*
+	* @param def Atscript array type
+	* @returns Set of property names marked as keys; empty set if none
+	*/ static getKeyProps(def) {
+		if (def.type.of.type.kind === "object") {
+			const objType = def.type.of.type;
+			const keyProps = new Set();
+			for (const [key, val] of objType.props.entries()) if (val.metadata.get("meta.isKey")) keyProps.add(key);
+			return keyProps;
+		}
+		return new Set();
+	}
+	/**
+	* Build a runtime *Validator* that understands the extended patch payload.
+	*
+	*  * Adds per‑array *patch* wrappers (the `$replace`, `$insert`, … fields).
+	*  * Honors `mongo.patch.strategy === "merge"` metadata.
+	*
+	* @param collection Target collection wrapper
+	* @returns Atscript Validator
+	*/ static prepareValidator(collection) {
+		return collection.type.validator({
+			plugins: [validateMongoIdPlugin, validateMongoUniqueArrayItemsPlugin],
+			replace: (def, path) => {
+				if (path === "" && def.type.kind === "object") {
+					const obj = (0, __atscript_typescript.defineAnnotatedType)("object").copyMetadata(def.metadata);
+					for (const [prop, type] of def.type.props.entries()) obj.prop(prop, (0, __atscript_typescript.defineAnnotatedType)().refTo(type).copyMetadata(type.metadata).optional(prop !== "_id").$type);
+					return obj.$type;
+				}
+				if (def.type.kind === "array" && collection.flatMap.get(path)?.metadata.get("mongo.__topLevelArray") && !def.metadata.has("mongo.__patchArrayValue")) {
+					const defArray = def;
+					const mergeStrategy = defArray.metadata.get("mongo.patch.strategy") === "merge";
+					function getPatchType() {
+						const isPrimitive$1 = (0, __atscript_typescript.isAnnotatedTypeOfPrimitive)(defArray.type.of);
+						if (isPrimitive$1) return (0, __atscript_typescript.defineAnnotatedType)().refTo(def).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+						if (defArray.type.of.type.kind === "object") {
+							const objType = defArray.type.of.type;
+							const t = (0, __atscript_typescript.defineAnnotatedType)("object").copyMetadata(defArray.type.of.metadata);
+							const keyProps = CollectionPatcher.getKeyProps(defArray);
+							for (const [key, val] of objType.props.entries()) if (keyProps.size) if (keyProps.has(key)) t.prop(key, (0, __atscript_typescript.defineAnnotatedType)().refTo(val).copyMetadata(def.metadata).$type);
+else t.prop(key, (0, __atscript_typescript.defineAnnotatedType)().refTo(val).copyMetadata(def.metadata).optional().$type);
+else t.prop(key, (0, __atscript_typescript.defineAnnotatedType)().refTo(val).copyMetadata(def.metadata).optional(!!val.optional).$type);
+							return (0, __atscript_typescript.defineAnnotatedType)("array").of(t.$type).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+						}
+						return undefined;
+					}
+					const fullType = (0, __atscript_typescript.defineAnnotatedType)().refTo(def).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+					const patchType = getPatchType();
+					return patchType ? (0, __atscript_typescript.defineAnnotatedType)("object").prop("$replace", fullType).prop("$insert", fullType).prop("$upsert", fullType).prop("$update", mergeStrategy ? patchType : fullType).prop("$remove", patchType).optional().$type : (0, __atscript_typescript.defineAnnotatedType)("object").prop("$replace", fullType).prop("$insert", fullType).optional().$type;
+				}
+				return def;
+			},
+			partial: (def, path) => {
+				return path !== "" && def.metadata.get("mongo.patch.strategy") === "merge";
+			}
+		});
+	}
+	/**
+	* Entry point – walk the payload, build `filter`, `update` and `options`.
+	*
+	* @returns Helper object exposing both individual parts and
+	*          a `.toArgs()` convenience callback.
+	*/ preparePatch() {
+		this.filterObj = { _id: this.collection.prepareId(this.payload._id) };
+		this.flattenPayload(this.payload);
+		let updateFilter = this.updatePipeline;
+		return {
+			toArgs: () => [
+				this.filterObj,
+				updateFilter,
+				this.optionsObj
+			],
+			filter: this.filterObj,
+			updateFilter,
+			updateOptions: this.optionsObj
+		};
+	}
+	/**
+	* Helper – lazily create `$set` section and assign *key* → *value*.
+	*
+	* @param key Fully‑qualified dotted path
+	* @param val Value to be written
+	* @private
+	*/ _set(key, val) {
+		for (const pipe of this.updatePipeline) {
+			if (!pipe.$set) pipe.$set = {};
+			if (!pipe.$set[key]) {
+				pipe.$set[key] = val;
+				return;
+			}
+		}
+		this.updatePipeline.push({ $set: { [key]: val } });
+	}
+	/**
+	* Recursively walk through the patch *payload* and convert it into `$set`/…
+	* statements. Top‑level arrays are delegated to {@link parseArrayPatch}.
+	*
+	* @param payload Current payload chunk
+	* @param prefix  Dotted path accumulated so far
+	* @private
+	*/ flattenPayload(payload, prefix = "") {
+		const evalKey = (k) => prefix ? `${prefix}.${k}` : k;
+		for (const [_key, value] of Object.entries(payload)) {
+			const key = evalKey(_key);
+			const flatType = this.collection.flatMap.get(key);
+			const topLevelArray = flatType?.metadata?.get("mongo.__topLevelArray");
+			if (typeof value === "object" && topLevelArray) this.parseArrayPatch(key, value);
+else if (typeof value === "object" && this.collection.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge") this.flattenPayload(value, key);
+else if (key !== "_id") this._set(key, value);
+		}
+		return this.updatePipeline;
+	}
+	/**
+	* Dispatch a *single* array patch. Exactly one of `$replace`, `$insert`,
+	* `$upsert`, `$update`, `$remove` must be present – otherwise we throw.
+	*
+	* @param key   Dotted path to the array field
+	* @param value Payload slice for that field
+	* @private
+	*/ parseArrayPatch(key, value) {
+		const flatType = this.collection.flatMap.get(key);
+		const toRemove = value.$remove;
+		const toReplace = value.$replace;
+		const toInsert = value.$insert;
+		const toUpsert = value.$upsert;
+		const toUpdate = value.$update;
+		const keyProps = flatType?.type.kind === "array" ? CollectionPatcher.getKeyProps(flatType) : new Set();
+		this._remove(key, toRemove, keyProps);
+		this._replace(key, toReplace);
+		this._insert(key, toInsert, keyProps);
+		this._upsert(key, toUpsert, keyProps);
+		this._update(key, toUpdate, keyProps);
+	}
+	/**
+	* Build an *aggregation‐expression* that checks equality by **all** keys in
+	* `keys`.  Example output for keys `["id", "lang"]` and bases `a`, `b`:
+	* ```json
+	* { "$and": [ { "$eq": ["$$a.id", "$$b.id"] }, { "$eq": ["$$a.lang", "$$b.lang"] } ] }
+	* ```
+	*
+	* @param keys  Ordered list of key property names
+	* @param left  Base token for *left* expression (e.g. `"$$el"`)
+	* @param right Base token for *right* expression (e.g. `"$$this"`)
+	*/ _keysEqual(keys, left, right) {
+		return keys.map((k) => ({ $eq: [`${left}.${k}`, `${right}.${k}`] })).reduce((acc, cur) => acc ? { $and: [acc, cur] } : cur);
+	}
+	/**
+	* `$replace` – overwrite the entire array with `input`.
+	*
+	* @param key   Dotted path to the array
+	* @param input New array value (may be `undefined`)
+	* @private
+	*/ _replace(key, input) {
+		if (input) this._set(key, input);
+	}
+	/**
+	* `$insert`
+	* - plain append      → $concatArrays
+	* - unique / keyed    → delegate to _upsert (insert-or-update)
+	*/ _insert(key, input, keyProps) {
+		if (!input?.length) return;
+		const uniqueItems = this.collection.flatMap.get(key)?.metadata?.has("mongo.array.uniqueItems");
+		if (uniqueItems || keyProps.size > 0) this._upsert(key, input, keyProps);
+else this._set(key, { $concatArrays: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$upsert`
+	* - keyed  → remove existing matching by key(s) then append candidate
+	* - unique → $setUnion (deep equality)
+	*/ _upsert(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const keys = [...keyProps];
+			this._set(key, { $reduce: {
+				input,
+				initialValue: { $ifNull: [`$${key}`, []] },
+				in: { $let: {
+					vars: {
+						acc: "$$value",
+						cand: "$$this"
+					},
+					in: { $concatArrays: [{ $filter: {
+						input: "$$acc",
+						as: "el",
+						cond: { $not: this._keysEqual(keys, "$$el", "$$cand") }
+					} }, ["$$cand"]] }
+				} }
+			} });
+			return;
+		}
+		this._set(key, { $setUnion: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$update`
+	* - keyed       → map array and merge / replace matching element(s)
+	* - non-keyed   → behave like `$addToSet` (insert only when not present)
+	*/ _update(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const mergeStrategy = this.collection.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge";
+			const keys = [...keyProps];
+			this._set(key, { $reduce: {
+				input,
+				initialValue: { $ifNull: [`$${key}`, []] },
+				in: { $map: {
+					input: "$$value",
+					as: "el",
+					in: { $cond: [
+						this._keysEqual(keys, "$$el", "$$this"),
+						mergeStrategy ? { $mergeObjects: ["$$el", "$$this"] } : "$$this",
+						"$$el"
+					] }
+				} }
+			} });
+		} else this._set(key, { $setUnion: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$remove`
+	* - keyed     → filter out any element whose key set matches a payload item
+	* - non-keyed → deep equality remove (`$setDifference`)
+	*/ _remove(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const keys = [...keyProps];
+			this._set(key, { $let: {
+				vars: { rem: input },
+				in: { $filter: {
+					input: { $ifNull: [`$${key}`, []] },
+					as: "el",
+					cond: { $not: { $anyElementTrue: { $map: {
+						input: "$$rem",
+						as: "r",
+						in: this._keysEqual(keys, "$$el", "$$r")
+					} } } }
+				} }
+			} });
+		} else this._set(key, { $setDifference: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	constructor(collection, payload) {
+		_define_property$2(this, "collection", void 0);
+		_define_property$2(this, "payload", void 0);
+		/**
+		* Internal accumulator: filter passed to `updateOne()`.
+		* Filled only with the `_id` field right now.
+		*/ _define_property$2(this, "filterObj", void 0);
+		/** MongoDB *update* document being built. */ _define_property$2(this, "updatePipeline", void 0);
+		/** Additional *options* (mainly `arrayFilters`). */ _define_property$2(this, "optionsObj", void 0);
+		this.collection = collection;
+		this.payload = payload;
+		this.filterObj = {};
+		this.updatePipeline = [];
+		this.optionsObj = {};
+	}
+};
+
 //#endregion
 //#region packages/mongo/src/lib/as-collection.ts
 function _define_property$1(obj, key, value) {
@@ -370,17 +699,24 @@ var AsCollection = class {
 	*/ getValidator(purpose) {
 		if (!this.validators.has(purpose)) switch (purpose) {
 			case "insert": {
-				this.validators.set(purpose, this.type.validator(this.idType === "objectId" ? { skipList: new Set(["_id"]) } : {}));
+				this.validators.set(purpose, this.type.validator({
+					plugins: [validateMongoIdPlugin, validateMongoUniqueArrayItemsPlugin],
+					replace(type, path) {
+						if (path === "_id" && type.type.tags.has("objectId")) return {
+							...type,
+							optional: true
+						};
+						return type;
+					}
+				}));
 				break;
 			}
 			case "update": {
-				this.validators.set(purpose, this.type.validator());
+				this.validators.set(purpose, this.type.validator({ plugins: [validateMongoIdPlugin] }));
 				break;
 			}
 			case "patch": {
-				this.validators.set(purpose, this.type.validator({ partial: (def, path) => {
-					return path === "" || def.metadata.get("mongo.patch.strategy") === "merge";
-				} }));
+				this.validators.set(purpose, CollectionPatcher.prepareValidator(this));
 				break;
 			}
 			default: throw new Error(`Unknown validator purpose: ${purpose}`);
@@ -431,20 +767,24 @@ else {
 			if (analyzer) index.definition.mappings.fields[fieldName].analyzer = analyzer;
 		}
 	}
-	_flattenType(type, prefix) {
+	_flattenType(type, prefix = "", inComplexTypeOrArray = false) {
 		switch (type.type.kind) {
 			case "object":
 				this._flatMap?.set(prefix || "", type);
 				const items = Array.from(type.type.props.entries());
-				for (const [key, value] of items) this._flattenType(value, prefix ? `${prefix}.${key}` : key);
+				for (const [key, value] of items) this._flattenType(value, prefix ? `${prefix}.${key}` : key, inComplexTypeOrArray);
 				break;
 			case "array":
-				this._flatMap?.set(prefix || "", type);
-				if (type.type.of.type.kind) this._flattenType(type.type.of, prefix);
+				let typeArray = type;
+				if (!inComplexTypeOrArray) {
+					typeArray = (0, __atscript_typescript.defineAnnotatedType)().refTo(type).copyMetadata(type.metadata).$type;
+					typeArray.metadata.set("mongo.__topLevelArray", true);
+				}
+				this._flatMap?.set(prefix || "", typeArray);
 				break;
 			case "intersection":
 			case "tuple":
-			case "union": for (const item of type.type.items) this._flattenType(item, prefix);
+			case "union": for (const item of type.type.items) this._flattenType(item, prefix, true);
 			default:
 				this._flatMap?.set(prefix || "", type);
 				break;
@@ -465,6 +805,9 @@ else {
 			text: { fuzzy: { maxEdits: textSearch.fuzzy || 0 } }
 		});
 	}
+	get uniqueProps() {
+		return this._uniqueProps;
+	}
 	_finalizeIndexesForCollection() {
 		for (const [key, value] of Array.from(this._vectorFilters.entries())) {
 			const index = this._indexes.get(key);
@@ -473,6 +816,10 @@ else {
 				path: value
 			});
 		}
+		for (const [, value] of Array.from(this._indexes.entries())) if (value.type === "unique") {
+			const keys = Object.keys(value.fields);
+			if (keys.length === 1) this._uniqueProps.add(keys[0]);
+		}
 	}
 	_prepareIndexesForField(fieldName, metadata) {
 		for (const index of metadata.get("mongo.index.plain") || []) this._addIndexField("plain", index === true ? fieldName : index, fieldName);
@@ -588,39 +935,62 @@ else toUpdate.add(remote.name);
 			default:
 		}
 	}
+	insert(payload, options) {
+		const toInsert = this.prepareInsert(payload);
+		return Array.isArray(toInsert) ? this.collection.insertMany(toInsert, options) : this.collection.insertOne(toInsert, options);
+	}
+	replace(payload, options) {
+		const [filter, replace, opts] = this.prepareReplace(payload).toArgs();
+		return this.collection.replaceOne(filter, replace, {
+			...opts,
+			...options
+		});
+	}
+	update(payload, options) {
+		const [filter, update, opts] = this.prepareUpdate(payload).toArgs();
+		return this.collection.updateOne(filter, update, {
+			...opts,
+			...options
+		});
+	}
 	prepareInsert(payload) {
 		const v = this.getValidator("insert");
-		if (v.validate(payload)) {
-			const data = { ...payload };
+		const arr = Array.isArray(payload) ? payload : [payload];
+		const prepared = [];
+		for (const item of arr) if (v.validate(item)) {
+			const data = { ...item };
 			if (data._id) data._id = this.prepareId(data._id);
 else if (this.idType !== "objectId") throw new Error("Missing \"_id\" field");
-			return data;
-		}
-		throw new Error("Invalid payload");
+			prepared.push(data);
+		} else throw new Error("Invalid payload");
+		return prepared.length === 1 ? prepared[0] : prepared;
 	}
-	prepareUpdate(payload) {
-		const v = this.getValidator("insert");
+	prepareReplace(payload) {
+		const v = this.getValidator("update");
 		if (v.validate(payload)) {
-			const data = { ...payload };
-			data._id = this.prepareId(data._id);
-			return data;
+			const _id = this.prepareId(payload._id);
+			const data = {
+				...payload,
+				_id
+			};
+			return {
+				toArgs: () => [
+					{ _id },
+					data,
+					{}
+				],
+				filter: { _id },
+				updateFilter: data,
+				updateOptions: {}
+			};
 		}
 		throw new Error("Invalid payload");
 	}
-	preparePatch(payload) {
+	prepareUpdate(payload) {
 		const v = this.getValidator("patch");
-		if (v.validate(payload)) return { $set: this._flattenPayload(payload) };
+		if (v.validate(payload)) return new CollectionPatcher(this, payload).preparePatch();
 		throw new Error("Invalid payload");
 	}
-	_flattenPayload(payload, prefix = "", obj = {}) {
-		const evalKey = (k) => prefix ? `${prefix}.${k}` : k;
-		for (const [_key, value] of Object.entries(payload)) {
-			const key = evalKey(_key);
-			if (typeof value === "object" && this.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge") this._flattenPayload(value, key, obj);
-else obj[key] = value;
-		}
-		return obj;
-	}
 	constructor(asMongo, _type, logger = NoopLogger) {
 		_define_property$1(this, "asMongo", void 0);
 		_define_property$1(this, "_type", void 0);
@@ -631,12 +1001,14 @@ else obj[key] = value;
 		_define_property$1(this, "_indexes", void 0);
 		_define_property$1(this, "_vectorFilters", void 0);
 		_define_property$1(this, "_flatMap", void 0);
+		_define_property$1(this, "_uniqueProps", void 0);
 		this.asMongo = asMongo;
 		this._type = _type;
 		this.logger = logger;
 		this.validators = new Map();
 		this._indexes = new Map();
 		this._vectorFilters = new Map();
+		this._uniqueProps = new Set();
 		if (!(0, __atscript_typescript.isAnnotatedType)(_type)) throw new Error("Atscript Annotated Type expected");
 		const name = _type.metadata.get("mongo.collection");
 		if (!name) throw new Error("@mongo.collection annotation expected with collection name");
@@ -715,13 +1087,20 @@ var AsMongo = class {
 		return list.has(name);
 	}
 	getCollection(type, logger) {
-		return new AsCollection(this, type, logger || this.logger);
+		let collection = this._collections.get(type);
+		if (!collection) {
+			collection = new AsCollection(this, type, logger || this.logger);
+			this._collections.set(type, collection);
+		}
+		return collection;
 	}
 	constructor(client, logger = NoopLogger) {
 		_define_property(this, "logger", void 0);
 		_define_property(this, "client", void 0);
 		_define_property(this, "collectionsList", void 0);
+		_define_property(this, "_collections", void 0);
 		this.logger = logger;
+		this._collections = new WeakMap();
 		if (typeof client === "string") this.client = new mongodb.MongoClient(client);
 else this.client = client;
 	}

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { TAtscriptPlugin } from '@atscript/core';
-import { TAtscriptAnnotatedTypeConstructor, TAtscriptAnnotatedType, TAtscriptTypeObject, TMetadataMap } from '@atscript/typescript';
 import * as mongodb from 'mongodb';
-import { MongoClient, Collection, ObjectId, WithId, UpdateFilter, MatchKeysAndValues } from 'mongodb';
+import { MongoClient, Collection, ObjectId, InsertOneOptions, ReplaceOptions, UpdateOptions, OptionalUnlessRequiredId, Filter, WithoutId } from 'mongodb';
+import { TAtscriptAnnotatedTypeConstructor, TAtscriptAnnotatedType, TAtscriptTypeObject, TMetadataMap } from '@atscript/typescript';
 
 declare const MongoPlugin: () => TAtscriptPlugin;
 
@@ -22,6 +22,7 @@ declare class AsMongo {
     protected getCollectionsList(): Promise<Set<string>>;
     collectionExists(name: string): Promise<boolean>;
     getCollection<T extends TAtscriptAnnotatedTypeConstructor>(type: T, logger?: TGenericLogger): AsCollection<T>;
+    private _collections;
 }
 
 type TPlainIndex = {
@@ -48,25 +49,74 @@ declare class AsCollection<T extends TAtscriptAnnotatedTypeConstructor> {
     constructor(asMongo: AsMongo, _type: T, logger?: TGenericLogger);
     exists(): Promise<boolean>;
     ensureExists(): Promise<void>;
+    /**
+     * Returns the a type definition of the "_id" prop.
+     */
     get idType(): 'string' | 'number' | 'objectId';
+    /**
+     * Transforms an "_id" value to the expected type (`ObjectId`, `number`, or `string`).
+     * Assumes input has already been validated.
+     *
+     * @param {string | number | ObjectId} id - The validated ID.
+     * @returns {string | number | ObjectId} - The transformed ID.
+     * @throws {Error} If the `_id` type is unknown.
+     */
     prepareId<D = string | number | ObjectId>(id: string | number | ObjectId): D;
+    /**
+     * Retrieves a validator for a given purpose. If the validator is not already cached,
+     * it creates and stores a new one based on the purpose.
+     *
+     * @param {TValidatorPurpose} purpose - The validation purpose (`input`, `update`, `patch`).
+     * @returns {Validator} The corresponding validator instance.
+     * @throws {Error} If an unknown purpose is provided.
+     */
     getValidator(purpose: TValidatorPurpose): any;
     get type(): TAtscriptAnnotatedType<TAtscriptTypeObject>;
     get indexes(): Map<string, TIndex>;
     protected _addIndexField(type: TPlainIndex['type'], name: string, field: string, weight?: number): void;
     protected _setSearchIndex(type: TSearchIndex['type'], name: string | undefined, definition: TMongoSearchIndexDefinition): void;
     protected _addFieldToSearchIndex(type: TSearchIndex['type'], _name: string | undefined, fieldName: string, analyzer?: string): void;
-    protected _flattenType(type: TAtscriptAnnotatedType, prefix?: string): void;
+    protected _flattenType(type: TAtscriptAnnotatedType, prefix?: string, inComplexTypeOrArray?: boolean): void;
     protected _prepareIndexesForCollection(): void;
+    protected _uniqueProps: Set<string>;
+    get uniqueProps(): Set<string>;
     protected _finalizeIndexesForCollection(): void;
     protected _prepareIndexesForField(fieldName: string, metadata: TMetadataMap<AtscriptMetadata>): void;
     protected _flatten(): void;
     get flatMap(): Map<string, TAtscriptAnnotatedType>;
     syncIndexes(): Promise<void>;
-    prepareInsert(payload: any): InstanceType<T>;
-    prepareUpdate(payload: any): WithId<InstanceType<T>>;
-    preparePatch(payload: any): UpdateFilter<InstanceType<T>>;
-    protected _flattenPayload(payload: T, prefix?: string, obj?: MatchKeysAndValues<InstanceType<T>>): MatchKeysAndValues<InstanceType<T>>;
+    insert(payload: (Omit<InstanceType<T>, '_id'> & {
+        _id?: InstanceType<T>['_id'] | ObjectId;
+    }) | (Omit<InstanceType<T>, '_id'> & {
+        _id?: InstanceType<T>['_id'] | ObjectId;
+    })[], options?: InsertOneOptions): Promise<mongodb.InsertManyResult<InstanceType<T>>> | Promise<mongodb.InsertOneResult<InstanceType<T>>>;
+    replace(payload: Omit<InstanceType<T>, '_id'> & {
+        _id: InstanceType<T>['_id'] | ObjectId;
+    }, options?: ReplaceOptions): Promise<mongodb.UpdateResult<InstanceType<T>>>;
+    update(payload: AsMongoPatch<Omit<InstanceType<T>, '_id'>> & {
+        _id: InstanceType<T>['_id'] | ObjectId;
+    }, options?: UpdateOptions): Promise<mongodb.UpdateResult<InstanceType<T>>>;
+    prepareInsert(payload: (Omit<InstanceType<T>, '_id'> & {
+        _id?: InstanceType<T>['_id'] | ObjectId;
+    }) | (Omit<InstanceType<T>, '_id'> & {
+        _id?: InstanceType<T>['_id'] | ObjectId;
+    })[]): OptionalUnlessRequiredId<InstanceType<T>> | OptionalUnlessRequiredId<InstanceType<T>>[];
+    prepareReplace(payload: Omit<InstanceType<T>, '_id'> & {
+        _id: InstanceType<T>['_id'] | ObjectId;
+    }): {
+        toArgs: () => [Filter<InstanceType<T>>, WithoutId<InstanceType<T>>, ReplaceOptions];
+        filter: Filter<InstanceType<T>>;
+        updateFilter: WithoutId<InstanceType<T>>;
+        updateOptions: ReplaceOptions;
+    };
+    prepareUpdate(payload: AsMongoPatch<Omit<InstanceType<T>, '_id'>> & {
+        _id: InstanceType<T>['_id'] | ObjectId;
+    }): {
+        toArgs: () => [Filter<InstanceType<T>>, mongodb.Document[] | mongodb.UpdateFilter<InstanceType<T>>, UpdateOptions];
+        filter: Filter<InstanceType<T>>;
+        updateFilter: mongodb.Document[];
+        updateOptions: UpdateOptions;
+    };
 }
 type TVectorSimilarity = 'cosine' | 'euclidean' | 'dotProduct';
 type TMongoSearchIndexDefinition = {
@@ -90,5 +140,26 @@ type TMongoSearchIndexDefinition = {
         };
     };
 };
+type TArrayPatch<A extends readonly unknown[]> = {
+    $replace?: A;
+    $insert?: A;
+    $upsert?: A;
+    $update?: Partial<TArrayElement<A>>[];
+    $remove?: Partial<TArrayElement<A>>[];
+};
+type TArrayElement<ArrayType extends readonly unknown[]> = ArrayType extends readonly (infer ElementType)[] ? ElementType : never;
+/**
+ * AsMongoPatch<T>
+ * ─────────────────
+ * - For every key K in T:
+ *     • if T[K] is `X[]`, rewrite it to `TArrayPatch<X[]>`
+ *     • otherwise omit the key (feel free to keep it if you want)
+ *
+ * The result is an *optional* property bag that matches a patch payload
+ * for array fields only.
+ */
+type AsMongoPatch<T> = {
+    [K in keyof T]?: T[K] extends Array<infer _> ? TArrayPatch<T[K]> : Partial<T[K]>;
+};
 
 export { AsCollection, AsMongo, MongoPlugin };

package/dist/index.mjs

@@ -1,5 +1,5 @@
 import { AnnotationSpec, isArray, isInterface, isPrimitive, isRef, isStructure } from "@atscript/core";
-import { isAnnotatedType } from "@atscript/typescript";
+import { defineAnnotatedType, isAnnotatedType, isAnnotatedTypeOfPrimitive } from "@atscript/typescript";
 import { MongoClient, ObjectId } from "mongodb";
 
 //#region packages/mongo/src/plugin/primitives.ts
@@ -82,6 +82,15 @@ const annotations = { mongo: {
 			});
 		}
 	}),
+	autoIndexes: new AnnotationSpec({
+		description: "Switch on/off the automatic index creation. Works with as-mongo moost controller.\n\nDefault: true",
+		nodeType: ["interface"],
+		argument: {
+			name: "type",
+			type: "boolean",
+			description: "On/Off the automatic index creation"
+		}
+	}),
 	index: {
 		plain: new AnnotationSpec({
 			description: "Defines a **standard MongoDB index** on a field.\n\n- Improves query performance on indexed fields.\n- Can be used for **single-field** or **compound** indexes.\n\n**Example:**\n```atscript\n@mongo.index.plain \"departmentIndex\"\ndepartment: string\n```\n",
@@ -254,6 +263,28 @@ const annotations = { mongo: {
 			});
 			return errors;
 		}
+	}) },
+	array: { uniqueItems: new AnnotationSpec({
+		description: "Marks an **array field** as containing *globally unique items* when handling **patch `$insert` operations**.\n\n- Forces the patcher to use **set-semantics** (`$setUnion`) instead of a plain append, so duplicates are silently skipped.\n- Has **no effect** on `$replace`, `$update`, or `$remove`.\n- If the array’s element type already defines one or more `@meta.isKey` properties, *uniqueness is implied* and this annotation is unnecessary (but harmless).\n\n**Example:**\n```atscript\n@mongo.array.uniqueItems\ntags: string[]\n\n// Later in a patch payload …\n{\n  $insert: {\n    tags: [\"mongo\", \"mongo\"] // second \"mongo\" is ignored\n  }\n}\n```\n",
+		nodeType: ["prop"],
+		multiple: false,
+		validate(token, args, doc) {
+			const field = token.parentNode;
+			const errors = [];
+			const definition = field.getDefinition();
+			if (!definition) return errors;
+			let wrongType = false;
+			if (isRef(definition)) {
+				const def = doc.unwindType(definition.id, definition.chain)?.def;
+				if (!isArray(def)) wrongType = true;
+			} else if (!isArray(definition)) wrongType = true;
+			if (wrongType) errors.push({
+				message: `[mongo] type of array expected when using @mongo.array.uniqueItems`,
+				severity: 1,
+				range: token.range
+			});
+			return errors;
+		}
 	}) }
 } };
 
@@ -281,6 +312,304 @@ const NoopLogger = {
 	debug: () => {}
 };
 
+//#endregion
+//#region packages/mongo/src/lib/validate-plugins.ts
+const validateMongoIdPlugin = (ctx, def, value) => {
+	if (ctx.path === "_id" && def.type.tags.has("objectId")) return ctx.validateAnnotatedType(def, value instanceof ObjectId ? value.toString() : value);
+};
+const validateMongoUniqueArrayItemsPlugin = (ctx, def, value) => {
+	if (def.metadata.has("mongo.array.uniqueItems") && def.type.kind === "array") {
+		if (Array.isArray(value)) {
+			const separator = "▼↩";
+			const seen = new Set();
+			const keyProps = CollectionPatcher.getKeyProps(def);
+			for (const item of value) {
+				let key = "";
+				if (keyProps.size) for (const prop of keyProps) key += JSON.stringify(item[prop]) + separator;
+else key = JSON.stringify(item);
+				if (seen.has(key)) {
+					ctx.error(`Duplicate items are not allowed`);
+					return false;
+				}
+				seen.add(key);
+			}
+		}
+	}
+	return undefined;
+};
+
+//#endregion
+//#region packages/mongo/src/lib/collection-patcher.ts
+function _define_property$2(obj, key, value) {
+	if (key in obj) Object.defineProperty(obj, key, {
+		value,
+		enumerable: true,
+		configurable: true,
+		writable: true
+	});
+else obj[key] = value;
+	return obj;
+}
+var CollectionPatcher = class CollectionPatcher {
+	/**
+	* Extract a set of *key properties* (annotated with `@meta.isKey`) from an
+	* array‐of‐objects type definition. These keys uniquely identify an element
+	* inside the array and are later used for `$update`, `$remove` and `$upsert`.
+	*
+	* @param def Atscript array type
+	* @returns Set of property names marked as keys; empty set if none
+	*/ static getKeyProps(def) {
+		if (def.type.of.type.kind === "object") {
+			const objType = def.type.of.type;
+			const keyProps = new Set();
+			for (const [key, val] of objType.props.entries()) if (val.metadata.get("meta.isKey")) keyProps.add(key);
+			return keyProps;
+		}
+		return new Set();
+	}
+	/**
+	* Build a runtime *Validator* that understands the extended patch payload.
+	*
+	*  * Adds per‑array *patch* wrappers (the `$replace`, `$insert`, … fields).
+	*  * Honors `mongo.patch.strategy === "merge"` metadata.
+	*
+	* @param collection Target collection wrapper
+	* @returns Atscript Validator
+	*/ static prepareValidator(collection) {
+		return collection.type.validator({
+			plugins: [validateMongoIdPlugin, validateMongoUniqueArrayItemsPlugin],
+			replace: (def, path) => {
+				if (path === "" && def.type.kind === "object") {
+					const obj = defineAnnotatedType("object").copyMetadata(def.metadata);
+					for (const [prop, type] of def.type.props.entries()) obj.prop(prop, defineAnnotatedType().refTo(type).copyMetadata(type.metadata).optional(prop !== "_id").$type);
+					return obj.$type;
+				}
+				if (def.type.kind === "array" && collection.flatMap.get(path)?.metadata.get("mongo.__topLevelArray") && !def.metadata.has("mongo.__patchArrayValue")) {
+					const defArray = def;
+					const mergeStrategy = defArray.metadata.get("mongo.patch.strategy") === "merge";
+					function getPatchType() {
+						const isPrimitive$1 = isAnnotatedTypeOfPrimitive(defArray.type.of);
+						if (isPrimitive$1) return defineAnnotatedType().refTo(def).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+						if (defArray.type.of.type.kind === "object") {
+							const objType = defArray.type.of.type;
+							const t = defineAnnotatedType("object").copyMetadata(defArray.type.of.metadata);
+							const keyProps = CollectionPatcher.getKeyProps(defArray);
+							for (const [key, val] of objType.props.entries()) if (keyProps.size) if (keyProps.has(key)) t.prop(key, defineAnnotatedType().refTo(val).copyMetadata(def.metadata).$type);
+else t.prop(key, defineAnnotatedType().refTo(val).copyMetadata(def.metadata).optional().$type);
+else t.prop(key, defineAnnotatedType().refTo(val).copyMetadata(def.metadata).optional(!!val.optional).$type);
+							return defineAnnotatedType("array").of(t.$type).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+						}
+						return undefined;
+					}
+					const fullType = defineAnnotatedType().refTo(def).copyMetadata(def.metadata).annotate("mongo.__patchArrayValue").optional().$type;
+					const patchType = getPatchType();
+					return patchType ? defineAnnotatedType("object").prop("$replace", fullType).prop("$insert", fullType).prop("$upsert", fullType).prop("$update", mergeStrategy ? patchType : fullType).prop("$remove", patchType).optional().$type : defineAnnotatedType("object").prop("$replace", fullType).prop("$insert", fullType).optional().$type;
+				}
+				return def;
+			},
+			partial: (def, path) => {
+				return path !== "" && def.metadata.get("mongo.patch.strategy") === "merge";
+			}
+		});
+	}
+	/**
+	* Entry point – walk the payload, build `filter`, `update` and `options`.
+	*
+	* @returns Helper object exposing both individual parts and
+	*          a `.toArgs()` convenience callback.
+	*/ preparePatch() {
+		this.filterObj = { _id: this.collection.prepareId(this.payload._id) };
+		this.flattenPayload(this.payload);
+		let updateFilter = this.updatePipeline;
+		return {
+			toArgs: () => [
+				this.filterObj,
+				updateFilter,
+				this.optionsObj
+			],
+			filter: this.filterObj,
+			updateFilter,
+			updateOptions: this.optionsObj
+		};
+	}
+	/**
+	* Helper – lazily create `$set` section and assign *key* → *value*.
+	*
+	* @param key Fully‑qualified dotted path
+	* @param val Value to be written
+	* @private
+	*/ _set(key, val) {
+		for (const pipe of this.updatePipeline) {
+			if (!pipe.$set) pipe.$set = {};
+			if (!pipe.$set[key]) {
+				pipe.$set[key] = val;
+				return;
+			}
+		}
+		this.updatePipeline.push({ $set: { [key]: val } });
+	}
+	/**
+	* Recursively walk through the patch *payload* and convert it into `$set`/…
+	* statements. Top‑level arrays are delegated to {@link parseArrayPatch}.
+	*
+	* @param payload Current payload chunk
+	* @param prefix  Dotted path accumulated so far
+	* @private
+	*/ flattenPayload(payload, prefix = "") {
+		const evalKey = (k) => prefix ? `${prefix}.${k}` : k;
+		for (const [_key, value] of Object.entries(payload)) {
+			const key = evalKey(_key);
+			const flatType = this.collection.flatMap.get(key);
+			const topLevelArray = flatType?.metadata?.get("mongo.__topLevelArray");
+			if (typeof value === "object" && topLevelArray) this.parseArrayPatch(key, value);
+else if (typeof value === "object" && this.collection.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge") this.flattenPayload(value, key);
+else if (key !== "_id") this._set(key, value);
+		}
+		return this.updatePipeline;
+	}
+	/**
+	* Dispatch a *single* array patch. Exactly one of `$replace`, `$insert`,
+	* `$upsert`, `$update`, `$remove` must be present – otherwise we throw.
+	*
+	* @param key   Dotted path to the array field
+	* @param value Payload slice for that field
+	* @private
+	*/ parseArrayPatch(key, value) {
+		const flatType = this.collection.flatMap.get(key);
+		const toRemove = value.$remove;
+		const toReplace = value.$replace;
+		const toInsert = value.$insert;
+		const toUpsert = value.$upsert;
+		const toUpdate = value.$update;
+		const keyProps = flatType?.type.kind === "array" ? CollectionPatcher.getKeyProps(flatType) : new Set();
+		this._remove(key, toRemove, keyProps);
+		this._replace(key, toReplace);
+		this._insert(key, toInsert, keyProps);
+		this._upsert(key, toUpsert, keyProps);
+		this._update(key, toUpdate, keyProps);
+	}
+	/**
+	* Build an *aggregation‐expression* that checks equality by **all** keys in
+	* `keys`.  Example output for keys `["id", "lang"]` and bases `a`, `b`:
+	* ```json
+	* { "$and": [ { "$eq": ["$$a.id", "$$b.id"] }, { "$eq": ["$$a.lang", "$$b.lang"] } ] }
+	* ```
+	*
+	* @param keys  Ordered list of key property names
+	* @param left  Base token for *left* expression (e.g. `"$$el"`)
+	* @param right Base token for *right* expression (e.g. `"$$this"`)
+	*/ _keysEqual(keys, left, right) {
+		return keys.map((k) => ({ $eq: [`${left}.${k}`, `${right}.${k}`] })).reduce((acc, cur) => acc ? { $and: [acc, cur] } : cur);
+	}
+	/**
+	* `$replace` – overwrite the entire array with `input`.
+	*
+	* @param key   Dotted path to the array
+	* @param input New array value (may be `undefined`)
+	* @private
+	*/ _replace(key, input) {
+		if (input) this._set(key, input);
+	}
+	/**
+	* `$insert`
+	* - plain append      → $concatArrays
+	* - unique / keyed    → delegate to _upsert (insert-or-update)
+	*/ _insert(key, input, keyProps) {
+		if (!input?.length) return;
+		const uniqueItems = this.collection.flatMap.get(key)?.metadata?.has("mongo.array.uniqueItems");
+		if (uniqueItems || keyProps.size > 0) this._upsert(key, input, keyProps);
+else this._set(key, { $concatArrays: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$upsert`
+	* - keyed  → remove existing matching by key(s) then append candidate
+	* - unique → $setUnion (deep equality)
+	*/ _upsert(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const keys = [...keyProps];
+			this._set(key, { $reduce: {
+				input,
+				initialValue: { $ifNull: [`$${key}`, []] },
+				in: { $let: {
+					vars: {
+						acc: "$$value",
+						cand: "$$this"
+					},
+					in: { $concatArrays: [{ $filter: {
+						input: "$$acc",
+						as: "el",
+						cond: { $not: this._keysEqual(keys, "$$el", "$$cand") }
+					} }, ["$$cand"]] }
+				} }
+			} });
+			return;
+		}
+		this._set(key, { $setUnion: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$update`
+	* - keyed       → map array and merge / replace matching element(s)
+	* - non-keyed   → behave like `$addToSet` (insert only when not present)
+	*/ _update(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const mergeStrategy = this.collection.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge";
+			const keys = [...keyProps];
+			this._set(key, { $reduce: {
+				input,
+				initialValue: { $ifNull: [`$${key}`, []] },
+				in: { $map: {
+					input: "$$value",
+					as: "el",
+					in: { $cond: [
+						this._keysEqual(keys, "$$el", "$$this"),
+						mergeStrategy ? { $mergeObjects: ["$$el", "$$this"] } : "$$this",
+						"$$el"
+					] }
+				} }
+			} });
+		} else this._set(key, { $setUnion: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	/**
+	* `$remove`
+	* - keyed     → filter out any element whose key set matches a payload item
+	* - non-keyed → deep equality remove (`$setDifference`)
+	*/ _remove(key, input, keyProps) {
+		if (!input?.length) return;
+		if (keyProps.size) {
+			const keys = [...keyProps];
+			this._set(key, { $let: {
+				vars: { rem: input },
+				in: { $filter: {
+					input: { $ifNull: [`$${key}`, []] },
+					as: "el",
+					cond: { $not: { $anyElementTrue: { $map: {
+						input: "$$rem",
+						as: "r",
+						in: this._keysEqual(keys, "$$el", "$$r")
+					} } } }
+				} }
+			} });
+		} else this._set(key, { $setDifference: [{ $ifNull: [`$${key}`, []] }, input] });
+	}
+	constructor(collection, payload) {
+		_define_property$2(this, "collection", void 0);
+		_define_property$2(this, "payload", void 0);
+		/**
+		* Internal accumulator: filter passed to `updateOne()`.
+		* Filled only with the `_id` field right now.
+		*/ _define_property$2(this, "filterObj", void 0);
+		/** MongoDB *update* document being built. */ _define_property$2(this, "updatePipeline", void 0);
+		/** Additional *options* (mainly `arrayFilters`). */ _define_property$2(this, "optionsObj", void 0);
+		this.collection = collection;
+		this.payload = payload;
+		this.filterObj = {};
+		this.updatePipeline = [];
+		this.optionsObj = {};
+	}
+};
+
 //#endregion
 //#region packages/mongo/src/lib/as-collection.ts
 function _define_property$1(obj, key, value) {
@@ -346,17 +675,24 @@ var AsCollection = class {
 	*/ getValidator(purpose) {
 		if (!this.validators.has(purpose)) switch (purpose) {
 			case "insert": {
-				this.validators.set(purpose, this.type.validator(this.idType === "objectId" ? { skipList: new Set(["_id"]) } : {}));
+				this.validators.set(purpose, this.type.validator({
+					plugins: [validateMongoIdPlugin, validateMongoUniqueArrayItemsPlugin],
+					replace(type, path) {
+						if (path === "_id" && type.type.tags.has("objectId")) return {
+							...type,
+							optional: true
+						};
+						return type;
+					}
+				}));
 				break;
 			}
 			case "update": {
-				this.validators.set(purpose, this.type.validator());
+				this.validators.set(purpose, this.type.validator({ plugins: [validateMongoIdPlugin] }));
 				break;
 			}
 			case "patch": {
-				this.validators.set(purpose, this.type.validator({ partial: (def, path) => {
-					return path === "" || def.metadata.get("mongo.patch.strategy") === "merge";
-				} }));
+				this.validators.set(purpose, CollectionPatcher.prepareValidator(this));
 				break;
 			}
 			default: throw new Error(`Unknown validator purpose: ${purpose}`);
@@ -407,20 +743,24 @@ else {
 			if (analyzer) index.definition.mappings.fields[fieldName].analyzer = analyzer;
 		}
 	}
-	_flattenType(type, prefix) {
+	_flattenType(type, prefix = "", inComplexTypeOrArray = false) {
 		switch (type.type.kind) {
 			case "object":
 				this._flatMap?.set(prefix || "", type);
 				const items = Array.from(type.type.props.entries());
-				for (const [key, value] of items) this._flattenType(value, prefix ? `${prefix}.${key}` : key);
+				for (const [key, value] of items) this._flattenType(value, prefix ? `${prefix}.${key}` : key, inComplexTypeOrArray);
 				break;
 			case "array":
-				this._flatMap?.set(prefix || "", type);
-				if (type.type.of.type.kind) this._flattenType(type.type.of, prefix);
+				let typeArray = type;
+				if (!inComplexTypeOrArray) {
+					typeArray = defineAnnotatedType().refTo(type).copyMetadata(type.metadata).$type;
+					typeArray.metadata.set("mongo.__topLevelArray", true);
+				}
+				this._flatMap?.set(prefix || "", typeArray);
 				break;
 			case "intersection":
 			case "tuple":
-			case "union": for (const item of type.type.items) this._flattenType(item, prefix);
+			case "union": for (const item of type.type.items) this._flattenType(item, prefix, true);
 			default:
 				this._flatMap?.set(prefix || "", type);
 				break;
@@ -441,6 +781,9 @@ else {
 			text: { fuzzy: { maxEdits: textSearch.fuzzy || 0 } }
 		});
 	}
+	get uniqueProps() {
+		return this._uniqueProps;
+	}
 	_finalizeIndexesForCollection() {
 		for (const [key, value] of Array.from(this._vectorFilters.entries())) {
 			const index = this._indexes.get(key);
@@ -449,6 +792,10 @@ else {
 				path: value
 			});
 		}
+		for (const [, value] of Array.from(this._indexes.entries())) if (value.type === "unique") {
+			const keys = Object.keys(value.fields);
+			if (keys.length === 1) this._uniqueProps.add(keys[0]);
+		}
 	}
 	_prepareIndexesForField(fieldName, metadata) {
 		for (const index of metadata.get("mongo.index.plain") || []) this._addIndexField("plain", index === true ? fieldName : index, fieldName);
@@ -564,39 +911,62 @@ else toUpdate.add(remote.name);
 			default:
 		}
 	}
+	insert(payload, options) {
+		const toInsert = this.prepareInsert(payload);
+		return Array.isArray(toInsert) ? this.collection.insertMany(toInsert, options) : this.collection.insertOne(toInsert, options);
+	}
+	replace(payload, options) {
+		const [filter, replace, opts] = this.prepareReplace(payload).toArgs();
+		return this.collection.replaceOne(filter, replace, {
+			...opts,
+			...options
+		});
+	}
+	update(payload, options) {
+		const [filter, update, opts] = this.prepareUpdate(payload).toArgs();
+		return this.collection.updateOne(filter, update, {
+			...opts,
+			...options
+		});
+	}
 	prepareInsert(payload) {
 		const v = this.getValidator("insert");
-		if (v.validate(payload)) {
-			const data = { ...payload };
+		const arr = Array.isArray(payload) ? payload : [payload];
+		const prepared = [];
+		for (const item of arr) if (v.validate(item)) {
+			const data = { ...item };
 			if (data._id) data._id = this.prepareId(data._id);
 else if (this.idType !== "objectId") throw new Error("Missing \"_id\" field");
-			return data;
-		}
-		throw new Error("Invalid payload");
+			prepared.push(data);
+		} else throw new Error("Invalid payload");
+		return prepared.length === 1 ? prepared[0] : prepared;
 	}
-	prepareUpdate(payload) {
-		const v = this.getValidator("insert");
+	prepareReplace(payload) {
+		const v = this.getValidator("update");
 		if (v.validate(payload)) {
-			const data = { ...payload };
-			data._id = this.prepareId(data._id);
-			return data;
+			const _id = this.prepareId(payload._id);
+			const data = {
+				...payload,
+				_id
+			};
+			return {
+				toArgs: () => [
+					{ _id },
+					data,
+					{}
+				],
+				filter: { _id },
+				updateFilter: data,
+				updateOptions: {}
+			};
 		}
 		throw new Error("Invalid payload");
 	}
-	preparePatch(payload) {
+	prepareUpdate(payload) {
 		const v = this.getValidator("patch");
-		if (v.validate(payload)) return { $set: this._flattenPayload(payload) };
+		if (v.validate(payload)) return new CollectionPatcher(this, payload).preparePatch();
 		throw new Error("Invalid payload");
 	}
-	_flattenPayload(payload, prefix = "", obj = {}) {
-		const evalKey = (k) => prefix ? `${prefix}.${k}` : k;
-		for (const [_key, value] of Object.entries(payload)) {
-			const key = evalKey(_key);
-			if (typeof value === "object" && this.flatMap.get(key)?.metadata?.get("mongo.patch.strategy") === "merge") this._flattenPayload(value, key, obj);
-else obj[key] = value;
-		}
-		return obj;
-	}
 	constructor(asMongo, _type, logger = NoopLogger) {
 		_define_property$1(this, "asMongo", void 0);
 		_define_property$1(this, "_type", void 0);
@@ -607,12 +977,14 @@ else obj[key] = value;
 		_define_property$1(this, "_indexes", void 0);
 		_define_property$1(this, "_vectorFilters", void 0);
 		_define_property$1(this, "_flatMap", void 0);
+		_define_property$1(this, "_uniqueProps", void 0);
 		this.asMongo = asMongo;
 		this._type = _type;
 		this.logger = logger;
 		this.validators = new Map();
 		this._indexes = new Map();
 		this._vectorFilters = new Map();
+		this._uniqueProps = new Set();
 		if (!isAnnotatedType(_type)) throw new Error("Atscript Annotated Type expected");
 		const name = _type.metadata.get("mongo.collection");
 		if (!name) throw new Error("@mongo.collection annotation expected with collection name");
@@ -691,13 +1063,20 @@ var AsMongo = class {
 		return list.has(name);
 	}
 	getCollection(type, logger) {
-		return new AsCollection(this, type, logger || this.logger);
+		let collection = this._collections.get(type);
+		if (!collection) {
+			collection = new AsCollection(this, type, logger || this.logger);
+			this._collections.set(type, collection);
+		}
+		return collection;
 	}
 	constructor(client, logger = NoopLogger) {
 		_define_property(this, "logger", void 0);
 		_define_property(this, "client", void 0);
 		_define_property(this, "collectionsList", void 0);
+		_define_property(this, "_collections", void 0);
 		this.logger = logger;
+		this._collections = new WeakMap();
 		if (typeof client === "string") this.client = new MongoClient(client);
 else this.client = client;
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atscript/mongo",
-  "version": "0.0.17",
+  "version": "0.0.19",
   "description": "Mongodb plugin for atscript.",
   "type": "module",
   "main": "dist/index.mjs",
@@ -14,8 +14,7 @@
     "./package.json": "./package.json"
   },
   "files": [
-    "dist",
-    "cli.cjs"
+    "dist"
   ],
   "keywords": [
     "atscript",
@@ -33,13 +32,13 @@
   },
   "homepage": "https://github.com/moostjs/atscript/tree/main/packages/mongo#readme",
   "license": "ISC",
-  "dependencies": {
-    "mongodb": "^6.13.0",
-    "@atscript/core": "^0.0.17",
-    "@atscript/typescript": "^0.0.17"
+  "peerDependencies": {
+    "mongodb": "^6.17.0",
+    "@atscript/core": "^0.0.19",
+    "@atscript/typescript": "^0.0.19"
   },
   "devDependencies": {
-    "vitest": "^3.0.0"
+    "vitest": "3.2.4"
   },
   "scripts": {
     "pub": "pnpm publish --access public",