@aws-amplify/api-graphql 4.4.3-local-stack.0aa9031.0 → 4.4.3-unstable.847fb51.0

package/src/internals/graphqlAuth.ts

@@ -0,0 +1,87 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import { GraphQLAuthMode } from '@aws-amplify/core/internals/utils';
+import { AmplifyClassV6 } from '@aws-amplify/core';
+
+import { GraphQLApiError } from '../utils/errors';
+import {
+	NO_API_KEY,
+	NO_AUTH_TOKEN_HEADER,
+	NO_SIGNED_IN_USER,
+	NO_VALID_AUTH_TOKEN,
+	NO_VALID_CREDENTIALS,
+} from '../utils/errors/constants';
+
+export async function headerBasedAuth(
+	amplify: AmplifyClassV6,
+	authMode: GraphQLAuthMode,
+	apiKey: string | undefined,
+	additionalHeaders: Record<string, string> = {},
+) {
+	let headers = {};
+
+	switch (authMode) {
+		case 'apiKey':
+			if (!apiKey) {
+				throw new GraphQLApiError(NO_API_KEY);
+			}
+			headers = {
+				'X-Api-Key': apiKey,
+			};
+			break;
+		case 'iam': {
+			const session = await amplify.Auth.fetchAuthSession();
+			if (session.credentials === undefined) {
+				throw new GraphQLApiError(NO_VALID_CREDENTIALS);
+			}
+			break;
+		}
+		case 'oidc':
+		case 'userPool': {
+			let token: string | undefined;
+
+			try {
+				token = (
+					await amplify.Auth.fetchAuthSession()
+				).tokens?.accessToken.toString();
+			} catch (e) {
+				// fetchAuthSession failed
+				throw new GraphQLApiError({
+					...NO_SIGNED_IN_USER,
+					underlyingError: e,
+				});
+			}
+
+			// `fetchAuthSession()` succeeded but didn't return `tokens`.
+			// This may happen when unauthenticated access is enabled and there is
+			// no user signed in.
+			if (!token) {
+				throw new GraphQLApiError(NO_VALID_AUTH_TOKEN);
+			}
+
+			headers = {
+				Authorization: token,
+			};
+			break;
+		}
+		case 'lambda':
+			if (
+				typeof additionalHeaders === 'object' &&
+				!additionalHeaders.Authorization
+			) {
+				throw new GraphQLApiError(NO_AUTH_TOKEN_HEADER);
+			}
+
+			headers = {
+				Authorization: additionalHeaders.Authorization,
+			};
+			break;
+		case 'none':
+			break;
+		default:
+			break;
+	}
+
+	return headers;
+}

package/src/internals/graphqlRequest.ts

@@ -0,0 +1,30 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import { AmplifyUrl } from '@aws-amplify/core/internals/utils';
+import { AmplifyClassV6 } from '@aws-amplify/core';
+import {
+	// cancel as cancelREST,
+	post,
+	// updateRequestToBeCancellable,
+} from '@aws-amplify/api-rest/internals';
+
+export async function graphqlRequest(
+	amplify: AmplifyClassV6,
+	url: string,
+	options: any,
+	abortController: AbortController,
+	_post?: typeof post,
+) {
+	const p = _post ?? post;
+
+	const { body: responseBody } = await p(amplify, {
+		url: new AmplifyUrl(url),
+		options,
+		abortController,
+	});
+
+	const response = await responseBody.json();
+
+	return response;
+}

package/src/utils/ConnectionStateMonitor.ts

@@ -17,18 +17,7 @@ interface LinkedConnectionStates {
 	keepAliveState: LinkedHealthState;
 }
 
-export const CONNECTION_CHANGE: {
-	[key in
-		| 'KEEP_ALIVE_MISSED'
-		| 'KEEP_ALIVE'
-		| 'CONNECTION_ESTABLISHED'
-		| 'CONNECTION_FAILED'
-		| 'CLOSING_CONNECTION'
-		| 'OPENING_CONNECTION'
-		| 'CLOSED'
-		| 'ONLINE'
-		| 'OFFLINE']: Partial<LinkedConnectionStates>;
-} = {
+export const CONNECTION_CHANGE = {
 	KEEP_ALIVE_MISSED: { keepAliveState: 'unhealthy' },
 	KEEP_ALIVE: { keepAliveState: 'healthy' },
 	CONNECTION_ESTABLISHED: { connectionState: 'connected' },
@@ -44,7 +33,7 @@ export const CONNECTION_CHANGE: {
 	CLOSED: { connectionState: 'disconnected' },
 	ONLINE: { networkState: 'connected' },
 	OFFLINE: { networkState: 'disconnected' },
-};
+} as const;
 
 export class ConnectionStateMonitor {
 	/**

package/src/utils/ReconnectionMonitor.ts

@@ -11,7 +11,7 @@ export enum ReconnectEvent {
 
 /**
  * Captures the reconnect event logic used to determine when to reconnect to PubSub providers.
- *   Reconnnect attempts are delayed by 5 seconds to let the interface settle.
+ *   Reconnect attempts are delayed by 5 seconds to let the interface settle.
  *   Attempting to reconnect only once creates unrecoverable states when the network state isn't
  *   supported by the browser, so this keeps retrying every minute until halted.
  */