use crate::{
    config::{Config, UrlKind},
    data::State,
    db::Actor,
    error::MyError,
};
use activitystreams::{
    activity::{Follow as AsFollow, Undo as AsUndo},
    context,
    prelude::*,
    security,
    url::Url,
};
use std::convert::TryInto;

mod announce;
mod follow;
mod forward;
mod reject;
mod undo;

pub use self::{announce::Announce, follow::Follow, forward::Forward, reject::Reject, undo::Undo};

async fn get_inboxes(state: &State, actor: &Actor, object_id: &Url) -> Result<Vec<Url>, MyError> {
    let domain = object_id.host().ok_or(MyError::Domain)?.to_string();

    state.inboxes_without(&actor.inbox, &domain).await
}

fn prepare_activity<T, U, V, Kind>(
    mut t: T,
    id: impl TryInto<Url, Error = U>,
    to: impl TryInto<Url, Error = V>,
) -> Result<T, MyError>
where
    T: ObjectExt<Kind> + BaseExt<Kind>,
    MyError: From<U> + From<V>,
{
    t.set_id(id.try_into()?)
        .set_many_tos(vec![to.try_into()?])
        .set_many_contexts(vec![context(), security()]);
    Ok(t)
}

// Generate a type that says "I want to stop following you"
fn generate_undo_follow(config: &Config, actor_id: &Url, my_id: &Url) -> Result<AsUndo, MyError> {
    let mut follow = AsFollow::new(my_id.clone(), actor_id.clone());

    follow.set_id(config.generate_url(UrlKind::Activity));

    let undo = AsUndo::new(my_id.clone(), follow.into_any_base()?);

    prepare_activity(undo, config.generate_url(UrlKind::Actor), actor_id.clone())
}