crates/api_crud/src/site/read.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   person::Register,
   5   site::{CreateSite, GetSite, GetSiteResponse, MyUserInfo},
   6   utils::{blocking, build_federated_instances, get_local_user_settings_view_from_jwt_opt},
   7 };
   8 use lemmy_db_views::structs::SiteView;
   9 use lemmy_db_views_actor::structs::{
  10   CommunityBlockView,
  11   CommunityFollowerView,
  12   CommunityModeratorView,
  13   PersonBlockView,
  14   PersonViewSafe,
  15 };
  16 use lemmy_utils::{error::LemmyError, version, ConnectionId};
  17 use lemmy_websocket::{messages::GetUsersOnline, LemmyContext};
  18 use tracing::info;
  19
  20 #[async_trait::async_trait(?Send)]
  21 impl PerformCrud for GetSite {
  22   type Response = GetSiteResponse;
  23
  24   #[tracing::instrument(skip(context, websocket_id))]
  25   async fn perform(
  26     &self,
  27     context: &Data<LemmyContext>,
  28     websocket_id: Option<ConnectionId>,
  29   ) -> Result<GetSiteResponse, LemmyError> {
  30     let data: &GetSite = self;
  31
  32     let site_view = match blocking(context.pool(), SiteView::read_local).await? {
  33       Ok(site_view) => Some(site_view),
  34       // If the site isn't created yet, check the setup
  35       Err(_) => {
  36         if let Some(setup) = context.settings().setup.as_ref() {
  37           let register = Register {
  38             username: setup.admin_username.to_owned(),
  39             email: setup.admin_email.clone().map(|s| s.into()),
  40             password: setup.admin_password.clone().into(),
  41             password_verify: setup.admin_password.clone().into(),
  42             show_nsfw: true,
  43             captcha_uuid: None,
  44             captcha_answer: None,
  45             honeypot: None,
  46             answer: None,
  47           };
  48           let admin_jwt = register
  49             .perform(context, websocket_id)
  50             .await?
  51             .jwt
  52             .expect("jwt is returned from registration on newly created site");
  53           info!("Admin {} created", setup.admin_username);
  54
  55           let create_site = CreateSite {
  56             name: setup.site_name.to_owned(),
  57             auth: admin_jwt,
  58             ..CreateSite::default()
  59           };
  60           create_site.perform(context, websocket_id).await?;
  61           info!("Site {} created", setup.site_name);
  62           Some(blocking(context.pool(), SiteView::read_local).await??)
  63         } else {
  64           None
  65         }
  66       }
  67     };
  68
  69     let admins = blocking(context.pool(), PersonViewSafe::admins).await??;
  70
  71     let online = context
  72       .chat_server()
  73       .send(GetUsersOnline)
  74       .await
  75       .unwrap_or(1);
  76
  77     // Build the local user
  78     let my_user = if let Some(local_user_view) = get_local_user_settings_view_from_jwt_opt(
  79       data.auth.as_ref(),
  80       context.pool(),
  81       context.secret(),
  82     )
  83     .await?
  84     {
  85       let person_id = local_user_view.person.id;
  86       let follows = blocking(context.pool(), move |conn| {
  87         CommunityFollowerView::for_person(conn, person_id)
  88       })
  89       .await?
  90       .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?;
  91
  92       let person_id = local_user_view.person.id;
  93       let community_blocks = blocking(context.pool(), move |conn| {
  94         CommunityBlockView::for_person(conn, person_id)
  95       })
  96       .await?
  97       .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?;
  98
  99       let person_id = local_user_view.person.id;
 100       let person_blocks = blocking(context.pool(), move |conn| {
 101         PersonBlockView::for_person(conn, person_id)
 102       })
 103       .await?
 104       .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?;
 105
 106       let moderates = blocking(context.pool(), move |conn| {
 107         CommunityModeratorView::for_person(conn, person_id)
 108       })
 109       .await?
 110       .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?;
 111
 112       Some(MyUserInfo {
 113         local_user_view,
 114         follows,
 115         moderates,
 116         community_blocks,
 117         person_blocks,
 118       })
 119     } else {
 120       None
 121     };
 122
 123     let federated_instances = build_federated_instances(context.pool(), context.settings()).await?;
 124
 125     Ok(GetSiteResponse {
 126       site_view,
 127       admins,
 128       online,
 129       version: version::VERSION.to_string(),
 130       my_user,
 131       federated_instances,
 132     })
 133   }
 134 }