crates/api_crud/src/site/update.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   blocking,
   5   get_local_user_view_from_jwt,
   6   is_admin,
   7   site::{EditSite, SiteResponse},
   8   site_description_length_check,
   9 };
  10 use lemmy_db_queries::{
  11   diesel_option_overwrite,
  12   diesel_option_overwrite_to_url,
  13   source::site::Site_,
  14   Crud,
  15 };
  16 use lemmy_db_schema::{
  17   naive_now,
  18   source::site::{Site, SiteForm},
  19 };
  20 use lemmy_db_views::site_view::SiteView;
  21 use lemmy_utils::{
  22   utils::{check_slurs, check_slurs_opt},
  23   ApiError,
  24   ConnectionId,
  25   LemmyError,
  26 };
  27 use lemmy_websocket::{messages::SendAllMessage, LemmyContext, UserOperationCrud};
  28
  29 #[async_trait::async_trait(?Send)]
  30 impl PerformCrud for EditSite {
  31   type Response = SiteResponse;
  32   async fn perform(
  33     &self,
  34     context: &Data<LemmyContext>,
  35     websocket_id: Option<ConnectionId>,
  36   ) -> Result<SiteResponse, LemmyError> {
  37     let data: &EditSite = &self;
  38     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?;
  39
  40     check_slurs(&data.name)?;
  41     check_slurs_opt(&data.description)?;
  42
  43     // Make sure user is an admin
  44     is_admin(&local_user_view)?;
  45
  46     let found_site = blocking(context.pool(), move |conn| Site::read_simple(conn)).await??;
  47
  48     let sidebar = diesel_option_overwrite(&data.sidebar);
  49     let description = diesel_option_overwrite(&data.description);
  50     let icon = diesel_option_overwrite_to_url(&data.icon)?;
  51     let banner = diesel_option_overwrite_to_url(&data.banner)?;
  52
  53     if let Some(Some(desc)) = &description {
  54       site_description_length_check(desc)?;
  55     }
  56
  57     let site_form = SiteForm {
  58       name: data.name.to_owned(),
  59       sidebar,
  60       description,
  61       icon,
  62       banner,
  63       creator_id: found_site.creator_id,
  64       updated: Some(naive_now()),
  65       enable_downvotes: data.enable_downvotes,
  66       open_registration: data.open_registration,
  67       enable_nsfw: data.enable_nsfw,
  68     };
  69
  70     let update_site = move |conn: &'_ _| Site::update(conn, 1, &site_form);
  71     if blocking(context.pool(), update_site).await?.is_err() {
  72       return Err(ApiError::err("couldnt_update_site").into());
  73     }
  74
  75     let site_view = blocking(context.pool(), move |conn| SiteView::read(conn)).await??;
  76
  77     let res = SiteResponse { site_view };
  78
  79     context.chat_server().do_send(SendAllMessage {
  80       op: UserOperationCrud::EditSite,
  81       response: res.clone(),
  82       websocket_id,
  83     });
  84
  85     Ok(res)
  86   }
  87 }