agliyo

1 files changed, 5 insertions, 26 deletions

diff --git a/src/auth.rs b/src/auth.rs
index ced9e8e..03930f0 100644
--- a/src/auth.rs
+++ b/src/auth.rs
@@ -63,29 +63,10 @@ struct Claims {
     puk: String,
 }
 
-// #[derive(Error, Debug)]
-// pub enum Nope {
-//     #[error("Invalid header")]
-//     InvalidHeader {
-//         expected: String,
-//         found: String,
-//     },
-// }
-// impl warp::reject::Reject for Nope {}
+#[derive(Debug)]
+struct RateLimited;
 
-#[derive(Error, Debug)]
-pub enum DataStoreError {
-    #[error("invalid header")]
-    InvalidHeader {},
-}
-
-impl Reject for DataStoreError {}
-
-// impl From<LessThanTenError> for Rejection {
-//     fn from(other: LessThanTenError) -> Self {
-//         warp::reject::custom(other)
-//     }
-// }
+impl Reject for RateLimited {}
 
 pub fn with_auth(
     db: Db,
@@ -94,10 +75,9 @@ pub fn with_auth(
     headers_cloned()
         .map(move |headers: HeaderMap<HeaderValue>| (db.clone(), headers))
         .and_then(authorize)
-        .recover(handle_rejection())
 }
 
-async fn authorize((db, headers): (Db, HeaderMap<HeaderValue>)) -> Result<String, Infallible> {
+async fn authorize((db, headers): (Db, HeaderMap<HeaderValue>)) -> Result<String, Rejection> {
     match jwt_from_header(&headers) {
         Ok(jwt) => {
             let decoded = decode::<Claims>(
@@ -111,8 +91,7 @@ async fn authorize((db, headers): (Db, HeaderMap<HeaderValue>)) -> Result<String
 
             Ok(decoded.claims.puk)
         }
-        Err(e) => return (StatusCode::UNAUTHORIZED, e.to_string()),
-        // warp error
+        Err(e) => return Err(warp::reject::custom(RateLimited)),
     }
 }