Plume/plume-models/src/blogs.rs

use crate::{
    ap_url, db_conn::DbConn, instance::*, medias::Media, posts::Post, safe_string::SafeString,
    schema::blogs, users::User, Connection, Error, PlumeRocket, Result, CONFIG, ITEMS_PER_PAGE,
};
use activitypub::{
    actor::Group,
    collection::{OrderedCollection, OrderedCollectionPage},
    object::Image,
    CustomObject,
};
use activitystreams::{
    actor::AsApActor,
    base::AnyBase,
    object::{kind::ImageType, ApObject, ApObjectExt, Image as Image07, ObjectExt},
    prelude::*,
};
use chrono::NaiveDateTime;
use diesel::{self, ExpressionMethods, OptionalExtension, QueryDsl, RunQueryDsl, SaveChangesDsl};
use openssl::{
    hash::MessageDigest,
    pkey::{PKey, Private},
    rsa::Rsa,
    sign::{Signer, Verifier},
};
use plume_common::activity_pub::{
    inbox::{AsActor, FromId, FromId07},
    sign, ActivityStream, ApSignature, CustomGroup as CustomGroup07, Id, IntoId, PublicKey, Source,
    ToAsString, ToAsUri,
};
use url::Url;
use webfinger::*;

pub type CustomGroup = CustomObject<ApSignature, Group>;

#[derive(Queryable, Identifiable, Clone, AsChangeset)]
#[changeset_options(treat_none_as_null = "true")]
pub struct Blog {
    pub id: i32,
    pub actor_id: String,
    pub title: String,
    pub summary: String,
    pub outbox_url: String,
    pub inbox_url: String,
    pub instance_id: i32,
    pub creation_date: NaiveDateTime,
    pub ap_url: String,
    pub private_key: Option<String>,
    pub public_key: String,
    pub fqn: String,
    pub summary_html: SafeString,
    pub icon_id: Option<i32>,
    pub banner_id: Option<i32>,
    pub theme: Option<String>,
}

#[derive(Default, Insertable)]
#[table_name = "blogs"]
pub struct NewBlog {
    pub actor_id: String,
    pub title: String,
    pub summary: String,
    pub outbox_url: String,
    pub inbox_url: String,
    pub instance_id: i32,
    pub ap_url: String,
    pub private_key: Option<String>,
    pub public_key: String,
    pub summary_html: SafeString,
    pub icon_id: Option<i32>,
    pub banner_id: Option<i32>,
    pub theme: Option<String>,
}

const BLOG_PREFIX: &str = "~";

impl Blog {
    insert!(blogs, NewBlog, |inserted, conn| {
        let instance = inserted.get_instance(conn)?;
        if inserted.outbox_url.is_empty() {
            inserted.outbox_url = instance.compute_box(BLOG_PREFIX, &inserted.actor_id, "outbox");
        }

        if inserted.inbox_url.is_empty() {
            inserted.inbox_url = instance.compute_box(BLOG_PREFIX, &inserted.actor_id, "inbox");
        }

        if inserted.ap_url.is_empty() {
            inserted.ap_url = instance.compute_box(BLOG_PREFIX, &inserted.actor_id, "");
        }

        if inserted.fqn.is_empty() {
            if instance.local {
                inserted.fqn = inserted.actor_id.clone();
            } else {
                inserted.fqn = format!("{}@{}", inserted.actor_id, instance.public_domain);
            }
        }

        inserted.save_changes(conn).map_err(Error::from)
    });
    get!(blogs);
    find_by!(blogs, find_by_ap_url, ap_url as &str);
    find_by!(blogs, find_by_name, actor_id as &str, instance_id as i32);

    pub fn get_instance(&self, conn: &Connection) -> Result<Instance> {
        Instance::get(conn, self.instance_id)
    }

    pub fn list_authors(&self, conn: &Connection) -> Result<Vec<User>> {
        use crate::schema::blog_authors;
        use crate::schema::users;
        let authors_ids = blog_authors::table
            .filter(blog_authors::blog_id.eq(self.id))
            .select(blog_authors::author_id);
        users::table
            .filter(users::id.eq_any(authors_ids))
            .load::<User>(conn)
            .map_err(Error::from)
    }

