dinah 0.12.0 → 0.13.1

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { i as resolveAttrType, n as extractTableDesc, r as removeUndefined, t as chunk } from "./util-BfKXObBN.mjs";
+import { a as resolveAttrType, i as removeUndefined, n as extractTableDesc, r as matchesPartial, t as chunk } from "./util-0kc6E2VD.mjs";
 import { ConditionalCheckFailedException, CreateTableCommand, DeleteTableCommand, DynamoDB, DynamoDBClient, ListTablesCommand, TransactionCanceledException } from "@aws-sdk/client-dynamodb";
 import * as Lib from "@aws-sdk/lib-dynamodb";
 //#region src/error.ts
@@ -7,6 +7,7 @@ function dinahErrorMessage(details) {
 		case "NOT_FOUND": return `${details.resource ?? "Item"} not found: ${JSON.stringify(details.key)}`;
 		case "CONDITIONAL_CHECK_FAILED": return details.key ? `${details.resource ?? "Item"} condition check failed: ${JSON.stringify(details.key)}` : `${details.resource ?? "Item"} condition check failed`;
 		case "VALIDATION": return `Validation error: ${details.message}`;
+		case "DATA_INTEGRITY": return `Data integrity error: ${details.message}`;
 		case "TRANSACTION_CANCELED": return `Transaction canceled: ${details.reasons.map((r) => r.type).join(", ")}`;
 	}
 }
@@ -249,8 +250,8 @@ var Repo = class {
 		const t = this.table.def.name;
 		return t.charAt(0).toUpperCase() + t.slice(1);
 	}
