use crate::Perform; use actix_web::web::Data; use lemmy_api_common::{utils::get_local_user_view_from_jwt, websocket::*}; use lemmy_utils::{error::LemmyError, ConnectionId}; use lemmy_websocket::{ messages::{JoinCommunityRoom, JoinModRoom, JoinPostRoom, JoinUserRoom}, LemmyContext, }; #[async_trait::async_trait(?Send)] impl Perform for UserJoin { type Response = UserJoinResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &UserJoin = self; let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?; if let Some(ws_id) = websocket_id { context.chat_server().do_send(JoinUserRoom { local_user_id: local_user_view.local_user.id, id: ws_id, }); } Ok(UserJoinResponse { joined: true }) } } #[async_trait::async_trait(?Send)] impl Perform for CommunityJoin { type Response = CommunityJoinResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &CommunityJoin = self; if let Some(ws_id) = websocket_id { context.chat_server().do_send(JoinCommunityRoom { community_id: data.community_id, id: ws_id, }); } Ok(CommunityJoinResponse { joined: true }) } } #[async_trait::async_trait(?Send)] impl Perform for ModJoin { type Response = ModJoinResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &ModJoin = self; if let Some(ws_id) = websocket_id { context.chat_server().do_send(JoinModRoom { community_id: data.community_id, id: ws_id, }); } Ok(ModJoinResponse { joined: true }) } } #[async_trait::async_trait(?Send)] impl Perform for PostJoin { type Response = PostJoinResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &PostJoin = self; if let Some(ws_id) = websocket_id { context.chat_server().do_send(JoinPostRoom { post_id: data.post_id, id: ws_id, }); } Ok(PostJoinResponse { joined: true }) } }