    pub fn count_authors(&self, conn: &Connection) -> Result<i64> {
        use crate::schema::blog_authors;
        blog_authors::table
            .filter(blog_authors::blog_id.eq(self.id))
            .count()
            .get_result(conn)
            .map_err(Error::from)
    }

    pub fn find_for_author(conn: &Connection, author: &User) -> Result<Vec<Blog>> {
        use crate::schema::blog_authors;
        let author_ids = blog_authors::table
            .filter(blog_authors::author_id.eq(author.id))
            .select(blog_authors::blog_id);
        blogs::table
            .filter(blogs::id.eq_any(author_ids))
            .load::<Blog>(conn)
            .map_err(Error::from)
    }

    pub fn find_by_fqn(conn: &DbConn, fqn: &str) -> Result<Blog> {
        let from_db = blogs::table
            .filter(blogs::fqn.eq(fqn))
            .first(&**conn)
            .optional()?;
        if let Some(from_db) = from_db {
            Ok(from_db)
        } else {
            Blog::fetch_from_webfinger(conn, fqn)
        }
    }

    fn fetch_from_webfinger(conn: &DbConn, acct: &str) -> Result<Blog> {
        resolve_with_prefix(Prefix::Group, acct.to_owned(), true)?
            .links
            .into_iter()
            .find(|l| l.mime_type == Some(String::from("application/activity+json")))
            .ok_or(Error::Webfinger)
            .and_then(|l| {
                Blog::from_id(
                    conn,
                    &l.href.ok_or(Error::MissingApProperty)?,
                    None,
                    CONFIG.proxy(),
                )
                .map_err(|(_, e)| e)
            })
    }

    pub fn to_activity(&self, conn: &Connection) -> Result<CustomGroup> {
        let mut blog = Group::default();
        blog.ap_actor_props
            .set_preferred_username_string(self.actor_id.clone())?;
        blog.object_props.set_name_string(self.title.clone())?;
        blog.ap_actor_props
            .set_outbox_string(self.outbox_url.clone())?;
        blog.ap_actor_props
            .set_inbox_string(self.inbox_url.clone())?;
        blog.object_props
            .set_summary_string(self.summary_html.to_string())?;
        blog.ap_object_props.set_source_object(Source {
            content: self.summary.clone(),
            media_type: String::from("text/markdown"),
        })?;

        let mut icon = Image::default();
        icon.object_props.set_url_string(
            self.icon_id
                .and_then(|id| Media::get(conn, id).and_then(|m| m.url()).ok())
                .unwrap_or_default(),
        )?;
        icon.object_props.set_attributed_to_link(
            self.icon_id
                .and_then(|id| {
                    Media::get(conn, id)
                        .and_then(|m| Ok(User::get(conn, m.owner_id)?.into_id()))
                        .ok()
                })
                .unwrap_or_else(|| Id::new(String::new())),
        )?;
        blog.object_props.set_icon_object(icon)?;

        let mut banner = Image::default();
        banner.object_props.set_url_string(
            self.banner_id
                .and_then(|id| Media::get(conn, id).and_then(|m| m.url()).ok())
                .unwrap_or_default(),
        )?;
        banner.object_props.set_attributed_to_link(
            self.banner_id
                .and_then(|id| {
                    Media::get(conn, id)
                        .and_then(|m| Ok(User::get(conn, m.owner_id)?.into_id()))
                        .ok()
                })
                .unwrap_or_else(|| Id::new(String::new())),
        )?;
        blog.object_props.set_image_object(banner)?;

        blog.object_props.set_id_string(self.ap_url.clone())?;

        let mut public_key = PublicKey::default();
        public_key.set_id_string(format!("{}#main-key", self.ap_url))?;
        public_key.set_owner_string(self.ap_url.clone())?;
        public_key.set_public_key_pem_string(self.public_key.clone())?;
        let mut ap_signature = ApSignature::default();
        ap_signature.set_public_key_publickey(public_key)?;

        Ok(CustomGroup::new(blog, ap_signature))
    }

