dynamodb-toolbox
Version:
Lightweight and type-safe query builder for DynamoDB and TypeScript.
164 lines (163 loc) • 6.06 kB
JavaScript
import { ifThenElse } from '../../utils/ifThenElse.js';
import { overwrite } from '../../utils/overwrite.js';
import { AnySchema } from './schema.js';
/**
* Define a new schema of any type
*
* @param props _(optional)_ Attribute Props
*/
export const any = (props = {}) => new AnySchema_(props);
/**
* Any attribute (warm)
*/
export class AnySchema_ extends AnySchema {
/**
* Tag attribute as required. Possible values are:
* - `'atLeastOnce'` _(default)_: Required in PUTs, optional in UPDATEs
* - `'never'`: Optional in PUTs and UPDATEs
* - `'always'`: Required in PUTs and UPDATEs
*
* @param nextRequired SchemaRequiredProp
*/
required(nextRequired = 'atLeastOnce') {
return new AnySchema_(overwrite(this.props, { required: nextRequired }));
}
/**
* Shorthand for `required('never')`
*/
optional() {
return this.required('never');
}
/**
* Hide attribute after fetch commands and formatting
*/
hidden(nextHidden = true) {
return new AnySchema_(overwrite(this.props, { hidden: nextHidden }));
}
/**
* Tag attribute as a primary key attribute or linked to a primary attribute
*/
key(nextKey = true) {
return new AnySchema_(overwrite(this.props, { key: nextKey, required: 'always' }));
}
/**
* Rename attribute before save commands
*/
savedAs(nextSavedAs) {
return new AnySchema_(overwrite(this.props, { savedAs: nextSavedAs }));
}
/**
* Cast attribute TS type
*/
castAs(nextCastAs = undefined) {
return new AnySchema_(overwrite(this.props, { castAs: nextCastAs }));
}
/**
* Transform the attribute value in PUT commands OR Primary Key computing if attribute is tagged as key
*/
transform(transform) {
return new AnySchema_(overwrite(this.props, { transform }));
}
/**
* Provide a default value for attribute in Primary Key computing
*
* @param nextKeyDefault `keyAttributeInput | (() => keyAttributeInput)`
*/
keyDefault(nextKeyDefault) {
return new AnySchema_(overwrite(this.props, { keyDefault: nextKeyDefault }));
}
/**
* Provide a default value for attribute in PUT commands
*
* @param nextPutDefault `putAttributeInput | (() => putAttributeInput)`
*/
putDefault(nextPutDefault) {
return new AnySchema_(overwrite(this.props, { putDefault: nextPutDefault }));
}
/**
* Provide a default value for attribute in UPDATE commands
*
* @param nextUpdateDefault `updateAttributeInput | (() => updateAttributeInput)`
*/
updateDefault(nextUpdateDefault) {
return new AnySchema_(overwrite(this.props, { updateDefault: nextUpdateDefault }));
}
/**
* Provide a default value for attribute in PUT commands OR Primary Key computing if attribute is tagged as key
*
* @param nextDefault `key/putAttributeInput | (() => key/putAttributeInput)`
*/
default(nextDefault) {
return ifThenElse(this.props.key, new AnySchema_(overwrite(this.props, { keyDefault: nextDefault })), new AnySchema_(overwrite(this.props, { putDefault: nextDefault })));
}
/**
* Provide a **linked** default value for attribute in Primary Key computing
*
* @param nextKeyLink `keyAttributeInput | ((keyInput) => keyAttributeInput)`
*/
keyLink(nextKeyLink) {
return new AnySchema_(overwrite(this.props, { keyLink: nextKeyLink }));
}
/**
* Provide a **linked** default value for attribute in PUT commands
*
* @param nextPutLink `putAttributeInput | ((putItemInput) => putAttributeInput)`
*/
putLink(nextPutLink) {
return new AnySchema_(overwrite(this.props, { putLink: nextPutLink }));
}
/**
* Provide a **linked** default value for attribute in UPDATE commands
*
* @param nextUpdateLink `unknown | ((updateItemInput) => updateAttributeInput)`
*/
updateLink(nextUpdateLink) {
return new AnySchema_(overwrite(this.props, { updateLink: nextUpdateLink }));
}
/**
* Provide a **linked** default value for attribute in PUT commands OR Primary Key computing if attribute is tagged as key
*
* @param nextLink `key/putAttributeInput | (() => key/putAttributeInput)`
*/
link(nextLink) {
return ifThenElse(this.props.key, new AnySchema_(overwrite(this.props, { keyLink: nextLink })), new AnySchema_(overwrite(this.props, { putLink: nextLink })));
}
/**
* Provide a custom validator for attribute in Primary Key computing
*
* @param nextKeyValidator `(keyAttributeInput) => boolean | string`
*/
keyValidate(nextKeyValidator) {
return new AnySchema_(overwrite(this.props, { keyValidator: nextKeyValidator }));
}
/**
* Provide a custom validator for attribute in PUT commands
*
* @param nextPutValidator `(putAttributeInput) => boolean | string`
*/
putValidate(nextPutValidator) {
return new AnySchema_(overwrite(this.props, { putValidator: nextPutValidator }));
}
/**
* Provide a custom validator for attribute in UPDATE commands
*
* @param nextUpdateValidator `(updateAttributeInput) => boolean | string`
*/
updateValidate(nextUpdateValidator) {
return new AnySchema_(overwrite(this.props, { updateValidator: nextUpdateValidator }));
}
/**
* Provide a custom validator for attribute in PUT commands OR Primary Key computing if attribute is tagged as key
*
* @param nextValidator `(key/putAttributeInput) => boolean | string`
*/
validate(nextValidator) {
return ifThenElse(this.props.key, new AnySchema_(overwrite(this.props, { keyValidator: nextValidator })), new AnySchema_(overwrite(this.props, { putValidator: nextValidator })));
}
clone(nextProps = {}) {
return new AnySchema_(overwrite(this.props, nextProps));
}
build(Action) {
return new Action(this);
}
}