Improve error messages for dispatch failures

Author: kkashilk

crates/chat-cli/src/auth/pkce.rs

@@ -56,6 +56,8 @@ use crate::auth::{
     AuthError,
     START_URL,
 };
+use crate::aws_common::SdkErrorDisplay;
+use aws_smithy_runtime_api::client::result::SdkError;
 use crate::database::Database;
 
 const DEFAULT_AUTHORIZATION_TIMEOUT: Duration = Duration::from_secs(60 * 3);
@@ -139,7 +141,10 @@ impl PkceClient for Client {
         for scope in Self::scopes() {
             register = register.scopes(scope);
         }
-        let output = register.send().await?;
+        let output = register.send().await.map_err(|e| match e {
+            SdkError::DispatchFailure(_) => AuthError::OAuthCustomError(format!("Authentication failed: {}", SdkErrorDisplay(&e))),
+            other => other.into()
+        })?;
         Ok(RegisterClientResponse { output })
     }
 
@@ -153,7 +158,11 @@ impl PkceClient for Client {
             .code_verifier(args.code_verifier)
             .code(args.code)
             .send()
-            .await?;
+            .await
+            .map_err(|e| match e {
+                SdkError::DispatchFailure(_) => AuthError::OAuthCustomError(format!("Authentication failed: {}", SdkErrorDisplay(&e))),
+                other => other.into()
+            })?;
         Ok(CreateTokenResponse { output })
     }
 }