use rocket::{
    request::Form,
    response::{Redirect, Flash}
};
use rocket_contrib::Template;

use activity_pub::{broadcast, IntoId, inbox::Notify};
use db_conn::DbConn;
use models::{
    blogs::Blog,
    comments::*,
    posts::Post,
    users::User
};

use utils;
use safe_string::SafeString;

#[get("/~/<blog>/<slug>/comment")]
fn new(blog: String, slug: String, user: User, conn: DbConn) -> Template {
    may_fail!(Blog::find_by_fqn(&*conn, blog), "Couldn't find this blog", |blog| {
        may_fail!(Post::find_by_slug(&*conn, slug, blog.id), "Couldn't find this post", |post| {
            Template::render("comments/new", json!({
                "post": post,
                "account": user
            }))
        })
    })
}

#[get("/~/<blog>/<slug>/comment", rank=2)]
fn new_auth(blog: String, slug: String) -> Flash<Redirect>{
    utils::requires_login("You need to be logged in order to post a comment", &format!("~/{}/{}/comment", blog, slug))
}

#[derive(FromForm)]
struct CommentQuery {
    responding_to: Option<i32>
}

#[derive(FromForm)]
struct NewCommentForm {
    pub content: String
}

#[post("/~/<blog_name>/<slug>/comment?<query>", data = "<data>")]
fn create(blog_name: String, slug: String, query: CommentQuery, data: Form<NewCommentForm>, user: User, conn: DbConn) -> Redirect {
    let blog = Blog::find_by_fqn(&*conn, blog_name.clone()).unwrap();
    let post = Post::find_by_slug(&*conn, slug.clone(), blog.id).unwrap();
    let form = data.get();
    let comment = Comment::insert(&*conn, NewComment {
        content: SafeString::new(&form.content.clone()),
        in_response_to_id: query.responding_to,
        post_id: post.id,
        author_id: user.id,
        ap_url: None,
        sensitive: false,
        spoiler_text: "".to_string()
    });

    Comment::notify(&*conn, comment.into_activity(&*conn), user.clone().into_id());
    broadcast(&*conn, &user, comment.create_activity(&*conn), user.get_followers(&*conn));

    Redirect::to(format!("/~/{}/{}/#comment-{}", blog_name, slug, comment.id))
}