110 lines
3.2 KiB
C++
110 lines
3.2 KiB
C++
#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"
|
|
|
|
namespace Kakoune
|
|
{
|
|
|
|
void HookManager::add_hook(StringView hook_name, String group, HookFunc hook)
|
|
{
|
|
auto& hooks = m_hooks[hook_name];
|
|
hooks.append({std::move(group), std::move(hook)});
|
|
}
|
|
|
|
void HookManager::remove_hooks(StringView group)
|
|
{
|
|
if (group.empty())
|
|
throw runtime_error("invalid id");
|
|
for (auto& hooks : m_hooks)
|
|
hooks.value.remove_all(group);
|
|
}
|
|
|
|
CandidateList HookManager::complete_hook_group(StringView prefix, ByteCount pos_in_token)
|
|
{
|
|
CandidateList res;
|
|
for (auto& list : m_hooks)
|
|
{
|
|
auto container = list.value | transform(decltype(list.value)::get_id);
|
|
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<std::pair<String, HookFunc>> hooks_to_run;
|
|
for (auto& hook : hook_list_it->value)
|
|
{
|
|
if (hook.key.empty() or disabled_hooks.empty() or
|
|
not regex_match(hook.key.begin(), hook.key.end(), disabled_hooks))
|
|
hooks_to_run.push_back({hook.key, hook.value});
|
|
}
|
|
|
|
bool hook_error = false;
|
|
for (auto& hook : hooks_to_run)
|
|
{
|
|
try
|
|
{
|
|
if (debug_flags & DebugFlags::Hooks)
|
|
write_to_debug_buffer(format("hook {}/{}", hook_name, hook.first));
|
|
hook.second(param, context);
|
|
}
|
|
catch (runtime_error& err)
|
|
{
|
|
hook_error = true;
|
|
write_to_debug_buffer(format("error running hook {}({})/{}: {}",
|
|
hook_name, param, hook.first, 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()));
|
|
}
|
|
}
|
|
|
|
}
|