@aws-amplify/api-graphql 4.0.2-unstable.ff85fa5.0 → 4.0.2

package/src/internals/APIClient.ts

@@ -0,0 +1,780 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+import { resolveOwnerFields } from '../utils/resolveOwnerFields';
+import {
+	GraphQLAuthMode,
+	ModelIntrospectionSchema,
+	ModelFieldType,
+	SchemaModel,
+	SchemaModels,
+	AssociationHasOne,
+	AssociationBelongsTo,
+} from '@aws-amplify/core/internals/utils';
+import {
+	AuthModeParams,
+	ClientWithModels,
+	ListArgs,
+	QueryArgs,
+	V6Client,
+	V6ClientSSRRequest,
+	__authMode,
+	__authToken,
+} from '../types';
+import { AmplifyServer } from '@aws-amplify/core/internals/adapter-core';
+
+type LazyLoadOptions = {
+	authMode?: GraphQLAuthMode;
+	authToken?: string | undefined;
+	limit?: number | undefined;
+	nextToken?: string | undefined | null;
+};
+
+const connectionType = {
+	HAS_ONE: 'HAS_ONE',
+	HAS_MANY: 'HAS_MANY',
+	BELONGS_TO: 'BELONGS_TO',
+};
+
+/**
+ *
+ * @param GraphQL response object
+ * @returns response object with `items` properties flattened
+ */
+export const flattenItems = (obj: Record<string, any>): Record<string, any> => {
+	const res: Record<string, any> = {};
+
+	Object.entries(obj).forEach(([prop, value]) => {
+		if (typeof value === 'object' && value !== null) {
+			if (value.items !== undefined) {
+				res[prop] = value.items.map((item: Record<string, any>) =>
+					flattenItems(item)
+				);
+				return;
+			}
+			res[prop] = flattenItems(value);
+			return;
+		}
+
+		res[prop] = value;
+	});
+
+	return res;
+};
+
+// TODO: this should accept single result to support CRUD methods; create helper for array/list
+export function initializeModel(
+	client: ClientWithModels,
+	modelName: string,
+	result: any[],
+	modelIntrospection: ModelIntrospectionSchema,
+	authMode: GraphQLAuthMode | undefined,
+	authToken: string | undefined,
+	context = false
+): any[] {
+	const introModel = modelIntrospection.models[modelName];
+	const introModelFields = introModel.fields;
+
+	const modelFields: string[] = Object.entries(introModelFields)
+		.filter(([_, field]: [string, any]) => field?.type?.model !== undefined)
+		.map(([fieldName]) => fieldName);
+
+	return result.map(record => {
+		const initializedRelationalFields: Record<string, any> = {};
+
+		for (const fieldName of modelFields) {
+			const modelField = introModelFields[fieldName];
+			const modelFieldType = modelField?.type as ModelFieldType;
+
+			const relatedModelName = modelFieldType.model;
+			const relatedModel = modelIntrospection.models[relatedModelName!];
+
+			const relatedModelPKFieldName =
+				relatedModel.primaryKeyInfo.primaryKeyFieldName;
+
+			const relatedModelSKFieldNames =
+				relatedModel.primaryKeyInfo.sortKeyFieldNames;
+
+			const relationType = modelField.association?.connectionType;
+
+			let connectionFields: string[] = [];
+			if (
+				modelField.association &&
+				'associatedWith' in modelField.association
+			) {
+				connectionFields = modelField.association.associatedWith;
+			}
+
+			let targetNames: string[] = [];
+			if (modelField.association && 'targetNames' in modelField.association) {
+				targetNames = modelField.association.targetNames;
+			}
+
+			switch (relationType) {
+				case connectionType.HAS_ONE:
+				case connectionType.BELONGS_TO:
+					const sortKeyValues = relatedModelSKFieldNames.reduce(
+						(acc: Record<string, any>, curVal) => {
+							if (record[curVal]) {
+								return (acc[curVal] = record[curVal]);
+							}
+						},
+						{}
+					);
+
+					if (context) {
+						initializedRelationalFields[fieldName] = (
+							contextSpec: AmplifyServer.ContextSpec,
+							options?: LazyLoadOptions
+						) => {
+							if (record[targetNames[0]]) {
+								return (
+									client as V6ClientSSRRequest<Record<string, any>>
+								).models[relatedModelName].get(
+									contextSpec,
+									{
+										[relatedModelPKFieldName]: record[targetNames[0]],
+										...sortKeyValues,
+									},
+									{
+										authMode: options?.authMode || authMode,
+										authToken: options?.authToken || authToken,
+									}
+								);
+							}
+							return undefined;
+						};
+					} else {
+						initializedRelationalFields[fieldName] = (
+							options?: LazyLoadOptions
+						) => {
+							if (record[targetNames[0]]) {
+								return (client as V6Client<Record<string, any>>).models[
+									relatedModelName
+								].get(
+									{
+										[relatedModelPKFieldName]: record[targetNames[0]],
+										...sortKeyValues,
+									},
+									{
+										authMode: options?.authMode || authMode,
+										authToken: options?.authToken || authToken,
+									}
+								);
+							}
+							return undefined;
+						};
+					}
+
+					break;
+				case connectionType.HAS_MANY:
+					const parentPk = introModel.primaryKeyInfo.primaryKeyFieldName;
+					const parentSK = introModel.primaryKeyInfo.sortKeyFieldNames;
+
+					// M:N check - TODO: refactor
+					const relatedModelField = relatedModel.fields[connectionFields[0]];
+					const relatedModelFieldType =
+						relatedModelField.type as ModelFieldType;
+					if (relatedModelFieldType.model) {
+						let relatedTargetNames: string[] = [];
+						if (
+							relatedModelField.association &&
+							'targetNames' in relatedModelField.association
+						) {
+							relatedTargetNames = relatedModelField.association?.targetNames;
+						}
+
+						const hasManyFilter: Record<string, any> = relatedTargetNames.map(
+							(field, idx) => {
+								if (idx === 0) {
+									return { [field]: { eq: record[parentPk] } };
+								}
+
+								return { [field]: { eq: record[parentSK[idx - 1]] } };
+							}
+						);
+
+						if (context) {
+							initializedRelationalFields[fieldName] = (
+								contextSpec: AmplifyServer.ContextSpec,
+								options?: LazyLoadOptions
+							) => {
+								if (record[parentPk]) {
+									return (
+										client as V6ClientSSRRequest<Record<string, any>>
+									).models[relatedModelName].list(contextSpec, {
+										filter: { and: hasManyFilter },
+										limit: options?.limit,
+										nextToken: options?.nextToken,
+										authMode: options?.authMode || authMode,
+										authToken: options?.authToken || authToken,
+									});
+								}
+								return [];
+							};
+						} else {
+							initializedRelationalFields[fieldName] = (
+								options?: LazyLoadOptions
+							) => {
+								if (record[parentPk]) {
+									return (client as V6Client<Record<string, any>>).models[
+										relatedModelName
+									].list({
+										filter: { and: hasManyFilter },
+										limit: options?.limit,
+										nextToken: options?.nextToken,
+										authMode: options?.authMode || authMode,
+										authToken: options?.authToken || authToken,
+									});
+								}
+								return [];
+							};
+						}
+
+						break;
+					}
+
+					const hasManyFilter: Record<string, any> = connectionFields.map(
+						(field, idx) => {
+							if (idx === 0) {
+								return { [field]: { eq: record[parentPk] } };
+							}
+
+							return { [field]: { eq: record[parentSK[idx - 1]] } };
+						}
+					);
+
+					if (context) {
+						initializedRelationalFields[fieldName] = (
+							contextSpec: AmplifyServer.ContextSpec,
+							options?: LazyLoadOptions
+						) => {
+							if (record[parentPk]) {
+								return (
+									client as V6ClientSSRRequest<Record<string, any>>
+								).models[relatedModelName].list(contextSpec, {
+									filter: { and: hasManyFilter },
+									limit: options?.limit,
+									nextToken: options?.nextToken,
+									authMode: options?.authMode || authMode,
+									authToken: options?.authToken || authToken,
+								});
+							}
+							return [];
+						};
+					} else {
+						initializedRelationalFields[fieldName] = (
+							options?: LazyLoadOptions
+						) => {
+							if (record[parentPk]) {
+								return (client as V6Client<Record<string, any>>).models[
+									relatedModelName
+								].list({
+									filter: { and: hasManyFilter },
+									limit: options?.limit,
+									nextToken: options?.nextToken,
+									authMode: options?.authMode || authMode,
+									authToken: options?.authToken || authToken,
+								});
+							}
+							return [];
+						};
+					}
+
+					break;
+				default:
+					break;
+			}
+		}
+
+		return { ...record, ...initializedRelationalFields };
+	});
+}
+
+export const graphQLOperationsInfo = {
+	CREATE: { operationPrefix: 'create' as const, usePlural: false },
+	READ: { operationPrefix: 'get' as const, usePlural: false },
+	UPDATE: { operationPrefix: 'update' as const, usePlural: false },
+	DELETE: { operationPrefix: 'delete' as const, usePlural: false },
+	LIST: { operationPrefix: 'list' as const, usePlural: true },
+	ONCREATE: { operationPrefix: 'onCreate' as const, usePlural: false },
+	ONUPDATE: { operationPrefix: 'onUpdate' as const, usePlural: false },
+	ONDELETE: { operationPrefix: 'onDelete' as const, usePlural: false },
+	OBSERVE_QUERY: { operationPrefix: 'observeQuery' as const, usePlural: false },
+};
+export type ModelOperation = keyof typeof graphQLOperationsInfo;
+
+type OperationPrefix =
+	(typeof graphQLOperationsInfo)[ModelOperation]['operationPrefix'];
+
+const graphQLDocumentsCache = new Map<string, Map<ModelOperation, string>>();
+const SELECTION_SET_WILDCARD = '*';
+
+function defaultSelectionSetForModel(modelDefinition: SchemaModel): string[] {
+	// fields that are explicitly part of the graphql schema; not
+	// inferred from owner auth rules.
+	const { fields } = modelDefinition;
+	const explicitFields = Object.values<any>(fields)
+		.map(({ type, name }) => typeof type === 'string' && name) // Default selection set omits model fields
+		.filter(Boolean);
+
+	// fields used for owner auth rules that may or may not also
+	// be explicit on the model.
+	const ownerFields = resolveOwnerFields(modelDefinition);
+
+	return Array.from(new Set(explicitFields.concat(ownerFields)));
+}
+
+const FIELD_IR = '';
+
+/**
+ * Generates nested Custom Selection Set IR from path
+ *
+ * @param modelDefinitions
+ * @param modelName
+ * @param selectionSet - array of object paths
+ * @example
+ * ### Given
+ * `selectionSet = ['id', 'comments.post.id']`
+ * ### Returns
+ * ```ts
+ * {
+ *   id: '',
+ *   comments: {
+ *     items: { post: { id: '' } }
+ *   }
+ * }
+ * ```
+ */
+export function customSelectionSetToIR(
+	modelDefinitions: SchemaModels,
+	modelName: string,
+	selectionSet: string[]
+): Record<string, string | object> {
+	const modelDefinition = modelDefinitions[modelName];
+	const { fields: modelFields } = modelDefinition;
+
+	return selectionSet.reduce((resultObj: Record<string, any>, path) => {
+		const [fieldName, nested, ...rest] = path.split('.');
+
+		if (nested) {
+			const fieldType = modelFields[fieldName]?.type as ModelFieldType;
+			const relatedModel = fieldType.model;
+
+			if (!relatedModel) {
+				// TODO: may need to change this to support custom types
+				throw Error(`${fieldName} is not a model field`);
+			}
+
+			const relatedModelDefinition = modelDefinitions[relatedModel];
+
+			const selectionSet =
+				nested === SELECTION_SET_WILDCARD
+					? defaultSelectionSetIR(relatedModelDefinition)
+					: // if we have a path like 'field.anotherField' recursively build up selection set IR
+					  customSelectionSetToIR(modelDefinitions, relatedModel, [
+							[nested, ...rest].join('.'),
+					  ]);
+
+			if (modelFields[fieldName]?.isArray) {
+				const existing = resultObj[fieldName] || {
+					items: {},
+				};
+				const merged = { ...existing.items, ...selectionSet };
+
+				resultObj[fieldName] = { items: merged };
+				return resultObj;
+			}
+
+			const existingItems = resultObj[fieldName] || {};
+			const merged = { ...existingItems, ...selectionSet };
+
+			resultObj[fieldName] = merged;
+			return resultObj;
+		}
+
+		const exists = Boolean(modelFields[fieldName]);
+
+		if (!exists) {
+			throw Error(`${fieldName} is not a field of model ${modelName}`);
+		}
+
+		resultObj[fieldName] = FIELD_IR;
+		return resultObj;
+	}, {});
+}
+
+const defaultSelectionSetIR = (relatedModelDefinition: SchemaModel) => {
+	const defaultSelectionSet = defaultSelectionSetForModel(
+		relatedModelDefinition
+	);
+
+	const reduced = defaultSelectionSet.reduce(
+		(acc: Record<string, any>, curVal) => {
+			acc[curVal] = FIELD_IR;
+			return acc;
+		},
+		{}
+	);
+
+	return reduced;
+};
+
+/**
+ * Stringifies selection set IR
+ * * @example
+ * ### Given
+ * ```ts
+ * {
+ *   id: '',
+ *   comments: {
+ *     items: { post: { id: '' } }
+ *   }
+ * }
+ * ```
+ * ### Returns
+ * `'id comments { items { post { id } } }'`
+ */
+export function selectionSetIRToString(
+	obj: Record<string, string | any>
+): string {
+	const res: string[] = [];
+
+	Object.entries(obj).forEach(([fieldName, value]) => {
+		if (value === FIELD_IR) {
+			res.push(fieldName);
+		} else if (typeof value === 'object' && value !== null) {
+			if (value?.items) {
+				res.push(
+					fieldName,
+					'{',
+					'items',
+					'{',
+					selectionSetIRToString(value.items),
+					'}',
+					'}'
+				);
+			} else {
+				res.push(fieldName, '{', selectionSetIRToString(value), '}');
+			}
+		}
+	});
+
+	return res.join(' ');
+}
+
+export function generateSelectionSet(
+	modelDefinitions: SchemaModels,
+	modelName: string,
+	selectionSet?: string[]
+) {
+	const modelDefinition = modelDefinitions[modelName];
+
+	if (!selectionSet) {
+		return defaultSelectionSetForModel(modelDefinition).join(' ');
+	}
+
+	const selSetIr = customSelectionSetToIR(
+		modelDefinitions,
+		modelName,
+		selectionSet
+	);
+	const selSetString = selectionSetIRToString(selSetIr);
+
+	return selSetString;
+}
+
+export function generateGraphQLDocument(
+	modelDefinitions: SchemaModels,
+	modelName: string,
+	modelOperation: ModelOperation,
+	listArgs?: ListArgs
+): string {
+	const modelDefinition = modelDefinitions[modelName];
+
+	const {
+		name,
+		pluralName,
+		fields,
+		primaryKeyInfo: {
+			isCustomPrimaryKey,
+			primaryKeyFieldName,
+			sortKeyFieldNames,
+		},
+	} = modelDefinition;
+
+	const { operationPrefix, usePlural } = graphQLOperationsInfo[modelOperation];
+
+	const { selectionSet } = listArgs || {};
+
+	const fromCache = graphQLDocumentsCache.get(name)?.get(modelOperation);
+
+	if (fromCache !== undefined) {
+		return fromCache;
+	}
+
+	if (!graphQLDocumentsCache.has(name)) {
+		graphQLDocumentsCache.set(name, new Map());
+	}
+
+	const graphQLFieldName = `${operationPrefix}${usePlural ? pluralName : name}`;
+	let graphQLOperationType: 'mutation' | 'query' | 'subscription' | undefined;
+	let graphQLSelectionSet: string | undefined;
+	let graphQLArguments: Record<string, any> | undefined;
+
+	const selectionSetFields = generateSelectionSet(
+		modelDefinitions,
+		modelName,
+		selectionSet
+	);
+
+	switch (modelOperation) {
+		case 'CREATE':
+		case 'UPDATE':
+		case 'DELETE':
+			graphQLArguments ??
+				(graphQLArguments = {
+					input: `${
+						operationPrefix.charAt(0).toLocaleUpperCase() +
+						operationPrefix.slice(1)
+					}${name}Input!`,
+				});
+			graphQLOperationType ?? (graphQLOperationType = 'mutation');
+		case 'READ':
+			graphQLArguments ??
+				(graphQLArguments = isCustomPrimaryKey
+					? [primaryKeyFieldName, ...sortKeyFieldNames].reduce(
+							(acc: Record<string, any>, fieldName) => {
+								acc[fieldName] = fields[fieldName].type;
+
+								return acc;
+							},
+							{}
+					  )
+					: {
+							[primaryKeyFieldName]: `${fields[primaryKeyFieldName].type}!`,
+					  });
+			graphQLSelectionSet ?? (graphQLSelectionSet = selectionSetFields);
+		case 'LIST':
+			graphQLArguments ??
+				(graphQLArguments = {
+					filter: `Model${name}FilterInput`,
+					limit: 'Int',
+					nextToken: 'String',
+				});
+			graphQLOperationType ?? (graphQLOperationType = 'query');
+			graphQLSelectionSet ??
+				(graphQLSelectionSet = `items { ${selectionSetFields} } nextToken __typename`);
+		case 'ONCREATE':
+		case 'ONUPDATE':
+		case 'ONDELETE':
+			graphQLArguments ??
+				(graphQLArguments = {
+					filter: `ModelSubscription${name}FilterInput`,
+				});
+			graphQLOperationType ?? (graphQLOperationType = 'subscription');
+			graphQLSelectionSet ?? (graphQLSelectionSet = selectionSetFields);
+			break;
+		case 'OBSERVE_QUERY':
+		default:
+			throw new Error(
+				'Internal error: Attempted to generate graphql document for observeQuery. Please report this error.'
+			);
+	}
+
+	const graphQLDocument = `${graphQLOperationType}${
+		graphQLArguments
+			? `(${Object.entries(graphQLArguments).map(
+					([fieldName, type]) => `\$${fieldName}: ${type}`
+			  )})`
+			: ''
+	} { ${graphQLFieldName}${
+		graphQLArguments
+			? `(${Object.keys(graphQLArguments).map(
+					fieldName => `${fieldName}: \$${fieldName}`
+			  )})`
+			: ''
+	} { ${graphQLSelectionSet} } }`;
+
+	graphQLDocumentsCache.get(name)?.set(modelOperation, graphQLDocument);
+
+	return graphQLDocument;
+}
+
+export function buildGraphQLVariables(
+	modelDefinition: SchemaModel,
+	operation: ModelOperation,
+	arg: QueryArgs | undefined,
+	modelIntrospection: ModelIntrospectionSchema
+): object {
+	const {
+		fields,
+		primaryKeyInfo: {
+			isCustomPrimaryKey,
+			primaryKeyFieldName,
+			sortKeyFieldNames,
+		},
+	} = modelDefinition;
+
+	let variables: Record<string, any> = {};
+
+	// TODO: process input
+	switch (operation) {
+		case 'CREATE':
+			variables = {
+				input: arg
+					? normalizeMutationInput(arg, modelDefinition, modelIntrospection)
+					: {},
+			};
+			break;
+		case 'UPDATE':
+			// readonly fields are not  updated
+			variables = {
+				input: arg
+					? Object.fromEntries(
+							Object.entries(
+								normalizeMutationInput(arg, modelDefinition, modelIntrospection)
+							).filter(([fieldName]) => {
+								const { isReadOnly } = fields[fieldName];
+
+								return !isReadOnly;
+							})
+					  )
+					: {},
+			};
+			break;
+		case 'READ':
+		case 'DELETE':
+			// only identifiers are sent
+			if (arg) {
+				variables = isCustomPrimaryKey
+					? [primaryKeyFieldName, ...sortKeyFieldNames].reduce(
+							(acc: Record<string, any>, fieldName) => {
+								acc[fieldName] = arg[fieldName];
+
+								return acc;
+							},
+							{}
+					  )
+					: { [primaryKeyFieldName]: arg[primaryKeyFieldName] };
+			}
+
+			if (operation === 'DELETE') {
+				variables = { input: variables };
+			}
+			break;
+		case 'LIST':
+			if (arg?.filter) {
+				variables.filter = arg.filter;
+			}
+			if (arg?.nextToken) {
+				variables.nextToken = arg.nextToken;
+			}
+			if (arg?.limit) {
+				variables.limit = arg.limit;
+			}
+			break;
+		case 'ONCREATE':
+		case 'ONUPDATE':
+		case 'ONDELETE':
+			if (arg?.filter) {
+				variables = { filter: arg.filter };
+			}
+			break;
+		case 'OBSERVE_QUERY':
+			throw new Error(
+				'Internal error: Attempted to build variables for observeQuery. Please report this error.'
+			);
+			break;
+		default:
+			const exhaustiveCheck: never = operation;
+			throw new Error(`Unhandled operation case: ${exhaustiveCheck}`);
+	}
+
+	return variables;
+}
+
+/**
+ * Iterates over mutation input values and resolves any model inputs to their corresponding join fields/values
+ *
+ * @example
+ * ### Usage
+ * ```ts
+ * const result = normalizeMutationInput({ post: post }, model, modelDefinition);
+ * ```
+ * ### Result
+ * ```ts
+ * { postId: "abc123" }
+ * ```
+ *
+ */
+export function normalizeMutationInput(
+	mutationInput: QueryArgs,
+	model: SchemaModel,
+	modelIntrospection: ModelIntrospectionSchema
+): QueryArgs {
+	const { fields } = model;
+
+	const normalized: Record<string, unknown> = {};
+
+	Object.entries(mutationInput).forEach(([inputFieldName, inputValue]) => {
+		const fieldType = fields[inputFieldName]?.type as ModelFieldType;
+		const relatedModelName = fieldType?.model;
+
+		if (relatedModelName) {
+			const association = fields[inputFieldName]?.association;
+			const relatedModelDef = modelIntrospection.models[relatedModelName];
+			const relatedModelPkInfo = relatedModelDef.primaryKeyInfo;
+
+			if (association?.connectionType === connectionType.HAS_ONE) {
+				const associationHasOne = association as AssociationHasOne;
+				associationHasOne.targetNames.forEach((targetName, idx) => {
+					const associatedFieldName = associationHasOne.associatedWith[idx];
+					normalized[targetName] = (inputValue as Record<string, unknown>)[
+						associatedFieldName
+					];
+				});
+			}
+
+			if (association?.connectionType === connectionType.BELONGS_TO) {
+				const associationBelongsTo = association as AssociationBelongsTo;
+				associationBelongsTo.targetNames.forEach((targetName, idx) => {
+					if (idx === 0) {
+						const associatedFieldName = relatedModelPkInfo.primaryKeyFieldName;
+						normalized[targetName] = (inputValue as Record<string, unknown>)[
+							associatedFieldName
+						];
+					} else {
+						const associatedFieldName =
+							relatedModelPkInfo.sortKeyFieldNames[idx - 1];
+						normalized[targetName] = (inputValue as Record<string, unknown>)[
+							associatedFieldName
+						];
+					}
+				});
+			}
+		} else {
+			normalized[inputFieldName] = inputValue;
+		}
+	});
+
+	return normalized;
+}
+
+/**
+ * Produces a parameter object that can contains auth mode/token overrides
+ * only if present in either `options` (first) or configured on the `client`
+ * as a fallback.
+ *
+ * @param client Configured client from `generateClient`
+ * @param options Args/Options obect from call site.
+ * @returns
+ */
+export function authModeParams(
+	client: ClientWithModels,
+	options: AuthModeParams = {}
+): AuthModeParams {
+	return {
+		authMode: options.authMode || client[__authMode],
+		authToken: options.authToken || client[__authToken],
+	};
+}