add identityPoolEndpoint config to allow using a custom Cognito-Ident…

…ity endpoint
aws-amplify · Jun 30, 2024 · aca9ec0 · aca9ec0
1 parent b6de5f9
commit aca9ec0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/packages/core/src/awsClients/cognitoIdentity/base.ts b/packages/core/src/awsClients/cognitoIdentity/base.ts
@@ -1,6 +1,8 @@
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 
+import { Amplify } from '@aws-amplify/core';
+
 import {
 	Endpoint,
 	EndpointResolverOptions,
@@ -29,11 +31,17 @@ const SERVICE_NAME = 'cognito-identity';
 /**
  * The endpoint resolver function that returns the endpoint URL for a given region.
  */
-const endpointResolver = ({ region }: EndpointResolverOptions) => ({
-	url: new AmplifyUrl(
-		`https://cognito-identity.${region}.${getDnsSuffix(region)}`,
-	),
-});
+const endpointResolver = ({ region }: EndpointResolverOptions) => {
+	const authConfig = Amplify.getConfig().Auth?.Cognito;
+	const customURL = authConfig?.userPoolEndpoint;
+	const defaultURL = new AmplifyUrl(
+		`https://${SERVICE_NAME}.${region}.${getDnsSuffix(region)}`,
+	);
+
+	return {
+		url: customURL ? new AmplifyUrl(customURL) : defaultURL,
+	};
+};
 
 /**
  * A Cognito Identity-specific middleware that disables caching for all requests.

diff --git a/packages/core/src/singleton/Auth/types.ts b/packages/core/src/singleton/Auth/types.ts
@@ -135,12 +135,14 @@ export interface AuthIdentityPoolConfig {
 
 export interface CognitoIdentityPoolConfig {
 	identityPoolId: string;
+	identityPoolEndpoint?: string;
 	allowGuestAccess?: boolean;
 }
 
 export interface AuthUserPoolConfig {
 	Cognito: CognitoUserPoolConfig & {
 		identityPoolId?: never;
+		identityPoolEndpoint?: never;
 		allowGuestAccess?: never;
 	};
 }