    pub fn outbox(&self, conn: &Connection) -> Result<ActivityStream<OrderedCollection>> {
        self.outbox_collection(conn)
            .map(|coll| ActivityStream::new(coll))
    }
    pub fn outbox_collection(&self, conn: &Connection) -> Result<OrderedCollection> {
        let mut coll = OrderedCollection::default();
        coll.collection_props.items = serde_json::to_value(self.get_activities(conn))?;
        coll.collection_props
            .set_total_items_u64(self.get_activities(conn).len() as u64)?;
        coll.collection_props
            .set_first_link(Id::new(ap_url(&format!("{}?page=1", &self.outbox_url))))?;
        coll.collection_props
            .set_last_link(Id::new(ap_url(&format!(
                "{}?page={}",
                &self.outbox_url,
                (self.get_activities(conn).len() as u64 + ITEMS_PER_PAGE as u64 - 1) as u64
                    / ITEMS_PER_PAGE as u64
            ))))?;
        Ok(coll)
    }
    pub fn outbox_page(
        &self,
        conn: &Connection,
        (min, max): (i32, i32),
    ) -> Result<ActivityStream<OrderedCollectionPage>> {
        let mut coll = OrderedCollectionPage::default();
        let acts = self.get_activity_page(conn, (min, max));
        //This still doesn't do anything because the outbox
        //doesn't do anything yet
        coll.collection_page_props.set_next_link(Id::new(&format!(
            "{}?page={}",
            &self.outbox_url,
            min / ITEMS_PER_PAGE + 1
        )))?;
        coll.collection_page_props.set_prev_link(Id::new(&format!(
            "{}?page={}",
            &self.outbox_url,
            min / ITEMS_PER_PAGE - 1
        )))?;
        coll.collection_props.items = serde_json::to_value(acts)?;
        Ok(ActivityStream::new(coll))
    }
    fn get_activities(&self, _conn: &Connection) -> Vec<serde_json::Value> {
        vec![]
    }
    fn get_activity_page(
        &self,
        _conn: &Connection,
        (_min, _max): (i32, i32),
    ) -> Vec<serde_json::Value> {
        vec![]
    }

    pub fn get_keypair(&self) -> Result<PKey<Private>> {
        PKey::from_rsa(Rsa::private_key_from_pem(
            self.private_key
                .clone()
                .ok_or(Error::MissingApProperty)?
                .as_ref(),
        )?)
        .map_err(Error::from)
    }

    pub fn webfinger(&self, conn: &Connection) -> Result<Webfinger> {
        Ok(Webfinger {
            subject: format!(
                "acct:{}@{}",
                self.actor_id,
                self.get_instance(conn)?.public_domain
            ),
            aliases: vec![self.ap_url.clone()],
            links: vec![
                Link {
                    rel: String::from("http://webfinger.net/rel/profile-page"),
                    mime_type: None,
                    href: Some(self.ap_url.clone()),
                    template: None,
                },
                Link {
                    rel: String::from("http://schemas.google.com/g/2010#updates-from"),
                    mime_type: Some(String::from("application/atom+xml")),
                    href: Some(self.get_instance(conn)?.compute_box(
                        BLOG_PREFIX,
                        &self.actor_id,
                        "feed.atom",
                    )),
                    template: None,
                },
                Link {
                    rel: String::from("self"),
                    mime_type: Some(String::from("application/activity+json")),
                    href: Some(self.ap_url.clone()),
                    template: None,
                },
            ],
        })
    }

    pub fn icon_url(&self, conn: &Connection) -> String {
        self.icon_id
            .and_then(|id| Media::get(conn, id).and_then(|m| m.url()).ok())
            .unwrap_or_else(|| "/static/images/default-avatar.png".to_string())
    }

    pub fn banner_url(&self, conn: &Connection) -> Option<String> {
        self.banner_id
            .and_then(|i| Media::get(conn, i).ok())
            .and_then(|c| c.url().ok())
    }

