use crate::Perform; use actix_web::web::Data; use lemmy_api_common::{ person::{BanPerson, BanPersonResponse}, utils::{blocking, 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, site::Site, }, traits::Crud, }; use lemmy_db_views_actor::structs::PersonViewSafe; use lemmy_utils::{utils::naive_from_unix, ConnectionId, LemmyError}; 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 ban_person = move |conn: &'_ _| Person::ban_person(conn, banned_person_id, ban, expires); let person = blocking(context.pool(), ban_person) .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()).await?; } // Mod tables let form = ModBanForm { mod_person_id: local_user_view.person.id, other_person_id: data.person_id, reason: data.reason.to_owned(), banned: Some(data.ban), expires, }; blocking(context.pool(), move |conn| ModBan::create(conn, &form)).await??; let person_id = data.person_id; let person_view = blocking(context.pool(), move |conn| { PersonViewSafe::read(conn, person_id) }) .await??; let site = SiteOrCommunity::Site( blocking(context.pool(), Site::read_local_site) .await?? .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) } }