mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-04-29 16:48:44 +00:00
ceff2ec686
* Use enum for registration mode setting * fix tests
71 lines
2 KiB
Rust
71 lines
2 KiB
Rust
use crate::Perform;
|
|
use actix_web::web::Data;
|
|
use lemmy_api_common::{
|
|
context::LemmyContext,
|
|
person::{LoginResponse, PasswordChangeAfterReset},
|
|
utils::password_length_check,
|
|
};
|
|
use lemmy_db_schema::source::{
|
|
local_site::RegistrationMode,
|
|
local_user::LocalUser,
|
|
password_reset_request::PasswordResetRequest,
|
|
};
|
|
use lemmy_db_views::structs::SiteView;
|
|
use lemmy_utils::{claims::Claims, error::LemmyError, ConnectionId};
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl Perform for PasswordChangeAfterReset {
|
|
type Response = LoginResponse;
|
|
|
|
#[tracing::instrument(skip(self, context, _websocket_id))]
|
|
async fn perform(
|
|
&self,
|
|
context: &Data<LemmyContext>,
|
|
_websocket_id: Option<ConnectionId>,
|
|
) -> Result<LoginResponse, LemmyError> {
|
|
let data: &PasswordChangeAfterReset = self;
|
|
|
|
// Fetch the user_id from the token
|
|
let token = data.token.clone();
|
|
let local_user_id = PasswordResetRequest::read_from_token(context.pool(), &token)
|
|
.await
|
|
.map(|p| p.local_user_id)?;
|
|
|
|
password_length_check(&data.password)?;
|
|
|
|
// Make sure passwords match
|
|
if data.password != data.password_verify {
|
|
return Err(LemmyError::from_message("passwords_dont_match"));
|
|
}
|
|
|
|
// Update the user with the new password
|
|
let password = data.password.clone();
|
|
let updated_local_user = LocalUser::update_password(context.pool(), local_user_id, &password)
|
|
.await
|
|
.map_err(|e| LemmyError::from_error_message(e, "couldnt_update_user"))?;
|
|
|
|
// Return the jwt if login is allowed
|
|
let site_view = SiteView::read_local(context.pool()).await?;
|
|
let jwt = if site_view.local_site.registration_mode == RegistrationMode::RequireApplication
|
|
&& !updated_local_user.accepted_application
|
|
{
|
|
None
|
|
} else {
|
|
Some(
|
|
Claims::jwt(
|
|
updated_local_user.id.0,
|
|
&context.secret().jwt_secret,
|
|
&context.settings().hostname,
|
|
)?
|
|
.into(),
|
|
)
|
|
};
|
|
|
|
Ok(LoginResponse {
|
|
jwt,
|
|
verify_email_sent: false,
|
|
registration_created: false,
|
|
})
|
|
}
|
|
}
|