    pub fn delete(&self, conn: &Connection) -> Result<()> {
        for post in Post::get_for_blog(conn, self)? {
            post.delete(conn)?;
        }
        diesel::delete(self)
            .execute(conn)
            .map(|_| ())
            .map_err(Error::from)
    }
}

impl IntoId for Blog {
    fn into_id(self) -> Id {
        Id::new(self.ap_url)
    }
}

impl FromId<DbConn> for Blog {
    type Error = Error;
    type Object = CustomGroup;

    fn from_db(conn: &DbConn, id: &str) -> Result<Self> {
        Self::find_by_ap_url(conn, id)
    }

    fn from_activity(conn: &DbConn, acct: CustomGroup) -> Result<Self> {
        let url = Url::parse(&acct.object.object_props.id_string()?)?;
        let inst = url.host_str().ok_or(Error::Url)?;
        let instance = Instance::find_by_domain(conn, inst).or_else(|_| {
            Instance::insert(
                conn,
                NewInstance {
                    public_domain: inst.to_owned(),
                    name: inst.to_owned(),
                    local: false,
                    // We don't really care about all the following for remote instances
                    long_description: SafeString::new(""),
                    short_description: SafeString::new(""),
                    default_license: String::new(),
                    open_registrations: true,
                    short_description_html: String::new(),
                    long_description_html: String::new(),
                },
            )
        })?;
        let icon_id = acct
            .object
            .object_props
            .icon_image()
            .ok()
            .and_then(|icon| {
                let owner = icon.object_props.attributed_to_link::<Id>().ok()?;
                Media::save_remote(
                    conn,
                    icon.object_props.url_string().ok()?,
                    &User::from_id(conn, &owner, None, CONFIG.proxy()).ok()?,
                )
                .ok()
            })
            .map(|m| m.id);

        let banner_id = acct
            .object
            .object_props
            .image_image()
            .ok()
            .and_then(|banner| {
                let owner = banner.object_props.attributed_to_link::<Id>().ok()?;
                Media::save_remote(
                    conn,
                    banner.object_props.url_string().ok()?,
                    &User::from_id(conn, &owner, None, CONFIG.proxy()).ok()?,
                )
                .ok()
            })
            .map(|m| m.id);

        let name = acct.object.ap_actor_props.preferred_username_string()?;
        if name.contains(&['<', '>', '&', '@', '\'', '"', ' ', '\t'][..]) {
            return Err(Error::InvalidValue);
        }

        Blog::insert(
            conn,
            NewBlog {
                actor_id: name.clone(),
                title: acct.object.object_props.name_string().unwrap_or(name),
                outbox_url: acct.object.ap_actor_props.outbox_string()?,
                inbox_url: acct.object.ap_actor_props.inbox_string()?,
                summary: acct
                    .object
                    .ap_object_props
                    .source_object::<Source>()
                    .map(|s| s.content)
                    .unwrap_or_default(),
                instance_id: instance.id,
                ap_url: acct.object.object_props.id_string()?,
                public_key: acct
                    .custom_props
                    .public_key_publickey()?
                    .public_key_pem_string()?,
                private_key: None,
                banner_id,
                icon_id,
                summary_html: SafeString::new(
                    &acct
                        .object
                        .object_props
                        .summary_string()
                        .unwrap_or_default(),
                ),
                theme: None,
            },
        )
    }

    fn get_sender() -> &'static dyn sign::Signer {
        Instance::get_local_instance_user().expect("Failed to local instance user")
    }
}

impl FromId07<DbConn> for Blog {
    type Error = Error;
    type Object = CustomGroup07;

    fn from_db07(conn: &DbConn, id: &str) -> Result<Self> {
        Self::find_by_ap_url(conn, id)
    }

