use crate::PerformCrud; use actix_web::web::Data; use lemmy_api_common::{ blocking, build_federated_instances, get_local_user_settings_view_from_jwt_opt, person::Register, site::*, }; use lemmy_db_views::site_view::SiteView; use lemmy_db_views_actor::{ community_block_view::CommunityBlockView, community_follower_view::CommunityFollowerView, community_moderator_view::CommunityModeratorView, person_block_view::PersonBlockView, person_view::PersonViewSafe, }; use lemmy_utils::{version, ConnectionId, LemmyError}; use lemmy_websocket::{messages::GetUsersOnline, LemmyContext}; use tracing::info; #[async_trait::async_trait(?Send)] impl PerformCrud for GetSite { type Response = GetSiteResponse; #[tracing::instrument(skip(context, websocket_id))] async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &GetSite = self; let site_view = match blocking(context.pool(), SiteView::read_local).await? { Ok(site_view) => Some(site_view), // If the site isn't created yet, check the setup Err(_) => { if let Some(setup) = context.settings().setup.as_ref() { let register = Register { username: setup.admin_username.to_owned(), email: setup.admin_email.clone().map(|s| s.into()), password: setup.admin_password.clone().into(), password_verify: setup.admin_password.clone().into(), show_nsfw: true, captcha_uuid: None, captcha_answer: None, honeypot: None, answer: None, }; let admin_jwt = register .perform(context, websocket_id) .await? .jwt .expect("jwt is returned from registration on newly created site"); info!("Admin {} created", setup.admin_username); let create_site = CreateSite { name: setup.site_name.to_owned(), sidebar: setup.sidebar.to_owned(), description: setup.description.to_owned(), icon: setup.icon.to_owned(), banner: setup.banner.to_owned(), enable_downvotes: setup.enable_downvotes, open_registration: setup.open_registration, enable_nsfw: setup.enable_nsfw, community_creation_admin_only: setup.community_creation_admin_only, require_email_verification: setup.require_email_verification, require_application: setup.require_application, application_question: setup.application_question.to_owned(), private_instance: setup.private_instance, default_theme: setup.default_theme.to_owned(), default_post_listing_type: setup.default_post_listing_type.to_owned(), auth: admin_jwt, }; create_site.perform(context, websocket_id).await?; info!("Site {} created", setup.site_name); Some(blocking(context.pool(), SiteView::read_local).await??) } else { None } } }; let admins = blocking(context.pool(), PersonViewSafe::admins).await??; let online = context .chat_server() .send(GetUsersOnline) .await .unwrap_or(1); // Build the local user let my_user = if let Some(local_user_view) = get_local_user_settings_view_from_jwt_opt( data.auth.as_ref(), context.pool(), context.secret(), ) .await? { let person_id = local_user_view.person.id; let follows = blocking(context.pool(), move |conn| { CommunityFollowerView::for_person(conn, person_id) }) .await? .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?; let person_id = local_user_view.person.id; let community_blocks = blocking(context.pool(), move |conn| { CommunityBlockView::for_person(conn, person_id) }) .await? .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?; let person_id = local_user_view.person.id; let person_blocks = blocking(context.pool(), move |conn| { PersonBlockView::for_person(conn, person_id) }) .await? .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?; let moderates = blocking(context.pool(), move |conn| { CommunityModeratorView::for_person(conn, person_id) }) .await? .map_err(|e| LemmyError::from_error_message(e, "system_err_login"))?; Some(MyUserInfo { local_user_view, follows, moderates, community_blocks, person_blocks, }) } else { None }; let federated_instances = build_federated_instances(context.pool(), &context.settings()).await?; Ok(GetSiteResponse { site_view, admins, online, version: version::VERSION.to_string(), my_user, federated_instances, }) } }