summaryrefslogtreecommitdiff
path: root/src/command_manager.hh
blob: b3f0917e64e9007a08745ef989ccd55d0e5a6639 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#ifndef command_manager_hh_INCLUDED
#define command_manager_hh_INCLUDED

#include <unordered_map>
#include <functional>
#include <initializer_list>

#include "string.hh"
#include "utils.hh"
#include "completion.hh"
#include "memoryview.hh"
#include "shell_manager.hh"

namespace Kakoune
{

struct Context;

struct wrong_argument_count : runtime_error
{
    wrong_argument_count() : runtime_error("wrong argument count") {}
};

typedef memoryview<String> CommandParameters;
typedef std::function<void (const CommandParameters&,
                            const Context& context)> Command;

typedef std::function<CandidateList (const CommandParameters&,
                                     size_t, size_t)> CommandCompleter;

class PerArgumentCommandCompleter
{
public:
    typedef std::function<CandidateList (const String&, size_t)> ArgumentCompleter;
    typedef memoryview<ArgumentCompleter> ArgumentCompleterList;

    PerArgumentCommandCompleter(const ArgumentCompleterList& completers)
        : m_completers(completers.begin(), completers.end()) {}

    CandidateList operator()(const CommandParameters& params,
                             size_t token_to_complete,
                             size_t pos_in_token) const;

private:
    std::vector<ArgumentCompleter> m_completers;
};

class CommandManager : public Singleton<CommandManager>
{
public:
    enum Flags
    {
        None = 0,
        IgnoreSemiColons = 1,
        DeferredShellEval = 2,
    };

    void execute(const String& command_line, const Context& context,
                 const EnvVarMap& env_vars = EnvVarMap());
    void execute(const CommandParameters& params, const Context& context,
                 const EnvVarMap& env_vars = EnvVarMap());

    Completions complete(const String& command_line, size_t cursor_pos);

    bool command_defined(const String& command_name) const;

    void register_command(const String& command_name,
                          Command command,
                          unsigned flags = None,
                          const CommandCompleter& completer = CommandCompleter());

    void register_commands(const memoryview<String>& command_names,
                           Command command,
                           unsigned flags = None,
                           const CommandCompleter& completer = CommandCompleter());

private:
    struct CommandDescriptor
    {
        Command command;
        unsigned flags;
        CommandCompleter completer;
    };
    std::unordered_map<String, CommandDescriptor> m_commands;
};

}

#endif // command_manager_hh_INCLUDED