Ask either cognitoIdpEndpoint or userPoolId (#204)

aws-samples · Sep 27, 2024 · b9d2423 · b9d2423
1 parent 0832b30
commit b9d2423
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/client/config.ts b/client/config.ts
@@ -92,11 +92,21 @@ export type ConfigWithDefaults = Config &
     Pick<Config, "storage" | "crypto" | "fetch" | "location" | "history">
   >;
 
+type ConfigInput = Omit<Config, "cognitoIdpEndpoint"> &
+  Partial<Pick<Config, "cognitoIdpEndpoint">>;
 let config_: ConfigWithDefaults | undefined = undefined;
-export function configure(config?: Config) {
+export function configure(config?: ConfigInput) {
   if (config) {
+    const cognitoIdpEndpoint =
+      config.cognitoIdpEndpoint || config.userPoolId?.split("_")[0];
+    if (!cognitoIdpEndpoint) {
+      throw new Error(
+        "Invalid configuration provided: either cognitoIdpEndpoint or userPoolId must be provided"
+      );
+    }
     config_ = {
       ...config,
+      cognitoIdpEndpoint,
       crypto: config.crypto ?? Defaults.crypto,
       storage: config.storage ?? Defaults.storage,
       fetch: config.fetch ?? Defaults.fetch,