94 lines
3.1 KiB
Rust
94 lines
3.1 KiB
Rust
|
use crate::{
|
||
|
|
State, get_user_from_token,
|
||
|
|
model::{ApiReturn, Error},
|
||
|
|
};
|
||
|
|
use axum::{Extension, Json, extract::Path, response::IntoResponse};
|
||
|
|
use axum_extra::extract::CookieJar;
|
||
|
|
use tetratto_core::model::auth::{UserBlock, UserFollow};
|
||
|
|
|
||
|
|
/// Toggle following on the given user.
|
||
|
|
pub async fn follow_request(
|
||
|
|
jar: CookieJar,
|
||
|
|
Path(id): Path<usize>,
|
||
|
|
Extension(data): Extension<State>,
|
||
|
|
) -> impl IntoResponse {
|
||
|
|
let data = &(data.read().await).0;
|
||
|
|
let user = match get_user_from_token!(jar, data) {
|
||
|
|
Some(ua) => ua,
|
||
|
|
None => return Json(Error::NotAllowed.into()),
|
||
|
|
};
|
||
|
|
|
||
|
|
if let Ok(userfollow) = data.get_userfollow_by_initiator_receiver(user.id, id).await {
|
||
|
|
// delete
|
||
|
|
match data.delete_userfollow(userfollow.id, user).await {
|
||
|
|
Ok(_) => Json(ApiReturn {
|
||
|
|
ok: true,
|
||
|
|
message: "User unfollowed".to_string(),
|
||
|
|
payload: (),
|
||
|
|
}),
|
||
|
|
Err(e) => return Json(e.into()),
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// create
|
||
|
|
match data.create_userfollow(UserFollow::new(user.id, id)).await {
|
||
|
|
Ok(_) => Json(ApiReturn {
|
||
|
|
ok: true,
|
||
|
|
message: "User followed".to_string(),
|
||
|
|
payload: (),
|
||
|
|
}),
|
||
|
|
Err(e) => return Json(e.into()),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Toggle blocking on the given user.
|
||
|
|
pub async fn block_request(
|
||
|
|
jar: CookieJar,
|
||
|
|
Path(id): Path<usize>,
|
||
|
|
Extension(data): Extension<State>,
|
||
|
|
) -> impl IntoResponse {
|
||
|
|
let data = &(data.read().await).0;
|
||
|
|
let user = match get_user_from_token!(jar, data) {
|
||
|
|
Some(ua) => ua,
|
||
|
|
None => return Json(Error::NotAllowed.into()),
|
||
|
|
};
|
||
|
|
|
||
|
|
if let Ok(userblock) = data.get_userblock_by_initiator_receiver(user.id, id).await {
|
||
|
|
// delete
|
||
|
|
match data.delete_userblock(userblock.id, user).await {
|
||
|
|
Ok(_) => Json(ApiReturn {
|
||
|
|
ok: true,
|
||
|
|
message: "User unblocked".to_string(),
|
||
|
|
payload: (),
|
||
|
|
}),
|
||
|
|
Err(e) => return Json(e.into()),
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// create
|
||
|
|
match data.create_userblock(UserBlock::new(user.id, id)).await {
|
||
|
|
Ok(_) => {
|
||
|
|
if let Ok(userfollow) = data.get_userfollow_by_initiator_receiver(user.id, id).await
|
||
|
|
{
|
||
|
|
// automatically unfollow
|
||
|
|
match data.delete_userfollow(userfollow.id, user).await {
|
||
|
|
Ok(_) => Json(ApiReturn {
|
||
|
|
ok: true,
|
||
|
|
message: "User unfollowed".to_string(),
|
||
|
|
payload: (),
|
||
|
|
}),
|
||
|
|
Err(e) => return Json(e.into()),
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// not following user, don't do anything else
|
||
|
|
Json(ApiReturn {
|
||
|
|
ok: true,
|
||
|
|
message: "User blocked".to_string(),
|
||
|
|
payload: (),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Err(e) => return Json(e.into()),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|