use serenity::{
    framework::standard::{
        macros::{command, group},
        Args, CommandResult,
    },
    model::{
        channel::{Channel, Message},
        id::UserId,
    },
};
use soft_ban::{SOFT_BAN_COMMAND, SOFT_BAN_INIT_COMMAND};
use std::{thread::sleep, time::Duration};
use youmubot_prelude::*;

mod soft_ban;
pub use soft_ban::watch_soft_bans;

#[group]
#[description = "Administrative commands for the server."]
#[commands(clean, ban, kick, soft_ban, soft_ban_init)]
struct Admin;

#[command]
#[aliases("cleanall")]
#[required_permissions(MANAGE_MESSAGES)]
#[description = "Clean at most X latest messages from the current channel (only clean Youmu's messages in DMs). Defaults to 10."]
#[usage = "clean 50"]
#[min_args(0)]
#[max_args(1)]
fn clean(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
    let limit = args.single().unwrap_or(10);
    let messages = msg
        .channel_id
        .messages(&ctx.http, |b| b.before(msg.id).limit(limit))?;
    let channel = msg.channel_id.to_channel(&ctx)?;
    match &channel {
        Channel::Private(_) | Channel::Group(_) => {
            let self_id = ctx.http.get_current_application_info()?.id;
            messages
                .into_iter()
                .filter(|v| v.author.id == self_id)
                .try_for_each(|m| m.delete(&ctx))?;
        }
        _ => {
            msg.channel_id
                .delete_messages(&ctx.http, messages.into_iter())?;
        }
    };
    msg.react(&ctx, "🌋")?;
    if let Channel::Guild(_) = &channel {
        sleep(Duration::from_secs(2));
        msg.delete(&ctx)?;
    }

    Ok(())
}

#[command]
#[required_permissions(ADMINISTRATOR)]
#[description = "Ban an user with a certain reason."]
#[usage = "@user#1234/spam"]
#[min_args(1)]
#[max_args(2)]
#[only_in("guilds")]
fn ban(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
    let user = args.single::<UserId>()?.to_user(&ctx)?;
    let reason = args
        .remains()
        .map(|v| format!("`{}`", v))
        .unwrap_or("no provided reason".to_owned());

    msg.reply(
        &ctx,
        format!("🔨 Banning user {} for reason `{}`.", user.tag(), reason),
    )?;

    msg.guild_id
        .ok_or("Can't get guild from message?")? // we had a contract
        .ban(&ctx.http, user, &reason)?;

    Ok(())
}

#[command]
#[required_permissions(ADMINISTRATOR)]
#[description = "Kick an user."]
#[usage = "@user#1234"]
#[num_args(1)]
#[only_in("guilds")]
fn kick(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
    let user = args.single::<UserId>()?.to_user(&ctx)?;

    msg.reply(&ctx, format!("🔫 Kicking user {}.", user.tag()))?;

    msg.guild_id
        .ok_or("Can't get guild from message?")? // we had a contract
        .kick(&ctx.http, user)?;

    Ok(())
}