#include "hook_manager.hh"

#include "clock.hh"
#include "containers.hh"
#include "context.hh"
#include "buffer_utils.hh"
#include "display_buffer.hh"
#include "face_registry.hh"
#include "regex.hh"
#include "option.hh"

namespace Kakoune
{

void HookManager::add_hook(StringView hook_name, String group, HookFunc func)
{
    auto& hooks = m_hooks[hook_name];
    hooks.push_back({std::move(group), std::move(func)});
}

void HookManager::remove_hooks(StringView group)
{
    if (group.empty())
        throw runtime_error("invalid id");
    for (auto& list : m_hooks)
        list.value.erase(std::remove_if(list.value.begin(), list.value.end(),
                                        [&](const Hook& h) { return h.group == group; }),
                         list.value.end());
}

CandidateList HookManager::complete_hook_group(StringView prefix, ByteCount pos_in_token)
{
    CandidateList res;
    for (auto& list : m_hooks)
    {
        auto container = list.value | transform(std::mem_fn(&decltype(list.value)::value_type::group));
        for (auto& c : complete(prefix, pos_in_token, container))
        {
            if (!contains(res, c))
                res.push_back(c);
        }
    }
    return res;
}

void HookManager::run_hook(StringView hook_name,
                           StringView param, Context& context) const
{
    if (context.hooks_disabled())
        return;

    if (m_parent)
        m_parent->run_hook(hook_name, param, context);

    auto hook_list_it = m_hooks.find(hook_name);
    if (hook_list_it == m_hooks.end())
        return;

    if (contains(m_running_hooks, std::make_pair(hook_name, param)))
    {
        auto error = format("recursive call of hook {}/{}, not executing", hook_name, param);
        write_to_debug_buffer(error);
        return;
    }

    m_running_hooks.emplace_back(hook_name, param);
    auto pop_running_hook = on_scope_end([this]{ m_running_hooks.pop_back(); });

    const DebugFlags debug_flags = context.options()["debug"].get<DebugFlags>();
    const bool profile = debug_flags & DebugFlags::Profile;
    auto start_time = profile ? Clock::now() : TimePoint{};

    auto& disabled_hooks = context.options()["disabled_hooks"].get<Regex>();
    Vector<Hook> hooks_to_run;
    for (auto& hook : hook_list_it->value)
    {
        if (hook.group.empty() or disabled_hooks.empty() or
            not regex_match(hook.group.begin(), hook.group.end(), disabled_hooks))
            hooks_to_run.push_back(hook);
    }

    bool hook_error = false;
    for (auto& hook : hooks_to_run)
    {
        try
        {
            if (debug_flags & DebugFlags::Hooks)
                write_to_debug_buffer(format("hook {}({})/{}", hook_name, param, hook.group));
            hook.func(param, context);
        }
        catch (runtime_error& err)
        {
            hook_error = true;
            write_to_debug_buffer(format("error running hook {}({})/{}: {}",
                               hook_name, param, hook.group, err.what()));
        }
    }

    if (hook_error)
        context.print_status({
            format("Error running hooks for '{}' '{}', see *debug* buffer",
                   hook_name, param), get_face("Error") });

    if (profile)
    {
        auto end_time = Clock::now();
        auto full = std::chrono::duration_cast<std::chrono::milliseconds>(end_time - start_time);
        write_to_debug_buffer(format("hook '{}({})' took {} ms", hook_name, param, (size_t)full.count()));
    }
}

}