Set up the authentication filter

This commit is contained in:
Savanni D'Gerinel 2022-11-18 08:44:17 -05:00
parent 23b28144a1
commit 977059c90d
2 changed files with 82 additions and 49 deletions

View File

@ -140,7 +140,7 @@ impl FromSql for Username {
}
}
pub trait AuthenticationDB {
pub trait AuthenticationDB: Send + Sync {
fn create_user(&mut self, username: Username) -> AppResult<UserId, AuthenticationError>;
fn create_invitation(&mut self, user: UserId) -> AppResult<Invitation, AuthenticationError>;

View File

@ -6,11 +6,43 @@ use std::{
use warp::Filter;
mod authentication;
use authentication::{AuthenticationDB, MemoryAuth, Username};
use authentication::{AuthenticationDB, AuthenticationError, MemoryAuth, UserId, Username};
mod database;
mod errors;
#[derive(Debug)]
struct AuthenticationRefused;
impl warp::reject::Reject for AuthenticationRefused {}
fn with_authentication(
auth_ctx: Arc<RwLock<impl AuthenticationDB>>,
) -> impl Filter<Extract = ((Username, UserId),), Error = warp::Rejection> + Clone {
let auth_ctx = auth_ctx.clone();
warp::header("authentication").and_then({
let auth_ctx = auth_ctx.clone();
move |auth_header: String| {
let auth_ctx = auth_ctx.clone();
async move {
if auth_header.starts_with("Basic ") {
let username = auth_header.split(" ").skip(1).collect::<String>();
match auth_ctx
.read()
.unwrap()
.get_user_id(Username::from(username.as_str()))
{
Ok(Ok(userid)) => Ok((Username::from(username.as_str()), userid)),
Ok(Err(_)) => Err(warp::reject::custom(AuthenticationRefused)),
Err(err) => panic!("{}", err),
}
} else {
Err(warp::reject::custom(AuthenticationRefused))
}
}
}
})
}
#[derive(Serialize)]
struct ErrorResponse {
error: String,
@ -26,18 +58,14 @@ struct MakeUserResponse {
userid: String,
}
#[tokio::main]
pub async fn main() {
let auth_ctx: Arc<RwLock<MemoryAuth>> = Arc::new(RwLock::new(Default::default()));
let make_user = {
let auth_ctx = auth_ctx.clone();
warp::path!("api" / "v1" / "user")
fn make_user(
auth_ctx: Arc<RwLock<impl AuthenticationDB>>,
) -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
warp::path!("api" / "v1" / "users")
.and(warp::put())
.and(warp::body::json())
.map(move |params: MakeUserParameters| {
let mut auth_ctx = auth_ctx.write().unwrap();
println!("{:?}", *auth_ctx);
match (*auth_ctx).create_user(Username::from(params.username.as_str())) {
Ok(Ok(userid)) => warp::reply::json(&MakeUserResponse {
userid: String::from(userid),
@ -48,15 +76,15 @@ pub async fn main() {
Err(err) => panic!("{}", err),
}
})
};
}
let list_users = {
let auth_ctx = auth_ctx.clone();
warp::path!("api" / "v1" / "user")
fn list_users(
auth_ctx: Arc<RwLock<impl AuthenticationDB>>,
) -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
warp::path!("api" / "v1" / "users")
.and(warp::get())
.map(move || {
let auth_ctx = auth_ctx.read().unwrap();
println!("{:?}", *auth_ctx);
match (*auth_ctx).list_users() {
Ok(Ok(users)) => warp::reply::json(
&users
@ -70,7 +98,11 @@ pub async fn main() {
Err(err) => panic!("{}", err),
}
})
};
}
#[tokio::main]
pub async fn main() {
let auth_ctx: Arc<RwLock<MemoryAuth>> = Arc::new(RwLock::new(Default::default()));
let echo_unauthenticated = warp::path!("api" / "v1" / "echo" / String).map(|param: String| {
println!("param: {}", param);
@ -85,15 +117,16 @@ pub async fn main() {
*/
let echo_authenticated = warp::path!("api" / "v1" / "echo" / String)
.and(warp::header::<String>("authentication"))
.map(|auth: String, param: String| {
println!("auth: {}", auth);
.and(with_authentication(auth_ctx.clone()))
.map(|param: String, (username, userid)| {
println!("param: {:?}", username);
println!("param: {:?}", userid);
println!("param: {}", param);
warp::reply::json(&vec!["authed", param.as_str()])
});
let filter = list_users
.or(make_user)
let filter = list_users(auth_ctx.clone())
.or(make_user(auth_ctx.clone()))
.or(echo_authenticated)
.or(echo_unauthenticated);