#ifndef id_map_hh_INCLUDED
#define id_map_hh_INCLUDED

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

#include <vector>

namespace Kakoune
{

template<typename Value>
class id_map
{
public:
    using value_type = std::pair<String, Value>;
    using container_type = std::vector<value_type>;
    using iterator = typename container_type::iterator;
    using const_iterator = typename container_type::const_iterator;

    id_map() = default;
    id_map(std::initializer_list<value_type> val) : m_content{val} {}

    void append(const value_type& value)
    {
        m_content.push_back(value);
    }

    void append(value_type&& value)
    {
        m_content.push_back(std::move(value));
    }

    iterator find(StringView id)
    {
        return find_if(m_content,
                       [&](const value_type& v){ return v.first == id; });
    }

    const_iterator find(StringView id) const
    {
        return find_if(m_content,
                       [&](const value_type& v){ return v.first == id; });
    }

    bool contains(StringView id) const
    {
        return find(id) != end();
    }

    void remove(StringView id)
    {
        auto it = find(id);
        if (it != end())
            m_content.erase(it);
    }

    void remove_all(StringView id)
    {
        auto it = std::remove_if(begin(), end(),
                                 [&](value_type& v){ return v.first == id; });
        m_content.erase(it, end());
    }

    template<typename Condition>
    CandidateList complete_id_if(StringView prefix,
                                 ByteCount cursor_pos,
                                 Condition condition) const
    {
        auto real_prefix = prefix.substr(0, cursor_pos);
        CandidateList result;
        for (auto& value : m_content)
        {
            if (not condition(value))
                continue;

            if (prefix_match(value.first, real_prefix))
                result.push_back(value.first);
        }
        return result;
    }

    CandidateList complete_id(StringView prefix,
                              ByteCount cursor_pos) const
    {
        return complete_id_if(
            prefix, cursor_pos, [](const value_type&) { return true; });
    }

    bool empty() const { return m_content.empty(); }

    iterator       begin()       { return m_content.begin(); }
    iterator       end()         { return m_content.end(); }
    const_iterator begin() const { return m_content.begin(); }
    const_iterator end()   const { return m_content.end(); }

private:
    container_type m_content;
};

}

#endif // id_map_hh_INCLUDED