This commit is contained in:
JMARyA 2024-07-26 14:14:08 +02:00
parent dcf546fa9c
commit 1e67f223f7
Signed by: jmarya
GPG key ID: 901B2ADDF27C2263
10 changed files with 164 additions and 92 deletions

View file

@ -1,19 +1,18 @@
use rocket::http::Status;
use crate::library::user::Session;
use crate::library::user::User;
use mongod::Model;
use mongodb::bson::doc;
use rocket::http::Status;
use rocket::outcome::Outcome;
use rocket::post;
use rocket::request::FromRequest;
use rocket::Request;
use serde_json::json;
use serde::Deserialize;
use rocket::serde::json::Json;
use crate::library::user::Session;
use crate::library::user::User;
use rocket::Request;
use serde::Deserialize;
use serde_json::json;
use super::FallibleApiResponse;
use super::api_error;
use super::FallibleApiResponse;
#[rocket::async_trait]
impl<'r> FromRequest<'r> for User {
@ -22,13 +21,13 @@ impl<'r> FromRequest<'r> for User {
async fn from_request(request: &'r Request<'_>) -> rocket::request::Outcome<Self, Self::Error> {
match request.headers().get_one("token") {
Some(key) => {
if let Some(session) = Session::find_one(doc! { "token": key} ).await {
if let Some(session) = Session::find_one(doc! { "token": key}).await {
let user = session.user.get().await;
Outcome::Success(user)
} else {
Outcome::Error((Status::Unauthorized, ()))
}
},
}
None => Outcome::Error((Status::Unauthorized, ())),
}
}
@ -37,15 +36,16 @@ impl<'r> FromRequest<'r> for User {
#[derive(Deserialize)]
pub struct LoginData {
pub username: String,
pub password: String
pub password: String,
}
#[post("/login", data = "<login>")]
pub async fn login_route(login: Json<LoginData>) -> FallibleApiResponse {
let ses = User::login(&login.username, &login.password).await.ok_or_else(|| api_error("Login failed"))?;
let ses = User::login(&login.username, &login.password)
.await
.ok_or_else(|| api_error("Login failed"))?;
Ok(json!({
"token": ses.token
}))
}