diff --git a/src/routes/bots/invite.rs b/src/routes/bots/invite.rs index ca9ce868..dacdd443 100644 --- a/src/routes/bots/invite.rs +++ b/src/routes/bots/invite.rs @@ -1,5 +1,5 @@ use revolt_quark::{ - models::{Channel, User}, + models::{Channel, User, Member, server_member::MemberCompositeKey}, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result, ServerPermission, }; @@ -43,7 +43,15 @@ pub async fn invite_bot( }); } - db.insert_member(&server.id, &bot.id) + let member = Member { + id: MemberCompositeKey { + server: server.id, + user: bot.id + }, + ..Default::default() + }; + + db.insert_member(&member) .await .map(|_| EmptyResponse) } diff --git a/src/routes/invites/invite_join.rs b/src/routes/invites/invite_join.rs index 7a05e1e9..63b39335 100644 --- a/src/routes/invites/invite_join.rs +++ b/src/routes/invites/invite_join.rs @@ -1,5 +1,5 @@ use revolt_quark::{ - models::{Invite, User}, + models::{Invite, User, server_member::MemberCompositeKey, Member}, Db, Error, Ref, Result, }; @@ -30,7 +30,15 @@ pub async fn req(db: &Db, user: User, target: Ref) -> Result { } let channel = db.fetch_channel(channel).await?; - db.insert_member(&server.id, &user.id).await?; + let member = Member { + id: MemberCompositeKey { + server: server.id.clone(), + user: user.id.clone() + }, + ..Default::default() + }; + + db.insert_member(&member).await?; Ok(json!({ "type": "Server",