From 3c021672d213d8b541f67d1881b5d17b33e6548d Mon Sep 17 00:00:00 2001
From: Maxime Coste <frrrwww@gmail.com>
Date: Mon, 21 Jul 2014 21:14:32 +0100
Subject: [PATCH] Add completion support for rmhooks

---
 src/commands.cc     | 40 +++++++++++++++++++++++++++++++---------
 src/hook_manager.cc | 15 +++++++++++++++
 src/hook_manager.hh |  1 +
 3 files changed, 47 insertions(+), 9 deletions(-)

diff --git a/src/commands.cc b/src/commands.cc
index 2fb93b7c..38fd871a 100644
--- a/src/commands.cc
+++ b/src/commands.cc
@@ -483,14 +483,21 @@ const CommandDesc rm_highlighter_cmd = {
     }
 };
 
-HookManager& get_hook_manager(const String& scope, Context& context)
+HookManager* get_hook_manager_ifp(const String& scope, const Context& context)
 {
     if (prefix_match("global", scope))
-        return GlobalHooks::instance();
+        return &GlobalHooks::instance();
     else if (prefix_match("buffer", scope))
-        return context.buffer().hooks();
+        return &context.buffer().hooks();
     else if (prefix_match("window", scope))
-        return context.window().hooks();
+        return &context.window().hooks();
+    return nullptr;
+}
+
+HookManager& get_hook_manager(const String& scope, const Context& context)
+{
+    if (auto manager = get_hook_manager_ifp(scope, context))
+        return *manager;
     throw runtime_error("error: no such hook container " + scope);
 }
 
@@ -520,17 +527,18 @@ const CommandDesc add_hook_cmd = {
     },
     CommandFlags::None,
     [](const Context& context, CompletionFlags flags,
-       CommandParameters params, size_t token_to_complete, ByteCount pos_in_token)
+       CommandParameters params, size_t token_to_complete,
+       ByteCount pos_in_token) -> Completions
     {
         if (token_to_complete == 0)
-            return Completions{ 0_byte, params[0].length(),
-                                complete_scope(params[0].substr(0_byte, pos_in_token)) };
+            return { 0_byte, params[0].length(),
+                     complete_scope(params[0].substr(0_byte, pos_in_token)) };
         else if (token_to_complete == 3)
         {
             auto& cm = CommandManager::instance();
             return cm.complete(context, flags, params[3], pos_in_token);
         }
-        return Completions{};
+        return {};
     },
     [](const ParametersParser& parser, Context& context)
     {
@@ -558,7 +566,21 @@ const CommandDesc rm_hook_cmd = {
     "rmhooks <group>: remove all hooks whose group is <group>",
     ParameterDesc{ SwitchMap{}, ParameterDesc::Flags::None, 2, 2 },
     CommandFlags::None,
-    CommandCompleter{},
+    [](const Context& context, CompletionFlags flags,
+       CommandParameters params, size_t token_to_complete,
+       ByteCount pos_in_token) -> Completions
+    {
+        if (token_to_complete == 0)
+            return { 0_byte, params[0].length(),
+                     complete_scope(params[0].substr(0_byte, pos_in_token)) };
+        else if (token_to_complete == 1)
+        {
+            if (auto manager = get_hook_manager_ifp(params[0], context))
+                return { 0_byte, params[0].length(),
+                         manager->complete_hook_group(params[1], pos_in_token) };
+        }
+        return {};
+    },
     [](const ParametersParser& parser, Context& context)
     {
         get_hook_manager(parser[0], context).remove_hooks(parser[1]);
diff --git a/src/hook_manager.cc b/src/hook_manager.cc
index e909a5b6..ae0d10ba 100644
--- a/src/hook_manager.cc
+++ b/src/hook_manager.cc
@@ -19,6 +19,21 @@ void HookManager::remove_hooks(StringView group)
         hooks.second.remove_all(group);
 }
 
+CandidateList HookManager::complete_hook_group(StringView prefix, ByteCount pos_in_token)
+{
+    CandidateList res;
+    for (auto& list : m_hook)
+    {
+        auto candidates = list.second.complete_id(prefix, pos_in_token);
+        for (auto& c : candidates)
+        {
+            if (!contains(res, c))
+                res.push_back(c);
+        }
+    }
+    return res;
+}
+
 void HookManager::run_hook(const String& hook_name,
                            const String& param,
                            Context& context) const
diff --git a/src/hook_manager.hh b/src/hook_manager.hh
index e7a902b7..4008e871 100644
--- a/src/hook_manager.hh
+++ b/src/hook_manager.hh
@@ -19,6 +19,7 @@ public:
 
     void add_hook(const String& hook_name, String group, HookFunc hook);
     void remove_hooks(StringView group);
+    CandidateList complete_hook_group(StringView prefix, ByteCount pos_in_token);
     void run_hook(const String& hook_name, const String& param,
                   Context& context) const;