#include "hecl/Console.hpp"
#include "hecl/CVarManager.hpp"
#include "hecl/CVar.hpp"
#include "athena/Utility.hpp"

namespace hecl
{
Console* Console::m_instance = nullptr;
Console::Console(CVarManager* cvarMgr)
    : m_overwrite(false)
    , m_cursorAtEnd(false)
{
    m_instance = this;
    registerCommand("help", "Prints information about a given function", "<command>", std::bind(&Console::help, this, std::placeholders::_1, std::placeholders::_2));
    registerCommand("listCommands", "Prints a list of all available Commands", "", std::bind(&Console::listCommands, this, std::placeholders::_1, std::placeholders::_2));
    registerCommand("listCVars", "Lists all available CVars", "", std::bind(&CVarManager::list, cvarMgr, std::placeholders::_1, std::placeholders::_2));
    registerCommand("setCVar", "Sets a given Console Variable to the specified value", "<cvar> <value>", std::bind(&CVarManager::setCVar, cvarMgr, std::placeholders::_1, std::placeholders::_2));
    registerCommand("getCVar", "Prints the value stored in the specified Console Variable", "<cvar>", std::bind(&CVarManager::getCVar, cvarMgr, std::placeholders::_1, std::placeholders::_2));
}

void Console::registerCommand(std::string_view name, std::string_view helpText, std::string_view usage, const std::function<void(Console*, const std::vector<std::string> &)>&& func)
{
    std::string lowName = name.data();
    athena::utility::tolower(lowName);
    if (m_commands.find(lowName) == m_commands.end())
        m_commands[lowName] = SConsoleCommand{name.data(), helpText.data(), usage.data(), std::move(func)};
}

void Console::executeString(const std::string& str)
{
    if (str.empty())
        return;

    /* First let's split semi-colon delimited commands */
    std::vector<std::string> commands = athena::utility::split(str, ';');

    if (commands.empty())
        return;

    for (std::string command : commands)
    {
        command = athena::utility::trim(command);
        std::vector<std::string> args = athena::utility::split(command, ' ');

        if (args.empty())
            continue;

        std::string commandName = args[0];
        args.erase(args.begin());

        std::string lowComName = commandName;
        athena::utility::tolower(lowComName);
        if (m_commands.find(lowComName) != m_commands.end())
            m_commands[lowComName].m_func(this, args);
        else
            report(Level::Error, "Command '%s' is not valid!", commandName.c_str());
    }
}

void Console::help(Console* /*con*/, const std::vector<std::string>& args)
{
    if (args.empty())
    {
        report(Level::Info, "Expected usage: help <command>");
        return;
    }
    std::string cmd = args.front();
    athena::utility::tolower(cmd);
    auto it = m_commands.find(cmd);
    if (it == m_commands.end())
    {
        report(Level::Error, "No such command '%s'", args.front().c_str());
        return;
    }

    report(Level::Info, "%s: %s", it->second.m_displayName.c_str(), it->second.m_helpString.c_str());
    if (!it->second.m_usage.empty())
        report(Level::Info, "Usage: %s %s", it->second.m_displayName.c_str(), it->second.m_usage.c_str());
}

void Console::listCommands(Console* /*con*/, const std::vector<std::string>& /*args*/)
{
    for (const auto& comPair : m_commands)
        report(Level::Info, "'%s': %s", comPair.second.m_displayName.c_str(), comPair.second.m_helpString.c_str());
}

bool Console::commandExists(std::string_view cmd)
{
    std::string cmdName = cmd.data();
    athena::utility::tolower(cmdName);

    return m_commands.find(cmdName) != m_commands.end();
}

void Console::report(Level level, const char* fmt, va_list list)
{
    char tmp[2048];
    vsnprintf(tmp, 2048, fmt, list);
    m_log.emplace_back(std::string(tmp), level);
    printf("%s\n", tmp);
}

void Console::report(Level level, const char* fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    report(level, fmt, ap);
    va_end(ap);
}

void Console::proc()
{
    if (m_state == State::Opened)
    {
        printf("\r%s                                   ", m_commandString.c_str());
        fflush(stdout);
    }
    else if (m_state == State::Opening)
        m_state = State::Opened;
    else if (m_state == State::Closing)
        m_state = State::Closed;


    if (m_cursorPosition > (int)m_commandString.size() - 1)
        m_cursorPosition = (int)m_commandString.size() - 1;
    if (m_cursorPosition < -1)
        m_cursorPosition = -1;

    if (m_logOffset > (int)m_log.size() - 1)
        m_logOffset = (int)m_log.size() - 1;
    if (m_logOffset < 0)
        m_logOffset = 0;
}

void Console::draw(boo::IGraphicsCommandQueue* gfxQ)
{
}

void Console::handleCharCode(unsigned long chr, boo::EModifierKey mod, bool repeat)
{
    if (chr == U'`' || chr == U'~')
    {
        if (m_state == State::Closed || m_state == State::Closing)
            m_state = State::Opening;
        else
            m_state = State::Closing;
    }

    if (m_state == State::Opened)
    {
        if (!m_commandString.empty() && m_cursorPosition + 1 < m_commandString.size())
        {
            if (m_overwrite)
                m_commandString[m_cursorPosition + 1] = chr;
            else
                m_commandString.insert(m_commandString.begin() + m_cursorPosition + 1, chr);
        }
        else
             m_commandString += chr;

        ++m_cursorPosition;
    }
}

void Console::handleSpecialKeyDown(boo::ESpecialKey sp, boo::EModifierKey mod, bool repeat)
{
    if (m_state != Opened)
        return;

    switch (sp)
    {
    case boo::ESpecialKey::Insert:
        m_overwrite ^= 1;
        break;
    case boo::ESpecialKey::Backspace:
    {
        if (!m_commandString.empty())
        {
            if (int(mod & boo::EModifierKey::Ctrl) != 0)
            {
                int index = m_commandString.rfind(' ', m_cursorPosition - 1);

                if (index == (int)std::string::npos)
                {
                    m_commandString.clear();
                    m_cursorPosition = -1;
                }
                else
                {
                    m_commandString.erase(index, (index - m_commandString.size()));
                    m_cursorPosition = index;
                }
                break;
            }
            if (m_cursorPosition < 0)
                break;

            m_commandString.erase(m_cursorPosition, 1);
            --m_cursorPosition;
        }
        break;
    }
    case boo::ESpecialKey::Delete:
    {
        if (!m_commandString.empty())
        {
            // Don't try to delete if the cursor is at the end of the line
            if ((m_cursorPosition + 1) >= (int)m_commandString.size())
                break;

            if (int(mod & boo::EModifierKey::Ctrl) != 0)
            {
                int index = m_commandString.find_first_of(' ', m_cursorPosition + 1);
                if (index != std::string::npos)
                    m_commandString.erase(m_cursorPosition + 1, index + 1);
                else
                    m_commandString.erase(m_cursorPosition + 1, (m_cursorPosition + 1) - m_commandString.size());
                break;
            }
            m_commandString.erase(m_cursorPosition + 1, 1);
        }
        break;
    }
    case boo::ESpecialKey::PgUp:
    {
        if (m_logOffset < (int)(m_log.size() - m_maxLines) - 1)
            m_logOffset++;
        break;
    }
    case boo::ESpecialKey::PgDown:
    {
        if (m_logOffset > 0)
            m_logOffset--;
        break;
    }
    case boo::ESpecialKey::Enter:
    {
        printf("\n");
        executeString(m_commandString);
        m_cursorPosition = -1;
        m_commandHistory.insert(m_commandHistory.begin(), m_commandString);
        m_commandString.clear();
        //m_showCursor = true;
        //m_cursorTime = 0;
        break;
    }
    case boo::ESpecialKey::Left:
    {
        if (m_cursorPosition < 0)
            break;

        if (int(mod & boo::EModifierKey::Ctrl) != 0)
            m_cursorPosition = (int)m_commandString.rfind(' ', m_cursorPosition) - 1;
        else
            m_cursorPosition--;

        //m_showCursor = true;
        //m_cursorTime = 0;
        break;
    }
    case boo::ESpecialKey::Right:
    {
        if (m_cursorPosition >= (int)m_commandString.size() - 1)
            break;

        if (int(mod & boo::EModifierKey::Ctrl) != 0)
        {
            if (m_commandString[m_cursorPosition] == ' ')
                m_cursorPosition++;

            int tmpPos = m_commandString.find(' ', m_cursorPosition);
            if (tmpPos == std::string::npos)
                m_cursorPosition = m_commandString.size() - 1;
            else
                m_cursorPosition = tmpPos;
        }
        else
            m_cursorPosition++;

//        m_showCursor = true;
//        m_cursorTime = 0;
        break;
    }

    case boo::ESpecialKey::Up:
    {
        if (m_commandHistory.size() == 0)
            break;

        m_currentCommand++;

        if (m_currentCommand > (int)m_commandHistory.size() - 1)
            m_currentCommand = (int)m_commandHistory.size() - 1;

        m_commandString = m_commandHistory[m_currentCommand];
        m_cursorPosition = m_commandString.size();
        break;
    }
    case boo::ESpecialKey::Down:
    {
        if (m_commandHistory.empty())
            break;
        m_currentCommand--;
        if (m_currentCommand >= 0)
        {
            m_commandString = m_commandHistory[m_currentCommand];
        }
        else if (m_currentCommand <= -1)
        {
            m_currentCommand = -1;
            m_commandString.clear();
        }
        m_cursorPosition = m_commandString.size();
        break;
    }
    case boo::ESpecialKey::Home:
        m_cursorPosition = -1;
        break;
    case boo::ESpecialKey::End:
        m_cursorPosition = m_commandString.size() - 1;
        break;
    default:
        break;
    }
}

void Console::handleSpecialKeyUp(boo::ESpecialKey sp, boo::EModifierKey mod)
{
}

void Console::LogVisorAdapter::report(const char* modName, logvisor::Level severity, const char *format, va_list ap)
{
    char tmp[2048];
    vsnprintf(tmp, 2048, format, ap);
    std::string v = athena::utility::sprintf("[%s] %s", modName, tmp);
    m_con->m_log.emplace_back(v, Console::Level(severity));
}

void Console::LogVisorAdapter::report(const char* modName, logvisor::Level severity, const wchar_t* format, va_list ap)
{
    wchar_t tmp[2048];
    vswprintf(tmp, 2048, format, ap);
    std::string v = athena::utility::sprintf("[%s] %s", modName, athena::utility::wideToUtf8(tmp).c_str());
    m_con->m_log.emplace_back(v, Console::Level(severity));
}

void Console::LogVisorAdapter::reportSource(const char* modName, logvisor::Level severity, const char* file, unsigned linenum, const char* format, va_list ap)
{
    char tmp[2048];
    vsnprintf(tmp, 2048, format, ap);
    std::string v = athena::utility::sprintf("[%s] %s %s:%i", modName, tmp, file, linenum);
    m_con->m_log.emplace_back(v, Console::Level(severity));
}

void Console::LogVisorAdapter::reportSource(const char* modName, logvisor::Level severity, const char* file, unsigned linenum, const wchar_t* format, va_list ap)
{
    wchar_t tmp[2048];
    vswprintf(tmp, 2048, format, ap);
    std::string v = athena::utility::sprintf("[%s] %s %s:%i", modName, athena::utility::wideToUtf8(tmp).c_str(), file, linenum);
    m_con->m_log.emplace_back(v, Console::Level(severity));
}

void Console::dumpLog()
{
    for (const auto& l : m_log)
    {
        switch(l.second)
        {
        case Level::Info:
            printf("%s\n", l.first.c_str());
        break;
        case Level::Warning:
            printf("[Warning] %s\n", l.first.c_str());
            break;
        case Level::Error:
            printf("[ Error ] %s\n", l.first.c_str());
            break;
        case Level::Fatal:
            printf("[ Fatal ] %s\n", l.first.c_str());
            break;
        }
    }
}

void Console::RegisterLogger(Console* con)
{
    /* Determine if console logger already added */
    for (auto& logger : logvisor::MainLoggers)
    {
        if (typeid(logger.get()) == typeid(LogVisorAdapter))
            return;
    }

    /* Otherwise construct new console logger */
    logvisor::MainLoggers.emplace_back(new LogVisorAdapter(con));
}

Console* Console::instance()
{
    return m_instance;
}
}