#include "normal.hh"

#include "buffer.hh"
#include "buffer_manager.hh"
#include "client_manager.hh"
#include "color_registry.hh"
#include "command_manager.hh"
#include "commands.hh"
#include "context.hh"
#include "file.hh"
#include "option_manager.hh"
#include "register_manager.hh"
#include "selectors.hh"
#include "shell_manager.hh"
#include "string.hh"
#include "window.hh"
#include "user_interface.hh"
#include "debug.hh"

namespace Kakoune
{

using namespace std::placeholders;

enum class SelectMode
{
    Replace,
    Extend,
    Append,
};

template<SelectMode mode = SelectMode::Replace, typename Func>
void select(Context& context, Func func)
{
    auto& buffer = context.buffer();
    auto& selections = context.selections();
    if (mode == SelectMode::Append)
    {
        auto& sel = selections.main();
        auto  res = func(buffer, sel);
        if (res.captures().empty())
            res.captures() = sel.captures();
        selections.push_back(res);
        selections.set_main_index(selections.size() - 1);
    }
    else
    {
        for (auto& sel : selections)
        {
            auto res = func(buffer, sel);
            if (mode == SelectMode::Extend)
                sel.merge_with(res);
            else
            {
                sel.anchor() = res.anchor();
                sel.cursor() = res.cursor();
            }
            if (not res.captures().empty())
                sel.captures() = std::move(res.captures());
        }
    }
    selections.sort_and_merge_overlapping();
    selections.check_invariant();
}

template<SelectMode mode, typename T>
class Select
{
public:
    constexpr Select(T t) : m_func(t) {}
    void operator() (Context& context, int) { select<mode>(context, m_func); }
private:
    T m_func;
};

template<SelectMode mode = SelectMode::Replace, typename T>
constexpr Select<mode, T> make_select(T func)
{
    return Select<mode, T>(func);
}

template<SelectMode mode = SelectMode::Replace>
void select_coord(Buffer& buffer, ByteCoord coord, SelectionList& selections)
{
    coord = buffer.clamp(coord);
    if (mode == SelectMode::Replace)
        selections = SelectionList{ buffer, coord };
    else if (mode == SelectMode::Extend)
    {
        for (auto& sel : selections)
            sel.cursor() = coord;
        selections.sort_and_merge_overlapping();
    }
}

template<InsertMode mode>
void enter_insert_mode(Context& context, int)
{
    context.input_handler().insert(mode);
}

void repeat_last_insert(Context& context, int)
{
    context.input_handler().repeat_last_insert();
}

bool show_auto_info_ifn(const String& title, const String& info,
                        const Context& context)
{
    if (context.options()["autoinfo"].get<int>() < 1 or not context.has_ui())
        return false;
    ColorPair col = get_color("Information");
    CharCoord pos = context.window().dimensions();
    pos.column -= 1;
    context.ui().info_show(title, info, pos , col, MenuStyle::Prompt);
    return true;
}

template<typename Cmd>
void on_next_key_with_autoinfo(const Context& context, Cmd cmd,
                               const String& title, const String& info)
{
    const bool hide = show_auto_info_ifn(title, info, context);
    context.input_handler().on_next_key([hide,cmd](Key key, Context& context) mutable {
            if (hide)
                context.ui().info_hide();
            cmd(key, context);
    });
}

template<SelectMode mode>
void goto_commands(Context& context, int line)
{
    if (line != 0)
    {
        context.push_jump();
        select_coord<mode>(context.buffer(), LineCount{line - 1}, context.selections());
        if (context.has_window())
            context.window().center_line(LineCount{line-1});
    }
    else
    {
        on_next_key_with_autoinfo(context, [](Key key, Context& context) {
            if (key.modifiers != Key::Modifiers::None)
                return;
            auto& buffer = context.buffer();
            switch (tolower(key.key))
            {
            case 'g':
            case 'k':
                context.push_jump();
                select_coord<mode>(buffer, ByteCoord{0,0}, context.selections());
                break;
            case 'l':
                select<mode>(context, select_to_eol);
                break;
            case 'h':
                select<mode>(context, select_to_eol_reverse);
                break;
            case 'j':
            {
                context.push_jump();
                select_coord<mode>(buffer, buffer.line_count() - 1, context.selections());
                break;
            }
            case 'e':
                context.push_jump();
                select_coord<mode>(buffer, buffer.back_coord(), context.selections());
                break;
            case 't':
                if (context.has_window())
                {
                    auto line = context.window().position().line;
                    select_coord<mode>(buffer, line, context.selections());
                }
                break;
            case 'b':
                if (context.has_window())
                {
                    auto& window = context.window();
                    auto line = window.position().line + window.dimensions().line - 1;
                    select_coord<mode>(buffer, line, context.selections());
                }
                break;
            case 'c':
                if (context.has_window())
                {
                    auto& window = context.window();
                    auto line = window.position().line + window.dimensions().line / 2;
                    select_coord<mode>(buffer, line, context.selections());
                }
                break;
            case 'a':
            {
                auto& buffer_manager = BufferManager::instance();
                auto it = buffer_manager.begin();
                if (it->get() == &buffer and ++it == buffer_manager.end())
                    break;
                context.push_jump();
                context.change_buffer(**it);
                break;
            }
            case 'f':
            {
                const Selection& sel = context.selections().main();
                String filename = content(buffer, sel);
                static constexpr char forbidden[] = { '\'', '\\', '\0' };
                for (auto c : forbidden)
                    if (contains(filename, c))
                        return;

                auto paths = context.options()["path"].get<std::vector<String>>();
                const String& buffer_name = buffer.name();
                auto it = find(reversed(buffer_name), '/');
                if (it != buffer_name.rend())
                    paths.insert(paths.begin(), String{buffer_name.begin(), it.base()});

                String path = find_file(filename, paths);
                if (path.empty())
                    throw runtime_error("unable to find file '" + filename + "'");

                Buffer* buffer = create_buffer_from_file(path);
                if (buffer == nullptr)
                    throw runtime_error("unable to open file '" + path + "'");

                if (buffer != &context.buffer())
                {
                    BufferManager::instance().set_last_used_buffer(*buffer);
                    context.push_jump();
                    context.change_buffer(*buffer);
                }
                break;
            }
            case '.':
            {
                context.push_jump();
                auto pos = buffer.last_modification_coord();
                if (buffer[pos.line].length() == pos.column + 1)
                    pos = ByteCoord{ pos.line+1, 0 };
                select_coord<mode>(buffer, pos, context.selections());
                break;
            }
            }
        }, "goto",
        "g,k:  buffer top        \n"
        "l:    line end          \n"
        "h:    line begin        \n"
        "j:    buffer bottom     \n"
        "e:    buffer end        \n"
        "t:    window top        \n"
        "b:    window bottom     \n"
        "c:    window center     \n"
        "a:    last buffer       \n"
        "f:    file              \n"
        ".:    last buffer change\n");
    }
}

void view_commands(Context& context, int param)
{
    on_next_key_with_autoinfo(context, [param](Key key, Context& context) {
        if (key.modifiers != Key::Modifiers::None or not context.has_window())
            return;

        LineCount cursor_line = context.selections().main().cursor().line;
        Window& window = context.window();
        switch (tolower(key.key))
        {
        case 'v':
        case 'c':
            context.window().center_line(cursor_line);
            break;
        case 't':
            context.window().display_line_at(cursor_line, 0);
            break;
        case 'b':
            context.window().display_line_at(cursor_line, window.dimensions().line-1);
            break;
        case 'h':
            context.window().scroll(-std::max<CharCount>(1, param));
            break;
        case 'j':
            context.window().scroll( std::max<LineCount>(1, param));
            break;
        case 'k':
            context.window().scroll(-std::max<LineCount>(1, param));
            break;
        case 'l':
            context.window().scroll( std::max<CharCount>(1, param));
            break;
        }
    }, "view",
    "v,c:  center cursor   \n"
    "t:    cursor on top   \n"
    "b:    cursor on bottom\n"
    "h:    scroll left     \n"
    "j:    scroll down     \n"
    "k:    scroll up       \n"
    "l:    scroll right    \n");
}

void replace_with_char(Context& context, int)
{
    on_next_key_with_autoinfo(context, [](Key key, Context& context) {
        if (not iswprint(key.key))
            return;
        ScopedEdition edition(context);
        Buffer& buffer = context.buffer();
        SelectionList& selections = context.selections();
        std::vector<String> strings;
        for (auto& sel : selections)
        {
            CharCount count = char_length(buffer, sel);
            strings.emplace_back(key.key, count);
        }
        selections.insert(strings, InsertMode::Replace);
    }, "replace with char", "enter char to replace with\n");
}

Codepoint to_lower(Codepoint cp) { return tolower(cp); }
Codepoint to_upper(Codepoint cp) { return toupper(cp); }

Codepoint swap_case(Codepoint cp)
{
    Codepoint res = std::tolower(cp);
    return res == cp ? std::toupper(cp) : res;
}

template<Codepoint (*func)(Codepoint)>
void for_each_char(Context& context, int)
{
    ScopedEdition edition(context);
    std::vector<String> sels = context.selections_content();
    for (auto& sel : sels)
    {
        for (auto& c : sel)
            c = func(c);
    }
    context.selections().insert(sels, InsertMode::Replace);
}

void command(Context& context, int)
{
    context.input_handler().prompt(
        ":", "", get_color("Prompt"),
        std::bind(&CommandManager::complete, &CommandManager::instance(), _1, _2, _3, _4),
        [](const String& cmdline, PromptEvent event, Context& context) {
            if (context.has_ui())
            {
                context.ui().info_hide();
                if (event == PromptEvent::Change and context.options()["autoinfo"].get<int>() > 0)
                {
                    auto info = CommandManager::instance().command_info(cmdline);
                    ColorPair col = get_color("Information");
                    CharCoord pos = context.window().dimensions();
                    pos.column -= 1;
                    if (not info.first.empty() and not info.second.empty())
                        context.ui().info_show(info.first, info.second, pos , col, MenuStyle::Prompt);
                }
            }
            if (event == PromptEvent::Validate)
                CommandManager::instance().execute(cmdline, context);
        });
}

template<InsertMode mode>
void pipe(Context& context, int)
{
    const char* prompt = mode == InsertMode::Replace ? "pipe:" : "pipe (ins):";
    context.input_handler().prompt(prompt, "", get_color("Prompt"), shell_complete,
        [](const String& cmdline, PromptEvent event, Context& context)
        {
            if (event != PromptEvent::Validate)
                return;

            StringView real_cmd;
            if (cmdline.empty())
                real_cmd = context.main_sel_register_value("|");
            else
            {
                RegisterManager::instance()['|'] = cmdline;
                real_cmd = cmdline;
            }

            if (real_cmd.empty())
                return;

            Buffer& buffer = context.buffer();
            SelectionList& selections = context.selections();
            std::vector<String> strings;
            for (auto& sel : selections)
            {
                auto str = content(buffer, sel);
                bool insert_eol = str.back() != '\n';
                if (insert_eol)
                    str += '\n';
                str = ShellManager::instance().pipe(str, real_cmd, context,
                                                    {}, EnvVarMap{});
                if (insert_eol and str.back() == '\n')
                    str = str.substr(0, str.length()-1);
                strings.push_back(str);
            }
            ScopedEdition edition(context);
            selections.insert(strings, mode);
        });
}

template<Direction direction, SelectMode mode>
void select_next_match(const Buffer& buffer, SelectionList& selections,
                       const Regex& regex)
{
    if (mode == SelectMode::Replace)
    {
        for (auto& sel : selections)
            sel = find_next_match<direction>(buffer, sel, regex);
    }
    if (mode == SelectMode::Extend)
    {
        for (auto& sel : selections)
            sel.merge_with(find_next_match<direction>(buffer, sel, regex));
    }
    else if (mode == SelectMode::Append)
    {
        selections.push_back(find_next_match<direction>(buffer, selections.main(), regex));
        selections.set_main_index(selections.size() - 1);
    }
    selections.sort_and_merge_overlapping();
}

void yank(Context& context, int)
{
    RegisterManager::instance()['"'] = context.selections_content();
    context.print_status({ "yanked " + to_string(context.selections().size()) +
                           " selections", get_color("Information") });
}

void erase_selections(Context& context, int)
{
    RegisterManager::instance()['"'] = context.selections_content();
    ScopedEdition edition(context);
    context.selections().erase();
    context.selections().avoid_eol();
}

void change(Context& context, int param)
{
    RegisterManager::instance()['"'] = context.selections_content();
    enter_insert_mode<InsertMode::Replace>(context, param);
}

constexpr InsertMode adapt_for_linewise(InsertMode mode)
{
    return ((mode == InsertMode::Append) ?
             InsertMode::InsertAtNextLineBegin :
             ((mode == InsertMode::Insert) ?
               InsertMode::InsertAtLineBegin :
               ((mode == InsertMode::Replace) ?
                 InsertMode::Replace : InsertMode::Insert)));
}

template<InsertMode mode>
void paste(Context& context, int)
{
    auto strings = RegisterManager::instance()['"'].values(context);
    InsertMode effective_mode = mode;
    for (auto& str : strings)
    {
        if (not str.empty() and str.back() == '\n')
        {
            effective_mode = adapt_for_linewise(mode);
            break;
        }
    }
    ScopedEdition edition(context);
    context.selections().insert(strings, effective_mode);
}

template<InsertMode mode>
void paste_all(Context& context, int)
{
    auto strings = RegisterManager::instance()['"'].values(context);
    InsertMode effective_mode = mode;
    String all;
    std::vector<ByteCount> offsets;
    for (auto& str : strings)
    {
        if (not str.empty() and str.back() == '\n')
            effective_mode = adapt_for_linewise(mode);
        all += str;
        offsets.push_back(all.length());
    }

    auto& selections = context.selections();
    {
        ScopedEdition edition(context);
        selections.insert(all, effective_mode, true);
    }

    const Buffer& buffer = context.buffer();
    std::vector<Selection> result;
    for (auto& selection : selections)
    {
        ByteCount pos = 0;
        for (auto offset : offsets)
        {
            result.push_back({ buffer.advance(selection.min(), pos),
                               buffer.advance(selection.min(), offset-1) });
            pos = offset;
        }
    }
    selections = std::move(result);
}

template<typename T>
void regex_prompt(Context& context, const String prompt, T func)
{
    SelectionList selections = context.selections();
    context.input_handler().prompt(prompt, "", get_color("Prompt"), complete_nothing,
        [=](const String& str, PromptEvent event, Context& context) mutable {
            try
            {
                if (event != PromptEvent::Change and context.has_ui())
                    context.ui().info_hide();
                selections.update();
                context.selections() = selections;
                context.input_handler().set_prompt_colors(get_color("Prompt"));
                if (event == PromptEvent::Abort)
                    return;
                if (event == PromptEvent::Change and
                    (str.empty() or not context.options()["incsearch"].get<bool>()))
                    return;

                if (event == PromptEvent::Validate)
                    context.push_jump();
                func(str.empty() ? Regex{} : Regex{str}, context);
            }
            catch (boost::regex_error& err)
            {
                if (event == PromptEvent::Validate)
                    throw runtime_error("regex error: "_str + err.what());
                else
                    context.input_handler().set_prompt_colors(get_color("Error"));
            }
            catch (std::runtime_error& err)
            {
                if (event == PromptEvent::Validate)
                    throw runtime_error("regex error: "_str + err.what());
                else
                {
                    context.input_handler().set_prompt_colors(get_color("Error"));
                    if (context.has_ui())
                    {
                        ColorPair col = get_color("Information");
                        CharCoord pos = context.window().dimensions();
                        pos.column -= 1;
                        context.ui().info_show("regex error", err.what(), pos, col, MenuStyle::Prompt);
                    }
                }
            }
            catch (runtime_error&)
            {
                context.selections() = selections;
                // only validation should propagate errors,
                // incremental search should not.
                if (event == PromptEvent::Validate)
                    throw;
            }
        });
}

template<SelectMode mode, Direction direction>
void search(Context& context, int)
{
    regex_prompt(context, direction == Forward ? "search:" : "reverse search:",
                 [](Regex ex, Context& context) {
                     if (ex.empty())
                         ex = Regex{context.main_sel_register_value("/").str()};
                     else
                         RegisterManager::instance()['/'] = String{ex.str()};
                     if (not ex.empty() and not ex.str().empty())
                         select_next_match<direction, mode>(context.buffer(), context.selections(), ex);
                 });
}

template<SelectMode mode, Direction direction>
void search_next(Context& context, int param)
{
    StringView str = context.main_sel_register_value("/");
    if (not str.empty())
    {
        try
        {
            Regex ex{str.begin(), str.end()};
            do {
                select_next_match<direction, mode>(context.buffer(), context.selections(), ex);
            } while (--param > 0);
        }
        catch (boost::regex_error& err)
        {
            throw runtime_error("regex error: "_str + err.what());
        }
    }
    else
        throw runtime_error("no search pattern");
}

template<bool smart>
void use_selection_as_search_pattern(Context& context, int)
{
    std::vector<String> patterns;
    auto& sels = context.selections();
    const auto& buffer = context.buffer();
    for (auto& sel : sels)
    {
        auto begin = utf8::make_iterator(buffer.iterator_at(sel.min()));
        auto end   = utf8::make_iterator(buffer.iterator_at(sel.max()))+1;
        auto content = "\\Q" + String{begin.base(), end.base()} + "\\E";
        if (smart)
        {
            if (begin == buffer.begin() or (is_word(*begin) and not is_word(*(begin-1))))
                content = "\\b" + content;
            if (end == buffer.end() or (is_word(*(end-1)) and not is_word(*end)))
                content = content + "\\b";
        }
        patterns.push_back(std::move(content));
    }
    RegisterManager::instance()['/'] = patterns;
}

void select_regex(Context& context, int)
{
    regex_prompt(context, "select:", [](Regex ex, Context& context) {
        if (ex.empty())
            ex = Regex{context.main_sel_register_value("/").str()};
        else
            RegisterManager::instance()['/'] = String{ex.str()};
        if (not ex.empty() and not ex.str().empty())
            select_all_matches(context.selections(), ex);
    });
}

void split_regex(Context& context, int)
{
    regex_prompt(context, "split:", [](Regex ex, Context& context) {
        if (ex.empty())
            ex = Regex{context.main_sel_register_value("/").str()};
        else
            RegisterManager::instance()['/'] = String{ex.str()};
        if (not ex.empty() and not ex.str().empty())
            split_selections(context.selections(), ex);
    });
}

void split_lines(Context& context, int)
{
    auto& selections = context.selections();
    auto& buffer = context.buffer();
    std::vector<Selection> res;
    for (auto& sel : selections)
    {
        if (sel.anchor().line == sel.cursor().line)
        {
             res.push_back(std::move(sel));
             continue;
        }
        auto min = sel.min();
        auto max = sel.max();
        res.push_back({min, {min.line, buffer[min.line].length()-1}});
        for (auto line = min.line+1; line < max.line; ++line)
            res.push_back({line, {line, buffer[line].length()-1}});
        res.push_back({max.line, max});
    }
    selections = std::move(res);
}

void join_select_spaces(Context& context, int)
{
    auto& buffer = context.buffer();
    std::vector<Selection> selections;
    for (auto& sel : context.selections())
    {
        for (LineCount line = sel.min().line; line <= sel.max().line; ++line)
        {
            if (line == buffer.line_count() - 1)
                continue;
            auto begin = buffer.iterator_at({line, buffer[line].length()-1});
            auto end = begin+1;
            skip_while(end, buffer.end(), is_horizontal_blank);
            selections.push_back({begin.coord(), (end-1).coord()});
        }
    }
    if (selections.empty())
        return;
    context.selections() = selections;
    ScopedEdition edition(context);
    context.selections().insert(" "_str, InsertMode::Replace);
}

void join(Context& context, int param)
{
    SelectionList sels{context.selections()};
    auto restore_sels = on_scope_end([&]{
        sels.update();
        context.selections() = std::move(sels);
    });

    join_select_spaces(context, param);
}

template<bool matching>
void keep(Context& context, int)
{
    constexpr const char* prompt = matching ? "keep matching:" : "keep not matching:";
    regex_prompt(context, prompt, [](const Regex& ex, Context& context) {
        if (ex.empty())
            return;
        const Buffer& buffer = context.buffer();
        std::vector<Selection> keep;
        for (auto& sel : context.selections())
        {
            if (boost::regex_search(buffer.iterator_at(sel.min()),
                                    utf8::next(buffer.iterator_at(sel.max()), buffer.end()), ex) == matching)
                keep.push_back(sel);
        }
        if (keep.empty())
            throw runtime_error("no selections remaining");
        context.set_selections(std::move(keep));
    });
}

void keep_pipe(Context& context, int)
{
    context.input_handler().prompt(
        "keep pipe:", "", get_color("Prompt"), shell_complete,
        [](const String& cmdline, PromptEvent event, Context& context) {
            if (event != PromptEvent::Validate)
                return;
            const Buffer& buffer = context.buffer();
            auto& shell_manager = ShellManager::instance();
            std::vector<Selection> keep;
            for (auto& sel : context.selections())
            {
                int status = 0;
                shell_manager.pipe(content(buffer, sel), cmdline, context,
                                   {}, EnvVarMap{}, &status);
                if (status == 0)
                    keep.push_back(sel);
            }
            if (keep.empty())
                throw runtime_error("no selections remaining");
            context.set_selections(std::move(keep));
    });
}
template<bool indent_empty = false>
void indent(Context& context, int)
{
    CharCount indent_width = context.options()["indentwidth"].get<int>();
    String indent = indent_width == 0 ? "\t" : String{' ', indent_width};

    auto& buffer = context.buffer();
    std::vector<Selection> sels;
    for (auto& sel : context.selections())
    {
        for (auto line = sel.min().line; line < sel.max().line+1; ++line)
        {
            if (indent_empty or buffer[line].length() > 1)
                sels.push_back({line, line});
        }
    }
    if (not sels.empty())
    {
        ScopedEdition edition(context);
        SelectionList selections{buffer, std::move(sels)};
        selections.insert(indent, InsertMode::Insert);
    }
}

template<bool deindent_incomplete = true>
void deindent(Context& context, int)
{
    CharCount tabstop = context.options()["tabstop"].get<int>();
    CharCount indent_width = context.options()["indentwidth"].get<int>();
    if (indent_width == 0)
        indent_width = tabstop;

    auto& buffer = context.buffer();
    std::vector<Selection> sels;
    for (auto& sel : context.selections())
    {
        for (auto line = sel.min().line; line < sel.max().line+1; ++line)
        {
            CharCount width = 0;
            auto& content = buffer[line];
            for (auto column = 0_byte; column < content.length(); ++column)
            {
                const char c = content[column];
                if (c == '\t')
                    width = (width / tabstop + 1) * tabstop;
                else if (c == ' ')
                    ++width;
                else
                {
                    if (deindent_incomplete and width != 0)
                        sels.push_back({ line, ByteCoord{line, column-1} });
                    break;
                }
                if (width == indent_width)
                {
                    sels.push_back({ line, ByteCoord{line, column} });
                    break;
                }
            }
        }
    }
    if (not sels.empty())
    {
        ScopedEdition edition(context);
        SelectionList selections{context.buffer(), std::move(sels)};
        selections.erase();
    }
}

template<ObjectFlags flags, SelectMode mode = SelectMode::Replace>
void select_object(Context& context, int param)
{
    int level = param <= 0 ? 0 : param - 1;
    on_next_key_with_autoinfo(context, [level](Key key, Context& context) {
        if (key.modifiers != Key::Modifiers::None)
            return;
        const Codepoint c = key.key;

        static constexpr struct
        {
            Codepoint key;
            Selection (*func)(const Buffer&, const Selection&, ObjectFlags);
        } selectors[] = {
            { 'w', select_word<Word> },
            { 'W', select_word<WORD> },
            { 's', select_sentence },
            { 'p', select_paragraph },
            { ' ', select_whitespaces },
            { 'i', select_indent },
            { 'n', select_number },
        };
        for (auto& sel : selectors)
        {
            if (c == sel.key)
                return select<mode>(context, std::bind(sel.func, _1, _2, flags));
        }

        static const struct
        {
            CodepointPair pair;
            Codepoint name;
        } surrounding_pairs[] = {
            { { '(', ')' }, 'b' },
            { { '{', '}' }, 'B' },
            { { '[', ']' }, 'r' },
            { { '<', '>' }, 'a' },
            { { '"', '"' }, 'Q' },
            { { '\'', '\'' }, 'q' },
            { { '`', '`' }, 'g' },
        };
        for (auto& sur : surrounding_pairs)
        {
            if (sur.pair.first == c or sur.pair.second == c or
                (sur.name != 0 and sur.name == c))
                return select<mode>(context, std::bind(select_surrounding, _1, _2,
                                                       sur.pair, level, flags));
        }
    }, "select object",
    "b,(,):  parenthesis block\n"
    "B,{,}:  braces block     \n"
    "r,[,]:  brackets block   \n"
    "a,<,>:  angle block      \n"
    "\",Q:  double quote string\n"
    "',q:  single quote string\n"
    "`,g:  grave quote string \n"
    "w:    word               \n"
    "W:    WORD               \n"
    "s:    sentence           \n"
    "p:    paragraph          \n"
    "␣:    whitespaces        \n"
    "i:    indent             \n");
}

template<Key::NamedKey key>
void scroll(Context& context, int)
{
    static_assert(key == Key::PageUp or key == Key::PageDown,
                  "scrool only implements PageUp and PageDown");
    Window& window = context.window();
    Buffer& buffer = context.buffer();
    CharCoord position = window.position();
    LineCount cursor_line = 0;

    if (key == Key::PageUp)
    {
        position.line -= (window.dimensions().line - 2);
        cursor_line = position.line;
    }
    else if (key == Key::PageDown)
    {
        position.line += (window.dimensions().line - 2);
        cursor_line = position.line + window.dimensions().line - 1;
    }
    auto cursor_pos = utf8::advance(buffer.iterator_at(position.line),
                                    buffer.iterator_at(position.line+1),
                                    position.column);
    select_coord(buffer, cursor_pos.coord(), context.selections());
    window.set_position(position);
}

void rotate_selections(Context& context, int count)
{
    context.selections().rotate_main(count != 0 ? count : 1);
}

void rotate_selections_content(Context& context, int group)
{
    int count = 1;
    auto strings = context.selections_content();
    if (group == 0 or group > (int)strings.size())
        group = (int)strings.size();
    count = count % group;
    for (auto it = strings.begin(); it != strings.end(); )
    {
        auto end = std::min(strings.end(), it + group);
        std::rotate(it, end-count, end);
        it = end;
    }
    context.selections().insert(strings, InsertMode::Replace);
    context.selections().rotate_main(count);
}

enum class SelectFlags
{
    None = 0,
    Reverse = 1,
    Inclusive = 2,
    Extend = 4
};
constexpr SelectFlags operator|(SelectFlags lhs, SelectFlags rhs)
{
    return (SelectFlags)((int) lhs | (int) rhs);
}
constexpr bool operator&(SelectFlags lhs, SelectFlags rhs)
{
    return ((int) lhs & (int) rhs) != 0;
}

template<SelectFlags flags>
void select_to_next_char(Context& context, int param)
{
    on_next_key_with_autoinfo(context, [param](Key key, Context& context) {
        select<flags & SelectFlags::Extend ? SelectMode::Extend : SelectMode::Replace>(
            context,
            std::bind(flags & SelectFlags::Reverse ? select_to_reverse : select_to,
                      _1, _2, key.key, param, flags & SelectFlags::Inclusive));
    }, "select to next char","enter char to select to");
}

void start_or_end_macro_recording(Context& context, int)
{
    if (context.input_handler().is_recording())
        context.input_handler().stop_recording();
    else
        on_next_key_with_autoinfo(context, [](Key key, Context& context) {
            if (key.modifiers == Key::Modifiers::None and isalpha(key.key))
                context.input_handler().start_recording(tolower(key.key));
        }, "record macro", "enter macro name ");
}

void replay_macro(Context& context, int count)
{
    on_next_key_with_autoinfo(context, [count](Key key, Context& context) mutable {
        if (key.modifiers == Key::Modifiers::None and isalpha(key.key))
        {
            static std::unordered_set<char> running_macros;
            const char name = tolower(key.key);
            if (contains(running_macros, name))
                throw runtime_error("recursive macros call detected");

            memoryview<String> reg_val = RegisterManager::instance()[name].values(context);
            if (not reg_val.empty())
            {
                running_macros.insert(name);
                auto stop = on_scope_end([&]{ running_macros.erase(name); });

                auto keys = parse_keys(reg_val[0]);
                ScopedEdition edition(context);
                do { exec_keys(keys, context); } while (--count > 0);
            }
        }
    }, "replay macro", "enter macro name");
}

template<Direction direction>
void jump(Context& context, int)
{
    auto jump = (direction == Forward) ?
                 context.jump_forward() : context.jump_backward();

    Buffer& buffer = const_cast<Buffer&>(jump.buffer());
    BufferManager::instance().set_last_used_buffer(buffer);
    if (&buffer != &context.buffer())
        context.change_buffer(buffer);
    context.selections() = jump;
}

void save_selections(Context& context, int)
{
    context.push_jump();
    context.print_status({ "saved " + to_string(context.selections().size()) +
                           " selections", get_color("Information") });
}

void align(Context& context, int)
{
    auto& selections = context.selections();
    auto& buffer = context.buffer();
    const CharCount tabstop = context.options()["tabstop"].get<int>();

    std::vector<std::vector<const Selection*>> columns;
    LineCount last_line = -1;
    size_t column = 0;
    for (auto& sel : selections)
    {
        auto line = sel.cursor().line;
        if (sel.anchor().line != line)
            throw runtime_error("align cannot work with multi line selections");

        column = (line == last_line) ? column + 1 : 0;
        if (column >= columns.size())
            columns.resize(column+1);
        columns[column].push_back(&sel);
        last_line = line;
    }

    const bool use_tabs = context.options()["aligntab"].get<bool>();
    for (auto& col : columns)
    {
        CharCount maxcol = 0;
        for (auto& sel : col)
            maxcol = std::max(get_column(buffer, tabstop, sel->cursor()), maxcol);
        for (auto& sel : col)
        {
            auto insert_coord = sel->min();
            auto lastcol = get_column(buffer, tabstop, sel->cursor());
            String padstr;
            if (not use_tabs)
                padstr = String{ ' ', maxcol - lastcol };
            else
            {
                auto inscol = get_column(buffer, tabstop, insert_coord);
                auto targetcol = maxcol - (lastcol - inscol);
                auto tabcol = inscol - (inscol % tabstop);
                auto tabs = (targetcol - tabcol) / tabstop;
                auto spaces = targetcol - (tabs ? (tabcol + tabs * tabstop) : inscol);
                padstr = String{ '\t', tabs } + String{ ' ', spaces };
            }
            buffer.insert(buffer.iterator_at(insert_coord), std::move(padstr));
        }
    }
}

void copy_indent(Context& context, int selection)
{
    auto& buffer = context.buffer();
    auto& selections = context.selections();
    std::vector<LineCount> lines;
    for (auto sel : selections)
    {
        for (LineCount l = sel.min().line; l < sel.max().line + 1; ++l)
            lines.push_back(l);
    }
    if (selection > selections.size())
        throw runtime_error("invalid selection index");
    if (selection == 0)
        selection = context.selections().main_index() + 1;

    auto ref_line = selections[selection-1].min().line;
    const String& line = buffer[ref_line];
    auto it = line.begin();
    while (it != line.end() and is_horizontal_blank(*it))
        ++it;
    const StringView indent = line.substr(0_byte, (int)(it-line.begin()));

    ScopedEdition edition{context};
    for (auto& l : lines)
    {
        if (l == ref_line)
            continue;

        auto& line = buffer[l];
        ByteCount i = 0;
        while (i < line.length() and is_horizontal_blank(line[i]))
            ++i;
        buffer.erase(buffer.iterator_at(l), buffer.iterator_at({l, i}));
        buffer.insert(buffer.iterator_at(l), indent);
    }
}

void tabs_to_spaces(Context& context, int ts)
{
    auto& buffer = context.buffer();
    const CharCount opt_tabstop = context.options()["tabstop"].get<int>();
    const CharCount tabstop = ts == 0 ? opt_tabstop : ts;
    std::vector<Selection> tabs;
    std::vector<String> spaces;
    for (auto& sel : context.selections())
    {
        for (auto it = buffer.iterator_at(sel.min()),
                  end = buffer.iterator_at(sel.max())+1; it != end; ++it)
        {
            if (*it == '\t')
            {
                CharCount col = get_column(buffer, opt_tabstop, it.coord());
                CharCount end_col = (col / tabstop + 1) * tabstop;
                tabs.push_back({ it.coord() });
                spaces.push_back(String{ ' ', end_col - col });
            }
        }
    }
    if (not tabs.empty())
        SelectionList{ buffer, std::move(tabs) }.insert(spaces, InsertMode::Replace);
}

void spaces_to_tabs(Context& context, int ts)
{
    auto& buffer = context.buffer();
    const CharCount opt_tabstop = context.options()["tabstop"].get<int>();
    const CharCount tabstop = ts == 0 ? opt_tabstop : ts;
    std::vector<Selection> spaces;
    for (auto& sel : context.selections())
    {
        for (auto it = buffer.iterator_at(sel.min()),
                  end = buffer.iterator_at(sel.max())+1; it != end;)
        {
            if (*it == ' ')
            {
                auto spaces_beg = it;
                auto spaces_end = spaces_beg+1;
                CharCount col = get_column(buffer, opt_tabstop, spaces_end.coord());
                while (*spaces_end == ' ' and (col % tabstop) != 0)
                {
                    ++spaces_end;
                    ++col;
                }
                if ((col % tabstop) == 0)
                    spaces.push_back({spaces_beg.coord(), (spaces_end-1).coord()});
                it = spaces_end;
            }
            else
                ++it;
        }
    }
    if (not spaces.empty())
        SelectionList{ buffer, std::move(spaces) }.insert("\t"_str, InsertMode::Replace);
}

void undo(Context& context, int)
{
    Buffer& buffer = context.buffer();
    size_t timestamp = buffer.timestamp();
    bool res = buffer.undo();
    if (res)
    {
        auto ranges = compute_modified_ranges(buffer, timestamp);
        if (not ranges.empty())
            context.set_selections(std::move(ranges));
    }
    else if (not res)
        context.print_status({ "nothing left to undo", get_color("Information") });
}

void redo(Context& context, int)
{
    using namespace std::placeholders;
    Buffer& buffer = context.buffer();
    size_t timestamp = buffer.timestamp();
    bool res = buffer.redo();
    if (res)
    {
        auto ranges = compute_modified_ranges(buffer, timestamp);
        if (not ranges.empty())
            context.set_selections(std::move(ranges));
    }

    else if (not res)
        context.print_status({ "nothing left to redo", get_color("Information") });
}

template<typename T>
class Repeated
{
public:
    constexpr Repeated(T t) : m_func(t) {}

