1 files changed, 36 insertions, 13 deletions

diff --git a/src/server/routes/users/get_user.rs b/src/server/routes/users/get_user.rs
index d86cc26..4079731 100644
--- a/src/server/routes/users/get_user.rs
+++ b/src/server/routes/users/get_user.rs
@@ -2,6 +2,7 @@ use activitypub_federation::{
     axum::json::FederationJson, config::Data, protocol::context::WithContext, traits::Object,
 };
 use axum::{debug_handler, extract::Path, http::StatusCode, response::IntoResponse};
+use tracing::trace;
 
 use crate::{error::AppError, state::AppHandle};
 
@@ -10,7 +11,7 @@ pub async fn http_get_user(
     Path(name): Path<String>,
     data: Data<AppHandle>,
 ) -> Result<impl IntoResponse, AppError> {
-    if let Some(a) = read_user(&name, &data).await {
+    if let Some(a) = read_user(&name, &data).await? {
         let json_user = a.into_json(&data).await?;
         Ok((
             StatusCode::OK,
@@ -22,11 +23,26 @@ pub async fn http_get_user(
     }
 }
 
-pub async fn read_user(name: &str, data: &Data<AppHandle>) -> Option<crate::entity::user::User> {
-    let read = data.users.read().await;
-    read.iter()
-        .find(|value| value.username.eq(&name))
-        .map(ToOwned::to_owned)
+pub async fn read_user(
+    name: &str,
+    data: &Data<AppHandle>,
+) -> Result<Option<crate::entity::user::User>, AppError> {
+    trace!(username = name, "getting user");
+    let read = sqlx::query_as!(
+        crate::entity::user::DbUser,
+        "select * from account where username = $1 and local = $2",
+        name,
+        true
+    )
+    .fetch_optional(&data.services.postgres)
+    .await?;
+
+    let user = read.into_iter().find(|value| value.username.eq(&name));
+    let user = match user {
+        Some(user) => Some(crate::entity::user::User::try_from(user)?),
+        None => None,
+    };
+    Ok(user)
 }
 
 #[cfg(test)]
@@ -35,13 +51,19 @@ mod tests {
         body::Body,
         http::{Request, StatusCode},
     };
+    use sqlx::PgPool;
+    use stack_up::Services;
     use tower::ServiceExt;
 
-    use crate::{server, state::AppState};
+    use crate::{
+        server::{self, test_config},
+        state::AppState,
+    };
 
-    #[tokio::test]
-    async fn get_user() {
-        let state = AppState::new().await.unwrap();
+    #[sqlx::test]
+    async fn get_user(pool: PgPool) {
+        let services = Services { postgres: pool };
+        let state = AppState::new(services, &test_config()).await.unwrap();
         let app = server::router(state);
 
         let response = app
@@ -57,9 +79,10 @@ mod tests {
         assert_eq!(response.status(), StatusCode::OK);
     }
 
-    #[tokio::test]
-    async fn get_user_not_found() {
-        let state = AppState::new().await.unwrap();
+    #[sqlx::test]
+    async fn get_user_not_found(pool: PgPool) {
+        let services = Services { postgres: pool };
+        let state = AppState::new(services, &test_config()).await.unwrap();
         let app = server::router(state);
 
         let response = app