refactor(quark): port add_friend, block_user, change_username, remove_friend, send_friend_request, unblock_user

#283
This commit is contained in:
Paul Makles
2024-04-07 18:30:59 +01:00
parent 4e7049d51e
commit 49bb235938
12 changed files with 89 additions and 72 deletions

View File

@@ -1,6 +1,7 @@
use revolt_quark::models::User;
use revolt_quark::{Database, Error, Ref, Result};
use revolt_database::util::reference::Reference;
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::{create_error, Result};
use rocket::serde::json::Json;
use rocket::State;
@@ -9,13 +10,17 @@ use rocket::State;
/// Accept another user's friend request.
#[openapi(tag = "Relationships")]
#[put("/<target>/friend")]
pub async fn req(db: &State<Database>, user: User, target: Ref) -> Result<Json<User>> {
pub async fn add(
db: &State<Database>,
mut user: User,
target: Reference,
) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?;
if user.bot.is_some() || target.bot.is_some() {
return Err(Error::IsBot);
return Err(create_error!(IsBot));
}
user.add_friend(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await))
Ok(Json(target.into(db, &user).await))
}

View File

@@ -1,6 +1,7 @@
use revolt_quark::models::User;
use revolt_quark::{Database, Result};
use revolt_database::util::reference::Reference;
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::Result;
use rocket::serde::json::Json;
use rocket::State;
@@ -9,8 +10,13 @@ use rocket::State;
/// Block another user by their id.
#[openapi(tag = "Relationships")]
#[put("/<target>/block")]
pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user(&target).await?;
pub async fn block(
db: &State<Database>,
mut user: User,
target: Reference,
) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?;
user.block_user(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await))
Ok(Json(target.into(db, &user).await))
}

View File

@@ -1,6 +1,9 @@
use authifier::models::Account;
use once_cell::sync::Lazy;
use regex::Regex;
use revolt_quark::{authifier::models::Account, models::User, Database, Error, Result};
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State};
use serde::{Deserialize, Serialize};
use validator::Validate;
@@ -32,15 +35,18 @@ pub async fn req(
account: Account,
mut user: User,
data: Json<DataChangeUsername>,
) -> Result<Json<User>> {
) -> Result<Json<v0::User>> {
let data = data.into_inner();
data.validate()
.map_err(|error| Error::FailedValidation { error })?;
data.validate().map_err(|error| {
create_error!(FailedValidation {
error: error.to_string()
})
})?;
account
.verify_password(&data.password)
.map_err(|_| Error::InvalidCredentials)?;
.map_err(|_| create_error!(InvalidCredentials))?;
user.update_username(db, data.username).await?;
Ok(Json(user.foreign()))
Ok(Json(user.into(db, None).await))
}

View File

@@ -32,10 +32,10 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
open_dm::req,
// Relationships
find_mutual::req,
add_friend::req,
remove_friend::req,
block_user::req,
unblock_user::req,
send_friend_request::req,
add_friend::add,
remove_friend::remove,
block_user::block,
unblock_user::unblock,
send_friend_request::send_friend_request,
]
}

View File

@@ -1,6 +1,7 @@
use revolt_quark::models::User;
use revolt_quark::{Database, Error, Result};
use revolt_database::util::reference::Reference;
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::{create_error, Result};
use rocket::serde::json::Json;
use rocket::State;
@@ -9,13 +10,17 @@ use rocket::State;
/// Denies another user's friend request or removes an existing friend.
#[openapi(tag = "Relationships")]
#[delete("/<target>/friend")]
pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user(&target).await?;
pub async fn remove(
db: &State<Database>,
mut user: User,
target: Reference,
) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?;
if user.bot.is_some() || target.bot.is_some() {
return Err(Error::IsBot);
return Err(create_error!(IsBot));
}
user.remove_friend(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await))
Ok(Json(target.into(db, &user).await))
}

View File

@@ -1,37 +1,30 @@
use revolt_quark::models::User;
use revolt_quark::{Database, Error, Result};
use revolt_database::util::reference::Reference;
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::{create_error, Result};
use rocket::serde::json::Json;
use rocket::State;
use serde::{Deserialize, Serialize};
/// # User Lookup Information
#[derive(Serialize, Deserialize, JsonSchema)]
pub struct DataSendFriendRequest {
/// Username and discriminator combo separated by #
username: String,
}
/// # Send Friend Request
///
/// Send a friend request to another user.
#[openapi(tag = "Relationships")]
#[post("/friend", data = "<data>")]
pub async fn req(
pub async fn send_friend_request(
db: &State<Database>,
user: User,
data: Json<DataSendFriendRequest>,
) -> Result<Json<User>> {
mut user: User,
data: Json<v0::DataSendFriendRequest>,
) -> Result<Json<v0::User>> {
if let Some((username, discriminator)) = data.username.split_once('#') {
let mut target = db.fetch_user_by_username(username, discriminator).await?;
if user.bot.is_some() || target.bot.is_some() {
return Err(Error::IsBot);
return Err(create_error!(IsBot));
}
user.add_friend(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await))
Ok(Json(target.into(db, &user).await))
} else {
Err(Error::InvalidProperty)
Err(create_error!(InvalidProperty))
}
}

View File

@@ -1,6 +1,7 @@
use revolt_quark::models::User;
use revolt_quark::{Database, Result};
use revolt_database::util::reference::Reference;
use revolt_database::{Database, User};
use revolt_models::v0;
use revolt_result::Result;
use rocket::serde::json::Json;
use rocket::State;
@@ -9,8 +10,13 @@ use rocket::State;
/// Unblock another user by their id.
#[openapi(tag = "Relationships")]
#[delete("/<target>/block")]
pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user(&target).await?;
pub async fn unblock(
db: &State<Database>,
mut user: User,
target: Reference,
) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?;
user.unblock_user(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await))
Ok(Json(target.into(db, &user).await))
}