    void operator() (Context& context, int count)
    {
        ScopedEdition edition(context);
        do { m_func(context, 0); } while(--count > 0);
    }
private:
    T m_func;
};

template<typename T>
constexpr Repeated<T> repeated(T func) { return Repeated<T>(func); }

template<typename Type, Direction direction, SelectMode mode = SelectMode::Replace>
void move(Context& context, int count)
{
    kak_assert(mode == SelectMode::Replace or mode == SelectMode::Extend);
    Type offset(std::max(count,1));
    if (direction == Backward)
        offset = -offset;
    auto& selections = context.selections();
    for (auto& sel : selections)
    {
        auto cursor = context.has_window() ? context.window().offset_coord(sel.cursor(), offset)
                                           : context.buffer().offset_coord(sel.cursor(), offset);

        sel.anchor() = mode == SelectMode::Extend ? sel.anchor() : cursor;
        sel.cursor() = cursor;
    }
    selections.avoid_eol();
    selections.sort_and_merge_overlapping();
}

KeyMap keymap =
{
    { 'h', { "move left", move<CharCount, Backward> } },
    { 'j', { "move down", move<LineCount, Forward> } },
    { 'k', { "move up",  move<LineCount, Backward> } },
    { 'l', { "move right", move<CharCount, Forward> } },

    { 'H', { "extend left", move<CharCount, Backward, SelectMode::Extend> } },
    { 'J', { "extend down", move<LineCount, Forward, SelectMode::Extend> } },
    { 'K', { "extend up", move<LineCount, Backward, SelectMode::Extend> } },
    { 'L', { "extend right", move<CharCount, Forward, SelectMode::Extend> } },

    { 't', { "select to next character", select_to_next_char<SelectFlags::None> } },
    { 'f', { "select to next character included", select_to_next_char<SelectFlags::Inclusive> } },
    { 'T', { "extend to next character", select_to_next_char<SelectFlags::Extend> } },
    { 'F', { "extend to next character included", select_to_next_char<SelectFlags::Inclusive | SelectFlags::Extend> } },
    { alt('t'), { "select to previous character", select_to_next_char<SelectFlags::Reverse> } },
    { alt('f'), { "select to previous character included", select_to_next_char<SelectFlags::Inclusive | SelectFlags::Reverse> } },
    { alt('T'), { "extend to previous character", select_to_next_char<SelectFlags::Extend | SelectFlags::Reverse> } },
    { alt('F'), { "extend to previous character included", select_to_next_char<SelectFlags::Inclusive | SelectFlags::Extend | SelectFlags::Reverse> } },

    { 'd', { "erase selected text", erase_selections } },
    { 'c', { "change selected text", change } },
    { 'i', { "insert before selected text", enter_insert_mode<InsertMode::Insert> } },
    { 'I', { "insert at line begin", enter_insert_mode<InsertMode::InsertAtLineBegin> } },
    { 'a', { "insert after selected text", enter_insert_mode<InsertMode::Append> } },
    { 'A', { "insert at line end", enter_insert_mode<InsertMode::AppendAtLineEnd> } },
    { 'o', { "insert on new line below", enter_insert_mode<InsertMode::OpenLineBelow> } },
    { 'O', { "insert on new line above", enter_insert_mode<InsertMode::OpenLineAbove> } },
    { 'r', { "replace with character", replace_with_char } },

    { 'g', { "go to location", goto_commands<SelectMode::Replace> } },
    { 'G', { "extend to location", goto_commands<SelectMode::Extend> } },

    { 'v', { "move view", view_commands } },

    { 'y', { "yank selected text", yank } },
    { 'p', { "paste after selected text", repeated(paste<InsertMode::Append>) } },
    { 'P', { "paste before selected text", repeated(paste<InsertMode::Insert>) } },
    { alt('p'), { "paste every yanked selection after selected text", paste_all<InsertMode::Append> } },
    { alt('P'), { "paste every yanked selection before selected text", paste_all<InsertMode::Insert> } },
    { 'R', { "replace selected text with yanked text", paste<InsertMode::Replace> } },

    { 's', { "select regex matches in selected text", select_regex } },
    { 'S', { "split selected text on regex matches", split_regex } },
    { alt('s'), { "split selected text on line ends", split_lines } },

    { '.', { "repeat last insert command", repeat_last_insert } },

    { '%', { "select whole buffer", [](Context& context, int) { select_buffer(context.selections()); } } },

    { ':', { "enter command prompt", command } },
    { '|', { "pipe each selection through filter and replace with output", pipe<InsertMode::Replace> } },
    { alt('|'), { "pipe each selection through filter and append with output", pipe<InsertMode::Append> } },
    { ' ', { "remove all selection except main", [](Context& context, int count) { keep_selection(context.selections(), count ? count-1 : context.selections().main_index()); } } },
    { alt(' '), { "remove main selection", [](Context& context, int count) { remove_selection(context.selections(), count ? count-1 : context.selections().main_index()); } } },
    { ';', { "reduce selections to their cursor", [](Context& context, int count) { clear_selections(context.selections()); } } },
    { alt(';'), { "swap selections cursor and anchor", [](Context& context, int count) { flip_selections(context.selections()); } } },

    { 'w', { "select to next word start", repeated(make_select<SelectMode::Replace>(select_to_next_word<Word>)) } },
    { 'e', { "select to next word end", repeated(make_select<SelectMode::Replace>(select_to_next_word_end<Word>)) } },
    { 'b', { "select to prevous word start", repeated(make_select<SelectMode::Replace>(select_to_previous_word<Word>)) } },
    { 'W', { "extend to next word start", repeated(make_select<SelectMode::Extend>(select_to_next_word<Word>)) } },
    { 'E', { "extend to next word end", repeated(make_select<SelectMode::Extend>(select_to_next_word_end<Word>)) } },
    { 'B', { "extend to prevous word start", repeated(make_select<SelectMode::Extend>(select_to_previous_word<Word>)) } },

    { alt('w'), { "select to next WORD start", repeated(make_select<SelectMode::Replace>(select_to_next_word<WORD>)) } },
    { alt('e'), { "select to next WORD end", repeated(make_select<SelectMode::Replace>(select_to_next_word_end<WORD>)) } },
    { alt('b'), { "select to prevous WORD start", repeated(make_select<SelectMode::Replace>(select_to_previous_word<WORD>)) } },
    { alt('W'), { "extend to next WORD start", repeated(make_select<SelectMode::Extend>(select_to_next_word<WORD>)) } },
    { alt('E'), { "extend to next WORD end", repeated(make_select<SelectMode::Extend>(select_to_next_word_end<WORD>)) } },
    { alt('B'), { "extend to prevous WORD start", repeated(make_select<SelectMode::Extend>(select_to_previous_word<WORD>)) } },

    { alt('l'), { "select to line end", repeated(make_select<SelectMode::Replace>(select_to_eol)) } },
    { alt('L'), { "extend to line end", repeated(make_select<SelectMode::Extend>(select_to_eol)) } },
    { alt('h'), { "select to line begin", repeated(make_select<SelectMode::Replace>(select_to_eol_reverse)) } },
    { alt('H'), { "extend to line begin", repeated(make_select<SelectMode::Extend>(select_to_eol_reverse)) } },

    { 'x', { "select line", repeated(make_select<SelectMode::Replace>(select_line)) } },
    { 'X', { "extend line", repeated(make_select<SelectMode::Extend>(select_line)) } },
    { alt('x'), { "extend selections to whole lines", make_select<SelectMode::Replace>(select_lines) } },
    { alt('X'), { "crop selections to whole lines", make_select<SelectMode::Replace>(trim_partial_lines) } },

    { 'm', { "select to matching character", make_select<SelectMode::Replace>(select_matching) } },
    { 'M', { "extend to matching character", make_select<SelectMode::Extend>(select_matching) } },

    { '/', { "select next given regex match", search<SelectMode::Replace, Forward> } },
    { '?', { "extend with next given regex match", search<SelectMode::Extend, Forward> } },
    { alt('/'), { "select previous given regex match", search<SelectMode::Replace, Backward> } },
    { alt('?'), { "extend with previous given regex match", search<SelectMode::Extend, Backward> } },
    { 'n', { "select next current search pattern match", search_next<SelectMode::Replace, Forward> } },
    { 'N', { "extend with next current search pattern match", search_next<SelectMode::Append, Forward> } },
    { alt('n'), { "select previous current search pattern match", search_next<SelectMode::Replace, Backward> } },
    { alt('N'), { "extend with previous current search pattern match", search_next<SelectMode::Append, Backward> } },
    { '*', { "set search pattern to main selection content", use_selection_as_search_pattern<true> } },
    { alt('*'), { "set search pattern to main selection content, do not detect words", use_selection_as_search_pattern<false> } },

    { 'u', { "undo", undo } },
    { 'U', { "redo", redo } },

    { alt('i'), { "select inner object", select_object<ObjectFlags::ToBegin | ObjectFlags::ToEnd | ObjectFlags::Inner> } },
    { alt('a'), { "select whole object", select_object<ObjectFlags::ToBegin | ObjectFlags::ToEnd> } },
    { ']', { "select to object end", select_object<ObjectFlags::ToEnd> } },
    { '[', { "select to object start", select_object<ObjectFlags::ToBegin> } },
    { '}', { "extend to object end", select_object<ObjectFlags::ToEnd, SelectMode::Extend> } },
    { '{', { "extend to object start", select_object<ObjectFlags::ToBegin, SelectMode::Extend> } },

    { alt('j'), { "join lines", join } },
    { alt('J'), { "join lines and select spaces", join_select_spaces } },

    { alt('k'), { "keep selections matching given regex", keep<true> } },
    { alt('K'), { "keep selections not matching given regex", keep<false> } },
    { '$', { "pipe each selection through shell command and keep the ones whose command succeed", keep_pipe } },

    { '<', { "deindent", deindent<true> } },
    { '>', { "indent", indent<false> } },
    { alt('>'), { "indent, including empty lines", indent<true> } },
    { alt('<'), { "deindent, not including incomplete indent", deindent<false> } },

    { ctrl('i'), { "jump forward in jump list",jump<Forward> } },
    { ctrl('o'), { "jump backward in jump list", jump<Backward> } },
    { ctrl('s'), { "push current selections in jump list", save_selections } },

    { alt('r'), { "rotate main selection", rotate_selections } },
    { alt('R'), { "rotate selections content", rotate_selections_content } },

    { 'q', { "replay recorded macro", replay_macro } },
    { 'Q', { "start or end macro recording", start_or_end_macro_recording } },

    { '`', { "convert to lower case in selections", for_each_char<to_lower> } },
    { '~', { "convert to upper case in selections", for_each_char<to_upper> } },
    { alt('`'),  { "swap case in selections", for_each_char<swap_case> } },

    { '&', { "align selection cursors", align } },
    { alt('&'), { "copy indentation", copy_indent } },

    { '@', { "convert tabs to spaces in selections", tabs_to_spaces } },
    { alt('@'), { "convert spaces to tabs in selections", spaces_to_tabs } },

    { Key::Left,  { "move left", move<CharCount, Backward> } },
    { Key::Down,  { "move down", move<LineCount, Forward> } },
    { Key::Up,    { "move up", move<LineCount, Backward> } },
    { Key::Right, { "move right", move<CharCount, Forward> } },

    { Key::PageUp,   { "scroll one page up", scroll<Key::PageUp> } },
    { Key::PageDown, { "scroll one page down", scroll<Key::PageDown> } },
};

}