src/api/v1/me/friends/mod.rs

@@ -8,7 +8,7 @@ use crate::{
     api::v1::auth::check_access_token,
     error::Error,
     objects::Me,
-    utils::{get_auth_header, global_checks, user_uuid_from_identifier},
+    utils::{get_auth_header, global_checks, user_uuid_from_username}
 };
 
 /// Returns a list of users that are your friends
@@ -73,7 +73,7 @@ pub async fn post(
 
     let me = Me::get(&mut conn, uuid).await?;
 
-    let target_uuid = user_uuid_from_identifier(&mut conn, &json.username).await?;
+    let target_uuid = user_uuid_from_username(&mut conn, &json.username).await?;
     me.add_friend(&mut conn, target_uuid).await?;
 
     Ok(HttpResponse::Ok().finish())

src/utils.rs

@@ -168,6 +168,26 @@ pub async fn user_uuid_from_identifier(
     }
 }
 
+pub async fn user_uuid_from_username(
+    conn: &mut Conn,
+    identifier: &String,
+) -> Result<Uuid, Error> {
+    if USERNAME_REGEX.is_match(identifier) {
+        use users::dsl;
+        let user_uuid = dsl::users
+            .filter(dsl::username.eq(identifier))
+            .select(dsl::uuid)
+            .get_result(conn)
+            .await?;
+
+        Ok(user_uuid)
+    } else {
+        Err(Error::BadRequest(
+            "Please provide a valid username".to_string(),
+        ))
+    }
+}
+
 pub async fn global_checks(data: &Data, user_uuid: Uuid) -> Result<(), Error> {
     if data.config.instance.require_email_verification {
         let mut conn = data.pool.get().await?;