use crate::Perform; use actix_web::web::Data; use lemmy_api_common::{ person::{BanPerson, BanPersonResponse}, utils::{get_local_user_view_from_jwt, is_admin, remove_user_data}, }; use lemmy_apub::{ activities::block::SiteOrCommunity, protocol::activities::block::{block_user::BlockUser, undo_block_user::UndoBlockUser}, }; use lemmy_db_schema::{ source::{ moderator::{ModBan, ModBanForm}, person::{Person, PersonUpdateForm}, }, traits::Crud, }; use lemmy_db_views::structs::SiteView; use lemmy_db_views_actor::structs::PersonViewSafe; use lemmy_utils::{error::LemmyError, utils::naive_from_unix, ConnectionId}; use lemmy_websocket::{messages::SendAllMessage, LemmyContext, UserOperation}; #[async_trait::async_trait(?Send)] impl Perform for BanPerson { type Response = BanPersonResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &BanPerson = self; let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?; // Make sure user is an admin is_admin(&local_user_view)?; let ban = data.ban; let banned_person_id = data.person_id; let expires = data.expires.map(naive_from_unix); let person = Person::update( context.pool(), banned_person_id, &PersonUpdateForm::builder() .banned(Some(ban)) .ban_expires(Some(expires)) .build(), ) .await .map_err(|e| LemmyError::from_error_message(e, "couldnt_update_user"))?; // Remove their data if that's desired let remove_data = data.remove_data.unwrap_or(false); if remove_data { remove_user_data( person.id, context.pool(), context.settings(), context.client(), ) .await?; } // Mod tables let form = ModBanForm { mod_person_id: local_user_view.person.id, other_person_id: data.person_id, reason: data.reason.clone(), banned: Some(data.ban), expires, }; ModBan::create(context.pool(), &form).await?; let person_id = data.person_id; let person_view = PersonViewSafe::read(context.pool(), person_id).await?; let site = SiteOrCommunity::Site(SiteView::read_local(context.pool()).await?.site.into()); // if the action affects a local user, federate to other instances if person.local { if ban { BlockUser::send( &site, &person.into(), &local_user_view.person.into(), remove_data, data.reason.clone(), expires, context, ) .await?; } else { UndoBlockUser::send( &site, &person.into(), &local_user_view.person.into(), data.reason.clone(), context, ) .await?; } } let res = BanPersonResponse { person_view, banned: data.ban, }; context.chat_server().do_send(SendAllMessage { op: UserOperation::BanPerson, response: res.clone(), websocket_id, }); Ok(res) } }