Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update users and invite count after an invite gets redeemed

Browse source
This commit is contained in:
Antonio Scandurra 2022-05-20 12:02:31 +02:00
parent eedb8ba59f
commit 6f2c3f1e37
4 changed files with 34 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
crates/collab/src/rpc.rs
View file

@ -412,6 +412,23 @@ impl Server {
Ok(())
}
pub async fn invite_code_redeemed(self: &Arc<Self>, code: &str, invitee_id: UserId) -> Result<()> {
let user = self.app_state.db.get_user_for_invite_code(code).await?;
let store = self.store().await;
let invitee_contact = store.contact_for_user(invitee_id, true);
for connection_id in store.connection_ids_for_user(user.id) {
self.peer.send(connection_id, proto::UpdateContacts {
contacts: vec![invitee_contact.clone()],
..Default::default()
})?;
self.peer.send(connection_id, proto::UpdateInviteInfo {
url: format!("{}{}", self.app_state.invite_link_prefix, code),
count: user.invite_count as u32,
})?;
}
Ok(())
}
async fn ping(
self: Arc<Server>,
_: TypedEnvelope<proto::Ping>,
@ -1530,13 +1547,12 @@ impl Header for ProtocolVersion {
}
}
pub fn routes(app_state: Arc<AppState>) -> Router<Body> {
let server = Server::new(app_state.clone(), None);
pub fn routes(server: Arc<Server>) -> Router<Body> {
Router::new()
.route("/rpc", get(handle_websocket_request))
.layer(
ServiceBuilder::new()
.layer(Extension(app_state))
.layer(Extension(server.app_state.clone()))
.layer(middleware::from_fn(auth::validate_header))
.layer(Extension(server)),
)