summaryrefslogtreecommitdiff
path: root/src/alias_registry.hh
blob: 32f9e2fb428fa08d8d85f5af5330755f054de4c9 (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
#ifndef alias_registry_hh_INCLUDED
#define alias_registry_hh_INCLUDED

#include "safe_ptr.hh"
#include "string.hh"
#include "ranges.hh"
#include "hash_map.hh"

namespace Kakoune
{

class AliasRegistry : public SafeCountable
{
public:
    AliasRegistry(AliasRegistry& parent) : SafeCountable{}, m_parent(&parent) {}

    void reparent(AliasRegistry& parent) { m_parent = &parent; }

    void add_alias(String alias, String command);
    void remove_alias(StringView alias);
    StringView operator[](StringView alias) const;

    Vector<StringView> aliases_for(StringView command) const;

    auto flatten_aliases() const
    {
        auto merge = [](auto&& first, const AliasMap& second) {
            return concatenated(std::forward<decltype(first)>(first)
                                | filter([&second](auto& i) { return not second.contains(i.key); }),
                                second);
        };
        static const AliasMap empty;
        auto& parent = m_parent ? m_parent->m_aliases : empty;
        auto& grand_parent = (m_parent and m_parent->m_parent) ? m_parent->m_parent->m_aliases : empty;
        return merge(merge(grand_parent, parent), m_aliases);
    }

private:
    friend class Scope;
    AliasRegistry() = default;

    SafePtr<AliasRegistry> m_parent;
    using AliasMap = HashMap<String, String, MemoryDomain::Aliases>;
    AliasMap m_aliases;
};

}

#endif // alias_registry_hh_INCLUDED