Handle unauthorized error

prisma · Oct 2, 2024 · 7c69bf5 · 7c69bf5
1 parent e99704a
commit 7c69bf5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/quaint/src/connector/postgres/native/websocket.rs b/quaint/src/connector/postgres/native/websocket.rs
@@ -2,15 +2,20 @@ use std::str::FromStr;
 
 use async_tungstenite::{
     tokio::connect_async,
-    tungstenite::{self, client::IntoClientRequest, http::HeaderValue, Error as TungsteniteError},
+    tungstenite::{
+        self,
+        client::IntoClientRequest,
+        http::{HeaderValue, StatusCode},
+        Error as TungsteniteError,
+    },
 };
 use futures::FutureExt;
 use tokio_postgres::{Client, Config, NoTls};
 use ws_stream_tungstenite::WsStream;
 
 use crate::{
     connector::PostgresWebSocketUrl,
-    error::{self, Error, ErrorKind, NativeErrorKind},
+    error::{self, Error, ErrorKind, Name, NativeErrorKind},
 };
 
 const CONNECTION_PARAMS_HEADER: &str = "Prisma-Connection-Parameters";
@@ -60,6 +65,12 @@ impl From<TungsteniteError> for error::Error {
                 message: tls_error.to_string(),
             })),
 
+            TungsteniteError::Http(response) if response.status() == StatusCode::UNAUTHORIZED => {
+                Error::builder(ErrorKind::DatabaseAccessDenied {
+                    db_name: Name::Unavailable,
+                })
+            }
+
             _ => Error::builder(ErrorKind::Native(NativeErrorKind::ConnectionError(Box::new(value)))),
         };
 

diff --git a/schema-engine/connectors/sql-schema-connector/src/flavour/postgres.rs b/schema-engine/connectors/sql-schema-connector/src/flavour/postgres.rs
@@ -51,10 +51,10 @@ impl MigratePostgresUrl {
                     "Required `apiKey` query string parameter was not provided in a connection URL",
                 ));
             };
-            PostgresUrl::new_websocket(ws_url, api_key.into_owned()).map_err(ConnectorError::url_parse_error)?
+            PostgresUrl::new_websocket(ws_url, api_key.into_owned())
         } else {
-            PostgresUrl::new_native(url).map_err(ConnectorError::url_parse_error)?
-        };
+            PostgresUrl::new_native(url)
+        }.map_err(ConnectorError::url_parse_error)?;
 
         Ok(Self(postgres_url))
     }