feat: add more errors

jzeuzs · Jun 18, 2022 · bcb7e12 · bcb7e12
1 parent 308332e
commit bcb7e12
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/error.rs b/src/error.rs
@@ -26,6 +26,8 @@ pub enum Error {
     FailedToConnect,
     #[display(fmt = "An error occured when accessing the website.")]
     WebsiteError,
+    #[display(fmt = "You provided an invalid window size.")]
+    InvalidWindowSize,
 }
 
 impl ResponseError for Error {
@@ -40,7 +42,8 @@ impl ResponseError for Error {
             Error::InvalidUrl
             | Error::MissingAuthToken
             | Error::FailedToConnect
-            | Error::WebsiteError => StatusCode::BAD_REQUEST,
+            | Error::WebsiteError
+            | Error::InvalidWindowSize => StatusCode::BAD_REQUEST,
             Error::Unauthorized => StatusCode::UNAUTHORIZED,
             Error::ScreenshotNotFound => StatusCode::NOT_FOUND,
             Error::UrlNotSafeForWork => StatusCode::FORBIDDEN,

diff --git a/src/routes/screenshot.rs b/src/routes/screenshot.rs
@@ -65,8 +65,8 @@ pub async fn screenshot(
 
     let client = &data.browser;
 
-    client.goto(url.as_str()).await.expect("Failed navigating to site");
-    client.set_window_size(1980, 1080).await.expect("Failed setting window size");
+    client.goto(url.as_str()).await.map_err(|_| Error::WebsiteError)?;
+    client.set_window_size(1980, 1080).await.map_err(|_| Error::InvalidWindowSize)?;
     client
         .execute(
             "\