    fn from_activity07(conn: &DbConn, acct: CustomGroup07) -> Result<Self> {
        let (name, outbox_url, inbox_url) = {
            let actor = acct.ap_actor_ref();
            let name = actor
                .preferred_username()
                .ok_or(Error::MissingApProperty)?
                .to_string();
            if name.contains(&['<', '>', '&', '@', '\'', '"', ' ', '\t'][..]) {
                return Err(Error::InvalidValue);
            }
            (
                name,
                actor.outbox()?.ok_or(Error::MissingApProperty)?.to_string(),
                actor.inbox()?.to_string(),
            )
        };

        let mut new_blog = NewBlog {
            actor_id: name.to_string(),
            outbox_url,
            inbox_url,
            public_key: acct.ext_one.public_key.public_key_pem.to_string(),
            private_key: None,
            theme: None,
            ..NewBlog::default()
        };

        let object = ApObject::new(acct.inner);
        new_blog.title = object
            .name()
            .and_then(|name| name.to_as_string())
            .unwrap_or_else(|| name);
        new_blog.summary_html = SafeString::new(
            &object
                .summary()
                .and_then(|summary| summary.to_as_string())
                .unwrap_or_default(),
        );

        let icon_id = object
            .icon()
            .and_then(|icons| {
                icons.iter().next().and_then(|icon| {
                    let icon = icon.to_owned().extend::<Image07, ImageType>().ok()??;
                    let owner = icon.attributed_to()?.to_as_uri()?;
                    Media::save_remote(
                        conn,
                        icon.url()?.to_as_uri()?,
                        &User::from_id07(conn, &owner, None, CONFIG.proxy()).ok()?,
                    )
                    .ok()
                })
            })
            .map(|m| m.id);
        new_blog.icon_id = icon_id;

        let banner_id = object
            .image()
            .and_then(|banners| {
                banners.iter().next().and_then(|banner| {
                    let banner = banner.to_owned().extend::<Image07, ImageType>().ok()??;
                    let owner = banner.attributed_to()?.to_as_uri()?;
                    Media::save_remote(
                        conn,
                        banner.url()?.to_as_uri()?,
                        &User::from_id(conn, &owner, None, CONFIG.proxy()).ok()?,
                    )
                    .ok()
                })
            })
            .map(|m| m.id);
        new_blog.banner_id = banner_id;

        let source = object
            .source()
            .and_then(|s| s.as_xsd_string())
            .unwrap_or_default()
            .to_string();
        new_blog.summary = source;

        let any_base = AnyBase::from_extended(object)?;
        let id = any_base.id().ok_or(Error::MissingApProperty)?;
        new_blog.ap_url = id.to_string();

        let inst = id
            .authority_components()
            .ok_or(Error::Url)?
            .host()
            .to_string();
        let instance = Instance::find_by_domain(conn, &inst).or_else(|_| {
            Instance::insert(
                conn,
                NewInstance {
                    public_domain: inst.to_owned(),
                    name: inst.to_owned(),
                    local: false,
                    // We don't really care about all the following for remote instances
                    long_description: SafeString::new(""),
                    short_description: SafeString::new(""),
                    default_license: String::new(),
                    open_registrations: true,
                    short_description_html: String::new(),
                    long_description_html: String::new(),
                },
            )
        })?;
        new_blog.instance_id = instance.id;

        Blog::insert(conn, new_blog)
    }

    fn get_sender07() -> &'static dyn sign::Signer {
        Instance::get_local_instance_user().expect("Failed to local instance user")
    }
}

impl AsActor<&PlumeRocket> for Blog {
    fn get_inbox_url(&self) -> String {
        self.inbox_url.clone()
    }

    fn get_shared_inbox_url(&self) -> Option<String> {
        None
    }

    fn is_local(&self) -> bool {
        Instance::get_local()
            .map(|i| self.instance_id == i.id)
            .unwrap_or(false)
    }
}

impl sign::Signer for Blog {
    fn get_key_id(&self) -> String {
        format!("{}#main-key", self.ap_url)
    }

    fn sign(&self, to_sign: &str) -> sign::Result<Vec<u8>> {
        let key = self.get_keypair().map_err(|_| sign::Error())?;
        let mut signer = Signer::new(MessageDigest::sha256(), &key)?;
        signer.update(to_sign.as_bytes())?;
        signer.sign_to_vec().map_err(sign::Error::from)
    }

