Compare commits
8 commits
wip/member
...
main
Author | SHA1 | Date | |
---|---|---|---|
447c577a2a | |||
5b24d0052b | |||
8a7711cabc | |||
4cb89645fe | |||
946085a18f | |||
53451e67c7 | |||
027649a060 | |||
314b9ee011 |
8 changed files with 82 additions and 10 deletions
2
migrations/2025-08-04-180235_add_status_to_user/down.sql
Normal file
2
migrations/2025-08-04-180235_add_status_to_user/down.sql
Normal file
|
@ -0,0 +1,2 @@
|
|||
-- This file should undo anything in `up.sql`
|
||||
ALTER TABLE users DROP COLUMN online_status;
|
2
migrations/2025-08-04-180235_add_status_to_user/up.sql
Normal file
2
migrations/2025-08-04-180235_add_status_to_user/up.sql
Normal file
|
@ -0,0 +1,2 @@
|
|||
-- Your SQL goes here
|
||||
ALTER TABLE users ADD COLUMN online_status INT2 NOT NULL DEFAULT 0;
|
|
@ -49,6 +49,7 @@ struct NewInfo {
|
|||
email: Option<String>,
|
||||
pronouns: Option<String>,
|
||||
about: Option<String>,
|
||||
online_status: Option<i16>,
|
||||
}
|
||||
|
||||
pub async fn update(
|
||||
|
@ -110,5 +111,10 @@ pub async fn update(
|
|||
.await?;
|
||||
}
|
||||
|
||||
if let Some(online_status) = &json.online_status {
|
||||
me.set_online_status(&mut conn, &app_state.cache_pool, *online_status)
|
||||
.await?;
|
||||
}
|
||||
|
||||
Ok(StatusCode::OK)
|
||||
}
|
||||
|
|
|
@ -29,6 +29,7 @@ pub struct Me {
|
|||
avatar: Option<String>,
|
||||
pronouns: Option<String>,
|
||||
about: Option<String>,
|
||||
online_status: i16,
|
||||
pub email: String,
|
||||
pub email_verified: bool,
|
||||
}
|
||||
|
@ -277,6 +278,35 @@ impl Me {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn set_online_status(
|
||||
&mut self,
|
||||
conn: &mut Conn,
|
||||
cache_pool: &redis::Client,
|
||||
new_status: i16,
|
||||
) -> Result<(), Error> {
|
||||
if !(0..=4).contains(&new_status) {
|
||||
return Err(Error::BadRequest("Invalid status code".to_string()));
|
||||
}
|
||||
self.online_status = new_status;
|
||||
|
||||
use users::dsl;
|
||||
update(users::table)
|
||||
.filter(dsl::uuid.eq(self.uuid))
|
||||
.set(dsl::online_status.eq(new_status))
|
||||
.execute(conn)
|
||||
.await?;
|
||||
|
||||
if cache_pool
|
||||
.get_cache_key::<User>(self.uuid.to_string())
|
||||
.await
|
||||
.is_ok()
|
||||
{
|
||||
cache_pool.del_cache_key(self.uuid.to_string()).await?
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn friends_with(
|
||||
&self,
|
||||
conn: &mut Conn,
|
||||
|
|
|
@ -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,
|
||||
}
|
||||
|
|
|
@ -18,6 +18,7 @@ pub struct UserBuilder {
|
|||
avatar: Option<String>,
|
||||
pronouns: Option<String>,
|
||||
about: Option<String>,
|
||||
online_status: i16,
|
||||
}
|
||||
|
||||
impl UserBuilder {
|
||||
|
@ -29,6 +30,7 @@ impl UserBuilder {
|
|||
avatar: self.avatar,
|
||||
pronouns: self.pronouns,
|
||||
about: self.about,
|
||||
online_status: self.online_status,
|
||||
friends_since: None,
|
||||
}
|
||||
}
|
||||
|
@ -42,6 +44,7 @@ pub struct User {
|
|||
avatar: Option<String>,
|
||||
pronouns: Option<String>,
|
||||
about: Option<String>,
|
||||
online_status: i16,
|
||||
pub friends_since: Option<DateTime<Utc>>,
|
||||
}
|
||||
|
||||
|
|
|
@ -157,6 +157,7 @@ diesel::table! {
|
|||
pronouns -> Nullable<Varchar>,
|
||||
#[max_length = 200]
|
||||
about -> Nullable<Varchar>,
|
||||
online_status -> Int2,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue