feat: Implement token-based authentication for user data endpoints.
This commit is contained in:
@@ -18,6 +18,38 @@ impl AuthState {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
#[allow(dead_code)]
|
||||
pub struct Token(pub String);
|
||||
|
||||
#[rocket::async_trait]
|
||||
impl<'r> rocket::request::FromRequest<'r> for Token {
|
||||
type Error = ();
|
||||
|
||||
async fn from_request(
|
||||
request: &'r rocket::Request<'_>,
|
||||
) -> rocket::request::Outcome<Self, Self::Error> {
|
||||
let token = request.headers().get_one("Authorization");
|
||||
|
||||
match token {
|
||||
Some(token) => {
|
||||
// Check if token starts with "Bearer "
|
||||
if let Some(token) = token.strip_prefix("Bearer ") {
|
||||
let state = request.guard::<&State<AuthState>>().await.unwrap();
|
||||
let tokens = state.tokens.lock().unwrap();
|
||||
|
||||
if tokens.contains_key(token) {
|
||||
return rocket::request::Outcome::Success(Token(token.to_string()));
|
||||
}
|
||||
}
|
||||
|
||||
rocket::request::Outcome::Error((rocket::http::Status::Unauthorized, ()))
|
||||
}
|
||||
None => rocket::request::Outcome::Error((rocket::http::Status::Unauthorized, ())),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[post("/login", data = "<request>")]
|
||||
pub fn login(
|
||||
state: &State<AuthState>,
|
||||
|
||||
+6
-2
@@ -25,7 +25,11 @@ impl std::ops::Deref for User {
|
||||
}
|
||||
|
||||
#[get("/<name>")]
|
||||
fn get_user(user_list: &rocket::State<Vec<User>>, name: String) -> Option<items::Person> {
|
||||
fn get_user(
|
||||
_token: auth::Token,
|
||||
user_list: &rocket::State<Vec<User>>,
|
||||
name: String,
|
||||
) -> Option<items::Person> {
|
||||
user_list
|
||||
.iter()
|
||||
.find(|user| user.person.name == name)
|
||||
@@ -33,7 +37,7 @@ fn get_user(user_list: &rocket::State<Vec<User>>, name: String) -> Option<items:
|
||||
}
|
||||
|
||||
#[get("/")]
|
||||
fn get_users(user_list: &rocket::State<Vec<User>>) -> items::PersonList {
|
||||
fn get_users(_token: auth::Token, user_list: &rocket::State<Vec<User>>) -> items::PersonList {
|
||||
items::PersonList {
|
||||
person: user_list
|
||||
.inner()
|
||||
|
||||
Reference in New Issue
Block a user