feat: add way to fetch and join invites from /invites/{id}

This commit is contained in:
Radical 2025-05-10 00:09:59 +02:00
parent c693e89853
commit 8935c2d496
3 changed files with 130 additions and 4 deletions

79
src/api/v1/invites/id.rs Normal file
View file

@ -0,0 +1,79 @@
use actix_web::{get, post, web, Error, HttpRequest, HttpResponse};
use crate::{api::v1::auth::check_access_token, structs::{Guild, Invite, Member}, utils::get_auth_header, Data};
#[get("{id}")]
pub async fn get(req: HttpRequest, path: web::Path<(String,)>, data: web::Data<Data>) -> Result<HttpResponse, Error> {
let headers = req.headers();
let auth_header = get_auth_header(headers);
if let Err(error) = auth_header {
return Ok(error)
}
let invite_id = path.into_inner().0;
let result = Invite::fetch_one(&data.pool, invite_id).await;
if let Err(error) = result {
return Ok(error)
}
let invite = result.unwrap();
let guild_result = Guild::fetch_one(&data.pool, invite.guild_uuid).await;
if let Err(error) = guild_result {
return Ok(error);
}
let guild = guild_result.unwrap();
Ok(HttpResponse::Ok().json(guild))
}
#[post("{id}")]
pub async fn join(req: HttpRequest, path: web::Path<(String,)>, data: web::Data<Data>) -> Result<HttpResponse, Error> {
let headers = req.headers();
let auth_header = get_auth_header(headers);
if let Err(error) = auth_header {
return Ok(error)
}
let invite_id = path.into_inner().0;
let authorized = check_access_token(auth_header.unwrap(), &data.pool).await;
if let Err(error) = authorized {
return Ok(error)
}
let uuid = authorized.unwrap();
let result = Invite::fetch_one(&data.pool, invite_id).await;
if let Err(error) = result {
return Ok(error)
}
let invite = result.unwrap();
let guild_result = Guild::fetch_one(&data.pool, invite.guild_uuid).await;
if let Err(error) = guild_result {
return Ok(error);
}
let guild = guild_result.unwrap();
let member = Member::new(&data.pool, uuid, guild.uuid).await;
if let Err(error) = member {
return Ok(error);
}
Ok(HttpResponse::Ok().json(guild))
}

View file

@ -0,0 +1,9 @@
use actix_web::{web, Scope};
mod id;
pub fn web() -> Scope {
web::scope("/invites")
.service(id::get)
.service(id::join)
}

View file

@ -444,7 +444,7 @@ impl Role {
pub struct Member {
pub uuid: Uuid,
pub nickname: String,
pub nickname: Option<String>,
pub user_uuid: Uuid,
pub guild_uuid: Uuid,
}
@ -475,15 +475,36 @@ impl Member {
return Err(HttpResponse::InternalServerError().finish())
}
let (uuid, nickname): (String, String) = row.unwrap();
let (uuid, nickname): (String, Option<String>) = row.unwrap();
Ok(Member {
Ok(Self {
uuid: Uuid::from_str(&uuid).unwrap(),
nickname,
user_uuid,
guild_uuid,
})
}
pub async fn new(pool: &Pool<Postgres>, user_uuid: Uuid, guild_uuid: Uuid) -> Result<Self, HttpResponse> {
let member_uuid = Uuid::now_v7();
let row = sqlx::query(&format!("INSERT INTO guild_members uuid, guild_uuid, user_uuid VALUES ('{}', '{}', '{}')", member_uuid, guild_uuid, user_uuid))
.execute(pool)
.await;
if let Err(error) = row {
error!("{}", error);
return Err(HttpResponse::InternalServerError().finish())
}
Ok(Self {
uuid: member_uuid,
nickname: None,
user_uuid,
guild_uuid,
})
}
}
#[derive(FromRow)]
@ -538,5 +559,22 @@ pub struct Invite {
/// User that created the invite
user_uuid: Uuid,
/// UUID of the guild that the invite belongs to
guild_uuid: Uuid,
pub guild_uuid: Uuid,
}
impl Invite {
pub async fn fetch_one(pool: &Pool<Postgres>, invite_id: String) -> Result<Self, HttpResponse> {
let invite: Result<InviteBuilder, sqlx::Error> = sqlx::query_as("SELECT id, user_uuid, guild_uuid FROM invites WHERE id = $1")
.bind(invite_id)
.fetch_one(pool)
.await;
if let Err(error) = invite {
error!("{}", error);
return Err(HttpResponse::InternalServerError().finish())
}
Ok(invite.unwrap().build())
}
}