Plume/src/routes/comments.rs

173 lines
5.8 KiB
Rust
Raw Normal View History

use activitypub::object::Note;
2019-03-20 17:56:17 +01:00
use rocket::{request::LenientForm, response::Redirect};
use rocket_i18n::I18n;
use template_utils::Ructe;
2019-03-20 17:56:17 +01:00
use validator::Validate;
2018-05-10 11:44:57 +02:00
use std::time::Duration;
2019-03-20 17:56:17 +01:00
use plume_common::{
activity_pub::{
broadcast,
inbox::{Deletable, Notify},
ActivityStream, ApRequest,
},
2019-03-20 17:56:17 +01:00
utils,
};
use plume_models::{
2019-03-20 17:56:17 +01:00
blogs::Blog, comments::*, db_conn::DbConn, instance::Instance, mentions::Mention, posts::Post,
safe_string::SafeString, tags::Tag, users::User,
2018-05-19 09:39:59 +02:00
};
use routes::errors::ErrorPage;
2019-03-20 17:56:17 +01:00
use Worker;
2018-05-10 11:44:57 +02:00
#[derive(Default, FromForm, Debug, Validate)]
pub struct NewCommentForm {
pub responding_to: Option<i32>,
#[validate(length(min = "1", message = "Your comment can't be empty"))]
pub content: String,
pub warning: String,
2018-05-10 11:44:57 +02:00
}
#[post("/~/<blog_name>/<slug>/comment", data = "<form>")]
2019-03-20 17:56:17 +01:00
pub fn create(
blog_name: String,
slug: String,
form: LenientForm<NewCommentForm>,
user: User,
conn: DbConn,
worker: Worker,
intl: I18n,
) -> Result<Redirect, Ructe> {
let blog = Blog::find_by_fqn(&*conn, &blog_name).expect("comments::create: blog error");
let post = Post::find_by_slug(&*conn, &slug, blog.id).expect("comments::create: post error");
2018-07-06 11:51:19 +02:00
form.validate()
.map(|_| {
let (html, mentions, _hashtags) = utils::md_to_html(
form.content.as_ref(),
2019-03-20 17:56:17 +01:00
&Instance::get_local(&conn)
.expect("comments::create: local instance error")
.public_domain,
true,
);
2019-03-20 17:56:17 +01:00
let comm = Comment::insert(
&*conn,
NewComment {
content: SafeString::new(html.as_ref()),
in_response_to_id: form.responding_to,
post_id: post.id,
author_id: user.id,
ap_url: None,
sensitive: !form.warning.is_empty(),
spoiler_text: form.warning.clone(),
public_visibility: true,
},
)
.expect("comments::create: insert error");
comm.notify(&*conn).expect("comments::create: notify error");
2019-03-20 17:56:17 +01:00
let new_comment = comm
.create_activity(&*conn)
.expect("comments::create: activity error");
2018-05-19 00:04:30 +02:00
// save mentions
for ment in mentions {
Mention::from_activity(
&*conn,
2019-03-20 17:56:17 +01:00
&Mention::build_activity(&*conn, &ment)
.expect("comments::create: build mention error"),
comm.id,
false,
2019-03-20 17:56:17 +01:00
true,
)
.expect("comments::create: mention save error");
}
// federate
let dest = User::one_by_instance(&*conn).expect("comments::create: dest error");
2018-07-26 17:51:41 +02:00
let user_clone = user.clone();
worker.execute(move || broadcast(&user_clone, new_comment, dest));
2019-03-20 17:56:17 +01:00
Redirect::to(
uri!(super::posts::details: blog = blog_name, slug = slug, responding_to = _),
)
2018-07-06 11:51:19 +02:00
})
.map_err(|errors| {
// TODO: de-duplicate this code
2019-03-20 17:56:17 +01:00
let comments = CommentTree::from_post(&*conn, &post, Some(&user))
.expect("comments::create: comments error");
2018-05-10 17:36:32 +02:00
2019-03-20 17:56:17 +01:00
let previous = form
.responding_to
.and_then(|r| Comment::get(&*conn, r).ok());
render!(posts::details(
&(&*conn, &intl.catalog, Some(user.clone())),
post.clone(),
blog,
&*form,
errors,
Tag::for_post(&*conn, post.id).expect("comments::create: tags error"),
comments,
previous,
2019-03-20 17:56:17 +01:00
post.count_likes(&*conn)
.expect("comments::create: count likes error"),
post.count_reshares(&*conn)
.expect("comments::create: count reshares error"),
user.has_liked(&*conn, &post)
.expect("comments::create: liked error"),
user.has_reshared(&*conn, &post)
.expect("comments::create: reshared error"),
user.is_following(
&*conn,
post.get_authors(&*conn)
.expect("comments::create: authors error")[0]
.id
)
.expect("comments::create: following error"),
post.get_authors(&*conn)
.expect("comments::create: authors error")[0]
.clone()
))
})
2018-05-10 11:44:57 +02:00
}
#[post("/~/<blog>/<slug>/comment/<id>/delete")]
2019-03-20 17:56:17 +01:00
pub fn delete(
blog: String,
slug: String,
id: i32,
user: User,
conn: DbConn,
worker: Worker,
) -> Result<Redirect, ErrorPage> {
if let Ok(comment) = Comment::get(&*conn, id) {
if comment.author_id == user.id {
let dest = User::one_by_instance(&*conn)?;
let delete_activity = comment.delete(&*conn)?;
let user_c = user.clone();
worker.execute(move || broadcast(&user_c, delete_activity, dest));
2019-03-20 17:56:17 +01:00
worker.execute_after(Duration::from_secs(10 * 60), move || {
user.rotate_keypair(&conn)
.expect("Failed to rotate keypair");
});
}
}
2019-03-20 17:56:17 +01:00
Ok(Redirect::to(
uri!(super::posts::details: blog = blog, slug = slug, responding_to = _),
))
}
#[get("/~/<_blog>/<_slug>/comment/<id>")]
2019-03-20 17:56:17 +01:00
pub fn activity_pub(
_blog: String,
_slug: String,
id: i32,
_ap: ApRequest,
conn: DbConn,
) -> Option<ActivityStream<Note>> {
Comment::get(&*conn, id)
.and_then(|c| c.to_activity(&*conn))
.ok()
.map(ActivityStream::new)
}