#include "normal.hh" #include "buffer.hh" #include "buffer_manager.hh" #include "buffer_utils.hh" #include "client_manager.hh" #include "command_manager.hh" #include "commands.hh" #include "containers.hh" #include "context.hh" #include "face_registry.hh" #include "file.hh" #include "flags.hh" #include "option_manager.hh" #include "register_manager.hh" #include "selectors.hh" #include "shell_manager.hh" #include "string.hh" #include "user_interface.hh" #include "window.hh" namespace Kakoune { using namespace std::placeholders; enum class SelectMode { Replace, Extend, Append, }; template void select(Context& context, T 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 void select(Context& context, NormalParams) { select(context, func); } template 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 void enter_insert_mode(Context& context, NormalParams) { context.input_handler().insert(mode); } void repeat_last_insert(Context& context, NormalParams) { context.input_handler().repeat_last_insert(); } template void goto_commands(Context& context, NormalParams params) { if (params.count != 0) { context.push_jump(); select_coord(context.buffer(), LineCount{params.count - 1}, context.selections()); if (context.has_window()) context.window().center_line(LineCount{params.count-1}); } else { on_next_key_with_autoinfo(context, KeymapMode::Goto, [](Key key, Context& context) { auto cp = key.codepoint(); if (not cp) return; auto& buffer = context.buffer(); switch (to_lower(*cp)) { case 'g': case 'k': context.push_jump(); select_coord(buffer, ByteCoord{0,0}, context.selections()); break; case 'l': select>(context, {}); break; case 'h': select>(context, {}); break; case 'j': context.push_jump(); select_coord(buffer, buffer.line_count() - 1, context.selections()); break; case 'e': context.push_jump(); select_coord(buffer, buffer.back_coord(), context.selections()); break; case 't': if (context.has_window()) { auto line = context.window().position().line; select_coord(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(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(buffer, line, context.selections()); } break; case 'a': { Buffer* target = nullptr; if (not context.has_client() or not (target = context.client().last_buffer())) { context.print_status({"no last buffer", get_face("Error")}); break; } context.push_jump(); context.change_buffer(*target); break; } case 'f': { const Selection& sel = context.selections().main(); String filename = content(buffer, sel); static constexpr char forbidden[] = { '\'', '\\', '\0' }; for (auto c : filename) if (contains(forbidden, c)) return; auto paths = context.options()["path"].get>(); 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(format("unable to find file '{}'", filename)); Buffer* buffer = BufferManager::instance().get_buffer_ifp(path); if (not buffer) buffer = open_file_buffer(path); if (buffer != &context.buffer()) { context.push_jump(); context.change_buffer(*buffer); } break; } case '.': { context.push_jump(); auto pos = buffer.last_modification_coord(); if (pos >= buffer.back_coord()) pos = buffer.back_coord(); else if (buffer[pos.line].length() == pos.column + 1) pos = ByteCoord{ pos.line+1, 0 }; select_coord(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"); } } template void view_commands(Context& context, NormalParams params) { const int count = params.count; on_next_key_with_autoinfo(context, KeymapMode::View, [count](Key key, Context& context) { if (key == Key::Escape) return; if (lock) view_commands(context, {}); auto cp = key.codepoint(); if (not cp or not context.has_window()) return; const ByteCoord cursor = context.selections().main().cursor(); Window& window = context.window(); switch (to_lower(*cp)) { case 'v': case 'c': context.window().center_line(cursor.line); break; case 'm': context.window().center_column( context.buffer()[cursor.line].char_count_to(cursor.column)); 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(1, count)); break; case 'j': context.window().scroll( std::max(1, count)); break; case 'k': context.window().scroll(-std::max(1, count)); break; case 'l': context.window().scroll( std::max(1, count)); break; } }, "view", "v,c: center cursor (vertically)\n" "m: center cursor (horzontally)\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, NormalParams) { on_next_key_with_autoinfo(context, KeymapMode::None, [](Key key, Context& context) { auto cp = key.codepoint(); if (not cp) return; ScopedEdition edition(context); Buffer& buffer = context.buffer(); SelectionList& selections = context.selections(); Vector strings; for (auto& sel : selections) { CharCount count = char_length(buffer, sel); strings.emplace_back(*cp, count); } selections.insert(strings, InsertMode::Replace); }, "replace with char", "enter char to replace with\n"); } Codepoint swap_case(Codepoint cp) { Codepoint res = to_lower(cp); return res == cp ? to_upper(cp) : res; } template void for_each_codepoint(Context& context, NormalParams) { using Utf8It = utf8::iterator; ScopedEdition edition(context); Buffer& buffer = context.buffer(); SelectionList& selections = context.selections(); Vector strings; for (auto& sel : selections) { String str; for (auto begin = Utf8It{buffer.iterator_at(sel.min()), buffer}, end = Utf8It{buffer.iterator_at(sel.max()), buffer}+1; begin != end; ++begin) utf8::dump(std::back_inserter(str), func(*begin)); strings.push_back(std::move(str)); } selections.insert(strings, InsertMode::Replace); } void command(Context& context, NormalParams) { if (not CommandManager::has_instance()) return; context.input_handler().prompt( ":", "", get_face("Prompt"), [](const Context& context, CompletionFlags flags, StringView cmd_line, ByteCount pos) { return CommandManager::instance().complete(context, flags, cmd_line, pos); }, [](StringView cmdline, PromptEvent event, Context& context) { if (context.has_ui()) { context.ui().info_hide(); if (event == PromptEvent::Change and context.options()["autoinfo"].get() > 0) { Face face = get_face("Information"); if (cmdline.length() == 1 and is_horizontal_blank(cmdline[0_byte])) context.ui().info_show("prompt", "commands preceded by a blank wont be saved to history", CharCoord{}, face, InfoStyle::Prompt); auto info = CommandManager::instance().command_info(context, cmdline); if (not info.first.empty() and not info.second.empty()) context.ui().info_show(info.first, info.second, CharCoord{}, face, InfoStyle::Prompt); } } if (event == PromptEvent::Validate) CommandManager::instance().execute(cmdline, context); }); } template void pipe(Context& context, NormalParams) { const char* prompt = replace ? "pipe:" : "pipe-to:"; context.input_handler().prompt(prompt, "", get_face("Prompt"), shell_complete, [](StringView 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.str(); real_cmd = cmdline; } if (real_cmd.empty()) return; Buffer& buffer = context.buffer(); SelectionList& selections = context.selections(); if (replace) { Vector strings; for (auto& sel : selections) { auto in = content(buffer, sel); bool insert_eol = in.back() != '\n'; if (insert_eol) in += '\n'; auto out = ShellManager::instance().eval( real_cmd, context, in, ShellManager::Flags::WaitForStdout).first; if ((insert_eol or sel.max() == buffer.back_coord()) and out.back() == '\n') out = out.substr(0, out.length()-1).str(); strings.push_back(std::move(out)); } ScopedEdition edition(context); selections.insert(strings, InsertMode::Replace); } else { for (auto& sel : selections) ShellManager::instance().eval(real_cmd, context, content(buffer, sel), ShellManager::Flags::None); } }); } template void insert_output(Context& context, NormalParams) { const char* prompt = mode == InsertMode::Insert ? "insert-output:" : "append-output:"; context.input_handler().prompt(prompt, "", get_face("Prompt"), shell_complete, [](StringView 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.str(); real_cmd = cmdline; } if (real_cmd.empty()) return; auto str = ShellManager::instance().eval( real_cmd, context, {}, ShellManager::Flags::WaitForStdout).first; ScopedEdition edition(context); context.selections().insert(str, mode); }); } template void select_next_match(const Buffer& buffer, SelectionList& selections, const Regex& regex) { if (mode == SelectMode::Replace) { for (auto& sel : selections) sel = keep_direction(find_next_match(buffer, sel, regex), sel); } if (mode == SelectMode::Extend) { for (auto& sel : selections) sel.merge_with(find_next_match(buffer, sel, regex)); } else if (mode == SelectMode::Append) { auto sel = keep_direction( find_next_match(buffer, selections.main(), regex), selections.main()); selections.push_back(std::move(sel)); selections.set_main_index(selections.size() - 1); } selections.sort_and_merge_overlapping(); } void yank(Context& context, NormalParams params) { const char reg = params.reg ? params.reg : '"'; RegisterManager::instance()[reg] = context.selections_content(); context.print_status({ format("yanked {} selections to register {}", context.selections().size(), reg), get_face("Information") }); } void erase_selections(Context& context, NormalParams params) { const char reg = params.reg ? params.reg : '"'; RegisterManager::instance()[reg] = context.selections_content(); ScopedEdition edition(context); context.selections().erase(); context.selections().avoid_eol(); } void change(Context& context, NormalParams params) { const char reg = params.reg ? params.reg : '"'; RegisterManager::instance()[reg] = context.selections_content(); enter_insert_mode(context, params); } 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 void paste(Context& context, NormalParams params) { const char reg = params.reg ? params.reg : '"'; auto strings = RegisterManager::instance()[reg].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 void paste_all(Context& context, NormalParams params) { const char reg = params.reg ? params.reg : '"'; auto strings = RegisterManager::instance()[reg].values(context); InsertMode effective_mode = mode; String all; Vector offsets; for (auto& str : strings) { if (str.empty()) continue; if (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(); Vector 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; } } if (not result.empty()) selections = std::move(result); } template void regex_prompt(Context& context, const String prompt, T func) { CharCoord position = context.has_window() ? context.window().position() : CharCoord{}; SelectionList selections = context.selections(); context.input_handler().prompt(prompt, "", get_face("Prompt"), complete_nothing, [=](StringView str, PromptEvent event, Context& context) mutable { try { if (event != PromptEvent::Change and context.has_ui()) context.ui().info_hide(); const bool incsearch = context.options()["incsearch"].get(); if (incsearch) { selections.update(); context.selections_write_only() = selections; if (context.has_window()) context.window().set_position(position); context.input_handler().set_prompt_face(get_face("Prompt")); } if (event == PromptEvent::Abort or (event == PromptEvent::Change and (not incsearch or str.empty()))) return; if (event == PromptEvent::Validate) context.push_jump(); func(str.empty() ? Regex{} : Regex{str}, event, context); } catch (regex_error& err) { if (event == PromptEvent::Validate) throw; else context.input_handler().set_prompt_face(get_face("Error")); } catch (runtime_error&) { context.selections_write_only() = selections; // only validation should propagate errors, // incremental search should not. if (event == PromptEvent::Validate) throw; } }); } template void search(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '/'); regex_prompt(context, direction == Forward ? "search:" : "reverse search:", [reg](Regex ex, PromptEvent event, Context& context) { if (ex.empty()) ex = Regex{context.main_sel_register_value(reg)}; else if (event == PromptEvent::Validate) RegisterManager::instance()[reg] = ex.str(); if (not ex.empty() and not ex.str().empty()) select_next_match(context.buffer(), context.selections(), ex); }); } template void search_next(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '/'); StringView str = context.main_sel_register_value(reg); if (not str.empty()) { Regex ex{str}; do { select_next_match(context.buffer(), context.selections(), ex); } while (--params.count > 0); } else throw runtime_error("no search pattern"); } template void use_selection_as_search_pattern(Context& context, NormalParams params) { Vector patterns; auto& sels = context.selections(); const auto& buffer = context.buffer(); using Utf8It = utf8::iterator; for (auto& sel : sels) { Utf8It begin{buffer.iterator_at(sel.min()), buffer}; Utf8It end{buffer.iterator_at(sel.max()), buffer}; ++end; auto content = "\\Q" + buffer.string(begin.base().coord(), end.base().coord()) + "\\E"; if (smart) { if (is_word(*begin) and (begin == buffer.begin() or not is_word(*(begin-1)))) content = "\\b" + content; if (is_word(*(end-1)) and (end == buffer.end() or not is_word(*end))) content = content + "\\b"; } patterns.push_back(std::move(content)); } const char reg = to_lower(params.reg ? params.reg : '/'); context.print_status({ format("register '{}' set to '{}'", reg, patterns[sels.main_index()]), get_face("Information") }); RegisterManager::instance()[reg] = patterns; // Hack, as Window do not take register state into account if (context.has_window()) context.window().force_redraw(); } void select_regex(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '/'); regex_prompt(context, "select:", [reg](Regex ex, PromptEvent event, Context& context) { if (ex.empty()) ex = Regex{context.main_sel_register_value(reg)}; else if (event == PromptEvent::Validate) RegisterManager::instance()[reg] = ex.str(); if (not ex.empty() and not ex.str().empty()) select_all_matches(context.selections(), ex); }); } void split_regex(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '/'); regex_prompt(context, "split:", [reg](Regex ex, PromptEvent event, Context& context) { if (ex.empty()) ex = Regex{context.main_sel_register_value(reg)}; else if (event == PromptEvent::Validate) RegisterManager::instance()[reg] = ex.str(); if (not ex.empty() and not ex.str().empty()) split_selections(context.selections(), ex); }); } void split_lines(Context& context, NormalParams) { auto& selections = context.selections(); auto& buffer = context.buffer(); Vector 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(keep_direction({min, {min.line, buffer[min.line].length()-1}}, sel)); for (auto line = min.line+1; line < max.line; ++line) res.push_back(keep_direction({line, {line, buffer[line].length()-1}}, sel)); res.push_back(keep_direction({max.line, max}, sel)); } selections = std::move(res); } void join_lines_select_spaces(Context& context, NormalParams) { auto& buffer = context.buffer(); Vector selections; for (auto& sel : context.selections()) { const LineCount min_line = sel.min().line; const LineCount max_line = sel.max().line; auto end_line = std::min(buffer.line_count()-1, max_line + (min_line == max_line ? 1 : 0)); for (LineCount line = min_line; line < end_line; ++line) { auto begin = buffer.iterator_at({line, buffer[line].length()-1}); auto end = std::find_if_not(begin+1, buffer.end(), is_horizontal_blank); selections.push_back({begin.coord(), (end-1).coord()}); } } if (selections.empty()) return; context.selections_write_only() = std::move(selections); ScopedEdition edition(context); context.selections().insert(" "_str, InsertMode::Replace); } void join_lines(Context& context, NormalParams params) { SelectionList sels{context.selections()}; auto restore_sels = on_scope_end([&]{ sels.update(); context.selections_write_only() = std::move(sels); }); join_lines_select_spaces(context, params); } template void keep(Context& context, NormalParams) { constexpr const char* prompt = matching ? "keep matching:" : "keep not matching:"; regex_prompt(context, prompt, [](const Regex& ex, PromptEvent, Context& context) { if (ex.empty()) return; const Buffer& buffer = context.buffer(); Vector keep; for (auto& sel : context.selections()) { if (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.selections_write_only() = std::move(keep); }); } void keep_pipe(Context& context, NormalParams) { context.input_handler().prompt( "keep pipe:", "", get_face("Prompt"), shell_complete, [](StringView cmdline, PromptEvent event, Context& context) { if (event != PromptEvent::Validate) return; const Buffer& buffer = context.buffer(); auto& shell_manager = ShellManager::instance(); Vector keep; for (auto& sel : context.selections()) { if (shell_manager.eval(cmdline, context, content(buffer, sel), ShellManager::Flags::None).second == 0) keep.push_back(sel); } if (keep.empty()) throw runtime_error("no selections remaining"); context.selections_write_only() = std::move(keep); }); } template void indent(Context& context, NormalParams) { CharCount indent_width = context.options()["indentwidth"].get(); String indent = indent_width == 0 ? "\t" : String{' ', indent_width}; auto& buffer = context.buffer(); Vector sels; LineCount last_line = 0; for (auto& sel : context.selections()) { for (auto line = std::max(last_line, sel.min().line); line < sel.max().line+1; ++line) { if (indent_empty or buffer[line].length() > 1) sels.push_back({line, line}); } // avoid reindenting the same line if multiple selections are on it last_line = sel.max().line+1; } if (not sels.empty()) { ScopedEdition edition(context); SelectionList selections{buffer, std::move(sels)}; selections.insert(indent, InsertMode::Insert); } } template void deindent(Context& context, NormalParams) { CharCount tabstop = context.options()["tabstop"].get(); CharCount indent_width = context.options()["indentwidth"].get(); if (indent_width == 0) indent_width = tabstop; auto& buffer = context.buffer(); Vector sels; LineCount last_line = 0; for (auto& sel : context.selections()) { for (auto line = std::max(sel.min().line, last_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; } } } // avoid reindenting the same line if multiple selections are on it last_line = sel.max().line + 1; } if (not sels.empty()) { ScopedEdition edition(context); SelectionList selections{context.buffer(), std::move(sels)}; selections.erase(); } } template void select_object(Context& context, NormalParams params) { const int level = params.count <= 0 ? 0 : params.count - 1; on_next_key_with_autoinfo(context, KeymapMode::Object, [level](Key key, Context& context) { auto cp = key.codepoint(); if (not cp) return; static constexpr struct { Codepoint key; Selection (*func)(const Buffer&, const Selection&, ObjectFlags); } selectors[] = { { 'w', select_word }, { 'W', select_word }, { 's', select_sentence }, { 'p', select_paragraph }, { ' ', select_whitespaces }, { 'i', select_indent }, { 'n', select_number }, }; for (auto& sel : selectors) { if (*cp == sel.key) return select(context, std::bind(sel.func, _1, _2, flags)); } if (*cp == 'u') { return select(context, std::bind(select_argument, _1, _2, level, flags)); } static constexpr struct { MatchingPair pair; Codepoint name; } surrounding_pairs[] = { { { '(', ')' }, 'b' }, { { '{', '}' }, 'B' }, { { '[', ']' }, 'r' }, { { '<', '>' }, 'a' }, { { '"', '"' }, 'Q' }, { { '\'', '\'' }, 'q' }, { { '`', '`' }, 'g' }, }; for (auto& sur : surrounding_pairs) { if (sur.pair.opening == *cp or sur.pair.closing == *cp or (sur.name != 0 and sur.name == *cp)) return select(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" "u: argument \n" "n: number \n"); } template void scroll(Context& context, NormalParams) { 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); } template void copy_selections_on_next_lines(Context& context, NormalParams params) { auto& selections = context.selections(); auto& buffer = context.buffer(); const CharCount tabstop = context.options()["tabstop"].get(); Vector result; for (auto& sel : selections) { auto anchor = sel.anchor(); auto cursor = sel.cursor(); CharCount cursor_col = get_column(buffer, tabstop, cursor); CharCount anchor_col = get_column(buffer, tabstop, anchor); result.push_back(std::move(sel)); for (int i = 0; i < std::max(params.count, 1); ++i) { LineCount offset = (direction == Forward ? 1 : -1) * (i + 1); const LineCount anchor_line = anchor.line + offset; const LineCount cursor_line = cursor.line + offset; if (anchor_line < 0 or cursor_line < 0 or anchor_line >= buffer.line_count() or cursor_line >= buffer.line_count()) continue; ByteCount anchor_byte = get_byte_to_column(buffer, tabstop, {anchor_line, anchor_col}); ByteCount cursor_byte = get_byte_to_column(buffer, tabstop, {cursor_line, cursor_col}); if (anchor_byte != buffer[anchor_line].length() and cursor_byte != buffer[cursor_line].length()) result.emplace_back(ByteCoord{anchor_line, anchor_byte}, ByteCoordAndTarget{cursor_line, cursor_byte, cursor.target}); } } selections = std::move(result); selections.sort_and_merge_overlapping(); } void rotate_selections(Context& context, NormalParams params) { context.selections().rotate_main(params.count != 0 ? params.count : 1); } void rotate_selections_content(Context& context, NormalParams params) { int group = params.count; 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 }; template<> struct WithBitOps : std::true_type {}; template void select_to_next_char(Context& context, NormalParams params) { on_next_key_with_autoinfo(context, KeymapMode::None, [params](Key key, Context& context) { constexpr auto new_flags = flags & SelectFlags::Extend ? SelectMode::Extend : SelectMode::Replace; if (auto cp = key.codepoint()) select( context, std::bind(flags & SelectFlags::Reverse ? select_to_reverse : select_to, _1, _2, *cp, params.count, flags & SelectFlags::Inclusive)); }, "select to next char","enter char to select to"); } void start_or_end_macro_recording(Context& context, NormalParams params) { if (context.input_handler().is_recording()) context.input_handler().stop_recording(); else { const char reg = to_lower(params.reg ? params.reg : '@'); if (not is_basic_alpha(reg) and reg != '@') throw runtime_error("Macros can only use the '@' and alphabetic registers"); context.input_handler().start_recording(reg); } } void end_macro_recording(Context& context, NormalParams) { if (context.input_handler().is_recording()) context.input_handler().stop_recording(); } void replay_macro(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '@'); if (not is_basic_alpha(reg) and reg != '@') throw runtime_error("Macros can only use the '@' and alphabetic registers"); static bool running_macros[27] = {}; const size_t idx = reg != '@' ? (size_t)(reg - 'a') : 26; if (running_macros[idx]) throw runtime_error("recursive macros call detected"); ConstArrayView reg_val = RegisterManager::instance()[reg].values(context); if (reg_val.empty() or reg_val[0].empty()) throw runtime_error(format("Register '{}' is empty", reg)); running_macros[idx] = true; auto stop = on_scope_end([&]{ running_macros[idx] = false; }); auto keys = parse_keys(reg_val[0]); ScopedEdition edition(context); do { exec_keys(keys, context); } while (--params.count > 0); } template void jump(Context& context, NormalParams) { auto jump = (direction == Forward) ? context.jump_forward() : context.jump_backward(); Buffer* oldbuf = &context.buffer(); Buffer& buffer = const_cast(jump.buffer()); if (&buffer != oldbuf) context.change_buffer(buffer); context.selections_write_only() = jump; } void push_selections(Context& context, NormalParams) { context.push_jump(); context.print_status({ format("saved {} selections", context.selections().size()), get_face("Information") }); } void align(Context& context, NormalParams) { auto& selections = context.selections(); auto& buffer = context.buffer(); const CharCount tabstop = context.options()["tabstop"].get(); Vector> 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(); 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)); } selections.update(); } } void copy_indent(Context& context, NormalParams params) { int selection = params.count; auto& buffer = context.buffer(); auto& selections = context.selections(); Vector 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; auto 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, NormalParams params) { auto& buffer = context.buffer(); const CharCount opt_tabstop = context.options()["tabstop"].get(); const CharCount tabstop = params.count == 0 ? opt_tabstop : params.count; Vector tabs; Vector 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, NormalParams params) { auto& buffer = context.buffer(); const CharCount opt_tabstop = context.options()["tabstop"].get(); const CharCount tabstop = params.count == 0 ? opt_tabstop : params.count; Vector 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()}); else if (*spaces_end == '\t') spaces.push_back({spaces_beg.coord(), spaces_end.coord()}); it = spaces_end; } else ++it; } } if (not spaces.empty()) SelectionList{ buffer, std::move(spaces) }.insert("\t"_str, InsertMode::Replace); } void save_selections(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '^'); if (not is_basic_alpha(reg) and reg != '^') throw runtime_error("selections can only be saved to the '^' and alphabetic registers"); String desc = format("{}@{}%{}", selection_list_to_string(context.selections()), context.buffer().name(), context.buffer().timestamp()); RegisterManager::instance()[reg] = desc; context.print_status({format("Saved selections in register '{}'", reg), get_face("Information")}); } void restore_selections(Context& context, NormalParams params) { const char reg = to_lower(params.reg ? params.reg : '^'); if (not is_basic_alpha(reg) and reg != '^') throw runtime_error("selections can only be saved to the '^' and alphabetic registers"); auto content = RegisterManager::instance()[reg].values(context); if (content.size() != 1) throw runtime_error(format("Register {} does not contain a selections desc", reg)); StringView desc = content[0]; auto arobase = find(desc, '@'); auto percent = find(desc, '%'); if (arobase == desc.end() or percent == desc.end()) throw runtime_error(format("Register {} does not contain a selections desc", reg)); Buffer& buffer = BufferManager::instance().get_buffer({arobase+1, percent}); size_t timestamp = str_to_int({percent + 1, desc.end()}); Vector sels; for (auto sel_desc : split({desc.begin(), arobase}, ':')) sels.push_back(selection_from_string(sel_desc)); SelectionList sel_list{buffer, std::move(sels), timestamp}; if (&buffer != &context.buffer()) context.change_buffer(buffer); context.selections_write_only() = std::move(sel_list); context.print_status({format("Restored selections from register '{}'", reg), get_face("Information")}); } void undo(Context& context, NormalParams) { 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.selections_write_only() = std::move(ranges); context.selections().avoid_eol(); } else if (not res) context.print_status({ "nothing left to undo", get_face("Information") }); } void redo(Context& context, NormalParams) { 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.selections_write_only() = std::move(ranges); context.selections().avoid_eol(); } else if (not res) context.print_status({ "nothing left to redo", get_face("Information") }); } void exec_user_mappings(Context& context, NormalParams params) { on_next_key_with_autoinfo(context, KeymapMode::None, [params](Key key, Context& context) mutable { if (not context.keymaps().is_mapped(key, KeymapMode::User)) return; auto mapping = context.keymaps().get_mapping(key, KeymapMode::User); ScopedEdition edition(context); exec_keys(mapping, context); }, "user mapping", "enter user key"); } template class Repeated { public: constexpr Repeated(T t) : m_func(t) {} void operator() (Context& context, NormalParams params) { ScopedEdition edition(context); do { m_func(context, {0, params.reg}); } while(--params.count > 0); } private: T m_func; }; template void repeated(Context& context, NormalParams params) { ScopedEdition edition(context); do { func(context, {0, params.reg}); } while(--params.count > 0); } template void move(Context& context, NormalParams params) { kak_assert(mode == SelectMode::Replace or mode == SelectMode::Extend); Type offset(std::max(params.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(); } void select_whole_buffer(Context& context, NormalParams) { select_buffer(context.selections()); } void keep_selection(Context& context, NormalParams p) { auto& selections = context.selections(); const int index = p.count ? p.count-1 : selections.main_index(); if (index < selections.size()) selections = SelectionList{ selections.buffer(), std::move(selections[index]) }; selections.check_invariant(); } void remove_selection(Context& context, NormalParams p) { auto& selections = context.selections(); const int index = p.count ? p.count-1 : selections.main_index(); if (selections.size() > 1 and index < selections.size()) { selections.remove(index); size_t main_index = selections.main_index(); if (index < main_index or main_index == selections.size()) selections.set_main_index(main_index - 1); } selections.check_invariant(); } void clear_selections(Context& context, NormalParams) { for (auto& sel : context.selections()) sel.anchor() = sel.cursor(); } void flip_selections(Context& context, NormalParams) { for (auto& sel : context.selections()) { const ByteCoord tmp = sel.anchor(); sel.anchor() = sel.cursor(); sel.cursor() = tmp; } context.selections().check_invariant(); } void ensure_forward(Context& context, NormalParams) { for (auto& sel : context.selections()) { const ByteCoord min = sel.min(), max = sel.max(); sel.anchor() = min; sel.cursor() = max; } context.selections().check_invariant(); } static NormalCmdDesc cmds[] = { { 'h', "move left", move }, { 'j', "move down", move }, { 'k', "move up", move }, { 'l', "move right", move }, { 'H', "extend left", move }, { 'J', "extend down", move }, { 'K', "extend up", move }, { 'L', "extend right", move }, { 't', "select to next character", select_to_next_char }, { 'f', "select to next character included", select_to_next_char }, { 'T', "extend to next character", select_to_next_char }, { 'F', "extend to next character included", select_to_next_char }, { alt('t'), "select to previous character", select_to_next_char }, { alt('f'), "select to previous character included", select_to_next_char }, { alt('T'), "extend to previous character", select_to_next_char }, { alt('F'), "extend to previous character included", select_to_next_char }, { 'd', "erase selected text", erase_selections }, { 'c', "change selected text", change }, { 'i', "insert before selected text", enter_insert_mode }, { 'I', "insert at line begin", enter_insert_mode }, { 'a', "insert after selected text", enter_insert_mode }, { 'A', "insert at line end", enter_insert_mode }, { 'o', "insert on new line below", enter_insert_mode }, { 'O', "insert on new line above", enter_insert_mode }, { 'r', "replace with character", replace_with_char }, { 'g', "go to location", goto_commands }, { 'G', "extend to location", goto_commands }, { 'v', "move view", view_commands }, { 'V', "move view (locked)", view_commands }, { 'y', "yank selected text", yank }, { 'p', "paste after selected text", repeated> }, { 'P', "paste before selected text", repeated> }, { alt('p'), "paste every yanked selection after selected text", paste_all }, { alt('P'), "paste every yanked selection before selected text", paste_all }, { 'R', "replace selected text with yanked text", paste }, { alt('R'), "replace selected text with yanked text", paste_all }, { '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", select_whole_buffer }, { ':', "enter command prompt", command }, { '|', "pipe each selection through filter and replace with output", pipe }, { alt('|'), "pipe each selection through command and ignore output", pipe }, { '!', "insert command output", insert_output }, { alt('!'), "append command output", insert_output }, { ' ', "remove all selection except main", keep_selection }, { alt(' '), "remove main selection", remove_selection }, { ';', "reduce selections to their cursor", clear_selections }, { alt(';'), "swap selections cursor and anchor", flip_selections }, { alt(':'), "ensure selection cursor is after anchor", ensure_forward }, { 'w', "select to next word start", repeated<&select>> }, { 'e', "select to next word end", repeated>> }, { 'b', "select to previous word start", repeated>> }, { 'W', "extend to next word start", repeated>> }, { 'E', "extend to next word end", repeated>> }, { 'B', "extend to previous word start", repeated>> }, { alt('w'), "select to next WORD start", repeated>> }, { alt('e'), "select to next WORD end", repeated>> }, { alt('b'), "select to previous WORD start", repeated>> }, { alt('W'), "extend to next WORD start", repeated>> }, { alt('E'), "extend to next WORD end", repeated>> }, { alt('B'), "extend to previous WORD start", repeated>> }, { alt('l'), "select to line end", repeated>> }, { Key::End, "select to line end", repeated>> }, { alt('L'), "extend to line end", repeated>> }, { alt('h'), "select to line begin", repeated>> }, { Key::Home, "select to line begin", repeated>> }, { alt('H'), "extend to line begin", repeated>> }, { 'x', "select line", repeated> }, { 'X', "extend line", repeated> }, { alt('x'), "extend selections to whole lines", select }, { alt('X'), "crop selections to whole lines", select }, { 'm', "select to matching character", select }, { 'M', "extend to matching character", select }, { '/', "select next given regex match", search }, { '?', "extend with next given regex match", search }, { alt('/'), "select previous given regex match", search }, { alt('?'), "extend with previous given regex match", search }, { 'n', "select next current search pattern match", search_next }, { 'N', "extend with next current search pattern match", search_next }, { alt('n'), "select previous current search pattern match", search_next }, { alt('N'), "extend with previous current search pattern match", search_next }, { '*', "set search pattern to main selection content", use_selection_as_search_pattern }, { alt('*'), "set search pattern to main selection content, do not detect words", use_selection_as_search_pattern }, { 'u', "undo", undo }, { 'U', "redo", redo }, { alt('i'), "select inner object", select_object }, { alt('a'), "select whole object", select_object }, { '[', "select to object start", select_object }, { ']', "select to object end", select_object }, { '{', "extend to object start", select_object }, { '}', "extend to object end", select_object }, { alt('['), "select to inner object start", select_object }, { alt(']'), "select to inner object end", select_object }, { alt('{'), "extend to inner object start", select_object }, { alt('}'), "extend to inner object end", select_object }, { alt('j'), "join lines", join_lines }, { alt('J'), "join lines and select spaces", join_lines_select_spaces }, { alt('k'), "keep selections matching given regex", keep }, { alt('K'), "keep selections not matching given regex", keep }, { '$', "pipe each selection through shell command and keep the ones whose command succeed", keep_pipe }, { '<', "deindent", deindent }, { '>', "indent", indent }, { alt('>'), "indent, including empty lines", indent }, { alt('<'), "deindent, not including incomplete indent", deindent }, { ctrl('i'), "jump forward in jump list",jump }, { ctrl('o'), "jump backward in jump list", jump }, { ctrl('s'), "push current selections in jump list", push_selections }, { '\'', "rotate main selection", rotate_selections }, { alt('\''), "rotate selections content", rotate_selections_content }, { 'q', "replay recorded macro", replay_macro }, { 'Q', "start or end macro recording", start_or_end_macro_recording }, { Key::Escape, "end macro recording", end_macro_recording }, { '`', "convert to lower case in selections", for_each_codepoint }, { '~', "convert to upper case in selections", for_each_codepoint }, { alt('`'), "swap case in selections", for_each_codepoint }, { '&', "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 }, { 'C', "copy selection on next lines", copy_selections_on_next_lines }, { alt('C'), "copy selection on previous lines", copy_selections_on_next_lines }, { ',', "user mappings", exec_user_mappings }, { Key::Left, "move left", move }, { Key::Down, "move down", move }, { Key::Up, "move up", move }, { Key::Right, "move right", move }, { ctrl('b'), "scroll one page up", scroll }, { ctrl('f'), "scroll one page down", scroll }, { Key::PageUp, "scroll one page up", scroll }, { Key::PageDown, "scroll one page down", scroll }, { 'z', "restore selections", restore_selections }, { 'Z', "save selections", save_selections }, }; KeyMap keymap = cmds; }