-	get defaultPutData() {
-		return this.config.defaultPutData?.() ?? {};
+	get defaultCreateData() {
+		return this.config.defaultCreateData?.() ?? {};
 	}
 	get defaultUpdateData() {
 		return this.config.defaultUpdateData?.() ?? {};
@@ -258,9 +259,6 @@ var Repo = class {
 	transformOutput(item) {
 		return this.config.transformOutput?.(item) ?? item;
 	}
-	transformInput(item) {
-		return this.config.transformInput?.(item) ?? item;
-	}
 	extractKey(item) {
 		const { partitionKey, sortKey } = this.table.def;
 		if (sortKey) return {
@@ -270,43 +268,44 @@ var Repo = class {
 		return { [partitionKey]: item[partitionKey] };
 	}
 	async get(key, options) {
+		const { filter, ...restOptions } = options ?? {};
 		const item = await this.db.get({
 			table: this.tableName,
 			key: this.extractKey(key),
-			...options
+			filter,
+			...restOptions
 		});
 		return item && this.applyTransformIfNeeded(item, options);
 	}
 	async getOrThrow(key, options) {
+		const { filter, ...restOptions } = options ?? {};
 		const item = await this.db.getOrThrow({
 			table: this.tableName,
 			key: this.extractKey(key),
 			resource: this.resourceName,
-			...options
+			filter,
+			...restOptions
 		});
 		return this.applyTransformIfNeeded(item, options);
 	}
 	async put(item, options) {
-		const itemWithDefaults = this.applyPutTransforms(item);
 		const result = await this.db.put({
 			table: this.tableName,
-			item: itemWithDefaults,
+			item,
 			resource: this.resourceName,
 			...options
 		});
 		return this.applyTransformIfNeeded(result);
 	}
 	async update(key, update, options) {
-		const updateWithDefaults = this.applyNormalizersToExpression({
-			...this.defaultUpdateData,
-			...update
-		});
+		const updateWithDefaults = this.applyNormalizersToExpression(update);
 		const result = await this.db.update({
 			table: this.tableName,
 			key: this.extractKey(key),
 			resource: this.resourceName,
 			update: updateWithDefaults,
-			...options
+			...options,
+			condition: this.withDiscriminatorCondition(key, options?.condition)
 		});
 		return this.applyTransformIfNeeded(result);
 	}
@@ -314,10 +313,15 @@ var Repo = class {
 		const { condition: otherCondition, ...otherOptions } = options ?? {};
 		const condition = { $and: [{ [this.table.def.partitionKey]: { $exists: false } }] };
 		if (otherCondition) condition.$and.push(otherCondition);
-		return this.put(item, {
+		const normalizedItem = this.applyCreateTransforms(item);
+		const result = await this.db.put({
+			table: this.tableName,
+			item: normalizedItem,
+			resource: this.resourceName,
 			condition,
 			...otherOptions
 		});
+		return this.applyTransformIfNeeded(result);
 	}
 	async delete(key, options) {
 		const item = await this.db.delete({
@@ -375,6 +379,35 @@ var Repo = class {
 			gsi
 		});
 	}
+	async getGsi(gsi, query, options) {
+		const { filter, ...restOptions } = options ?? {};
+		const items = await this.db.query({
+			table: this.tableName,
+			index: gsi,
+			query,
+			...restOptions
+		});
+		if (items.length > 1) throw new DinahError({
+			type: "DATA_INTEGRITY",
+			message: `getGsi on "${gsi}" returned ${items.length} items; expected at most 1.`
+		});
+		const item = items[0];
+		if (!item) return void 0;
+		if (filter && !matchesPartial(filter, item)) return void 0;
+		return this.applyTransformIfNeeded(item, {
+			...options,
+			gsi
+		});
+	}
+	async getGsiOrThrow(gsi, query, options) {
+		const item = await this.getGsi(gsi, query, options);
+		if (item === void 0) throw new DinahError({
+			type: "NOT_FOUND",
+			key: query,
+			resource: this.resourceName
+		});
+		return item;
+	}
 	async scan(options) {
 		const items = await this.db.scan({
 			table: this.tableName,
@@ -425,9 +458,11 @@ var Repo = class {
 		});
 	}
 	async batchGet(keys, options) {
+		const { filter, ...restOptions } = options ?? {};
 		const { items, unprocessed } = await this.db.batchGet({ [this.tableName]: {
 			keys: keys.map((key) => this.extractKey(key)),
-			...options
+			filter,
+			...restOptions
 		} });
 		const tableItems = items[this.tableName];
 		return {
@@ -436,10 +471,12 @@ var Repo = class {
 		};
 	}
 	async batchGetOrThrow(keys, options) {
+		const { filter, ...restOptions } = options ?? {};
 		const result = await this.db.batchGetOrThrow({ [this.tableName]: {
 			keys: keys.map((key) => this.extractKey(key)),
 			resource: this.resourceName,
-			...options
+			filter,
+			...restOptions
 		} });
 		return this.applyTransformsIfNeeded(result[this.tableName] ?? [], options);
 	}
@@ -451,7 +488,7 @@ var Repo = class {
 			};
 			else return {
 				type: "PUT",
-				item: this.applyPutTransforms(request.item)
+				item: request.item
 			};
 		}) });
 		return {
@@ -460,28 +497,29 @@ var Repo = class {
 		};
 	}
 	async batchUpdate(keys, update) {
-		const updateWithDefaults = this.applyNormalizersToExpression({
-			...this.defaultUpdateData,
-			...update
-		});
+		const updateWithDefaults = this.applyNormalizersToExpression(update);
 		return { unprocessed: (await this.db.batchUpdate({ [this.tableName]: {
 			keys: keys.map((key) => this.extractKey(key)),
 			update: updateWithDefaults
 		} })).unprocessed?.[this.tableName]?.keys };
 	}
 	async trxGet(keys, options) {
+		const { filter, ...restOptions } = options ?? {};
 		return (await this.db.trxGet(...keys.map((key) => ({
 			table: this.tableName,
 			key: this.extractKey(key),
-			...options
+			filter,
+			...restOptions
 		})))).map((item) => item && this.applyTransformIfNeeded(item, options));
 	}
 	async trxGetOrThrow(keys, options) {
+		const { filter, ...restOptions } = options ?? {};
 		const items = await this.db.trxGetOrThrow(...keys.map((key) => ({
 			table: this.tableName,
 			key: this.extractKey(key),
 			resource: this.resourceName,
-			...options
+			filter,
+			...restOptions
 		})));
 		return this.applyTransformsIfNeeded(items, options);
 	}
@@ -489,8 +527,8 @@ var Repo = class {
 		await this.db.trxWrite(...requests.map((request) => {
 			switch (request.type) {
 				case "CONDITION": {
-					const { key, condition, ...options } = request;
-					return this.trxConditionRequest(key, condition, options);
+					const { key, condition } = request;
+					return this.trxConditionRequest(key, condition);
 				}
 				case "DELETE": {
 					const { key, ...options } = request;
@@ -535,85 +573,125 @@ var Repo = class {
 			...options
 		};
 	}
-	trxConditionRequest(key, condition, options) {
+	trxConditionRequest(key, condition) {
 		return {
 			table: this.tableName,
 			type: "CONDITION",
 			key: this.extractKey(key),
-			condition,
-			...options
+			condition
 		};
 	}
 	trxPutRequest(item, options) {
-		const itemWithDefaults = this.applyPutTransforms(item);
 		return {
 			table: this.tableName,
 			type: "PUT",
-			item: itemWithDefaults,
+			item,
 			...options
 		};
 	}
 	trxUpdateRequest(key, update, options) {
-		const updateWithDefaults = this.applyNormalizersToExpression({
-			...this.defaultUpdateData,
-			...update
-		});
+		const updateWithDefaults = this.applyNormalizersToExpression(update);
 		return {
 			table: this.tableName,
 			type: "UPDATE",
 			key: this.extractKey(key),
 			update: updateWithDefaults,
-			...options
+			...options,
+			condition: this.withDiscriminatorCondition(key, options?.condition)
 		};
 	}
 	trxCreateRequest(item, options) {
 		const { condition: otherCondition, ...otherOptions } = options ?? {};
 		const condition = { $and: [{ [this.table.def.partitionKey]: { $exists: false } }] };
 		if (otherCondition) condition.$and.push(otherCondition);
-		const itemWithDefaults = this.applyPutTransforms(item);
+		const normalizedItem = this.applyCreateTransforms(item);
 		return {
 			table: this.tableName,
 			type: "PUT",
-			item: itemWithDefaults,
+			item: normalizedItem,
 			condition,
 			...otherOptions
 		};
 	}
-	applyPutTransforms(item) {
-		const result = { ...this.transformInput({
-			...this.defaultPutData,
+	applyCreateTransforms(item) {
+		const { transformAttributes, computedAttributes } = this.config;
+		if (computedAttributes) {
+			for (const key of Object.keys(computedAttributes)) if (key in item) throw new DinahError({
+				type: "VALIDATION",
+				message: `Field "${key}" is a computed attribute and cannot be set directly. Remove it from the create input.`
+			});
+		}
+		const merged = {
+			...this.defaultCreateData,
 			...item
-		}) };
-		for (const key of this.config.derivedAttributes ?? []) delete result[key];
-		return result;
-	}
-	applyNormalizersToExpression(expression) {
-		const partial = {};
-		for (const [key, val] of Object.entries(expression)) if (val === void 0 || isOperation(val) && val.$remove === true) partial[key] = void 0;
-		else if (!isOperation(val)) partial[key] = val;
-		else if (val.$set !== void 0) partial[key] = val.$set;
-		else if (val.$ifNotExists !== void 0) partial[key] = Array.isArray(val.$ifNotExists) ? val.$ifNotExists[1] : val.$ifNotExists;
-		const normalized = this.transformInput(partial);
-		const result = { ...expression };
-		for (const [key, normalizedVal] of Object.entries(normalized)) {
-			if (normalizedVal === void 0) continue;
-			if (key in partial && partial[key] === void 0 || !(key in partial)) result[key] = normalizedVal;
-			else {
-				const original = expression[key];
-				if (!isOperation(original)) result[key] = normalizedVal;
-				else if (original.$set !== void 0) result[key] = {
-					...original,
-					$set: normalizedVal
-				};
-				else if (original.$ifNotExists !== void 0) result[key] = {
-					...original,
-					$ifNotExists: Array.isArray(original.$ifNotExists) ? [original.$ifNotExists[0], normalizedVal] : normalizedVal
+		};
+		if (transformAttributes) {
+			for (const [key, transform] of Object.entries(transformAttributes)) if (key in merged && merged[key] !== void 0) merged[key] = transform(merged[key]);
+		}
+		if (computedAttributes) for (const [key, def] of Object.entries(computedAttributes)) {
+			const { from, compute } = def;
+			const computed = compute(merged[from]);
+			if (computed !== void 0) merged[key] = computed;
+			else delete merged[key];
+		}
+		return merged;
+	}
+	applyNormalizersToExpression(userUpdate) {
+		const { transformAttributes, computedAttributes, immutableAttributes } = this.config;
+		if (computedAttributes) {
+			for (const key of Object.keys(computedAttributes)) if (key in userUpdate) throw new DinahError({
+				type: "VALIDATION",
+				message: `Field "${key}" is a computed attribute and cannot be set directly in an update. Update the source field instead.`
+			});
+		}
+		for (const key of immutableAttributes ?? []) if (key in userUpdate) throw new DinahError({
+			type: "VALIDATION",
+			message: `Field "${key}" is immutable and cannot be updated.`
+		});
+		const result = {
+			...this.defaultUpdateData,
+			...userUpdate
+		};
+		if (transformAttributes) for (const [key, transform] of Object.entries(transformAttributes)) {
+			if (!(key in result)) continue;
+			const val = result[key];
+			const fn = transform;
+			if (val === void 0 || isOperation(val) && val.$remove === true) {} else if (!isOperation(val)) result[key] = fn(val);
+			else if (val.$set !== void 0) result[key] = {
+				...val,
+				$set: fn(val.$set)
+			};
+			else if (val.$ifNotExists !== void 0) {
+				const ifne = val.$ifNotExists;
+				result[key] = {
+					...val,
+					$ifNotExists: Array.isArray(ifne) ? [ifne[0], fn(ifne[1])] : fn(ifne)
 				};
 			}
 		}
-		for (const key of [...this.config.derivedAttributes ?? [], ...this.config.immutableAttributes ?? []]) delete result[key];
+		if (computedAttributes) for (const [computedKey, def] of Object.entries(computedAttributes)) {
+			const { from, compute } = def;
+			if (!(from in result)) continue;
+			const val = result[from];
+			if (val === void 0 || isOperation(val) && val.$remove === true) result[computedKey] = void 0;
+			else if (!isOperation(val)) result[computedKey] = compute(val);
+			else if (val.$set !== void 0) result[computedKey] = compute(val.$set);
+			else throw new DinahError({
+				type: "VALIDATION",
+				message: `Field "${from}" drives computed field "${computedKey}" and cannot use arithmetic or list operators in updates.`
+			});
+		}
 		return result;
 	}
+	withDiscriminatorCondition(key, condition) {
+		const discriminator = this.config.discriminator;
+		if (!discriminator) return condition;
+		const value = key[discriminator];
+		if (value === void 0) return condition;
+		const guard = { [discriminator]: value };
+		if (!condition) return guard;
+		return { $and: [guard, condition] };
+	}
 	applyTransformsIfNeeded(items, options) {
 		if (options?.projection?.length) return items;
 		if (options?.gsi) {
@@ -683,7 +761,7 @@ var Db = class {
 			ExpressionAttributeNames: exp.attributeNames
 		});
 		const output = await this.client.send(input);
-		if (output.Item && data.filter && !data.filter(output.Item)) return;
+		if (output.Item && data.filter && !matchesPartial(data.filter, output.Item)) return;
 		return output.Item;
 	}
 	async getOrThrow(data) {
@@ -886,7 +964,7 @@ var Db = class {
 			const output = await this.client.send(input);
 			for (const [table, items] of Object.entries(output.Responses ?? {})) {
 				if (!result.items[table]) continue;
-				const filteredItems = tableData[table]?.filter ? items.filter(tableData[table].filter) : items;
+				const filteredItems = tableData[table]?.filter ? items.filter((item) => matchesPartial(tableData[table].filter, item)) : items;
 				result.items[table].push(...filteredItems);
 			}
 			let unprocessedKeyCount = 0;
@@ -1047,7 +1125,7 @@ var Db = class {
 		return (await this.client.send(input)).Responses?.map((response, i) => {
 			if (!response.Item) return;
 			if (!requests[i]?.filter) return response.Item;
-			return requests[i].filter(response.Item) ? response.Item : void 0;
+			return matchesPartial(requests[i].filter, response.Item) ? response.Item : void 0;
 		}) ?? [];
 	}
 	async trxGetOrThrow(...requests) {
@@ -1111,6 +1189,6 @@ var Table = class {
 	}
 };
 //#endregion
-export { Db, DinahError, Repo, Table, chunk, extractTableDesc, isDinahError, makeRepo, removeUndefined, resolveAttrType };
+export { Db, DinahError, Repo, Table, chunk, extractTableDesc, isDinahError, makeRepo, matchesPartial, removeUndefined, resolveAttrType };
 
 //# sourceMappingURL=index.mjs.map