feat: Implement game opinion functionality, update authentication to store username in token, and protect user list with a mutex.
This commit is contained in:
parent
3ecc971e8f
commit
f30af57934
@ -20,7 +20,10 @@ impl AuthState {
|
||||
|
||||
#[derive(Debug)]
|
||||
#[allow(dead_code)]
|
||||
pub struct Token(pub String);
|
||||
pub struct Token {
|
||||
pub token: String,
|
||||
pub username: String,
|
||||
}
|
||||
|
||||
#[rocket::async_trait]
|
||||
impl<'r> rocket::request::FromRequest<'r> for Token {
|
||||
@ -38,8 +41,11 @@ impl<'r> rocket::request::FromRequest<'r> for Token {
|
||||
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()));
|
||||
if let Some(username) = tokens.get(token) {
|
||||
return rocket::request::Outcome::Success(Token {
|
||||
token: token.to_string(),
|
||||
username: username.clone(),
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
@ -53,12 +59,13 @@ impl<'r> rocket::request::FromRequest<'r> for Token {
|
||||
#[post("/login", data = "<request>")]
|
||||
pub fn login(
|
||||
state: &State<AuthState>,
|
||||
user_list: &State<Vec<crate::User>>,
|
||||
user_list: &State<Mutex<Vec<crate::User>>>,
|
||||
request: Proto<items::LoginRequest>,
|
||||
) -> items::LoginResponse {
|
||||
let req = request.into_inner();
|
||||
let users = user_list.lock().unwrap();
|
||||
|
||||
if let Some(user) = user_list.iter().find(|u| u.name == req.username)
|
||||
if let Some(user) = users.iter().find(|u| u.person.name == req.username)
|
||||
&& bcrypt::verify(&req.password, &user.password_hash).unwrap_or(false)
|
||||
{
|
||||
let token = Uuid::new_v4().to_string();
|
||||
|
||||
@ -1,4 +1,5 @@
|
||||
use rocket::fs::FileServer;
|
||||
use std::sync::Mutex;
|
||||
|
||||
use crate::items::Game;
|
||||
|
||||
@ -29,19 +30,63 @@ impl std::ops::Deref for User {
|
||||
#[get("/<name>")]
|
||||
fn get_user(
|
||||
_token: auth::Token,
|
||||
user_list: &rocket::State<Vec<User>>,
|
||||
user_list: &rocket::State<Mutex<Vec<User>>>,
|
||||
name: String,
|
||||
) -> Option<items::Person> {
|
||||
user_list
|
||||
let users = user_list.lock().unwrap();
|
||||
users
|
||||
.iter()
|
||||
.find(|user| user.person.name == name)
|
||||
.map(|u| u.person.clone())
|
||||
}
|
||||
|
||||
#[get("/")]
|
||||
fn get_users(_token: auth::Token, user_list: &rocket::State<Vec<User>>) -> items::PersonList {
|
||||
fn get_users(
|
||||
_token: auth::Token,
|
||||
user_list: &rocket::State<Mutex<Vec<User>>>,
|
||||
) -> items::PersonList {
|
||||
let users = user_list.lock().unwrap();
|
||||
items::PersonList {
|
||||
person: user_list.inner().iter().map(|u| u.person.clone()).collect(),
|
||||
person: users.iter().map(|u| u.person.clone()).collect(),
|
||||
}
|
||||
}
|
||||
|
||||
#[get("/game/<title>")]
|
||||
fn get_game(
|
||||
_token: auth::Token,
|
||||
game_list: &rocket::State<Vec<Game>>,
|
||||
title: String,
|
||||
) -> Option<items::Game> {
|
||||
game_list.iter().find(|g| g.title == title).cloned()
|
||||
}
|
||||
|
||||
#[post("/opinion", data = "<req>")]
|
||||
fn add_opinion(
|
||||
token: auth::Token,
|
||||
user_list: &rocket::State<Mutex<Vec<User>>>,
|
||||
req: proto_utils::Proto<items::AddOpinionRequest>,
|
||||
) -> Option<items::Person> {
|
||||
let mut users = user_list.lock().unwrap();
|
||||
if let Some(user) = users.iter_mut().find(|u| u.person.name == token.username) {
|
||||
let req = req.into_inner();
|
||||
let opinion = items::Opinion {
|
||||
title: req.game_title.clone(),
|
||||
would_play: req.would_play,
|
||||
};
|
||||
|
||||
if let Some(existing) = user
|
||||
.person
|
||||
.opinion
|
||||
.iter_mut()
|
||||
.find(|o| o.title == req.game_title)
|
||||
{
|
||||
existing.would_play = req.would_play;
|
||||
} else {
|
||||
user.person.opinion.push(opinion);
|
||||
}
|
||||
Some(user.person.clone())
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
@ -84,9 +129,10 @@ fn rocket() -> _ {
|
||||
});
|
||||
|
||||
rocket::build()
|
||||
.manage(user_list)
|
||||
.manage(Mutex::new(user_list))
|
||||
.manage(auth::AuthState::new())
|
||||
.mount("/api", routes![get_users, get_user])
|
||||
.manage(game_list)
|
||||
.mount("/api", routes![get_users, get_user, get_game, add_opinion])
|
||||
.mount(
|
||||
"/auth",
|
||||
routes![auth::login, auth::logout, auth::get_auth_status],
|
||||
|
||||
@ -64,7 +64,12 @@ service AuthService {
|
||||
|
||||
message GameRequest { string title = 1; }
|
||||
|
||||
message AddOpinionRequest {
|
||||
string game_title = 1;
|
||||
bool would_play = 2;
|
||||
}
|
||||
|
||||
service MainService {
|
||||
rpc GetGame(GameRequest) returns (Game);
|
||||
rpc AddOpinion(GameRequest) returns (Person);
|
||||
rpc AddOpinion(AddOpinionRequest) returns (Person);
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user