From a100ea52a0fad67515f63c5b5368ad8659b8ff40 Mon Sep 17 00:00:00 2001
From: Jef Roosens <roosensjef@gmail.com>
Date: Fri, 27 Aug 2021 08:50:48 +0200
Subject: [PATCH] First draft stuff

---
 src/db/mod.rs       |  3 +++
 src/errors.rs       | 44 +++++++++++++++++++++++++-------------------
 src/routes/admin.rs | 19 ++++++++-----------
 3 files changed, 36 insertions(+), 30 deletions(-)

diff --git a/src/db/mod.rs b/src/db/mod.rs
index bf3b714..8979d35 100644
--- a/src/db/mod.rs
+++ b/src/db/mod.rs
@@ -1,2 +1,5 @@
 pub mod tokens;
 pub mod users;
+
+pub use users::{User, NewUser};
+pub use tokens::{RefreshToken, NewRefreshToken};
diff --git a/src/errors.rs b/src/errors.rs
index 7072dab..969e573 100644
--- a/src/errors.rs
+++ b/src/errors.rs
@@ -7,35 +7,41 @@ use rocket::{
 };
 
 #[derive(Debug)]
-pub enum RBError
+pub enum RbError
 {
-    /// When the login requests an unknown user
-    UnknownUser,
-    BlockedUser,
-    /// Invalid login password.
-    InvalidPassword,
-    /// When a non-admin user tries to use an admin endpoint
-    Unauthorized,
-    /// When an expired JWT token is used for auth.
-    JWTTokenExpired,
-    /// Umbrella error for when something goes wrong whilst creating a JWT token pair
-    JWTCreationError,
-    JWTError,
-    MissingJWTKey,
-    PWSaltError,
+    AuthUnknownUser,
+    AuthBlockedUser,
+    AuthInvalidPassword,
+    AuthUnauthorized,
+    AuthTokenExpired,
+    AuthRefreshTokenExpired,
+    AuthInvalidRefreshToken,
+    AuthDuplicateRefreshToken,
+
+    Custom(&'static str),
+
     AdminCreationError,
-    TokenExpired,
-    InvalidRefreshToken,
-    DuplicateRefreshToken,
     DBError,
     DuplicateUser,
 }
 
+impl RbError {
+    pub fn status(&self) -> Status {
+        Status::NotFound
+    }
+
+    pub fn message(&self) -> &'static str {
+        match self {
+
+        }
+    }
+}
+
 impl<'r> Responder<'r, 'static> for RBError
 {
     fn respond_to(self, _: &'r Request<'_>) -> response::Result<'static>
     {
-        let (status, message): (Status, &str) = match self {
+        let (status, message): (Status, &'static str) = match self {
             RBError::UnknownUser => (Status::NotFound, "Unknown user"),
             RBError::BlockedUser => (Status::Unauthorized, "This user is blocked"),
             RBError::InvalidPassword => (Status::Unauthorized, "Invalid password"),
diff --git a/src/routes/admin.rs b/src/routes/admin.rs
index b3db29b..87396ca 100644
--- a/src/routes/admin.rs
+++ b/src/routes/admin.rs
@@ -1,8 +1,5 @@
 use rb::{
-    db::{
-        users as db_users,
-        users::{NewUser, User},
-    },
+    db,
     errors::RBError,
 };
 use rocket::serde::json::Json;
@@ -12,29 +9,29 @@ use crate::{guards::Admin, RbDbConn};
 
 pub fn routes() -> Vec<rocket::Route>
 {
-    routes![get_users, get_user_info]
+    routes![get_users, get_user_info, create_user]
 }
 
 #[get("/users")]
-async fn get_users(admin: Admin, conn: RbDbConn) -> rb::Result<Json<Vec<User>>>
+async fn get_users(admin: Admin, conn: RbDbConn) -> rb::Result<Json<Vec<db::User>>>
 {
-    Ok(Json(conn.run(|c| rb::db::users::all(c)).await?))
+    Ok(Json(conn.run(|c| db::users::all(c)).await?))
 }
 
 #[post("/users", data = "<user>")]
-async fn create_user(admin: Admin, conn: RbDbConn, user: Json<NewUser>) -> rb::Result<()>
+async fn create_user(admin: Admin, conn: RbDbConn, user: Json<db::NewUser>) -> rb::Result<()>
 {
     Ok(conn
-        .run(move |c| db_users::create(c, &user.into_inner()))
+        .run(move |c| db::users::create(c, &user.into_inner()))
         .await?)
 }
 
 #[get("/users/<user_id_str>")]
-async fn get_user_info(_admin: Admin, conn: RbDbConn, user_id_str: &str) -> rb::Result<Json<User>>
+async fn get_user_info(_admin: Admin, conn: RbDbConn, user_id_str: &str) -> rb::Result<Json<db::User>>
 {
     let user_id = Uuid::parse_str(user_id_str).map_err(|_| RBError::UnknownUser)?;
 
-    match conn.run(move |c| db_users::find(c, user_id)).await {
+    match conn.run(move |c| db::users::find(c, user_id)).await {
         Some(user) => Ok(Json(user)),
         None => Err(RBError::UnknownUser),
     }