use regex_command_attr::command;

use serenity::{
    constants::MESSAGE_CODE_LIMIT,
    client::Context,
    model::{
        id::{
            UserId, GuildId, ChannelId,
        },
        channel::{
            Message,
        },
    },
    framework::standard::CommandResult,
};

use std::fmt;

use crate::{
    models::{
        UserData,
        GuildData,
    },
    SQLPool,
};
use sqlx::MySqlPool;

#[derive(Debug)]
struct TodoNotFound;

impl std::error::Error for TodoNotFound {}
impl fmt::Display for TodoNotFound {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "Todo not found")
    }
}

#[derive(Debug)]
struct Todo {
    id: u32,
    user_id: Option<u32>,
    guild_id: Option<u32>,
    channel_id: Option<u32>,
    value: String,
}

struct TodoTarget {
    user: UserId,
    guild: Option<GuildId>,
    channel: Option<ChannelId>,
}

impl TodoTarget {
    pub fn command(&self, subcommand_opt: Option<SubCommand>) -> String {
        let context = if self.channel.is_some() {
            "channel"
        }
        else if self.guild.is_some() {
            "guild"
        }
        else {
            "user"
        };

        if let Some(subcommand) = subcommand_opt {
            format!("todo {} {}", context, subcommand.to_string())
        }
        else {
            format!("todo {}", context)
        }
    }

    pub fn name(&self) -> String {
        if self.channel.is_some() {
            "Channel"
        }
        else if self.guild.is_some() {
            "Guild"
        }
        else {
            "User"
        }.to_string()
    }

