1 files changed, 16 insertions, 20 deletions

diff --git a/src/routers/account.rs b/src/routers/account.rs
index a6fce99..71ba496 100644
--- a/src/routers/account.rs
+++ b/src/routers/account.rs
@@ -2,22 +2,30 @@ use axum::extract::State;
 use chrono::{DateTime, Duration, Utc};
 use entity::users::{self};
 use sea_orm::{
-    ActiveModelTrait, ActiveValue::Set, ColumnTrait, EntityTrait, IntoActiveModel, ModelTrait,
-    QueryFilter,
+    ActiveModelTrait, ActiveValue::Set, ColumnTrait, DatabaseConnection, EntityTrait,
+    IntoActiveModel, ModelTrait, QueryFilter,
 };
 use serde::{Deserialize, Serialize};
 use utoipa::ToSchema;
 use utoipa_axum::{router::OpenApiRouter, routes};
 
 use crate::{
-    ApiResult, AppState, ClientError, GlobalResponses, JwtClaims, ServerError, SuccessResponse,
-    create_jwt, create_password,
+    ApiError, ApiResult, AppState, ClientError, GlobalResponses, JwtClaims, ServerError,
+    SuccessResponse, create_jwt, create_password,
     extract::{ApiJson, Auth},
     models::Account,
     tags::ACCOUNT,
     validate_password,
 };
 
+async fn username_exists(username: &str, db: &DatabaseConnection) -> Result<bool, ApiError> {
+    Ok(users::Entity::find()
+        .filter(users::Column::Username.eq(username))
+        .one(db)
+        .await?
+        .is_some())
+}
+
 #[derive(Serialize, ToSchema)]
 #[schema(description = "Authorization token information")]
 struct Token {
@@ -130,14 +138,8 @@ async fn register(
     State(state): State<AppState>,
     ApiJson(req): ApiJson<RegisterRequest>,
 ) -> ApiResult<Account> {
-    let user_exists = users::Entity::find()
-        .filter(users::Column::Username.eq(&req.username))
-        .one(&state.db)
-        .await?
-        .is_some();
-
-    if user_exists {
-        return Err(ClientError::UserAlreadyExists {
+    if username_exists(&req.username, &state.db).await? {
+        return Err(ClientError::UsernameIsTaken {
             username: req.username,
         }
         .into());
@@ -255,14 +257,8 @@ async fn change_username(
     Auth(user): Auth,
     ApiJson(req): ApiJson<ChangeUsernameRequest>,
 ) -> ApiResult<Account> {
-    let user_exists = users::Entity::find()
-        .filter(users::Column::Username.eq(&req.new_username))
-        .one(&state.db)
-        .await?
-        .is_some();
-
-    if user_exists {
-        return Err(ClientError::UserAlreadyExists {
+    if username_exists(&req.new_username, &state.db).await? {
+        return Err(ClientError::UsernameIsTaken {
             username: req.new_username,
         }
         .into());