crates/api/src/site.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use anyhow::Context;
   4 use diesel::NotFound;
   5 use lemmy_api_common::{
   6   blocking,
   7   build_federated_instances,
   8   get_local_user_view_from_jwt,
   9   get_local_user_view_from_jwt_opt,
  10   is_admin,
  11   site::*,
  12 };
  13 use lemmy_apub::{
  14   fetcher::search::{search_by_apub_id, SearchableObjects},
  15   get_actor_id_from_name,
  16 };
  17 use lemmy_apub_lib::webfinger::WebfingerType;
  18 use lemmy_db_schema::{
  19   from_opt_str_to_opt_enum,
  20   newtypes::PersonId,
  21   source::{moderator::*, site::Site},
  22   traits::{Crud, DeleteableOrRemoveable},
  23   DbPool,
  24   ListingType,
  25   SearchType,
  26   SortType,
  27 };
  28 use lemmy_db_views::{
  29   comment_view::{CommentQueryBuilder, CommentView},
  30   post_view::{PostQueryBuilder, PostView},
  31   site_view::SiteView,
  32 };
  33 use lemmy_db_views_actor::{
  34   community_view::{CommunityQueryBuilder, CommunityView},
  35   person_view::{PersonQueryBuilder, PersonViewSafe},
  36 };
  37 use lemmy_db_views_moderator::{
  38   mod_add_community_view::ModAddCommunityView,
  39   mod_add_view::ModAddView,
  40   mod_ban_from_community_view::ModBanFromCommunityView,
  41   mod_ban_view::ModBanView,
  42   mod_lock_post_view::ModLockPostView,
  43   mod_remove_comment_view::ModRemoveCommentView,
  44   mod_remove_community_view::ModRemoveCommunityView,
  45   mod_remove_post_view::ModRemovePostView,
  46   mod_sticky_post_view::ModStickyPostView,
  47   mod_transfer_community_view::ModTransferCommunityView,
  48 };
  49 use lemmy_utils::{
  50   location_info,
  51   settings::structs::Settings,
  52   version,
  53   ApiError,
  54   ConnectionId,
  55   LemmyError,
  56 };
  57 use lemmy_websocket::LemmyContext;
  58
  59 #[async_trait::async_trait(?Send)]
  60 impl Perform for GetModlog {
  61   type Response = GetModlogResponse;
  62
  63   async fn perform(
  64     &self,
  65     context: &Data<LemmyContext>,
  66     _websocket_id: Option<ConnectionId>,
  67   ) -> Result<GetModlogResponse, LemmyError> {
  68     let data: &GetModlog = self;
  69
  70     let community_id = data.community_id;
  71     let mod_person_id = data.mod_person_id;
  72     let page = data.page;
  73     let limit = data.limit;
  74     let removed_posts = blocking(context.pool(), move |conn| {
  75       ModRemovePostView::list(conn, community_id, mod_person_id, page, limit)
  76     })
  77     .await??;
  78
  79     let locked_posts = blocking(context.pool(), move |conn| {
  80       ModLockPostView::list(conn, community_id, mod_person_id, page, limit)
  81     })
  82     .await??;
  83
  84     let stickied_posts = blocking(context.pool(), move |conn| {
  85       ModStickyPostView::list(conn, community_id, mod_person_id, page, limit)
  86     })
  87     .await??;
  88
  89     let removed_comments = blocking(context.pool(), move |conn| {
  90       ModRemoveCommentView::list(conn, community_id, mod_person_id, page, limit)
  91     })
  92     .await??;
  93
  94     let banned_from_community = blocking(context.pool(), move |conn| {
  95       ModBanFromCommunityView::list(conn, community_id, mod_person_id, page, limit)
  96     })
  97     .await??;
  98
  99     let added_to_community = blocking(context.pool(), move |conn| {
 100       ModAddCommunityView::list(conn, community_id, mod_person_id, page, limit)
 101     })
 102     .await??;
 103
 104     let transferred_to_community = blocking(context.pool(), move |conn| {
 105       ModTransferCommunityView::list(conn, community_id, mod_person_id, page, limit)
 106     })
 107     .await??;
 108
 109     // These arrays are only for the full modlog, when a community isn't given
 110     let (removed_communities, banned, added) = if data.community_id.is_none() {
 111       blocking(context.pool(), move |conn| {
 112         Ok((
 113           ModRemoveCommunityView::list(conn, mod_person_id, page, limit)?,
 114           ModBanView::list(conn, mod_person_id, page, limit)?,
 115           ModAddView::list(conn, mod_person_id, page, limit)?,
 116         )) as Result<_, LemmyError>
 117       })
 118       .await??
 119     } else {
 120       (Vec::new(), Vec::new(), Vec::new())
 121     };
 122
 123     // Return the jwt
 124     Ok(GetModlogResponse {
 125       removed_posts,
 126       locked_posts,
 127       stickied_posts,
 128       removed_comments,
 129       removed_communities,
 130       banned_from_community,
 131       banned,
 132       added_to_community,
 133       added,
 134       transferred_to_community,
 135     })
 136   }
 137 }
 138
 139 #[async_trait::async_trait(?Send)]
 140 impl Perform for Search {
 141   type Response = SearchResponse;
 142
 143   async fn perform(
 144     &self,
 145     context: &Data<LemmyContext>,
 146     _websocket_id: Option<ConnectionId>,
 147   ) -> Result<SearchResponse, LemmyError> {
 148     let data: &Search = self;
 149
 150     let local_user_view =
 151       get_local_user_view_from_jwt_opt(&data.auth, context.pool(), context.secret()).await?;
 152
 153     let show_nsfw = local_user_view.as_ref().map(|t| t.local_user.show_nsfw);
 154     let show_bot_accounts = local_user_view
 155       .as_ref()
 156       .map(|t| t.local_user.show_bot_accounts);
 157     let show_read_posts = local_user_view
 158       .as_ref()
 159       .map(|t| t.local_user.show_read_posts);
 160
 161     let person_id = local_user_view.map(|u| u.person.id);
 162
 163     let mut posts = Vec::new();
 164     let mut comments = Vec::new();
 165     let mut communities = Vec::new();
 166     let mut users = Vec::new();
 167
 168     // TODO no clean / non-nsfw searching rn
 169
 170     let q = data.q.to_owned();
 171     let page = data.page;
 172     let limit = data.limit;
 173     let sort: Option<SortType> = from_opt_str_to_opt_enum(&data.sort);
 174     let listing_type: Option<ListingType> = from_opt_str_to_opt_enum(&data.listing_type);
 175     let search_type: SearchType = from_opt_str_to_opt_enum(&data.type_).unwrap_or(SearchType::All);
 176     let community_id = data.community_id;
 177     let community_actor_id = if let Some(name) = &data.community_name {
 178       get_actor_id_from_name(WebfingerType::Group, name, context)
 179         .await
 180         .ok()
 181     } else {
 182       None
 183     };
 184     let creator_id = data.creator_id;
 185     match search_type {
 186       SearchType::Posts => {
 187         posts = blocking(context.pool(), move |conn| {
 188           PostQueryBuilder::create(conn)
 189             .sort(sort)
 190             .show_nsfw(show_nsfw)
 191             .show_bot_accounts(show_bot_accounts)
 192             .show_read_posts(show_read_posts)
 193             .listing_type(listing_type)
 194             .community_id(community_id)
 195             .community_actor_id(community_actor_id)
 196             .creator_id(creator_id)
 197             .my_person_id(person_id)
 198             .search_term(q)
 199             .page(page)
 200             .limit(limit)
 201             .list()
 202         })
 203         .await??;
 204       }
 205       SearchType::Comments => {
 206         comments = blocking(context.pool(), move |conn| {
 207           CommentQueryBuilder::create(conn)
 208             .sort(sort)
 209             .listing_type(listing_type)
 210             .search_term(q)
 211             .show_bot_accounts(show_bot_accounts)
 212             .community_id(community_id)
 213             .community_actor_id(community_actor_id)
 214             .creator_id(creator_id)
 215             .my_person_id(person_id)
 216             .page(page)
 217             .limit(limit)
 218             .list()
 219         })
 220         .await??;
 221       }
 222       SearchType::Communities => {
 223         communities = blocking(context.pool(), move |conn| {
 224           CommunityQueryBuilder::create(conn)
 225             .sort(sort)
 226             .listing_type(listing_type)
 227             .search_term(q)
 228             .my_person_id(person_id)
 229             .page(page)
 230             .limit(limit)
 231             .list()
 232         })
 233         .await??;
 234       }
 235       SearchType::Users => {
 236         users = blocking(context.pool(), move |conn| {
 237           PersonQueryBuilder::create(conn)
 238             .sort(sort)
 239             .search_term(q)
 240             .page(page)
 241             .limit(limit)
 242             .list()
 243         })
 244         .await??;
 245       }
 246       SearchType::All => {
 247         // If the community or creator is included, dont search communities or users
 248         let community_or_creator_included =
 249           data.community_id.is_some() || data.community_name.is_some() || data.creator_id.is_some();
 250         let community_actor_id_2 = community_actor_id.to_owned();
 251
 252         posts = blocking(context.pool(), move |conn| {
 253           PostQueryBuilder::create(conn)
 254             .sort(sort)
 255             .show_nsfw(show_nsfw)
 256             .show_bot_accounts(show_bot_accounts)
 257             .show_read_posts(show_read_posts)
 258             .listing_type(listing_type)
 259             .community_id(community_id)
 260             .community_actor_id(community_actor_id_2)
 261             .creator_id(creator_id)
 262             .my_person_id(person_id)
 263             .search_term(q)
 264             .page(page)
 265             .limit(limit)
 266             .list()
 267         })
 268         .await??;
 269
 270         let q = data.q.to_owned();
 271         let community_actor_id = community_actor_id.to_owned();
 272
 273         comments = blocking(context.pool(), move |conn| {
 274           CommentQueryBuilder::create(conn)
 275             .sort(sort)
 276             .listing_type(listing_type)
 277             .search_term(q)
 278             .show_bot_accounts(show_bot_accounts)
 279             .community_id(community_id)
 280             .community_actor_id(community_actor_id)
 281             .creator_id(creator_id)
 282             .my_person_id(person_id)
 283             .page(page)
 284             .limit(limit)
 285             .list()
 286         })
 287         .await??;
 288
 289         let q = data.q.to_owned();
 290
 291         communities = if community_or_creator_included {
 292           vec![]
 293         } else {
 294           blocking(context.pool(), move |conn| {
 295             CommunityQueryBuilder::create(conn)
 296               .sort(sort)
 297               .listing_type(listing_type)
 298               .search_term(q)
 299               .my_person_id(person_id)
 300               .page(page)
 301               .limit(limit)
 302               .list()
 303           })
 304           .await??
 305         };
 306
 307         let q = data.q.to_owned();
 308
 309         users = if community_or_creator_included {
 310           vec![]
 311         } else {
 312           blocking(context.pool(), move |conn| {
 313             PersonQueryBuilder::create(conn)
 314               .sort(sort)
 315               .search_term(q)
 316               .page(page)
 317               .limit(limit)
 318               .list()
 319           })
 320           .await??
 321         };
 322       }
 323       SearchType::Url => {
 324         posts = blocking(context.pool(), move |conn| {
 325           PostQueryBuilder::create(conn)
 326             .sort(sort)
 327             .show_nsfw(show_nsfw)
 328             .show_bot_accounts(show_bot_accounts)
 329             .show_read_posts(show_read_posts)
 330             .listing_type(listing_type)
 331             .my_person_id(person_id)
 332             .community_id(community_id)
 333             .community_actor_id(community_actor_id)
 334             .creator_id(creator_id)
 335             .url_search(q)
 336             .page(page)
 337             .limit(limit)
 338             .list()
 339         })
 340         .await??;
 341       }
 342     };
 343
 344     // Blank out deleted or removed info for non logged in users
 345     if person_id.is_none() {
 346       for cv in communities
 347         .iter_mut()
 348         .filter(|cv| cv.community.deleted || cv.community.removed)
 349       {
 350         cv.community = cv.to_owned().community.blank_out_deleted_or_removed_info();
 351       }
 352
 353       for pv in posts
 354         .iter_mut()
 355         .filter(|p| p.post.deleted || p.post.removed)
 356       {
 357         pv.post = pv.to_owned().post.blank_out_deleted_or_removed_info();
 358       }
 359
 360       for cv in comments
 361         .iter_mut()
 362         .filter(|cv| cv.comment.deleted || cv.comment.removed)
 363       {
 364         cv.comment = cv.to_owned().comment.blank_out_deleted_or_removed_info();
 365       }
 366     }
 367
 368     // Return the jwt
 369     Ok(SearchResponse {
 370       type_: search_type.to_string(),
 371       comments,
 372       posts,
 373       communities,
 374       users,
 375     })
 376   }
 377 }
 378
 379 #[async_trait::async_trait(?Send)]
 380 impl Perform for ResolveObject {
 381   type Response = ResolveObjectResponse;
 382
 383   async fn perform(
 384     &self,
 385     context: &Data<LemmyContext>,
 386     _websocket_id: Option<ConnectionId>,
 387   ) -> Result<ResolveObjectResponse, LemmyError> {
 388     let local_user_view =
 389       get_local_user_view_from_jwt_opt(&self.auth, context.pool(), context.secret()).await?;
 390     let res = search_by_apub_id(&self.q, context)
 391       .await
 392       .map_err(|e| ApiError::err("couldnt_find_object", e))?;
 393     convert_response(res, local_user_view.map(|l| l.person.id), context.pool())
 394       .await
 395       .map_err(|e| ApiError::err("couldnt_find_object", e).into())
 396   }
 397 }
 398
 399 async fn convert_response(
 400   object: SearchableObjects,
 401   user_id: Option<PersonId>,
 402   pool: &DbPool,
 403 ) -> Result<ResolveObjectResponse, LemmyError> {
 404   let removed_or_deleted;
 405   let mut res = ResolveObjectResponse {
 406     comment: None,
 407     post: None,
 408     community: None,
 409     person: None,
 410   };
 411   use SearchableObjects::*;
 412   match object {
 413     Person(p) => {
 414       removed_or_deleted = p.deleted;
 415       res.person = Some(blocking(pool, move |conn| PersonViewSafe::read(conn, p.id)).await??)
 416     }
 417     Community(c) => {
 418       removed_or_deleted = c.deleted || c.removed;
 419       res.community =
 420         Some(blocking(pool, move |conn| CommunityView::read(conn, c.id, user_id)).await??)
 421     }
 422     Post(p) => {
 423       removed_or_deleted = p.deleted || p.removed;
 424       res.post = Some(blocking(pool, move |conn| PostView::read(conn, p.id, user_id)).await??)
 425     }
 426     Comment(c) => {
 427       removed_or_deleted = c.deleted || c.removed;
 428       res.comment = Some(blocking(pool, move |conn| CommentView::read(conn, c.id, user_id)).await??)
 429     }
 430   };
 431   // if the object was deleted from database, dont return it
 432   if removed_or_deleted {
 433     return Err(NotFound {}.into());
 434   }
 435   Ok(res)
 436 }
 437
 438 #[async_trait::async_trait(?Send)]
 439 impl Perform for TransferSite {
 440   type Response = GetSiteResponse;
 441
 442   async fn perform(
 443     &self,
 444     context: &Data<LemmyContext>,
 445     _websocket_id: Option<ConnectionId>,
 446   ) -> Result<GetSiteResponse, LemmyError> {
 447     let data: &TransferSite = self;
 448     let local_user_view =
 449       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
 450
 451     is_admin(&local_user_view)?;
 452
 453     let read_site = blocking(context.pool(), Site::read_simple).await??;
 454
 455     // Make sure user is the creator
 456     if read_site.creator_id != local_user_view.person.id {
 457       return Err(ApiError::err_plain("not_an_admin").into());
 458     }
 459
 460     let new_creator_id = data.person_id;
 461     let transfer_site = move |conn: &'_ _| Site::transfer(conn, new_creator_id);
 462     blocking(context.pool(), transfer_site)
 463       .await?
 464       .map_err(|e| ApiError::err("couldnt_update_site", e))?;
 465
 466     // Mod tables
 467     let form = ModAddForm {
 468       mod_person_id: local_user_view.person.id,
 469       other_person_id: data.person_id,
 470       removed: Some(false),
 471     };
 472
 473     blocking(context.pool(), move |conn| ModAdd::create(conn, &form)).await??;
 474
 475     let site_view = blocking(context.pool(), SiteView::read).await??;
 476
 477     let mut admins = blocking(context.pool(), PersonViewSafe::admins).await??;
 478     let creator_index = admins
 479       .iter()
 480       .position(|r| r.person.id == site_view.creator.id)
 481       .context(location_info!())?;
 482     let creator_person = admins.remove(creator_index);
 483     admins.insert(0, creator_person);
 484
 485     let banned = blocking(context.pool(), PersonViewSafe::banned).await??;
 486     let federated_instances = build_federated_instances(
 487       context.pool(),
 488       &context.settings().federation,
 489       &context.settings().hostname,
 490     )
 491     .await?;
 492
 493     Ok(GetSiteResponse {
 494       site_view: Some(site_view),
 495       admins,
 496       banned,
 497       online: 0,
 498       version: version::VERSION.to_string(),
 499       my_user: None,
 500       federated_instances,
 501     })
 502   }
 503 }
 504
 505 #[async_trait::async_trait(?Send)]
 506 impl Perform for GetSiteConfig {
 507   type Response = GetSiteConfigResponse;
 508
 509   async fn perform(
 510     &self,
 511     context: &Data<LemmyContext>,
 512     _websocket_id: Option<ConnectionId>,
 513   ) -> Result<GetSiteConfigResponse, LemmyError> {
 514     let data: &GetSiteConfig = self;
 515     let local_user_view =
 516       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
 517
 518     // Only let admins read this
 519     is_admin(&local_user_view)?;
 520
 521     let config_hjson = Settings::read_config_file()?;
 522
 523     Ok(GetSiteConfigResponse { config_hjson })
 524   }
 525 }
 526
 527 #[async_trait::async_trait(?Send)]
 528 impl Perform for SaveSiteConfig {
 529   type Response = GetSiteConfigResponse;
 530
 531   async fn perform(
 532     &self,
 533     context: &Data<LemmyContext>,
 534     _websocket_id: Option<ConnectionId>,
 535   ) -> Result<GetSiteConfigResponse, LemmyError> {
 536     let data: &SaveSiteConfig = self;
 537     let local_user_view =
 538       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
 539
 540     // Only let admins read this
 541     is_admin(&local_user_view)?;
 542
 543     // Make sure docker doesn't have :ro at the end of the volume, so its not a read-only filesystem
 544     let config_hjson = Settings::save_config_file(&data.config_hjson)
 545       .map_err(|e| ApiError::err("couldnt_update_site", e))?;
 546
 547     Ok(GetSiteConfigResponse { config_hjson })
 548   }
 549 }