diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/src/main.rs b/src/main.rs index 81b416c..3ad9efd 100644 --- a/src/main.rs +++ b/src/main.rs @@ -125,7 +125,10 @@ async fn new_user(data: Json<RegisterForm>, db: &State<Mutex<DB>>) -> (Status, R // User has been created Ok(u) => return (Status::Created, Ok(Json(u.into()))), // Could not create user - Err(n) => return (Status::InternalServerError, Err(n.into())), + Err(e) => { + db.uid_generator.delete_uid(n); + return (Status::InternalServerError, Err(e.into())) + }, } } } @@ -239,16 +242,23 @@ async fn new_registration_key(data: Json<NewRegistrationKeyForm>, db: &State<Mut } } -#[post("/regkey/list", data="<data>", format="json")] -async fn list_registration_keys(data: Json<LogoutForm>, db: &State<Mutex<DB>>) -> (Status, Result<(), Json<String>>) { - (Status::NotFound, Err("NOT IMPLEMENTED".to_string().into())) -} - #[post("/regkey/del", data="<data>", format="json")] async fn del_registration_key(data: Json<NewRegistrationKeyForm>, db: &State<Mutex<DB>>) -> (Status, Result<(), Json<String>>) { - (Status::NotFound, Err("NOT IMPLEMENTED".to_string().into())) + let mut db = db.lock().await; + match db.del_registration_key(data.id, &data.session, &data.key).await { + Ok(_) => (Status::Ok, Ok(())), + Err(n) => (Status::InternalServerError, Err(n.into())) + } } +#[post("/regkey/list", data="<data>", format="json")] +async fn list_registration_keys(data: Json<LogoutForm>, db: &State<Mutex<DB>>) -> (Status, Result<Json<Vec<String>>, Json<String>>) { + let db = db.lock().await; + match db.list_registration_keys(data.id, &data.session).await { + Ok(n) => (Status::Ok, Ok(n.into())), + Err(n) => (Status::InternalServerError, Err(n.into())), + } +} #[derive(Deserialize, Debug)] struct TransferForm { |