1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4 site::{EditSite, SiteResponse},
6 get_local_user_view_from_jwt,
8 local_site_to_slur_regex,
9 site_description_length_check,
12 use lemmy_db_schema::{
14 actor_language::SiteLanguage,
15 federation_allowlist::FederationAllowList,
16 federation_blocklist::FederationBlockList,
17 local_site::{LocalSite, LocalSiteUpdateForm},
18 local_site_rate_limit::{LocalSiteRateLimit, LocalSiteRateLimitUpdateForm},
19 local_user::LocalUser,
20 site::{Site, SiteUpdateForm},
23 utils::{diesel_option_overwrite, diesel_option_overwrite_to_url, naive_now},
26 use lemmy_db_views::structs::SiteView;
29 utils::{check_application_question, check_slurs_opt},
32 use lemmy_websocket::{messages::SendAllMessage, LemmyContext, UserOperationCrud};
33 use std::str::FromStr;
35 #[async_trait::async_trait(?Send)]
36 impl PerformCrud for EditSite {
37 type Response = SiteResponse;
39 #[tracing::instrument(skip(context, websocket_id))]
42 context: &Data<LemmyContext>,
43 websocket_id: Option<ConnectionId>,
44 ) -> Result<SiteResponse, LemmyError> {
45 let data: &EditSite = self;
47 get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
48 let local_site = LocalSite::read(context.pool()).await?;
50 // Make sure user is an admin
51 is_admin(&local_user_view)?;
53 let slur_regex = local_site_to_slur_regex(&local_site);
55 check_slurs_opt(&data.name, &slur_regex)?;
56 check_slurs_opt(&data.description, &slur_regex)?;
58 if let Some(desc) = &data.description {
59 site_description_length_check(desc)?;
62 let application_question = diesel_option_overwrite(&data.application_question);
63 check_application_question(&application_question, &data.require_application)?;
65 if let Some(default_post_listing_type) = &data.default_post_listing_type {
66 // only allow all or local as default listing types
67 let val = ListingType::from_str(default_post_listing_type);
68 if val != Ok(ListingType::All) && val != Ok(ListingType::Local) {
69 return Err(LemmyError::from_message(
70 "invalid_default_post_listing_type",
75 let site_id = local_site.site_id;
76 if let Some(discussion_languages) = data.discussion_languages.clone() {
77 let site = Site::read(context.pool(), site_id).await?;
78 SiteLanguage::update(context.pool(), discussion_languages.clone(), &site).await?;
81 let name = data.name.to_owned();
82 let site_form = SiteUpdateForm::builder()
84 .sidebar(diesel_option_overwrite(&data.sidebar))
85 .description(diesel_option_overwrite(&data.description))
86 .icon(diesel_option_overwrite_to_url(&data.icon)?)
87 .banner(diesel_option_overwrite_to_url(&data.banner)?)
88 .updated(Some(Some(naive_now())))
91 Site::update(context.pool(), site_id, &site_form)
93 // Ignore errors for all these, so as to not throw errors if no update occurs
94 // Diesel will throw an error for empty update forms
97 let local_site_form = LocalSiteUpdateForm::builder()
98 .enable_downvotes(data.enable_downvotes)
99 .open_registration(data.open_registration)
100 .enable_nsfw(data.enable_nsfw)
101 .community_creation_admin_only(data.community_creation_admin_only)
102 .require_email_verification(data.require_email_verification)
103 .require_application(data.require_application)
104 .application_question(application_question)
105 .private_instance(data.private_instance)
106 .default_theme(data.default_theme.clone())
107 .default_post_listing_type(data.default_post_listing_type.clone())
108 .legal_information(diesel_option_overwrite(&data.legal_information))
109 .application_email_admins(data.application_email_admins)
110 .hide_modlog_mod_names(data.hide_modlog_mod_names)
111 .updated(Some(Some(naive_now())))
112 .slur_filter_regex(diesel_option_overwrite(&data.slur_filter_regex))
113 .actor_name_max_length(data.actor_name_max_length)
114 .federation_enabled(data.federation_enabled)
115 .federation_debug(data.federation_debug)
116 .federation_strict_allowlist(data.federation_strict_allowlist)
117 .federation_http_fetch_retry_limit(data.federation_http_fetch_retry_limit)
118 .federation_worker_count(data.federation_worker_count)
119 .captcha_enabled(data.captcha_enabled)
120 .captcha_difficulty(data.captcha_difficulty.to_owned())
123 let update_local_site = LocalSite::update(context.pool(), &local_site_form)
127 let local_site_rate_limit_form = LocalSiteRateLimitUpdateForm::builder()
128 .message(data.rate_limit_message)
129 .message_per_second(data.rate_limit_message_per_second)
130 .post(data.rate_limit_post)
131 .post_per_second(data.rate_limit_post_per_second)
132 .register(data.rate_limit_register)
133 .register_per_second(data.rate_limit_register_per_second)
134 .image(data.rate_limit_image)
135 .image_per_second(data.rate_limit_image_per_second)
136 .comment(data.rate_limit_comment)
137 .comment_per_second(data.rate_limit_comment_per_second)
138 .search(data.rate_limit_search)
139 .search_per_second(data.rate_limit_search_per_second)
142 LocalSiteRateLimit::update(context.pool(), &local_site_rate_limit_form)
146 // Replace the blocked and allowed instances
147 let allowed = data.allowed_instances.to_owned();
148 FederationAllowList::replace(context.pool(), allowed).await?;
149 let blocked = data.blocked_instances.to_owned();
150 FederationBlockList::replace(context.pool(), blocked).await?;
152 // TODO can't think of a better way to do this.
153 // If the server suddenly requires email verification, or required applications, no old users
154 // will be able to log in. It really only wants this to be a requirement for NEW signups.
155 // So if it was set from false, to true, you need to update all current users columns to be verified.
157 let new_require_application = update_local_site
159 .map(|ols| ols.require_application)
161 if !local_site.require_application && new_require_application {
162 LocalUser::set_all_users_registration_applications_accepted(context.pool())
164 .map_err(|e| LemmyError::from_error_message(e, "couldnt_set_all_registrations_accepted"))?;
167 let new_require_email_verification = update_local_site
169 .map(|ols| ols.require_email_verification)
171 if !local_site.require_email_verification && new_require_email_verification {
172 LocalUser::set_all_users_email_verified(context.pool())
174 .map_err(|e| LemmyError::from_error_message(e, "couldnt_set_all_email_verified"))?;
177 let site_view = SiteView::read_local(context.pool()).await?;
179 let res = SiteResponse { site_view };
181 context.chat_server().do_send(SendAllMessage {
182 op: UserOperationCrud::EditSite,
183 response: res.clone(),