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::*,
+ context::LemmyContext,
+ sensitive::Sensitive,
+ site::{GetSite, GetSiteResponse, MyUserInfo},
+ utils::{check_user_valid, check_validator_time},
};
-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_db_schema::{
+ newtypes::LocalUserId,
+ source::{
+ actor_language::{LocalUserLanguage, SiteLanguage},
+ language::Language,
+ tagline::Tagline,
+ },
+};
+use lemmy_db_views::structs::{CustomEmojiView, LocalUserView, SiteView};
+use lemmy_db_views_actor::structs::{
+ CommunityBlockView,
+ CommunityFollowerView,
+ CommunityModeratorView,
+ PersonBlockView,
+ PersonView,
+};
+use lemmy_utils::{
+ claims::Claims,
+ error::{LemmyError, LemmyErrorExt, LemmyErrorType},
+ version,
};
-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<LemmyContext>,
- websocket_id: Option<ConnectionId>,
- ) -> Result<GetSiteResponse, LemmyError> {
+ #[tracing::instrument(skip(context))]
+ async fn perform(&self, context: &Data<LemmyContext>) -> Result<GetSiteResponse, LemmyError> {
let data: &GetSite = self;
- let site_view = match blocking(context.pool(), SiteView::read).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,
- auth: admin_jwt,
- };
- create_site.perform(context, websocket_id).await?;
- info!("Site {} created", setup.site_name);
- Some(blocking(context.pool(), SiteView::read).await??)
- } else {
- None
- }
- }
- };
-
- let admins = blocking(context.pool(), PersonViewSafe::admins).await??;
+ let site_view = SiteView::read_local(&mut context.pool()).await?;
- let online = context
- .chat_server()
- .send(GetUsersOnline)
- .await
- .unwrap_or(1);
+ let admins = PersonView::admins(&mut context.pool()).await?;
// 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 my_user = if let Some(local_user_view) =
+ local_user_settings_view_from_jwt_opt(data.auth.as_ref(), context).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(LemmyError::from)
- .map_err(|e| e.with_message("system_err_login"))?;
+ let local_user_id = local_user_view.local_user.id;
+
+ let follows = CommunityFollowerView::for_person(&mut context.pool(), person_id)
+ .await
+ .with_lemmy_type(LemmyErrorType::SystemErrLogin)?;
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(LemmyError::from)
- .map_err(|e| e.with_message("system_err_login"))?;
+ let community_blocks = CommunityBlockView::for_person(&mut context.pool(), person_id)
+ .await
+ .with_lemmy_type(LemmyErrorType::SystemErrLogin)?;
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(LemmyError::from)
- .map_err(|e| e.with_message("system_err_login"))?;
+ let person_blocks = PersonBlockView::for_person(&mut context.pool(), person_id)
+ .await
+ .with_lemmy_type(LemmyErrorType::SystemErrLogin)?;
- let moderates = blocking(context.pool(), move |conn| {
- CommunityModeratorView::for_person(conn, person_id)
- })
- .await?
- .map_err(LemmyError::from)
- .map_err(|e| e.with_message("system_err_login"))?;
+ let moderates = CommunityModeratorView::for_person(&mut context.pool(), person_id)
+ .await
+ .with_lemmy_type(LemmyErrorType::SystemErrLogin)?;
+
+ let discussion_languages = LocalUserLanguage::read(&mut context.pool(), local_user_id)
+ .await
+ .with_lemmy_type(LemmyErrorType::SystemErrLogin)?;
Some(MyUserInfo {
local_user_view,
moderates,
community_blocks,
person_blocks,
+ discussion_languages,
})
} else {
None
};
- let federated_instances = build_federated_instances(
- context.pool(),
- &context.settings().federation,
- &context.settings().hostname,
- )
- .await?;
+ let all_languages = Language::read_all(&mut context.pool()).await?;
+ let discussion_languages = SiteLanguage::read_local_raw(&mut context.pool()).await?;
+ let taglines = Tagline::get_all(&mut context.pool(), site_view.local_site.id).await?;
+ let custom_emojis =
+ CustomEmojiView::get_all(&mut context.pool(), site_view.local_site.id).await?;
Ok(GetSiteResponse {
site_view,
admins,
- online,
version: version::VERSION.to_string(),
my_user,
- federated_instances,
+ all_languages,
+ discussion_languages,
+ taglines,
+ custom_emojis,
})
}
}
+
+#[tracing::instrument(skip_all)]
+async fn local_user_settings_view_from_jwt_opt(
+ jwt: Option<&Sensitive<String>>,
+ context: &LemmyContext,
+) -> Option<LocalUserView> {
+ match jwt {
+ Some(jwt) => {
+ let claims = Claims::decode(jwt.as_ref(), &context.secret().jwt_secret)
+ .ok()?
+ .claims;
+ let local_user_id = LocalUserId(claims.sub);
+ let local_user_view = LocalUserView::read(&mut context.pool(), local_user_id)
+ .await
+ .ok()?;
+ check_user_valid(
+ local_user_view.person.banned,
+ local_user_view.person.ban_expires,
+ local_user_view.person.deleted,
+ )
+ .ok()?;
+
+ check_validator_time(&local_user_view.local_user.validator_time, &claims).ok()?;
+
+ Some(local_user_view)
+ }
+ None => None,
+ }
+}