crates/api_crud/src/site/create.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::*,
   8   site_description_length_check,
   9 };
  10 use lemmy_db_schema::{
  11   diesel_option_overwrite,
  12   diesel_option_overwrite_to_url,
  13   source::site::{Site, SiteForm},
  14   traits::Crud,
  15 };
  16 use lemmy_db_views::site_view::SiteView;
  17 use lemmy_utils::{
  18   utils::{check_slurs, check_slurs_opt},
  19   ConnectionId,
  20   LemmyError,
  21 };
  22 use lemmy_websocket::LemmyContext;
  23
  24 #[async_trait::async_trait(?Send)]
  25 impl PerformCrud for CreateSite {
  26   type Response = SiteResponse;
  27
  28   #[tracing::instrument(skip(context, _websocket_id))]
  29   async fn perform(
  30     &self,
  31     context: &Data<LemmyContext>,
  32     _websocket_id: Option<ConnectionId>,
  33   ) -> Result<SiteResponse, LemmyError> {
  34     let data: &CreateSite = self;
  35
  36     let read_site = Site::read_simple;
  37     if blocking(context.pool(), read_site).await?.is_ok() {
  38       return Err(LemmyError::from_message("site_already_exists"));
  39     };
  40
  41     let local_user_view =
  42       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  43
  44     check_slurs(&data.name, &context.settings().slur_regex())?;
  45     check_slurs_opt(&data.description, &context.settings().slur_regex())?;
  46
  47     // Make sure user is an admin
  48     is_admin(&local_user_view)?;
  49
  50     let sidebar = diesel_option_overwrite(&data.sidebar);
  51     let description = diesel_option_overwrite(&data.description);
  52     let icon = diesel_option_overwrite_to_url(&data.icon)?;
  53     let banner = diesel_option_overwrite_to_url(&data.banner)?;
  54
  55     if let Some(Some(desc)) = &description {
  56       site_description_length_check(desc)?;
  57     }
  58
  59     let site_form = SiteForm {
  60       name: data.name.to_owned(),
  61       sidebar,
  62       description,
  63       icon,
  64       banner,
  65       enable_downvotes: data.enable_downvotes,
  66       open_registration: data.open_registration,
  67       enable_nsfw: data.enable_nsfw,
  68       community_creation_admin_only: data.community_creation_admin_only,
  69       ..SiteForm::default()
  70     };
  71
  72     let create_site = move |conn: &'_ _| Site::create(conn, &site_form);
  73     if blocking(context.pool(), create_site).await?.is_err() {
  74       return Err(LemmyError::from_message("site_already_exists"));
  75     }
  76
  77     let site_view = blocking(context.pool(), SiteView::read).await??;
  78
  79     Ok(SiteResponse { site_view })
  80   }
  81 }