Compare commits
3 commits
4cb89645fe
...
447c577a2a
Author | SHA1 | Date | |
---|---|---|---|
447c577a2a | |||
5b24d0052b | |||
8a7711cabc |
3 changed files with 39 additions and 11 deletions
|
@ -284,7 +284,7 @@ impl Me {
|
|||
cache_pool: &redis::Client,
|
||||
new_status: i16,
|
||||
) -> Result<(), Error> {
|
||||
if new_status > 4 || new_status < 0 {
|
||||
if !(0..=4).contains(&new_status) {
|
||||
return Err(Error::BadRequest("Invalid status code".to_string()));
|
||||
}
|
||||
self.online_status = new_status;
|
||||
|
|
|
@ -107,7 +107,7 @@ impl MemberBuilder {
|
|||
}
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
#[derive(Serialize, Deserialize, Clone)]
|
||||
pub struct Member {
|
||||
pub uuid: Uuid,
|
||||
pub nickname: Option<String>,
|
||||
|
@ -150,14 +150,17 @@ impl Member {
|
|||
pub async fn fetch_one(
|
||||
conn: &mut Conn,
|
||||
cache_pool: &redis::Client,
|
||||
me: &Me,
|
||||
me: Option<&Me>,
|
||||
user_uuid: Uuid,
|
||||
guild_uuid: Uuid,
|
||||
) -> Result<Self, Error> {
|
||||
let member: MemberBuilder;
|
||||
let user: UserBuilder;
|
||||
let friend: Option<Friend>;
|
||||
use friends::dsl as fdsl;
|
||||
use guild_members::dsl;
|
||||
let (member, user, friend): (MemberBuilder, UserBuilder, Option<Friend>) =
|
||||
dsl::guild_members
|
||||
if let Some(me) = me {
|
||||
(member, user, friend) = dsl::guild_members
|
||||
.filter(dsl::guild_uuid.eq(guild_uuid))
|
||||
.filter(dsl::user_uuid.eq(user_uuid))
|
||||
.inner_join(users::table)
|
||||
|
@ -174,6 +177,17 @@ impl Member {
|
|||
))
|
||||
.get_result(conn)
|
||||
.await?;
|
||||
} else {
|
||||
(member, user) = dsl::guild_members
|
||||
.filter(dsl::guild_uuid.eq(guild_uuid))
|
||||
.filter(dsl::user_uuid.eq(user_uuid))
|
||||
.inner_join(users::table)
|
||||
.select((MemberBuilder::as_select(), UserBuilder::as_select()))
|
||||
.get_result(conn)
|
||||
.await?;
|
||||
|
||||
friend = None;
|
||||
}
|
||||
|
||||
member
|
||||
.build_with_parts(conn, cache_pool, user, friend)
|
||||
|
|
|
@ -1,10 +1,15 @@
|
|||
use diesel::{Insertable, Queryable, Selectable};
|
||||
use diesel::{ExpressionMethods, Insertable, QueryDsl, Queryable, Selectable};
|
||||
use diesel_async::RunQueryDsl;
|
||||
use serde::Serialize;
|
||||
use uuid::Uuid;
|
||||
|
||||
use crate::{Conn, error::Error, schema::messages};
|
||||
use crate::{
|
||||
Conn,
|
||||
error::Error,
|
||||
schema::{channels, guilds, messages},
|
||||
};
|
||||
|
||||
use super::User;
|
||||
use super::Member;
|
||||
|
||||
#[derive(Clone, Queryable, Selectable, Insertable)]
|
||||
#[diesel(table_name = messages)]
|
||||
|
@ -23,7 +28,16 @@ impl MessageBuilder {
|
|||
conn: &mut Conn,
|
||||
cache_pool: &redis::Client,
|
||||
) -> Result<Message, Error> {
|
||||
let user = User::fetch_one(conn, cache_pool, self.user_uuid).await?;
|
||||
use channels::dsl;
|
||||
|
||||
let guild_uuid = dsl::channels
|
||||
.filter(dsl::uuid.eq(self.channel_uuid))
|
||||
.inner_join(guilds::table)
|
||||
.select(guilds::uuid)
|
||||
.get_result(conn)
|
||||
.await?;
|
||||
|
||||
let member = Member::fetch_one(conn, cache_pool, None, self.user_uuid, guild_uuid).await?;
|
||||
|
||||
Ok(Message {
|
||||
uuid: self.uuid,
|
||||
|
@ -31,7 +45,7 @@ impl MessageBuilder {
|
|||
user_uuid: self.user_uuid,
|
||||
message: self.message.clone(),
|
||||
reply_to: self.reply_to,
|
||||
user,
|
||||
member,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
@ -43,5 +57,5 @@ pub struct Message {
|
|||
user_uuid: Uuid,
|
||||
message: String,
|
||||
reply_to: Option<Uuid>,
|
||||
user: User,
|
||||
member: Member,
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue