use furbooru::Client;
use robespierre::framework::standard::{
    extractors::RawArgs, macros::command, CommandResult, FwContext,
};
use robespierre::model::MessageExt;
use robespierre::UserData;
use robespierre_models::channel::Message;
use std::sync::Arc;

pub struct ClientKey;

impl robespierre::typemap::Key for ClientKey {
    type Value = Arc<Client>;
}

pub fn make(username: String, api_key: String) -> Arc<Client> {
    Arc::new(Client::new(user_agent(username), api_key).unwrap())
}

pub fn user_agent(username: String) -> String {
    format!(
        "{}/{} (owner: {})",
        env!("CARGO_PKG_NAME"),
        env!("CARGO_PKG_VERSION"),
        username,
    )
}

#[command]
pub async fn search(ctx: &FwContext, msg: &Message, args: RawArgs) -> CommandResult {
    let data = ctx.data_lock_read().await;
    let client = data.get::<ClientKey>().unwrap().clone();
    let args = format!("{}", args.0);

    let mut imgs = client.image_search(args, 0_u64).await?;

    let mut response = String::new();

    imgs.truncate(3);

    for mut post in imgs.into_iter() {
        post.tags.sort();
        post.tags.truncate(8);

        response.push_str(&format!(
            "<{}> - {} - <{}>\n",
            format!("http://furbooru.org/{}", post.id),
            post.tags.join(", "),
            post.view_url,
        ));
    }

    msg.reply(ctx, response).await?;

    Ok(())
}