    fn verify(&self, data: &str, signature: &[u8]) -> sign::Result<bool> {
        let key = PKey::from_rsa(Rsa::public_key_from_pem(self.public_key.as_ref())?)?;
        let mut verifier = Verifier::new(MessageDigest::sha256(), &key)?;
        verifier.update(data.as_bytes())?;
        verifier.verify(signature).map_err(sign::Error::from)
    }
}

impl NewBlog {
    pub fn new_local(
        actor_id: String,
        title: String,
        summary: String,
        instance_id: i32,
    ) -> Result<NewBlog> {
        let (pub_key, priv_key) = sign::gen_keypair();
        Ok(NewBlog {
            actor_id,
            title,
            summary,
            instance_id,
            public_key: String::from_utf8(pub_key).or(Err(Error::Signature))?,
            private_key: Some(String::from_utf8(priv_key).or(Err(Error::Signature))?),
            ..NewBlog::default()
        })
    }
}

#[cfg(test)]
pub(crate) mod tests {
    use super::*;
    use crate::{
        blog_authors::*,
        instance::tests as instance_tests,
        medias::NewMedia,
        post_authors::{NewPostAuthor, PostAuthor},
        posts::{NewPost, Post},
        tests::db,
        users::tests as usersTests,
        Connection as Conn, ITEMS_PER_PAGE,
    };
    use assert_json_diff::assert_json_eq;
    use diesel::Connection;
    use serde_json::to_value;

    pub(crate) fn fill_database(conn: &Conn) -> (Vec<User>, Vec<Blog>) {
        instance_tests::fill_database(conn);
        let users = usersTests::fill_database(conn);
        let blog1 = Blog::insert(
            conn,
            NewBlog::new_local(
                "BlogName".to_owned(),
                "Blog name".to_owned(),
                "This is a small blog".to_owned(),
                Instance::get_local().unwrap().id,
            )
            .unwrap(),
        )
        .unwrap();
        let blog2 = Blog::insert(
            conn,
            NewBlog::new_local(
                "MyBlog".to_owned(),
                "My blog".to_owned(),
                "Welcome to my blog".to_owned(),
                Instance::get_local().unwrap().id,
            )
            .unwrap(),
        )
        .unwrap();
        let blog3 = Blog::insert(
            conn,
            NewBlog::new_local(
                "WhyILikePlume".to_owned(),
                "Why I like Plume".to_owned(),
                "In this blog I will explay you why I like Plume so much".to_owned(),
                Instance::get_local().unwrap().id,
            )
            .unwrap(),
        )
        .unwrap();

        BlogAuthor::insert(
            conn,
            NewBlogAuthor {
                blog_id: blog1.id,
                author_id: users[0].id,
                is_owner: true,
            },
        )
        .unwrap();

        BlogAuthor::insert(
            conn,
            NewBlogAuthor {
                blog_id: blog1.id,
                author_id: users[1].id,
                is_owner: false,
            },
        )
        .unwrap();

        BlogAuthor::insert(
            conn,
            NewBlogAuthor {
                blog_id: blog2.id,
                author_id: users[1].id,
                is_owner: true,
            },
        )
        .unwrap();

        BlogAuthor::insert(
            conn,
            NewBlogAuthor {
                blog_id: blog3.id,
                author_id: users[2].id,
                is_owner: true,
            },
        )
        .unwrap();

        for i in 1..(ITEMS_PER_PAGE * 4 + 3) {
            let title = format!("Post {}", i);
            let content = format!("Content for post {}.", i);
            let post = Post::insert(
                conn,
                NewPost {
                    blog_id: blog1.id,
                    slug: title.clone(),
                    title: title.clone(),
                    content: SafeString::new(&content),
                    published: true,
                    license: "CC-0".into(),
                    creation_date: None,
                    ap_url: format!("{}/{}", blog1.ap_url, title),
                    subtitle: "".into(),
                    source: content,
                    cover_id: None,
                },
            )
            .unwrap();
            PostAuthor::insert(
                conn,
                NewPostAuthor {
                    post_id: post.id,
                    author_id: users[0].id,
                },
            )
            .unwrap();
        }

        (users, vec![blog1, blog2, blog3])
    }

    #[test]
    fn get_instance() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            fill_database(&conn);

