Merge pull request #6 from FxQnLr/ping

Ping

1 files changed, 24 insertions, 8 deletions

diff --git a/src/error.rs b/src/error.rs
index db2fc86..5b82534 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -1,4 +1,4 @@
-use std::error::Error;
+use std::io;
 use axum::http::StatusCode;
 use axum::Json;
 use axum::response::{IntoResponse, Response};
@@ -8,25 +8,41 @@ use crate::auth::AuthError;
 
 #[derive(Debug)]
 pub enum WebolError {
-    Auth(AuthError),
     Generic,
-    Server(Box<dyn Error>),
+    Auth(AuthError),
+    DB(sqlx::Error),
+    IpParse(<std::net::IpAddr as std::str::FromStr>::Err),
+    BufferParse(std::num::ParseIntError),
+    Broadcast(io::Error),
 }
 
 impl IntoResponse for WebolError {
     fn into_response(self) -> Response {
         let (status, error_message) = match self {
-            WebolError::Auth(err) => err.get(),
-            WebolError::Generic => (StatusCode::INTERNAL_SERVER_ERROR, ""),
-            WebolError::Server(err) => {
+            Self::Auth(err) => {
+                err.get()
+            },
+            Self::Generic => (StatusCode::INTERNAL_SERVER_ERROR, ""),
+            Self::IpParse(err) => {
+                error!("server error: {}", err.to_string());
+                (StatusCode::INTERNAL_SERVER_ERROR, "Server Error")
+            },
+            Self::DB(err) => {
+                error!("server error: {}", err.to_string());
+                (StatusCode::INTERNAL_SERVER_ERROR, "Server Error")
+            },
+            Self::Broadcast(err) => {
+                error!("server error: {}", err.to_string());
+                (StatusCode::INTERNAL_SERVER_ERROR, "Server Error")
+            },
+            Self::BufferParse(err) => {
                 error!("server error: {}", err.to_string());
                 (StatusCode::INTERNAL_SERVER_ERROR, "Server Error")
             },
-
         };
         let body = Json(json!({
             "error": error_message,
         }));
         (status, body).into_response()
     }
-}
+}
\ No newline at end of file