2 build_federated_instances,
5 get_user_safe_settings_from_jwt,
6 get_user_safe_settings_from_jwt_opt,
11 use actix_web::web::Data;
13 use lemmy_apub::fetcher::search::search_by_apub_id;
14 use lemmy_db_queries::{
15 diesel_option_overwrite,
16 source::{category::Category_, site::Site_},
21 use lemmy_db_schema::{
30 comment_view::CommentQueryBuilder,
31 post_view::PostQueryBuilder,
34 use lemmy_db_views_actor::{
35 community_view::CommunityQueryBuilder,
36 user_view::{UserQueryBuilder, UserViewSafe},
38 use lemmy_db_views_moderator::{
39 mod_add_community_view::ModAddCommunityView,
40 mod_add_view::ModAddView,
41 mod_ban_from_community_view::ModBanFromCommunityView,
42 mod_ban_view::ModBanView,
43 mod_lock_post_view::ModLockPostView,
44 mod_remove_comment_view::ModRemoveCommentView,
45 mod_remove_community_view::ModRemoveCommunityView,
46 mod_remove_post_view::ModRemovePostView,
47 mod_sticky_post_view::ModStickyPostView,
49 use lemmy_structs::{blocking, site::*, user::Register};
53 utils::{check_slurs, check_slurs_opt},
58 use lemmy_websocket::{
59 messages::{GetUsersOnline, SendAllMessage},
63 use log::{debug, info};
64 use std::str::FromStr;
66 #[async_trait::async_trait(?Send)]
67 impl Perform for ListCategories {
68 type Response = ListCategoriesResponse;
72 context: &Data<LemmyContext>,
73 _websocket_id: Option<ConnectionId>,
74 ) -> Result<ListCategoriesResponse, LemmyError> {
75 let _data: &ListCategories = &self;
77 let categories = blocking(context.pool(), move |conn| Category::list_all(conn)).await??;
80 Ok(ListCategoriesResponse { categories })
84 #[async_trait::async_trait(?Send)]
85 impl Perform for GetModlog {
86 type Response = GetModlogResponse;
90 context: &Data<LemmyContext>,
91 _websocket_id: Option<ConnectionId>,
92 ) -> Result<GetModlogResponse, LemmyError> {
93 let data: &GetModlog = &self;
95 let community_id = data.community_id;
96 let mod_user_id = data.mod_user_id;
98 let limit = data.limit;
99 let removed_posts = blocking(context.pool(), move |conn| {
100 ModRemovePostView::list(conn, community_id, mod_user_id, page, limit)
104 let locked_posts = blocking(context.pool(), move |conn| {
105 ModLockPostView::list(conn, community_id, mod_user_id, page, limit)
109 let stickied_posts = blocking(context.pool(), move |conn| {
110 ModStickyPostView::list(conn, community_id, mod_user_id, page, limit)
114 let removed_comments = blocking(context.pool(), move |conn| {
115 ModRemoveCommentView::list(conn, community_id, mod_user_id, page, limit)
119 let banned_from_community = blocking(context.pool(), move |conn| {
120 ModBanFromCommunityView::list(conn, community_id, mod_user_id, page, limit)
124 let added_to_community = blocking(context.pool(), move |conn| {
125 ModAddCommunityView::list(conn, community_id, mod_user_id, page, limit)
129 // These arrays are only for the full modlog, when a community isn't given
130 let (removed_communities, banned, added) = if data.community_id.is_none() {
131 blocking(context.pool(), move |conn| {
133 ModRemoveCommunityView::list(conn, mod_user_id, page, limit)?,
134 ModBanView::list(conn, mod_user_id, page, limit)?,
135 ModAddView::list(conn, mod_user_id, page, limit)?,
136 )) as Result<_, LemmyError>
140 (Vec::new(), Vec::new(), Vec::new())
144 Ok(GetModlogResponse {
150 banned_from_community,
158 #[async_trait::async_trait(?Send)]
159 impl Perform for CreateSite {
160 type Response = SiteResponse;
164 context: &Data<LemmyContext>,
165 _websocket_id: Option<ConnectionId>,
166 ) -> Result<SiteResponse, LemmyError> {
167 let data: &CreateSite = &self;
169 let read_site = move |conn: &'_ _| Site::read_simple(conn);
170 if blocking(context.pool(), read_site).await?.is_ok() {
171 return Err(APIError::err("site_already_exists").into());
174 let user = get_user_from_jwt(&data.auth, context.pool()).await?;
176 check_slurs(&data.name)?;
177 check_slurs_opt(&data.description)?;
179 // Make sure user is an admin
180 is_admin(context.pool(), user.id).await?;
182 let site_form = SiteForm {
183 name: data.name.to_owned(),
184 description: data.description.to_owned(),
185 icon: Some(data.icon.to_owned()),
186 banner: Some(data.banner.to_owned()),
188 enable_downvotes: data.enable_downvotes,
189 open_registration: data.open_registration,
190 enable_nsfw: data.enable_nsfw,
194 let create_site = move |conn: &'_ _| Site::create(conn, &site_form);
195 if blocking(context.pool(), create_site).await?.is_err() {
196 return Err(APIError::err("site_already_exists").into());
199 let site_view = blocking(context.pool(), move |conn| SiteView::read(conn)).await??;
201 Ok(SiteResponse { site_view })
205 #[async_trait::async_trait(?Send)]
206 impl Perform for EditSite {
207 type Response = SiteResponse;
210 context: &Data<LemmyContext>,
211 websocket_id: Option<ConnectionId>,
212 ) -> Result<SiteResponse, LemmyError> {
213 let data: &EditSite = &self;
214 let user = get_user_from_jwt(&data.auth, context.pool()).await?;
216 check_slurs(&data.name)?;
217 check_slurs_opt(&data.description)?;
219 // Make sure user is an admin
220 is_admin(context.pool(), user.id).await?;
222 let found_site = blocking(context.pool(), move |conn| Site::read_simple(conn)).await??;
224 let icon = diesel_option_overwrite(&data.icon);
225 let banner = diesel_option_overwrite(&data.banner);
227 let site_form = SiteForm {
228 name: data.name.to_owned(),
229 description: data.description.to_owned(),
232 creator_id: found_site.creator_id,
233 updated: Some(naive_now()),
234 enable_downvotes: data.enable_downvotes,
235 open_registration: data.open_registration,
236 enable_nsfw: data.enable_nsfw,
239 let update_site = move |conn: &'_ _| Site::update(conn, 1, &site_form);
240 if blocking(context.pool(), update_site).await?.is_err() {
241 return Err(APIError::err("couldnt_update_site").into());
244 let site_view = blocking(context.pool(), move |conn| SiteView::read(conn)).await??;
246 let res = SiteResponse { site_view };
248 context.chat_server().do_send(SendAllMessage {
249 op: UserOperation::EditSite,
250 response: res.clone(),
258 #[async_trait::async_trait(?Send)]
259 impl Perform for GetSite {
260 type Response = GetSiteResponse;
264 context: &Data<LemmyContext>,
265 websocket_id: Option<ConnectionId>,
266 ) -> Result<GetSiteResponse, LemmyError> {
267 let data: &GetSite = &self;
269 let site_view = match blocking(context.pool(), move |conn| SiteView::read(conn)).await? {
270 Ok(site_view) => Some(site_view),
271 // If the site isn't created yet, check the setup
273 if let Some(setup) = Settings::get().setup.as_ref() {
274 let register = Register {
275 username: setup.admin_username.to_owned(),
276 email: setup.admin_email.to_owned(),
277 password: setup.admin_password.to_owned(),
278 password_verify: setup.admin_password.to_owned(),
281 captcha_answer: None,
283 let login_response = register.perform(context, websocket_id).await?;
284 info!("Admin {} created", setup.admin_username);
286 let create_site = CreateSite {
287 name: setup.site_name.to_owned(),
291 enable_downvotes: true,
292 open_registration: true,
294 auth: login_response.jwt,
296 create_site.perform(context, websocket_id).await?;
297 info!("Site {} created", setup.site_name);
298 Some(blocking(context.pool(), move |conn| SiteView::read(conn)).await??)
305 let mut admins = blocking(context.pool(), move |conn| UserViewSafe::admins(conn)).await??;
307 // Make sure the site creator is the top admin
308 if let Some(site_view) = site_view.to_owned() {
309 let site_creator_id = site_view.creator.id;
310 // TODO investigate why this is sometimes coming back null
311 // Maybe user_.admin isn't being set to true?
312 if let Some(creator_index) = admins.iter().position(|r| r.user.id == site_creator_id) {
313 let creator_user = admins.remove(creator_index);
314 admins.insert(0, creator_user);
318 let banned = blocking(context.pool(), move |conn| UserViewSafe::banned(conn)).await??;
322 .send(GetUsersOnline)
326 let my_user = get_user_safe_settings_from_jwt_opt(&data.auth, context.pool()).await?;
327 let federated_instances = build_federated_instances(context.pool()).await?;
334 version: version::VERSION.to_string(),
341 #[async_trait::async_trait(?Send)]
342 impl Perform for Search {
343 type Response = SearchResponse;
347 context: &Data<LemmyContext>,
348 _websocket_id: Option<ConnectionId>,
349 ) -> Result<SearchResponse, LemmyError> {
350 let data: &Search = &self;
352 match search_by_apub_id(&data.q, context).await {
353 Ok(r) => return Ok(r),
354 Err(e) => debug!("Failed to resolve search query as activitypub ID: {}", e),
357 let user = get_user_from_jwt_opt(&data.auth, context.pool()).await?;
358 let user_id = user.map(|u| u.id);
360 let type_ = SearchType::from_str(&data.type_)?;
362 let mut posts = Vec::new();
363 let mut comments = Vec::new();
364 let mut communities = Vec::new();
365 let mut users = Vec::new();
367 // TODO no clean / non-nsfw searching rn
369 let q = data.q.to_owned();
370 let page = data.page;
371 let limit = data.limit;
372 let sort = SortType::from_str(&data.sort)?;
373 let community_id = data.community_id;
374 let community_name = data.community_name.to_owned();
376 SearchType::Posts => {
377 posts = blocking(context.pool(), move |conn| {
378 PostQueryBuilder::create(conn)
381 .community_id(community_id)
382 .community_name(community_name)
391 SearchType::Comments => {
392 comments = blocking(context.pool(), move |conn| {
393 CommentQueryBuilder::create(&conn)
403 SearchType::Communities => {
404 communities = blocking(context.pool(), move |conn| {
405 CommunityQueryBuilder::create(conn)
415 SearchType::Users => {
416 users = blocking(context.pool(), move |conn| {
417 UserQueryBuilder::create(conn)
427 posts = blocking(context.pool(), move |conn| {
428 PostQueryBuilder::create(conn)
431 .community_id(community_id)
432 .community_name(community_name)
441 let q = data.q.to_owned();
442 let sort = SortType::from_str(&data.sort)?;
444 comments = blocking(context.pool(), move |conn| {
445 CommentQueryBuilder::create(conn)
455 let q = data.q.to_owned();
456 let sort = SortType::from_str(&data.sort)?;
458 communities = blocking(context.pool(), move |conn| {
459 CommunityQueryBuilder::create(conn)
469 let q = data.q.to_owned();
470 let sort = SortType::from_str(&data.sort)?;
472 users = blocking(context.pool(), move |conn| {
473 UserQueryBuilder::create(conn)
483 posts = blocking(context.pool(), move |conn| {
484 PostQueryBuilder::create(conn)
488 .community_id(community_id)
489 .community_name(community_name)
501 type_: data.type_.to_owned(),
510 #[async_trait::async_trait(?Send)]
511 impl Perform for TransferSite {
512 type Response = GetSiteResponse;
516 context: &Data<LemmyContext>,
517 _websocket_id: Option<ConnectionId>,
518 ) -> Result<GetSiteResponse, LemmyError> {
519 let data: &TransferSite = &self;
520 let user = get_user_safe_settings_from_jwt(&data.auth, context.pool()).await?;
522 is_admin(context.pool(), user.id).await?;
524 let read_site = blocking(context.pool(), move |conn| Site::read_simple(conn)).await??;
526 // Make sure user is the creator
527 if read_site.creator_id != user.id {
528 return Err(APIError::err("not_an_admin").into());
531 let new_creator_id = data.user_id;
532 let transfer_site = move |conn: &'_ _| Site::transfer(conn, new_creator_id);
533 if blocking(context.pool(), transfer_site).await?.is_err() {
534 return Err(APIError::err("couldnt_update_site").into());
538 let form = ModAddForm {
539 mod_user_id: user.id,
540 other_user_id: data.user_id,
541 removed: Some(false),
544 blocking(context.pool(), move |conn| ModAdd::create(conn, &form)).await??;
546 let site_view = blocking(context.pool(), move |conn| SiteView::read(conn)).await??;
548 let mut admins = blocking(context.pool(), move |conn| UserViewSafe::admins(conn)).await??;
549 let creator_index = admins
551 .position(|r| r.user.id == site_view.creator.id)
552 .context(location_info!())?;
553 let creator_user = admins.remove(creator_index);
554 admins.insert(0, creator_user);
556 let banned = blocking(context.pool(), move |conn| UserViewSafe::banned(conn)).await??;
557 let federated_instances = build_federated_instances(context.pool()).await?;
560 site_view: Some(site_view),
564 version: version::VERSION.to_string(),
571 #[async_trait::async_trait(?Send)]
572 impl Perform for GetSiteConfig {
573 type Response = GetSiteConfigResponse;
577 context: &Data<LemmyContext>,
578 _websocket_id: Option<ConnectionId>,
579 ) -> Result<GetSiteConfigResponse, LemmyError> {
580 let data: &GetSiteConfig = &self;
581 let user = get_user_from_jwt(&data.auth, context.pool()).await?;
583 // Only let admins read this
584 is_admin(context.pool(), user.id).await?;
586 let config_hjson = Settings::read_config_file()?;
588 Ok(GetSiteConfigResponse { config_hjson })
592 #[async_trait::async_trait(?Send)]
593 impl Perform for SaveSiteConfig {
594 type Response = GetSiteConfigResponse;
598 context: &Data<LemmyContext>,
599 _websocket_id: Option<ConnectionId>,
600 ) -> Result<GetSiteConfigResponse, LemmyError> {
601 let data: &SaveSiteConfig = &self;
602 let user = get_user_from_jwt(&data.auth, context.pool()).await?;
604 // Only let admins read this
605 let user_id = user.id;
606 is_admin(context.pool(), user_id).await?;
608 // Make sure docker doesn't have :ro at the end of the volume, so its not a read-only filesystem
609 let config_hjson = match Settings::save_config_file(&data.config_hjson) {
610 Ok(config_hjson) => config_hjson,
611 Err(_e) => return Err(APIError::err("couldnt_update_site").into()),
614 Ok(GetSiteConfigResponse { config_hjson })