@atscript/typescript 0.1.32 → 0.1.33

package/dist/index.mjs

@@ -455,28 +455,72 @@ else {
 	*
 	* - **Single PK** (one `@meta.id`) → `static __pk: <scalar type>`
 	* - **Compound PK** (multiple `@meta.id`) → `static __pk: { field1: Type1; field2: Type2 }`
-	* - **No PK** → no `__pk` emitted
+	* - **No PK** → no `__pk` emitted (unless unique indexes exist)
+	* - **Unique indexes** (`@db.index.unique`) → appended as union members
+	* - **Mongo collection** → always includes `string` (ObjectId) in the union;
+	*   if no `@meta.id` fields, `__pk` is just `string`
 	*/ renderPk(node) {
+		const isMongoCollection = !!node.annotations?.some((a) => a.name === "db.mongo.collection");
 		let struct;
 		if (node.hasExtends) struct = this.doc.resolveInterfaceExtends(node);
 		if (!struct) struct = node.getDefinition();
 		if (!struct || !isStructure(struct)) return;
+		const structNode = struct;
 		const pkProps = [];
-		for (const [name, prop] of struct.props) {
+		const uniqueByIndex = new Map();
+		for (const [name, prop] of structNode.props) {
 			if (prop.token("identifier")?.pattern) continue;
+			if (isMongoCollection && name === "_id") continue;
 			if (prop.countAnnotations("meta.id") > 0) pkProps.push({
 				name,
 				prop
 			});
+			if (prop.annotations) {
+				for (const ann of prop.annotations) if (ann.name === "db.index.unique") {
+					const indexName = ann.args[0]?.text ?? name;
+					let group = uniqueByIndex.get(indexName);
+					if (!group) {
+						group = [];
+						uniqueByIndex.set(indexName, group);
+					}
+					group.push({
+						name,
+						prop
+					});
+				}
+			}
+		}
+		const uniqueProps = [];
+		const pkNames = new Set(pkProps.map((p) => p.name));
+		for (const fields of uniqueByIndex.values()) if (fields.length === 1 && !pkNames.has(fields[0].name)) uniqueProps.push(fields[0]);
+		if (pkProps.length === 0 && uniqueProps.length === 0 && !isMongoCollection) return;
+		let mongoIdType;
+		if (isMongoCollection) {
+			const idProp = structNode.props.get("_id");
+			if (idProp) mongoIdType = this.renderTypeDefString(idProp.getDefinition()).trim();
+			mongoIdType ?? (mongoIdType = "string");
+		}
+		const uniqueTypes = [];
+		const seenTypes = new Set();
+		for (const { prop } of uniqueProps) {
+			const rendered = this.renderTypeDefString(prop.getDefinition()).trim();
+			if (!seenTypes.has(rendered)) {
+				seenTypes.add(rendered);
+				uniqueTypes.push(rendered);
+			}
 		}
-		if (pkProps.length === 0) return;
 		this.writeln();
-		if (pkProps.length === 1) {
+		const uniqueSuffix = uniqueTypes.length > 0 ? ` | ${uniqueTypes.join(" | ")}` : "";
+		if (pkProps.length === 0 && !isMongoCollection) this.writeln(`static __pk: ${uniqueTypes.join(" | ")}`);
+else if (pkProps.length === 0) this.writeln(`static __pk: ${mongoIdType}${uniqueSuffix}`);
+else if (pkProps.length === 1) {
 			this.write("static __pk: ");
-			const renderedDef = this.renderTypeDefString(pkProps[0].prop.getDefinition());
-			renderedDef.split("\n").forEach((l) => this.writeln(l));
+			if (isMongoCollection) this.write(`${mongoIdType} | `);
+			const renderedDef = this.renderTypeDefString(pkProps[0].prop.getDefinition()).trim();
+			this.writeln(`${renderedDef}${uniqueSuffix}`);
 		} else {
 			this.write("static __pk: ");
+			if (isMongoCollection) this.write(`${mongoIdType} | `);
 			this.blockln("{}");
 			for (const { name, prop } of pkProps) {
 				this.write(wrapProp(name), ": ");
@@ -484,6 +528,7 @@ else {
 				renderedDef.split("\n").forEach((l) => this.writeln(l));
 			}
 			this.pop();
+			this.writeln(uniqueSuffix);
 		}
 	}
 	phantomPropType(def) {
@@ -971,6 +1016,24 @@ var ValidatorError = class extends Error {
 
 //#endregion
 //#region packages/typescript/src/annotated-type.ts
+const COMPLEX_KINDS = new Set([
+	"union",
+	"intersection",
+	"tuple"
+]);
+/** Shared validator method reused by all annotated type nodes. */ function validatorMethod(opts) {
+	return new Validator(this, opts);
+}
+function createAnnotatedTypeNode(type, metadata, opts) {
+	return {
+		__is_atscript_annotated_type: true,
+		type,
+		metadata,
+		validator: validatorMethod,
+		id: opts?.id,
+		optional: opts?.optional
+	};
+}
 function isAnnotatedType(type) {
 	return type && type.__is_atscript_annotated_type;
 }
@@ -987,38 +1050,24 @@ function defineAnnotatedType(_kind, base) {
 	const kind = _kind || "";
 	const type = base?.type || {};
 	type.kind = kind;
-	if ([
-		"union",
-		"intersection",
-		"tuple"
-	].includes(kind)) type.items = [];
+	if (COMPLEX_KINDS.has(kind)) type.items = [];
 	if (kind === "object") {
 		type.props = new Map();
 		type.propsPatterns = [];
 	}
 	type.tags = new Set();
 	const metadata = base?.metadata || new Map();
-	if (base) Object.assign(base, {
+	const payload = {
 		__is_atscript_annotated_type: true,
 		metadata,
 		type,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
-	});
-else base = {
-		__is_atscript_annotated_type: true,
-		metadata,
-		type,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
+		validator: validatorMethod
 	};
+	base = base ? Object.assign(base, payload) : payload;
 	const handle = {
 		$type: base,
 		$def: type,
 		$metadata: metadata,
-		_existingObject: undefined,
 		tags(...tags) {
 			for (const tag of tags) this.$def.tags.add(tag);
 			return this;
@@ -1059,27 +1108,18 @@ else base = {
 			return this;
 		},
 		refTo(type$1, chain) {
+			if (!isAnnotatedType(type$1)) throw new Error(`${type$1} is not annotated type`);
 			let newBase = type$1;
 			const typeName = type$1.name || "Unknown";
-			if (isAnnotatedType(newBase)) {
-				let keys = "";
-				for (const c of chain || []) {
-					keys += `["${c}"]`;
-					if (newBase.type.kind === "object" && newBase.type.props.has(c)) newBase = newBase.type.props.get(c);
-else throw new Error(`Can't find prop ${typeName}${keys}`);
+			if (chain) for (let i = 0; i < chain.length; i++) {
+				const c = chain[i];
+				if (newBase.type.kind === "object" && newBase.type.props.has(c)) newBase = newBase.type.props.get(c);
+else {
+					const keys = chain.slice(0, i + 1).map((k) => `["${k}"]`).join("");
+					throw new Error(`Can't find prop ${typeName}${keys}`);
 				}
-				if (!newBase && keys) throw new Error(`Can't find prop ${typeName}${keys}`);
-else if (!newBase) throw new Error(`"${typeName}" is not annotated type`);
-				this.$type = {
-					__is_atscript_annotated_type: true,
-					type: newBase.type,
-					metadata,
-					id: newBase.id,
-					validator(opts) {
-						return new Validator(this, opts);
-					}
-				};
-			} else throw new Error(`${type$1} is not annotated type`);
+			}
+			this.$type = createAnnotatedTypeNode(newBase.type, metadata, { id: newBase.id });
 			return this;
 		},
 		annotate(key, value, asArray) {
@@ -1129,10 +1169,10 @@ function forAnnotatedType(def, handlers) {
 	const firstObj = items[0].type;
 	const candidates = [];
 	for (const [propName, propType] of firstObj.props.entries()) if (propType.type.kind === "" && propType.type.value !== undefined) candidates.push(propName);
-	const validCandidates = [];
+	let result = null;
 	for (const candidate of candidates) {
 		const values = new Set();
-		const mapping = {};
+		const indexMapping = {};
 		let valid = true;
 		for (let i = 0; i < items.length; i++) {
 			const obj = items[i].type;
@@ -1147,19 +1187,21 @@ function forAnnotatedType(def, handlers) {
 				break;
 			}
 			values.add(val);
-			mapping[String(val)] = `#/oneOf/${i}`;
+			indexMapping[String(val)] = i;
+		}
+		if (valid) {
+			if (result) return null;
+			result = {
+				propertyName: candidate,
+				indexMapping
+			};
 		}
-		if (valid) validCandidates.push({
-			propertyName: candidate,
-			mapping
-		});
 	}
-	if (validCandidates.length === 1) return validCandidates[0];
-	return null;
+	return result;
 }
 function buildJsonSchema(type) {
 	const defs = {};
-	let isRoot = true;
+	let hasDefs = false;
 	const buildObject = (d) => {
 		const properties = {};
 		const required = [];
@@ -1176,15 +1218,15 @@ function buildJsonSchema(type) {
 		return schema$1;
 	};
 	const build = (def) => {
-		if (def.id && def.type.kind === "object" && !isRoot) {
+		if (def.id && def.type.kind === "object" && def !== type) {
 			const name = def.id;
 			if (!defs[name]) {
+				hasDefs = true;
 				defs[name] = {};
 				defs[name] = buildObject(def);
 			}
 			return { $ref: `#/$defs/${name}` };
 		}
-		isRoot = false;
 		const meta = def.metadata;
 		return forAnnotatedType(def, {
 			phantom() {
@@ -1209,11 +1251,9 @@ function buildJsonSchema(type) {
 				if (disc) {
 					const oneOf = d.type.items.map(build);
 					const mapping = {};
-					for (const [val, origPath] of Object.entries(disc.mapping)) {
-						const idx = Number.parseInt(origPath.split("/").pop(), 10);
+					for (const [val, idx] of Object.entries(disc.indexMapping)) {
 						const item = d.type.items[idx];
-						if (item.id && defs[item.id]) mapping[val] = `#/$defs/${item.id}`;
-else mapping[val] = origPath;
+						mapping[val] = item.id && defs[item.id] ? `#/$defs/${item.id}` : `#/oneOf/${idx}`;
 					}
 					return {
 						oneOf,
@@ -1263,7 +1303,7 @@ else schema$1.allOf = (schema$1.allOf || []).concat(patterns.map((p) => ({ patte
 		});
 	};
 	const schema = build(type);
-	if (Object.keys(defs).length > 0) return {
+	if (hasDefs) return {
 		...schema,
 		$defs: defs
 	};

package/dist/utils.cjs

@@ -436,6 +436,25 @@ var ValidatorError = class extends Error {
 
 //#endregion
 //#region packages/typescript/src/annotated-type.ts
+const COMPLEX_KINDS = new Set([
+	"union",
+	"intersection",
+	"tuple"
+]);
+const NON_PRIMITIVE_KINDS = new Set(["array", "object"]);
+/** Shared validator method reused by all annotated type nodes. */ function validatorMethod(opts) {
+	return new Validator(this, opts);
+}
+function createAnnotatedTypeNode(type, metadata, opts) {
+	return {
+		__is_atscript_annotated_type: true,
+		type,
+		metadata,
+		validator: validatorMethod,
+		id: opts?.id,
+		optional: opts?.optional
+	};
+}
 function isAnnotatedType(type) {
 	return type && type.__is_atscript_annotated_type;
 }
@@ -454,32 +473,22 @@ function cloneRefProp(parentType, propName) {
 	const existing = objType.props.get(propName);
 	if (!existing) return;
 	const clonedType = cloneTypeDef(existing.type);
-	objType.props.set(propName, {
-		__is_atscript_annotated_type: true,
-		type: clonedType,
-		metadata: new Map(existing.metadata),
+	objType.props.set(propName, createAnnotatedTypeNode(clonedType, new Map(existing.metadata), {
 		id: existing.id,
-		optional: existing.optional,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
-	});
+		optional: existing.optional
+	}));
 }
 function cloneTypeDef(type) {
 	if (type.kind === "object") {
 		const obj = type;
+		const props = new Map();
+		for (const [k, v] of obj.props) props.set(k, createAnnotatedTypeNode(v.type, new Map(v.metadata), {
+			id: v.id,
+			optional: v.optional
+		}));
 		return {
 			kind: "object",
-			props: new Map(Array.from(obj.props.entries()).map(([k, v]) => [k, {
-				__is_atscript_annotated_type: true,
-				type: v.type,
-				metadata: new Map(v.metadata),
-				id: v.id,
-				optional: v.optional,
-				validator(opts) {
-					return new Validator(this, opts);
-				}
-			}])),
+			props,
 			propsPatterns: [...obj.propsPatterns],
 			tags: new Set(obj.tags)
 		};
@@ -509,38 +518,24 @@ function defineAnnotatedType(_kind, base) {
 	const kind = _kind || "";
 	const type = base?.type || {};
 	type.kind = kind;
-	if ([
-		"union",
-		"intersection",
-		"tuple"
-	].includes(kind)) type.items = [];
+	if (COMPLEX_KINDS.has(kind)) type.items = [];
 	if (kind === "object") {
 		type.props = new Map();
 		type.propsPatterns = [];
 	}
 	type.tags = new Set();
 	const metadata = base?.metadata || new Map();
-	if (base) Object.assign(base, {
-		__is_atscript_annotated_type: true,
-		metadata,
-		type,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
-	});
-else base = {
+	const payload = {
 		__is_atscript_annotated_type: true,
 		metadata,
 		type,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
+		validator: validatorMethod
 	};
+	base = base ? Object.assign(base, payload) : payload;
 	const handle = {
 		$type: base,
 		$def: type,
 		$metadata: metadata,
-		_existingObject: undefined,
 		tags(...tags) {
 			for (const tag of tags) this.$def.tags.add(tag);
 			return this;
@@ -581,27 +576,18 @@ else base = {
 			return this;
 		},
 		refTo(type$1, chain) {
+			if (!isAnnotatedType(type$1)) throw new Error(`${type$1} is not annotated type`);
 			let newBase = type$1;
 			const typeName = type$1.name || "Unknown";
-			if (isAnnotatedType(newBase)) {
-				let keys = "";
-				for (const c of chain || []) {
-					keys += `["${c}"]`;
-					if (newBase.type.kind === "object" && newBase.type.props.has(c)) newBase = newBase.type.props.get(c);
-else throw new Error(`Can't find prop ${typeName}${keys}`);
+			if (chain) for (let i = 0; i < chain.length; i++) {
+				const c = chain[i];
+				if (newBase.type.kind === "object" && newBase.type.props.has(c)) newBase = newBase.type.props.get(c);
+else {
+					const keys = chain.slice(0, i + 1).map((k) => `["${k}"]`).join("");
+					throw new Error(`Can't find prop ${typeName}${keys}`);
 				}
-				if (!newBase && keys) throw new Error(`Can't find prop ${typeName}${keys}`);
-else if (!newBase) throw new Error(`"${typeName}" is not annotated type`);
-				this.$type = {
-					__is_atscript_annotated_type: true,
-					type: newBase.type,
-					metadata,
-					id: newBase.id,
-					validator(opts) {
-						return new Validator(this, opts);
-					}
-				};
-			} else throw new Error(`${type$1} is not annotated type`);
+			}
+			this.$type = createAnnotatedTypeNode(newBase.type, metadata, { id: newBase.id });
 			return this;
 		},
 		annotate(key, value, asArray) {
@@ -619,13 +605,9 @@ function isPhantomType(def) {
 	return def.type.kind === "" && def.type.designType === "phantom";
 }
 function isAnnotatedTypeOfPrimitive(t) {
-	if (["array", "object"].includes(t.type.kind)) return false;
+	if (NON_PRIMITIVE_KINDS.has(t.type.kind)) return false;
 	if (!t.type.kind) return true;
-	if ([
-		"union",
-		"tuple",
-		"intersection"
-	].includes(t.type.kind)) {
+	if (COMPLEX_KINDS.has(t.type.kind)) {
 		for (const item of t.type.items) if (!isAnnotatedTypeOfPrimitive(item)) return false;
 		return true;
 	}
@@ -664,10 +646,10 @@ function forAnnotatedType(def, handlers) {
 	const firstObj = items[0].type;
 	const candidates = [];
 	for (const [propName, propType] of firstObj.props.entries()) if (propType.type.kind === "" && propType.type.value !== undefined) candidates.push(propName);
-	const validCandidates = [];
+	let result = null;
 	for (const candidate of candidates) {
 		const values = new Set();
-		const mapping = {};
+		const indexMapping = {};
 		let valid = true;
 		for (let i = 0; i < items.length; i++) {
 			const obj = items[i].type;
@@ -682,19 +664,21 @@ function forAnnotatedType(def, handlers) {
 				break;
 			}
 			values.add(val);
-			mapping[String(val)] = `#/oneOf/${i}`;
+			indexMapping[String(val)] = i;
+		}
+		if (valid) {
+			if (result) return null;
+			result = {
+				propertyName: candidate,
+				indexMapping
+			};
 		}
-		if (valid) validCandidates.push({
-			propertyName: candidate,
-			mapping
-		});
 	}
-	if (validCandidates.length === 1) return validCandidates[0];
-	return null;
+	return result;
 }
 function buildJsonSchema(type) {
 	const defs = {};
-	let isRoot = true;
+	let hasDefs = false;
 	const buildObject = (d) => {
 		const properties = {};
 		const required = [];
@@ -711,15 +695,15 @@ function buildJsonSchema(type) {
 		return schema$1;
 	};
 	const build$1 = (def) => {
-		if (def.id && def.type.kind === "object" && !isRoot) {
+		if (def.id && def.type.kind === "object" && def !== type) {
 			const name = def.id;
 			if (!defs[name]) {
+				hasDefs = true;
 				defs[name] = {};
 				defs[name] = buildObject(def);
 			}
 			return { $ref: `#/$defs/${name}` };
 		}
-		isRoot = false;
 		const meta = def.metadata;
 		return forAnnotatedType(def, {
 			phantom() {
@@ -744,11 +728,9 @@ function buildJsonSchema(type) {
 				if (disc) {
 					const oneOf = d.type.items.map(build$1);
 					const mapping = {};
-					for (const [val, origPath] of Object.entries(disc.mapping)) {
-						const idx = Number.parseInt(origPath.split("/").pop(), 10);
+					for (const [val, idx] of Object.entries(disc.indexMapping)) {
 						const item = d.type.items[idx];
-						if (item.id && defs[item.id]) mapping[val] = `#/$defs/${item.id}`;
-else mapping[val] = origPath;
+						mapping[val] = item.id && defs[item.id] ? `#/$defs/${item.id}` : `#/oneOf/${idx}`;
 					}
 					return {
 						oneOf,
@@ -798,7 +780,7 @@ else schema$1.allOf = (schema$1.allOf || []).concat(patterns.map((p) => ({ patte
 		});
 	};
 	const schema = build$1(type);
-	if (Object.keys(defs).length > 0) return {
+	if (hasDefs) return {
 		...schema,
 		$defs: defs
 	};
@@ -813,6 +795,8 @@ function fromJsonSchema(schema) {
 			const refName = s.$ref.replace(/^#\/(\$defs|definitions)\//, "");
 			if (resolved.has(refName)) return resolved.get(refName);
 			if (defsSource[refName]) {
+				const placeholder = defineAnnotatedType().designType("any").$type;
+				resolved.set(refName, placeholder);
 				const type = convert(defsSource[refName]);
 				resolved.set(refName, type);
 				return type;
@@ -1223,16 +1207,10 @@ function deserializeAnnotatedType(data) {
 function deserializeNode(data) {
 	const metadata = new Map(Object.entries(data.metadata));
 	const type = deserializeTypeDef(data.type);
-	const result = {
-		__is_atscript_annotated_type: true,
-		type,
-		metadata,
-		validator(opts) {
-			return new Validator(this, opts);
-		}
-	};
-	if (data.optional) result.optional = true;
-	if (data.id) result.id = data.id;
+	const result = createAnnotatedTypeNode(type, metadata, {
+		optional: data.optional || undefined,
+		id: data.id || undefined
+	});
 	return result;
 }
 function deserializeTypeDef(t) {
@@ -1284,6 +1262,7 @@ exports.ValidatorError = ValidatorError
 exports.annotate = annotate
 exports.buildJsonSchema = buildJsonSchema
 exports.cloneRefProp = cloneRefProp
+exports.createAnnotatedTypeNode = createAnnotatedTypeNode
 exports.createDataFromAnnotatedType = createDataFromAnnotatedType
 exports.defineAnnotatedType = defineAnnotatedType
 exports.deserializeAnnotatedType = deserializeAnnotatedType

package/dist/utils.d.ts

@@ -96,6 +96,14 @@ declare class ValidatorError extends Error {
     constructor(errors: TError[]);
 }
 
+/**
+ * Creates a minimal annotated type node from a type def and metadata map.
+ * Centralises the shape so callers never duplicate the boilerplate.
+ */
+declare function createAnnotatedTypeNode(type: TAtscriptTypeDef, metadata: TMetadataMap<AtscriptMetadata>, opts?: {
+    id?: string;
+    optional?: boolean;
+}): TAtscriptAnnotatedType;
 /** Type definition for union, intersection, or tuple types. */
 interface TAtscriptTypeComplex<DataType = unknown> {
     kind: 'union' | 'intersection' | 'tuple';
@@ -235,7 +243,6 @@ interface TAnnotatedTypeHandle {
         kind: TKind;
     } & Omit<TAtscriptTypeComplex, 'kind'> & Omit<TAtscriptTypeFinal, 'kind'> & Omit<TAtscriptTypeArray, 'kind'> & Omit<TAtscriptTypeObject<string>, 'kind'>;
     $metadata: TMetadataMap<AtscriptMetadata>;
-    _existingObject: TAtscriptAnnotatedType | undefined;
     tags(...tags: string[]): TAnnotatedTypeHandle;
     designType(value: TAtscriptTypeFinal['designType']): TAnnotatedTypeHandle;
     value(value: string | number | boolean): TAnnotatedTypeHandle;
@@ -559,5 +566,5 @@ type PrimaryKeyOf<T> = T extends {
     __pk: infer PK;
 } ? PK : unknown;
 
-export { SERIALIZE_VERSION, Validator, ValidatorError, annotate, buildJsonSchema, cloneRefProp, createDataFromAnnotatedType, defineAnnotatedType, deserializeAnnotatedType, flattenAnnotatedType, forAnnotatedType, fromJsonSchema, isAnnotatedType, isAnnotatedTypeOfPrimitive, isPhantomType, mergeJsonSchemas, serializeAnnotatedType, throwFeatureDisabled };
+export { SERIALIZE_VERSION, Validator, ValidatorError, annotate, buildJsonSchema, cloneRefProp, createAnnotatedTypeNode, createDataFromAnnotatedType, defineAnnotatedType, deserializeAnnotatedType, flattenAnnotatedType, forAnnotatedType, fromJsonSchema, isAnnotatedType, isAnnotatedTypeOfPrimitive, isPhantomType, mergeJsonSchemas, serializeAnnotatedType, throwFeatureDisabled };
 export type { FlatOf, InferDataType, PrimaryKeyOf, TAnnotatedTypeHandle, TAtscriptAnnotatedType, TAtscriptAnnotatedTypeConstructor, TAtscriptDataType, TAtscriptTypeArray, TAtscriptTypeComplex, TAtscriptTypeDef, TAtscriptTypeFinal, TAtscriptTypeObject, TCreateDataOptions, TFlattenOptions, TJsonSchema, TMetadataMap, TProcessAnnotationContext, TSerializeOptions, TSerializedAnnotatedType, TSerializedAnnotatedTypeInner, TSerializedTypeArray, TSerializedTypeComplex, TSerializedTypeDef, TSerializedTypeFinal, TSerializedTypeObject, TValidatorOptions, TValidatorPlugin, TValidatorPluginContext, TValueResolver };