lemmy/crates/api/src/post.rs

323 lines
8.5 KiB
Rust
Raw Normal View History

use crate::Perform;
use actix_web::web::Data;
use lemmy_api_common::{
blocking,
check_community_ban,
check_community_deleted_or_removed,
check_downvotes_enabled,
check_person_block,
2021-03-10 22:33:55 +00:00
get_local_user_view_from_jwt,
is_mod_or_admin,
mark_post_as_read,
post::*,
};
2021-07-31 17:26:17 +00:00
use lemmy_apub::{
fetcher::post_or_comment::PostOrComment,
objects::post::ApubPost,
protocol::activities::{
create_or_update::post::CreateOrUpdatePost,
voting::{
undo_vote::UndoVote,
vote::{Vote, VoteType},
},
CreateOrUpdateType,
},
2021-07-31 17:26:17 +00:00
};
2021-10-16 13:33:38 +00:00
use lemmy_db_schema::{
source::{moderator::*, post::*},
traits::{Crud, Likeable, Saveable},
};
use lemmy_db_views::post_view::PostView;
use lemmy_utils::{request::fetch_site_metadata, ApiError, ConnectionId, LemmyError};
use lemmy_websocket::{send::send_post_ws_message, LemmyContext, UserOperation};
use std::convert::TryInto;
2019-05-05 05:20:38 +00:00
#[async_trait::async_trait(?Send)]
impl Perform for CreatePostLike {
type Response = PostResponse;
async fn perform(
&self,
context: &Data<LemmyContext>,
websocket_id: Option<ConnectionId>,
) -> Result<PostResponse, LemmyError> {
2021-07-05 16:07:26 +00:00
let data: &CreatePostLike = self;
let local_user_view =
get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
2019-05-05 05:20:38 +00:00
// Don't do a downvote if site has downvotes disabled
check_downvotes_enabled(data.score, context.pool()).await?;
2019-05-05 05:20:38 +00:00
// Check for a community ban
let post_id = data.post_id;
let post: ApubPost = blocking(context.pool(), move |conn| Post::read(conn, post_id))
.await??
.into();
2021-03-10 22:33:55 +00:00
check_community_ban(local_user_view.person.id, post.community_id, context.pool()).await?;
check_community_deleted_or_removed(post.community_id, context.pool()).await?;
2019-05-05 05:20:38 +00:00
check_person_block(local_user_view.person.id, post.creator_id, context.pool()).await?;
2019-05-05 05:20:38 +00:00
let like_form = PostLikeForm {
post_id: data.post_id,
2021-03-10 22:33:55 +00:00
person_id: local_user_view.person.id,
score: data.score,
2019-05-05 05:20:38 +00:00
};
// Remove any likes first
2021-03-10 22:33:55 +00:00
let person_id = local_user_view.person.id;
blocking(context.pool(), move |conn| {
2021-03-10 22:33:55 +00:00
PostLike::remove(conn, person_id, post_id)
})
.await??;
2019-05-05 05:20:38 +00:00
let community_id = post.community_id;
let object = PostOrComment::Post(Box::new(post));
2019-05-05 05:20:38 +00:00
// Only add the like if the score isnt 0
let do_add = like_form.score != 0 && (like_form.score == 1 || like_form.score == -1);
2019-05-15 16:46:39 +00:00
if do_add {
let like_form2 = like_form.clone();
let like = move |conn: &'_ _| PostLike::like(conn, &like_form2);
blocking(context.pool(), like)
.await?
.map_err(|e| ApiError::err("couldnt_like_post", e))?;
2020-04-28 02:46:09 +00:00
Vote::send(
&object,
&local_user_view.person.clone().into(),
community_id,
like_form.score.try_into()?,
context,
)
.await?;
2020-04-28 02:46:09 +00:00
} else {
// API doesn't distinguish between Undo/Like and Undo/Dislike
UndoVote::send(
&object,
&local_user_view.person.clone().into(),
community_id,
VoteType::Like,
context,
)
.await?;
2019-05-05 05:20:38 +00:00
}
// Mark the post as read
mark_post_as_read(person_id, post_id, context.pool()).await?;
send_post_ws_message(
data.post_id,
UserOperation::CreatePostLike,
websocket_id,
Some(local_user_view.person.id),
context,
)
.await
2019-05-05 05:20:38 +00:00
}
}
#[async_trait::async_trait(?Send)]
impl Perform for LockPost {
type Response = PostResponse;
async fn perform(
&self,
context: &Data<LemmyContext>,
websocket_id: Option<ConnectionId>,
) -> Result<PostResponse, LemmyError> {
2021-07-05 16:07:26 +00:00
let data: &LockPost = self;
let local_user_view =
get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
let post_id = data.post_id;
let orig_post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
2021-03-11 04:43:11 +00:00
check_community_ban(
local_user_view.person.id,
orig_post.community_id,
context.pool(),
)
.await?;
check_community_deleted_or_removed(orig_post.community_id, context.pool()).await?;
// Verify that only the mods can lock
2021-03-11 04:43:11 +00:00
is_mod_or_admin(
context.pool(),
local_user_view.person.id,
orig_post.community_id,
)
.await?;
// Update the post
let post_id = data.post_id;
let locked = data.locked;
let updated_post: ApubPost = blocking(context.pool(), move |conn| {
Post::update_locked(conn, post_id, locked)
})
.await??
.into();
// Mod tables
let form = ModLockPostForm {
2021-03-10 22:33:55 +00:00
mod_person_id: local_user_view.person.id,
post_id: data.post_id,
locked: Some(locked),
};
blocking(context.pool(), move |conn| ModLockPost::create(conn, &form)).await??;
// apub updates
2021-07-31 17:26:17 +00:00
CreateOrUpdatePost::send(
&updated_post,
&local_user_view.person.clone().into(),
2021-07-31 17:26:17 +00:00
CreateOrUpdateType::Update,
context,
)
.await?;
send_post_ws_message(
data.post_id,
UserOperation::LockPost,
websocket_id,
Some(local_user_view.person.id),
context,
)
.await
}
}
#[async_trait::async_trait(?Send)]
impl Perform for StickyPost {
type Response = PostResponse;
async fn perform(
&self,
context: &Data<LemmyContext>,
websocket_id: Option<ConnectionId>,
) -> Result<PostResponse, LemmyError> {
2021-07-05 16:07:26 +00:00
let data: &StickyPost = self;
let local_user_view =
get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
let post_id = data.post_id;
let orig_post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
2021-03-11 04:43:11 +00:00
check_community_ban(
local_user_view.person.id,
orig_post.community_id,
context.pool(),
)
.await?;
check_community_deleted_or_removed(orig_post.community_id, context.pool()).await?;
// Verify that only the mods can sticky
2021-03-11 04:43:11 +00:00
is_mod_or_admin(
context.pool(),
local_user_view.person.id,
orig_post.community_id,
)
.await?;
// Update the post
let post_id = data.post_id;
let stickied = data.stickied;
let updated_post: ApubPost = blocking(context.pool(), move |conn| {
Post::update_stickied(conn, post_id, stickied)
})
.await??
.into();
// Mod tables
let form = ModStickyPostForm {
2021-03-10 22:33:55 +00:00
mod_person_id: local_user_view.person.id,
post_id: data.post_id,
stickied: Some(stickied),
};
blocking(context.pool(), move |conn| {
ModStickyPost::create(conn, &form)
})
.await??;
// Apub updates
// TODO stickied should pry work like locked for ease of use
2021-07-31 17:26:17 +00:00
CreateOrUpdatePost::send(
&updated_post,
&local_user_view.person.clone().into(),
2021-07-31 17:26:17 +00:00
CreateOrUpdateType::Update,
context,
)
.await?;
send_post_ws_message(
data.post_id,
UserOperation::StickyPost,
websocket_id,
Some(local_user_view.person.id),
context,
)
.await
2019-05-05 05:20:38 +00:00
}
}
#[async_trait::async_trait(?Send)]
impl Perform for SavePost {
type Response = PostResponse;
async fn perform(
&self,
context: &Data<LemmyContext>,
_websocket_id: Option<ConnectionId>,
) -> Result<PostResponse, LemmyError> {
2021-07-05 16:07:26 +00:00
let data: &SavePost = self;
let local_user_view =
get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
2019-05-05 05:20:38 +00:00
let post_saved_form = PostSavedForm {
post_id: data.post_id,
2021-03-10 22:33:55 +00:00
person_id: local_user_view.person.id,
2019-05-05 05:20:38 +00:00
};
if data.save {
let save = move |conn: &'_ _| PostSaved::save(conn, &post_saved_form);
blocking(context.pool(), save)
.await?
.map_err(|e| ApiError::err("couldnt_save_post", e))?;
2019-05-05 05:20:38 +00:00
} else {
let unsave = move |conn: &'_ _| PostSaved::unsave(conn, &post_saved_form);
blocking(context.pool(), unsave)
.await?
.map_err(|e| ApiError::err("couldnt_save_post", e))?;
2019-05-05 05:20:38 +00:00
}
let post_id = data.post_id;
2021-03-10 22:33:55 +00:00
let person_id = local_user_view.person.id;
let post_view = blocking(context.pool(), move |conn| {
2021-03-10 22:33:55 +00:00
PostView::read(conn, post_id, Some(person_id))
})
.await??;
2019-05-05 05:20:38 +00:00
// Mark the post as read
mark_post_as_read(person_id, post_id, context.pool()).await?;
2020-12-11 15:27:33 +00:00
Ok(PostResponse { post_view })
2019-05-05 05:20:38 +00:00
}
}
#[async_trait::async_trait(?Send)]
impl Perform for GetSiteMetadata {
type Response = GetSiteMetadataResponse;
async fn perform(
&self,
context: &Data<LemmyContext>,
_websocket_id: Option<ConnectionId>,
) -> Result<GetSiteMetadataResponse, LemmyError> {
let data: &Self = self;
let metadata = fetch_site_metadata(context.client(), &data.url).await?;
Ok(GetSiteMetadataResponse { metadata })
}
}