            let blog = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "SomeName".to_owned(),
                    "Some name".to_owned(),
                    "This is some blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();

            assert_eq!(
                blog.get_instance(&conn).unwrap().id,
                Instance::get_local().unwrap().id
            );
            // TODO add tests for remote instance
            Ok(())
        })
    }

    #[test]
    fn authors() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            let (user, _) = fill_database(&conn);

            let b1 = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "SomeName".to_owned(),
                    "Some name".to_owned(),
                    "This is some blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();
            let b2 = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "Blog".to_owned(),
                    "Blog".to_owned(),
                    "I've named my blog Blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();
            let blog = vec![b1, b2];

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[0].id,
                    author_id: user[0].id,
                    is_owner: true,
                },
            )
            .unwrap();

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[0].id,
                    author_id: user[1].id,
                    is_owner: false,
                },
            )
            .unwrap();

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[1].id,
                    author_id: user[0].id,
                    is_owner: true,
                },
            )
            .unwrap();

            assert!(blog[0]
                .list_authors(&conn)
                .unwrap()
                .iter()
                .any(|a| a.id == user[0].id));
            assert!(blog[0]
                .list_authors(&conn)
                .unwrap()
                .iter()
                .any(|a| a.id == user[1].id));
            assert!(blog[1]
                .list_authors(&conn)
                .unwrap()
                .iter()
                .any(|a| a.id == user[0].id));
            assert!(!blog[1]
                .list_authors(&conn)
                .unwrap()
                .iter()
                .any(|a| a.id == user[1].id));

            assert!(Blog::find_for_author(&conn, &user[0])
                .unwrap()
                .iter()
                .any(|b| b.id == blog[0].id));
            assert!(Blog::find_for_author(&conn, &user[1])
                .unwrap()
                .iter()
                .any(|b| b.id == blog[0].id));
            assert!(Blog::find_for_author(&conn, &user[0])
                .unwrap()
                .iter()
                .any(|b| b.id == blog[1].id));
            assert!(!Blog::find_for_author(&conn, &user[1])
                .unwrap()
                .iter()
                .any(|b| b.id == blog[1].id));
            Ok(())
        })
    }

    #[test]
    fn find_local() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            fill_database(&conn);

            let blog = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "SomeName".to_owned(),
                    "Some name".to_owned(),
                    "This is some blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();

            assert_eq!(Blog::find_by_fqn(&conn, "SomeName").unwrap().id, blog.id);
            Ok(())
        })
    }

    #[test]
    fn get_fqn() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            fill_database(&conn);

            let blog = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "SomeName".to_owned(),
                    "Some name".to_owned(),
                    "This is some blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();

            assert_eq!(blog.fqn, "SomeName");
            Ok(())
        })
    }

    #[test]
    fn delete() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            let (_, blogs) = fill_database(&conn);

            blogs[0].delete(&conn).unwrap();
            assert!(Blog::get(&conn, blogs[0].id).is_err());
            Ok(())
        })
    }

    #[test]
    fn delete_via_user() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            let (user, _) = fill_database(&conn);

            let b1 = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "SomeName".to_owned(),
                    "Some name".to_owned(),
                    "This is some blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();
            let b2 = Blog::insert(
                &conn,
                NewBlog::new_local(
                    "Blog".to_owned(),
                    "Blog".to_owned(),
                    "I've named my blog Blog".to_owned(),
                    Instance::get_local().unwrap().id,
                )
                .unwrap(),
            )
            .unwrap();
            let blog = vec![b1, b2];

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[0].id,
                    author_id: user[0].id,
                    is_owner: true,
                },
            )
            .unwrap();

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[0].id,
                    author_id: user[1].id,
                    is_owner: false,
                },
            )
            .unwrap();

            BlogAuthor::insert(
                &conn,
                NewBlogAuthor {
                    blog_id: blog[1].id,
                    author_id: user[0].id,
                    is_owner: true,
                },
            )
            .unwrap();

            user[0].delete(&conn).unwrap();
            assert!(Blog::get(&conn, blog[0].id).is_ok());
            assert!(Blog::get(&conn, blog[1].id).is_err());
            user[1].delete(&conn).unwrap();
            assert!(Blog::get(&conn, blog[0].id).is_err());
            Ok(())
        })
    }

    #[test]
    fn self_federation() {
        let conn = &db();
        conn.test_transaction::<_, (), _>(|| {
            let (users, mut blogs) = fill_database(&conn);
            blogs[0].icon_id = Some(
                Media::insert(
                    &conn,
                    NewMedia {
                        file_path: "aaa.png".into(),
                        alt_text: String::new(),
                        is_remote: false,
                        remote_url: None,
                        sensitive: false,
                        content_warning: None,
                        owner_id: users[0].id,
                    },
                )
                .unwrap()
                .id,
            );
            blogs[0].banner_id = Some(
                Media::insert(
                    &conn,
                    NewMedia {
                        file_path: "bbb.png".into(),
                        alt_text: String::new(),
                        is_remote: false,
                        remote_url: None,
                        sensitive: false,
                        content_warning: None,
                        owner_id: users[0].id,
                    },
                )
                .unwrap()
                .id,
            );
            let _: Blog = blogs[0].save_changes(&**conn).unwrap();

            let ap_repr = blogs[0].to_activity(&conn).unwrap();
            blogs[0].delete(&conn).unwrap();
            let blog = Blog::from_activity(&conn, ap_repr).unwrap();

            assert_eq!(blog.actor_id, blogs[0].actor_id);
            assert_eq!(blog.title, blogs[0].title);
            assert_eq!(blog.summary, blogs[0].summary);
            assert_eq!(blog.outbox_url, blogs[0].outbox_url);
            assert_eq!(blog.inbox_url, blogs[0].inbox_url);
            assert_eq!(blog.instance_id, blogs[0].instance_id);
            assert_eq!(blog.ap_url, blogs[0].ap_url);
            assert_eq!(blog.public_key, blogs[0].public_key);
            assert_eq!(blog.fqn, blogs[0].fqn);
            assert_eq!(blog.summary_html, blogs[0].summary_html);
            assert_eq!(blog.icon_url(&conn), blogs[0].icon_url(&conn));
            assert_eq!(blog.banner_url(&conn), blogs[0].banner_url(&conn));

            Ok(())
        })
    }

    #[test]
    fn to_activity() {
        let conn = &db();
        conn.test_transaction::<_, Error, _>(|| {
            let (_users, blogs) = fill_database(&conn);
            let blog = &blogs[0];
            let act = blog.to_activity(conn)?;

            let expected = json!({
                "followers": null,
                "following": null,
                "icon": {
                    "attributedTo": "",
                    "type": "Image",
                    "url": ""
                },
                "id": "https://plu.me/~/BlogName/",
                "image": {
                    "attributedTo": "",
                    "type": "Image",
                    "url": ""
                },
                "inbox": "https://plu.me/~/BlogName/inbox",
                "liked": null,
                "name": "Blog name",
                "outbox": "https://plu.me/~/BlogName/outbox",
                "preferredUsername": "BlogName",
                "publicKey": {
                    "id": "https://plu.me/~/BlogName/#main-key",
                    "owner": "https://plu.me/~/BlogName/",
                    "publicKeyPem": blog.public_key
                },
                "source": {
                    "content": "This is a small blog",
                    "mediaType": "text/markdown"
                },
                "summary": "",
                "type": "Group"
            });

            assert_json_eq!(to_value(act)?, expected);

            Ok(())
        });
    }

    #[test]
    fn outbox_collection() {
        let conn = &db();
        conn.test_transaction::<_, Error, _>(|| {
            let (_users, blogs) = fill_database(conn);
            let blog = &blogs[0];
            let act = blog.outbox_collection(conn)?;

            let expected = json!({
                "items": [],
                "totalItems": 0,
                "first": "https://plu.me/~/BlogName/outbox?page=1",
                "last": "https://plu.me/~/BlogName/outbox?page=0",
                "type": "OrderedCollection"
            });

            assert_json_eq!(to_value(act)?, expected);

            Ok(())
        });
    }
}