    pub async fn view(&self, pool: MySqlPool) -> Result<Vec<Todo>, Box<dyn std::error::Error + Send + Sync>> {
        Ok(if let Some(cid) = self.channel {
            sqlx::query_as!(Todo,
                "
SELECT * FROM todos WHERE channel_id = (SELECT id FROM channels WHERE channel = ?)
                ", cid.as_u64())
                .fetch_all(&pool)
                .await?
        }
        else if let Some(gid) = self.guild {
            sqlx::query_as!(Todo,
                "
SELECT * FROM todos WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND channel_id IS NULL
                ", gid.as_u64())
                .fetch_all(&pool)
                .await?
        }
        else {
            sqlx::query_as!(Todo,
                "
SELECT * FROM todos WHERE user_id = (SELECT id FROM users WHERE user = ?) AND guild_id IS NULL
                ", self.user.as_u64())
                .fetch_all(&pool)
                .await?
        })
    }

    pub async fn add(&self, value: String, pool: MySqlPool) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
        if let (Some(cid), Some(gid)) = (self.channel, self.guild) {
            sqlx::query!(
                "
INSERT INTO todos (user_id, guild_id, channel_id, value) VALUES (
    (SELECT id FROM users WHERE user = ?),
    (SELECT id FROM guilds WHERE guild = ?),
    (SELECT id FROM channels WHERE channel = ?),
    ?
)
                ", self.user.as_u64(), gid.as_u64(), cid.as_u64(), value)
                .execute(&pool)
                .await?;
        }
        else if let Some(gid) = self.guild {
            sqlx::query!(
                "
INSERT INTO todos (user_id, guild_id, value) VALUES (
    (SELECT id FROM users WHERE user = ?),
    (SELECT id FROM guilds WHERE guild = ?),
    ?
)
                ", self.user.as_u64(), gid.as_u64(), value)
                .execute(&pool)
                .await?;
        }
        else {
            sqlx::query!(
                "
INSERT INTO todos (user_id, value) VALUES (
    (SELECT id FROM users WHERE user = ?),
    ?
)
                ", self.user.as_u64(), value)
                .execute(&pool)
                .await?;
        }

        Ok(())
    }

    pub async fn remove(&self, num: usize, pool: &MySqlPool) -> Result<Todo, Box<dyn std::error::Error + Sync + Send>> {
        let todos = self.view(pool.clone()).await?;

        if let Some(removal_todo) = todos.get(num) {
            let deleting = sqlx::query_as!(Todo,
                "
SELECT * FROM todos WHERE id = ?
                ", removal_todo.id)
                .fetch_one(&pool.clone())
                .await?;

            sqlx::query!(
                "
DELETE FROM todos WHERE id = ?
                ", removal_todo.id)
                .execute(pool)
                .await?;

            Ok(deleting)
        }
        else {
            Err(Box::new(TodoNotFound))
        }
    }

    pub async fn clear(&self, pool: &MySqlPool) -> Result<(), Box<dyn std::error::Error + Sync + Send>> {
        if let Some(cid) = self.channel {
            sqlx::query!(
                "
DELETE FROM todos WHERE channel_id = (SELECT id FROM channels WHERE channel = ?)
                ", cid.as_u64())
                .execute(pool)
                .await?;
        }
        else if let Some(gid) = self.guild {
            sqlx::query!(
                "
DELETE FROM todos WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND channel_id IS NULL
                ", gid.as_u64())
                .execute(pool)
                .await?;
        }
        else {
            sqlx::query!(
                "
DELETE FROM todos WHERE user_id = (SELECT id FROM users WHERE user = ?) AND guild_id IS NULL
                ", self.user.as_u64())
                .execute(pool)
                .await?;
        }

        Ok(())
    }
}

enum SubCommand {
    View,
    Add,
    Remove,
    Clear,
}

impl ToString for SubCommand {
    fn to_string(&self) -> String {
        match self {
            SubCommand::View => "",
            SubCommand::Add => "add",
            SubCommand::Remove => "remove",
            SubCommand::Clear => "clear",
        }.to_string()
    }
}

#[command]
#[permission_level(Managed)]
async fn todo_parse(ctx: &Context, msg: &Message, args: String) -> CommandResult {

    let mut split = args.split(' ');

    if let Some(target) = split.next() {
        let target_opt = match target {
            "user" =>
                Some(TodoTarget {user: msg.author.id, guild: None, channel: None}),

            "channel" =>
                if let Some(gid) = msg.guild_id {
                    Some(TodoTarget {user: msg.author.id, guild: Some(gid), channel: Some(msg.channel_id)})
                }
                else {
                    None
                },

            "server" | "guild" => {
                if let Some(gid) = msg.guild_id {
                    Some(TodoTarget {user: msg.author.id, guild: Some(gid), channel: None})
                }
                else {
                    None
                }
            },

            _ => {
                 None
            },
        };

        if let Some(target) = target_opt {

            let subcommand_opt = match split.next() {

                Some("add") => Some(SubCommand::Add),

                Some("remove") => Some(SubCommand::Remove),

                Some("clear") => Some(SubCommand::Clear),

                None => Some(SubCommand::View),

                Some(_unrecognised) => None,
            };

            if let Some(subcommand) = subcommand_opt {
                todo(ctx, msg, target, subcommand, split.collect::<Vec<&str>>().join(" ")).await;
            }
            else {
                show_help(&ctx, msg, Some(target)).await;
            }

        }
        else {
            show_help(&ctx, msg, None).await;
        }

    }
    else {
        show_help(&ctx, msg, None).await;
    }

    Ok(())
}

async fn show_help(ctx: &Context, msg: &Message, target: Option<TodoTarget>) {
    let pool = ctx.data.read().await
        .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");

    let user_data = UserData::from_user(&msg.author, &ctx, &pool).await.unwrap();
    let prefix = GuildData::prefix_from_id(msg.guild_id, &pool).await;

    let content = user_data.response(&pool, "todo/help").await
        .replace("{prefix}", &prefix)
        .replace("{command}", target.map_or_else(|| "todo user".to_string(), |t| t.command(None)).as_str());

    let _ = msg.channel_id.say(&ctx, content).await;
}

async fn todo(ctx: &Context, msg: &Message, target: TodoTarget, subcommand: SubCommand, extra: String) {
    let pool = ctx.data.read().await
        .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");

    let user_data = UserData::from_user(&msg.author, &ctx, &pool).await.unwrap();
    let prefix = GuildData::prefix_from_id(msg.guild_id, &pool).await;

    match subcommand {
        SubCommand::View => {
            let todo_items = target.view(pool).await.unwrap();
            let mut todo_groups = vec!["".to_string()];
            let mut char_count = 0;

            todo_items.iter().enumerate().for_each(|(count, todo)| {
                let display = format!("{}: {}\n", count + 1, todo.value);

                if char_count + display.len() > MESSAGE_CODE_LIMIT as usize {
                    char_count = display.len();

                    todo_groups.push(display);
                }
                else {
                    char_count += display.len();

                    let last_group = todo_groups.pop().unwrap();

                    todo_groups.push(format!("{}{}", last_group, display));
                }
            });

            for group in todo_groups {
                let _ = msg.channel_id.send_message(&ctx, |m| m
                    .embed(|e| e
                        .title(format!("{} Todo", target.name()))
                        .description(group)
                    )
                ).await;
            }
        },

        SubCommand::Add => {
            let content = user_data.response(&pool, "todo/added").await
                .replacen("{name}", &extra, 1);

            target.add(extra, pool).await.unwrap();

            let _ = msg.channel_id.say(&ctx, content).await;
        },

        SubCommand::Remove => {
            let _ = if let Ok(num) = extra.parse::<usize>() {
                if let Ok(todo) = target.remove(num - 1, &pool).await {
                    let content = user_data.response(&pool, "todo/removed").await
                        .replacen("{}", &todo.value, 1);

                    msg.channel_id.say(&ctx, content)
                }
                else {
                    msg.channel_id.say(&ctx, user_data.response(&pool, "todo/error_index").await)
                }
            }
            else {
                let content = user_data.response(&pool, "todo/error_value").await
                    .replacen("{prefix}", &prefix, 1)
                    .replacen("{command}", &target.command(Some(subcommand)), 1);

                msg.channel_id.say(&ctx, content)
            }.await;
        },

        SubCommand::Clear => {
            target.clear(&pool).await.unwrap();

            let content = user_data.response(&pool, "todo/cleared").await;

            let _ = msg.channel_id.say(&ctx, content).await;
        },
    }
}