feat: implement message fetching
This commit is contained in:
parent
6374963e2f
commit
daf61e0275
2 changed files with 120 additions and 3 deletions
|
@ -0,0 +1,73 @@
|
||||||
|
use actix_web::{get, web, Error, HttpRequest, HttpResponse};
|
||||||
|
use serde::Deserialize;
|
||||||
|
use crate::{api::v1::auth::check_access_token, structs::{Channel, Member}, utils::get_auth_header, Data};
|
||||||
|
use ::uuid::Uuid;
|
||||||
|
use log::error;
|
||||||
|
|
||||||
|
#[derive(Deserialize)]
|
||||||
|
struct MessageRequest {
|
||||||
|
amount: i64,
|
||||||
|
offset: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[get("{uuid}/channels/{channel_uuid}/messages")]
|
||||||
|
pub async fn res(req: HttpRequest, path: web::Path<(Uuid, Uuid)>, message_request: web::Json<MessageRequest>, 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 (guild_uuid, channel_uuid) = path.into_inner();
|
||||||
|
|
||||||
|
let authorized = check_access_token(auth_header.unwrap(), &data.pool).await;
|
||||||
|
|
||||||
|
if let Err(error) = authorized {
|
||||||
|
return Ok(error)
|
||||||
|
}
|
||||||
|
|
||||||
|
let uuid = authorized.unwrap();
|
||||||
|
|
||||||
|
let member = Member::fetch_one(&data.pool, uuid, guild_uuid).await;
|
||||||
|
|
||||||
|
if let Err(error) = member {
|
||||||
|
return Ok(error);
|
||||||
|
}
|
||||||
|
|
||||||
|
let cache_result = data.get_cache_key(format!("{}", channel_uuid)).await;
|
||||||
|
|
||||||
|
let mut channel_raw: Option<Channel> = None;
|
||||||
|
|
||||||
|
if let Ok(cache_hit) = cache_result {
|
||||||
|
channel_raw = Some(serde_json::from_str(&cache_hit).unwrap())
|
||||||
|
}
|
||||||
|
|
||||||
|
if channel_raw.is_none() {
|
||||||
|
let channel_result = Channel::fetch_one(&data.pool, guild_uuid, channel_uuid).await;
|
||||||
|
|
||||||
|
if let Err(error) = channel_result {
|
||||||
|
return Ok(error)
|
||||||
|
}
|
||||||
|
|
||||||
|
channel_raw = Some(channel_result.unwrap());
|
||||||
|
|
||||||
|
let cache_result = data.set_cache_key(format!("{}", channel_uuid), channel_raw.clone().unwrap(), 60).await;
|
||||||
|
|
||||||
|
if let Err(error) = cache_result {
|
||||||
|
error!("{}", error);
|
||||||
|
return Ok(HttpResponse::InternalServerError().finish());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let channel = channel_raw.unwrap();
|
||||||
|
|
||||||
|
let messages = channel.fetch_messages(&data.pool, message_request.amount, message_request.offset).await;
|
||||||
|
|
||||||
|
if let Err(error) = messages {
|
||||||
|
return Ok(error)
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(HttpResponse::Ok().json(messages.unwrap()))
|
||||||
|
}
|
|
@ -1,6 +1,6 @@
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use serde::Serialize;
|
use serde::{Deserialize, Serialize};
|
||||||
use sqlx::{prelude::FromRow, Pool, Postgres};
|
use sqlx::{prelude::FromRow, Pool, Postgres};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
use actix_web::HttpResponse;
|
use actix_web::HttpResponse;
|
||||||
|
@ -8,7 +8,7 @@ use log::error;
|
||||||
|
|
||||||
use crate::Data;
|
use crate::Data;
|
||||||
|
|
||||||
#[derive(Serialize, Clone)]
|
#[derive(Serialize, Deserialize, Clone)]
|
||||||
pub struct Channel {
|
pub struct Channel {
|
||||||
pub uuid: Uuid,
|
pub uuid: Uuid,
|
||||||
pub guild_uuid: Uuid,
|
pub guild_uuid: Uuid,
|
||||||
|
@ -32,7 +32,7 @@ impl ChannelPermissionBuilder {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Clone, FromRow)]
|
#[derive(Serialize, Deserialize, Clone, FromRow)]
|
||||||
pub struct ChannelPermission {
|
pub struct ChannelPermission {
|
||||||
pub role_uuid: Uuid,
|
pub role_uuid: Uuid,
|
||||||
pub permissions: i32
|
pub permissions: i32
|
||||||
|
@ -155,6 +155,23 @@ impl Channel {
|
||||||
|
|
||||||
Ok(channel)
|
Ok(channel)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn fetch_messages(&self, pool: &Pool<Postgres>, amount: i64, offset: i64) -> Result<Vec<Message>, HttpResponse> {
|
||||||
|
let row = sqlx::query_as(&format!("SELECT uuid, user_uuid, message FROM channels WHERE channel_uuid = '{}' ORDER BY uuid LIMIT $1 OFFSET $2", self.uuid))
|
||||||
|
.bind(amount)
|
||||||
|
.bind(offset)
|
||||||
|
.fetch_all(pool)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
if let Err(error) = row {
|
||||||
|
error!("{}", error);
|
||||||
|
return Err(HttpResponse::InternalServerError().finish());
|
||||||
|
}
|
||||||
|
|
||||||
|
let message_builders: Vec<MessageBuilder> = row.unwrap();
|
||||||
|
|
||||||
|
Ok(message_builders.iter().map(|b| b.build()).collect())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, Copy)]
|
#[derive(Clone, Copy)]
|
||||||
|
@ -349,3 +366,30 @@ impl Member {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(FromRow)]
|
||||||
|
struct MessageBuilder {
|
||||||
|
uuid: String,
|
||||||
|
channel_uuid: String,
|
||||||
|
user_uuid: String,
|
||||||
|
message: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MessageBuilder {
|
||||||
|
fn build(&self) -> Message {
|
||||||
|
Message {
|
||||||
|
uuid: Uuid::from_str(&self.uuid).unwrap(),
|
||||||
|
channel_uuid: Uuid::from_str(&self.channel_uuid).unwrap(),
|
||||||
|
user_uuid: Uuid::from_str(&self.user_uuid).unwrap(),
|
||||||
|
message: self.message.clone(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
pub struct Message {
|
||||||
|
uuid: Uuid,
|
||||||
|
channel_uuid: Uuid,
|
||||||
|
user_uuid: Uuid,
|
||||||
|
message: String,
|
||||||
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue