#include "commands.hh" #include "buffer.hh" #include "buffer_manager.hh" #include "buffer_utils.hh" #include "client.hh" #include "client_manager.hh" #include "command_manager.hh" #include "completion.hh" #include "containers.hh" #include "context.hh" #include "event_manager.hh" #include "face_registry.hh" #include "file.hh" #include "highlighter.hh" #include "highlighters.hh" #include "option_manager.hh" #include "option_types.hh" #include "parameters_parser.hh" #include "register_manager.hh" #include "remote.hh" #include "shell_manager.hh" #include "string.hh" #include "user_interface.hh" #include "window.hh" #include #include #include #include #if defined(__GLIBC__) || defined(__CYGWIN__) #include #endif namespace Kakoune { namespace { Buffer* open_fifo(StringView name, StringView filename, bool scroll) { int fd = open(parse_filename(filename).c_str(), O_RDONLY | O_NONBLOCK); fcntl(fd, F_SETFD, FD_CLOEXEC); if (fd < 0) throw runtime_error(format("unable to open '{}'", filename)); return create_fifo_buffer(name.str(), fd, scroll); } const PerArgumentCommandCompleter filename_completer({ [](const Context& context, CompletionFlags flags, const String& prefix, ByteCount cursor_pos) { return Completions{ 0_byte, cursor_pos, complete_filename(prefix, context.options()["ignored_files"].get(), cursor_pos) }; } }); static CandidateList complete_buffer_name(StringView prefix, ByteCount cursor_pos) { prefix = prefix.substr(0, cursor_pos); const bool include_dirs = contains(prefix, '/'); CandidateList prefix_result, subsequence_result; for (auto& buffer : BufferManager::instance()) { String name = buffer->display_name(); StringView match_name = name; if (not include_dirs and buffer->flags() & Buffer::Flags::File) { auto it = find(reversed(name), '/'); if (it != name.rend()) match_name = StringView{it.base() + 2, name.end()}; } if (prefix_match(match_name, prefix)) prefix_result.push_back(name); if (subsequence_match(name, prefix)) subsequence_result.push_back(name); } return prefix_result.empty() ? subsequence_result : prefix_result; } const PerArgumentCommandCompleter buffer_completer({ [](const Context& context, CompletionFlags flags, const String& prefix, ByteCount cursor_pos) { return Completions{ 0_byte, cursor_pos, complete_buffer_name(prefix, cursor_pos) }; } }); const ParameterDesc no_params{ {}, ParameterDesc::Flags::None, 0, 0 }; const ParameterDesc single_name_param{ {}, ParameterDesc::Flags::None, 1, 1 }; const ParameterDesc single_optional_name_param{ {}, ParameterDesc::Flags::None, 0, 1 }; static constexpr auto scopes = { "global", "buffer", "window" }; Scope* get_scope_ifp(StringView scope, const Context& context) { if (prefix_match("global", scope)) return &GlobalScope::instance(); else if (prefix_match("buffer", scope)) return &context.buffer(); else if (prefix_match("window", scope)) return &context.window(); else if (prefix_match(scope, "buffer=")) return &BufferManager::instance().get_buffer(scope.substr(7_byte)); return nullptr; } Scope& get_scope(StringView scope, const Context& context) { if (auto s = get_scope_ifp(scope, context)) return *s; throw runtime_error(format("error: no such scope '{}'", scope)); } struct CommandDesc { const char* name; const char* alias; const char* docstring; ParameterDesc params; CommandFlags flags; CommandHelper helper; CommandCompleter completer; void (*func)(const ParametersParser&, Context&); }; template void edit(const ParametersParser& parser, Context& context) { if (parser.positional_count() == 0 and not force_reload) throw wrong_argument_count(); auto& name = parser.positional_count() > 0 ? parser[0] : context.buffer().name(); auto& buffer_manager = BufferManager::instance(); Buffer* buffer = nullptr; Buffer* oldbuf = &context.buffer(); if (not force_reload) buffer = buffer_manager.get_buffer_ifp(name); if (not buffer) { if (parser.get_switch("scratch")) { if (Buffer* buf = buffer_manager.get_buffer_ifp(name)) { buffer_manager.delete_buffer(*buf); if (buf == oldbuf) oldbuf = nullptr; } buffer = new Buffer(name, Buffer::Flags::None); } else if (auto fifo = parser.get_switch("fifo")) buffer = open_fifo(name, *fifo, (bool)parser.get_switch("scroll")); else { buffer = create_buffer_from_file(name); if (not buffer) { if (parser.get_switch("existing")) throw runtime_error(format("unable to open '{}'", name)); context.print_status({ format("new file '{}'", name), get_face("StatusLine") }); buffer = new Buffer(name, Buffer::Flags::File | Buffer::Flags::New); } } } if (oldbuf) buffer_manager.set_last_used_buffer(*oldbuf); const size_t param_count = parser.positional_count(); if (buffer != &context.buffer() or param_count > 1) context.push_jump(); if (buffer != &context.buffer()) context.change_buffer(*buffer); if (param_count > 1 and not parser[1].empty()) { int line = std::max(0, str_to_int(parser[1]) - 1); int column = param_count > 2 and not parser[2].empty() ? std::max(0, str_to_int(parser[2]) - 1) : 0; auto& buffer = context.buffer(); context.selections_write_only() = { buffer, buffer.clamp({ line, column }) }; if (context.has_window()) context.window().center_line(context.selections().main().cursor().line); } } ParameterDesc edit_params{ { { "existing", { false, "fail if the file does not exists, do not open a new file" } }, { "scratch", { false, "create a scratch buffer, not linked to a file" } }, { "fifo", { true, "create a buffer reading its content from a named fifo" } }, { "scroll", { false, "place the initial cursor so that the fifo will scroll to show new data" } } }, ParameterDesc::Flags::None, 0, 3 }; const CommandDesc edit_cmd = { "edit", "e", "edit [] [ []]: open the given filename in a buffer", edit_params, CommandFlags::None, CommandHelper{}, filename_completer, edit }; const CommandDesc force_edit_cmd = { "edit!", "e!", "edit! [] [ []]: open the given filename in a buffer, " "force reload if needed", edit_params, CommandFlags::None, CommandHelper{}, filename_completer, edit }; void write_buffer(const ParametersParser& parser, Context& context) { Buffer& buffer = context.buffer(); if (parser.positional_count() == 0 and !(buffer.flags() & Buffer::Flags::File)) throw runtime_error("cannot write a non file buffer without a filename"); auto filename = parser.positional_count() == 0 ? buffer.name() : parse_filename(parser[0]); write_buffer_to_file(buffer, filename); } const CommandDesc write_cmd = { "write", "w", "write [filename]: write the current buffer to it's file " "or to [filename] if specified", single_optional_name_param, CommandFlags::None, CommandHelper{}, filename_completer, write_buffer, }; void write_all_buffers() { for (auto& buffer : BufferManager::instance()) { if ((buffer->flags() & Buffer::Flags::File) and buffer->is_modified()) write_buffer_to_file(*buffer, buffer->name()); } } const CommandDesc writeall_cmd = { "writeall", "wa", "write all buffers that are associated to a file", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser&, Context&){ write_all_buffers(); } }; template void quit() { if (not force and ClientManager::instance().count() == 1) { Vector names; for (auto& buffer : BufferManager::instance()) { if ((buffer->flags() & Buffer::Flags::File) and buffer->is_modified()) names.push_back(buffer->name()); } if (not names.empty()) { String message = "modified buffers remaining: ["; for (auto it = names.begin(); it != names.end(); ++it) { if (it != names.begin()) message += ", "; message += *it; } message += "]"; throw runtime_error(message); } } // unwind back to this client event handler. throw client_removed{}; } const CommandDesc quit_cmd = { "quit", "q", "quit current client, and the kakoune session if the client is the last " "(if not running in daemon mode)", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser&, Context&){ quit(); } }; const CommandDesc force_quit_cmd = { "quit!", "q!", "quit current client, and the kakoune session if the client is the last " "(if not running in daemon mode). force quit even if the client is the " "last and some buffers are not saved.", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser&, Context&){ quit(); } }; const CommandDesc write_quit_cmd = { "wq", nullptr, "write current buffer and quit current client", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { write_buffer(parser, context); quit(); } }; const CommandDesc force_write_quit_cmd = { "wq!", nullptr, "write current buffer and quit current client, even if other buffers are " "not saved", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { write_buffer(parser, context); quit(); } }; const CommandDesc writeall_quit_cmd = { "waq", nullptr, "write all buffers associated to a file and quit current client", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { write_all_buffers(); quit(); } }; const CommandDesc buffer_cmd = { "buffer", "b", "buffer : set buffer to edit in current client", single_name_param, CommandFlags::None, CommandHelper{}, buffer_completer, [](const ParametersParser& parser, Context& context) { Buffer* oldbuf = &context.buffer(); Buffer& buffer = BufferManager::instance().get_buffer(parser[0]); if (&buffer != oldbuf) { BufferManager::instance().set_last_used_buffer(*oldbuf); context.push_jump(); context.change_buffer(buffer); } } }; template void cycle_buffer(const ParametersParser& parser, Context& context) { Buffer* oldbuf = &context.buffer(); auto it = find_if(BufferManager::instance(), [oldbuf](const SafePtr& lhs) { return lhs.get() == oldbuf; }); kak_assert(it != BufferManager::instance().end()); if (not next) { if (it == BufferManager::instance().begin()) it = BufferManager::instance().end(); --it; } else { if (++it == BufferManager::instance().end()) it = BufferManager::instance().begin(); } Buffer* newbuf = it->get(); if (newbuf != oldbuf) { BufferManager::instance().set_last_used_buffer(*oldbuf); context.push_jump(); context.change_buffer(*newbuf); } } const CommandDesc buffernext_cmd = { "buffernext", "bn", "buffernext: move to the next buffer in the list", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, cycle_buffer }; const CommandDesc bufferprev_cmd = { "bufferprev", "bp", "bufferprev: move to the previous buffer in the list", no_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, cycle_buffer }; template void delete_buffer(const ParametersParser& parser, Context& context) { BufferManager& manager = BufferManager::instance(); Buffer& buffer = parser.positional_count() == 0 ? context.buffer() : manager.get_buffer(parser[0]); if (not force and (buffer.flags() & Buffer::Flags::File) and buffer.is_modified()) throw runtime_error(format("buffer '{}' is modified", buffer.name())); if (manager.count() == 1) throw runtime_error(format("buffer '{}' is the last one", buffer.name())); manager.delete_buffer(buffer); } const CommandDesc delbuf_cmd = { "delbuf", "db", "delbuf [name]: delete current buffer or the buffer named if given", single_optional_name_param, CommandFlags::None, CommandHelper{}, buffer_completer, delete_buffer }; const CommandDesc force_delbuf_cmd = { "delbuf!", "db!", "delbuf! [name]: delete current buffer or the buffer named if " "given, even if the buffer is unsaved", single_optional_name_param, CommandFlags::None, CommandHelper{}, buffer_completer, delete_buffer }; const CommandDesc namebuf_cmd = { "namebuf", nullptr, "namebuf : change current buffer name", single_name_param, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { if (not context.buffer().set_name(parser[0])) throw runtime_error(format("unable to change buffer name to '{}'", parser[0])); } }; Completions complete_highlighter(const Context& context, StringView arg, ByteCount pos_in_token, bool only_group) { const bool shared = not arg.empty() and arg[0_byte] == '/'; if (shared) { auto& group = DefinedHighlighters::instance(); return offset_pos(group.complete_child(arg.substr(1_byte), pos_in_token-1, only_group), 1); } else { auto& group = context.window().highlighters(); return group.complete_child(arg, pos_in_token, only_group); } } Completions rm_highlighter_completer( const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { const String& arg = params[token_to_complete]; if (token_to_complete == 0 and not arg.empty() and arg.front() == '/') { auto& group = DefinedHighlighters::instance(); return offset_pos(group.complete_child(arg.substr(1_byte), pos_in_token-1, false), 1); } else if (token_to_complete == 0) return context.window().highlighters().complete_child(arg, pos_in_token, false); return {}; } Completions add_highlighter_completer( const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { StringView arg = params[token_to_complete]; if (token_to_complete == 1 and params[0] == "-group") return complete_highlighter(context, params[1], pos_in_token, true); else if (token_to_complete == 0 or (token_to_complete == 2 and params[0] == "-group")) return { 0_byte, arg.length(), complete(arg, pos_in_token, transformed(HighlighterRegistry::instance(), HighlighterRegistry::get_id)) }; return Completions{}; } Highlighter& get_highlighter(const Context& context, StringView path) { if (path.empty()) throw runtime_error("group path should not be empty"); Highlighter* root = nullptr; if (path[0_byte] == '/') { root = &DefinedHighlighters::instance(); path = path.substr(1_byte); } else root = &context.window().highlighters(); if (path.back() == '/') path = path.substr(0_byte, path.length() - 1); if (not path.empty()) return root->get_child(path); return *root; } const CommandDesc add_highlighter_cmd = { "addhl", "ah", "addhl ...: add an highlighter", ParameterDesc{ { { "group", { true, "Set the group in which to put the highlighter. " "If starting with /, search in shared highlighters, " "else search in the current window" } } }, ParameterDesc::Flags::SwitchesOnlyAtStart, 1 }, CommandFlags::None, [](const Context& context, CommandParameters params) -> String { if (params.size() > 0) { HighlighterRegistry& registry = HighlighterRegistry::instance(); auto it = registry.find(params[0]); if (it != registry.end()) return format("{}:\n{}", params[0], indent(it->second.docstring)); } return ""; }, add_highlighter_completer, [](const ParametersParser& parser, Context& context) { HighlighterRegistry& registry = HighlighterRegistry::instance(); auto begin = parser.begin(); const String& name = *begin++; Vector highlighter_params; for (; begin != parser.end(); ++begin) highlighter_params.push_back(*begin); auto group_name = parser.get_switch("group"); auto& group = group_name ? get_highlighter(context, *group_name) : context.window().highlighters(); auto it = registry.find(name); if (it == registry.end()) throw runtime_error(format("No such highlighter factory '{}'", name)); group.add_child(it->second.factory(highlighter_params)); } }; const CommandDesc rm_highlighter_cmd = { "rmhl", "rh", "rmhl : remove highlighter ", ParameterDesc{ {}, ParameterDesc::Flags::None, 1, 1 }, CommandFlags::None, CommandHelper{}, rm_highlighter_completer, [](const ParametersParser& parser, Context& context) { StringView path = parser[0]; auto sep_it = find(reversed(path), '/'); auto& group = sep_it != path.rend() ? get_highlighter(context, {path.begin(), sep_it.base()-1}) : context.window().highlighters(); group.remove_child({sep_it.base(), path.end()}); } }; const CommandDesc add_hook_cmd = { "hook", nullptr, "hook : add in " "to be executed on hook \n" "scope can be: \n" " * global: hook is executed for any buffer or window\n" " * buffer: hook is executed only for the current buffer\n" " (and any window for that buffer)\n" " * window: hook is executed only for the current window\n", ParameterDesc{ { { "group", { true, "set hook group, see rmhooks" } } }, ParameterDesc::Flags::None, 4, 4 }, CommandFlags::None, CommandHelper{}, [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) -> Completions { if (token_to_complete == 0) return { 0_byte, params[0].length(), complete(params[0], pos_in_token, scopes) }; else if (token_to_complete == 3) { auto& cm = CommandManager::instance(); return cm.complete(context, flags, params[3], pos_in_token); } return {}; }, [](const ParametersParser& parser, Context& context) { Regex regex(parser[2].begin(), parser[2].end(), Regex::optimize | Regex::nosubs | Regex::ECMAScript); const String& command = parser[3]; auto hook_func = [=](StringView param, Context& context) { if (context.user_hooks_support().is_disabled()) return; // Do not let hooks touch prompt history ScopedDisable disable_history{context.history_support()}; if (regex_match(param.begin(), param.end(), regex)) CommandManager::instance().execute(command, context, {}, { { "hook_param", param.str() } }); }; auto group = parser.get_switch("group").value_or(StringView{}); get_scope(parser[0], context).hooks().add_hook(parser[1], group.str(), hook_func); } }; const CommandDesc rm_hook_cmd = { "rmhooks", nullptr, "rmhooks : remove all hooks whose group is ", ParameterDesc{ {}, ParameterDesc::Flags::None, 2, 2 }, CommandFlags::None, CommandHelper{}, [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) -> Completions { if (token_to_complete == 0) return { 0_byte, params[0].length(), complete(params[0], pos_in_token, scopes) }; else if (token_to_complete == 1) { if (auto scope = get_scope_ifp(params[0], context)) return { 0_byte, params[0].length(), scope->hooks().complete_hook_group(params[1], pos_in_token) }; } return {}; }, [](const ParametersParser& parser, Context& context) { get_scope(parser[0], context).hooks().remove_hooks(parser[1]); } }; Vector params_to_shell(const ParametersParser& parser) { Vector vars; for (size_t i = 0; i < parser.positional_count(); ++i) vars.push_back(parser[i]); return vars; } void define_command(const ParametersParser& parser, Context& context) { const String& cmd_name = parser[0]; auto& cm = CommandManager::instance(); if (cm.command_defined(cmd_name) and not parser.get_switch("allow-override")) throw runtime_error(format("command '{}' already defined", cmd_name)); CommandFlags flags = CommandFlags::None; if (parser.get_switch("hidden")) flags = CommandFlags::Hidden; const String& commands = parser[1]; Command cmd; ParameterDesc desc; if (parser.get_switch("shell-params")) { desc = ParameterDesc{ {}, ParameterDesc::Flags::SwitchesAsPositional }; cmd = [=](const ParametersParser& parser, Context& context) { CommandManager::instance().execute(commands, context, params_to_shell(parser)); }; } else { desc = ParameterDesc{ {}, ParameterDesc::Flags::SwitchesAsPositional, 0, 0 }; cmd = [=](const ParametersParser& parser, Context& context) { CommandManager::instance().execute(commands, context); }; } CommandCompleter completer; if (parser.get_switch("file-completion")) { completer = [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { const String& prefix = params[token_to_complete]; auto& ignored_files = context.options()["ignored_files"].get(); return Completions{ 0_byte, pos_in_token, complete_filename(prefix, ignored_files, pos_in_token) }; }; } else if (parser.get_switch("client-completion")) { completer = [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { const String& prefix = params[token_to_complete]; auto& cm = ClientManager::instance(); return Completions{ 0_byte, pos_in_token, cm.complete_client_name(prefix, pos_in_token) }; }; } else if (parser.get_switch("buffer-completion")) { completer = [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { const String& prefix = params[token_to_complete]; return Completions{ 0_byte, pos_in_token, complete_buffer_name(prefix, pos_in_token) }; }; } else if (auto shell_cmd_opt = parser.get_switch("shell-completion")) { String shell_cmd = shell_cmd_opt->str(); completer = [=](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) { if (flags == CompletionFlags::Fast) // no shell on fast completion return Completions{}; EnvVarMap vars = { { "token_to_complete", to_string(token_to_complete) }, { "pos_in_token", to_string(pos_in_token) } }; String output = ShellManager::instance().eval(shell_cmd, context, {}, ShellManager::Flags::WaitForStdout, params, vars).first; return Completions{ 0_byte, pos_in_token, split(output, '\n', 0) }; }; } auto docstring = parser.get_switch("docstring").value_or(StringView{}); cm.register_command(cmd_name, cmd, docstring.str(), desc, flags, CommandHelper{}, completer); } const CommandDesc define_command_cmd = { "def", nullptr, "def : define a command executing ", ParameterDesc{ { { "shell-params", { false, "pass parameters to each shell escape as $0..$N" } }, { "allow-override", { false, "allow overriding an existing command" } }, { "hidden", { false, "do not display the command in completion candidates" } }, { "docstring", { true, "define the documentation string for command" } }, { "file-completion", { false, "complete parameters using filename completion" } }, { "client-completion", { false, "complete parameters using client name completion" } }, { "buffer-completion", { false, "complete parameters using buffer name completion" } }, { "shell-completion", { true, "complete the parameters using the given shell-script" } } }, ParameterDesc::Flags::None, 2, 2 }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, define_command }; const CommandDesc alias_cmd = { "alias", nullptr, "alias : alias to in \n", ParameterDesc{{}, ParameterDesc::Flags::None, 3, 3}, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { AliasRegistry& aliases = get_scope(parser[0], context).aliases(); aliases.add_alias(parser[1], parser[2]); } }; const CommandDesc unalias_cmd = { "unalias", nullptr, "unalias []: remove from \n" "If is specified, remove only if its value is ", ParameterDesc{{}, ParameterDesc::Flags::None, 2, 3}, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { AliasRegistry& aliases = get_scope(parser[0], context).aliases(); if (parser.positional_count() == 3 and aliases[parser[1]] != parser[2]) return; aliases.remove_alias(parser[1]); } }; const CommandDesc echo_cmd = { "echo", nullptr, "echo ...: display given parameters in the status line", ParameterDesc{ { { "color", { true, "set message color" } }, { "debug", { false, "write to debug buffer instead of status line" } } }, ParameterDesc::Flags::SwitchesOnlyAtStart }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { String message = join(parser, ' ', false); if (parser.get_switch("debug")) write_to_debug_buffer(message); else { auto face = get_face(parser.get_switch("color").value_or("StatusLine").str()); context.print_status({ std::move(message), face } ); } } }; const CommandDesc debug_cmd = { "debug", nullptr, "debug : write some debug informations in the debug buffer\n" "existing commands: info, buffers, options, memory, shared-strings", ParameterDesc{{}, ParameterDesc::Flags::SwitchesOnlyAtStart, 1}, CommandFlags::None, CommandHelper{}, PerArgumentCommandCompleter({ [](const Context& context, CompletionFlags flags, const String& prefix, ByteCount cursor_pos) -> Completions { auto c = {"info", "buffers", "options", "memory", "shared-strings"}; return { 0_byte, cursor_pos, complete(prefix, cursor_pos, c) }; } }), [](const ParametersParser& parser, Context& context) { if (parser[0] == "info") { write_to_debug_buffer(format("pid: {}", getpid())); write_to_debug_buffer(format("session: {}", Server::instance().session())); } else if (parser[0] == "buffers") { write_to_debug_buffer("Buffers:"); for (auto& buffer : BufferManager::instance()) write_to_debug_buffer(buffer->debug_description()); } else if (parser[0] == "options") { write_to_debug_buffer("Options:"); for (auto& option : context.options().flatten_options()) write_to_debug_buffer(format(" * {}: {}", option->name(), option->get_as_string())); } else if (parser[0] == "memory") { auto total = 0; write_to_debug_buffer("Memory usage:"); for (int domain = 0; domain < (int)MemoryDomain::Count; ++domain) { size_t count = domain_allocated_bytes[domain]; total += count; write_to_debug_buffer(format(" {}: {}", domain_name((MemoryDomain)domain), count)); } write_to_debug_buffer(format(" Total: {}", total)); #if defined(__GLIBC__) || defined(__CYGWIN__) write_to_debug_buffer(format(" Malloced: {}", mallinfo().uordblks)); #endif } else if (parser[0] == "shared-strings") { StringRegistry::instance().debug_stats(); } else throw runtime_error(format("unknown debug command '{}'", parser[0])); } }; const CommandDesc source_cmd = { "source", nullptr, "source : execute commands contained in ", single_name_param, CommandFlags::None, CommandHelper{}, filename_completer, [](const ParametersParser& parser, Context& context) { String file_content = read_file(parse_filename(parser[0]), true); try { CommandManager::instance().execute(file_content, context); } catch (Kakoune::runtime_error& err) { write_to_debug_buffer(format("{}:{}", parser[0], err.what())); throw; } } }; const CommandDesc set_option_cmd = { "set", nullptr, "set : set option in to ", ParameterDesc{ { { "add", { false, "add to option rather than replacing it" } } }, ParameterDesc::Flags::SwitchesOnlyAtStart, 3, 3 }, CommandFlags::None, [](const Context& context, CommandParameters params) -> String { if (params.size() < 2) return ""; try { OptionManager& options = get_scope(params[0], context).options(); const String& docstring = options[params[1]].docstring(); if (not docstring.empty()) return format("{}: {}", params[1], docstring); } catch (runtime_error&) {} return ""; }, [](const Context& context, CompletionFlags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) -> Completions { if (token_to_complete == 0) return { 0_byte, params[0].length(), complete(params[0], pos_in_token, scopes) }; else if (token_to_complete == 1) { OptionManager& options = get_scope(params[0], context).options(); return { 0_byte, params[1].length(), options.complete_option_name(params[1], pos_in_token) }; } else if (token_to_complete == 2 and GlobalScope::instance().option_registry().option_exists(params[1])) { OptionManager& options = get_scope(params[0], context).options(); String val = options[params[1]].get_as_string(); if (prefix_match(val, params[2])) return { 0_byte, params[2].length(), { std::move(val) } }; } return Completions{}; }, [](const ParametersParser& parser, Context& context) { Option& opt = get_scope(parser[0], context).options().get_local_option(parser[1]); if (parser.get_switch("add")) opt.add_from_string(parser[2]); else opt.set_from_string(parser[2]); } }; const CommandDesc declare_option_cmd = { "decl", nullptr, "decl [value]: declare option of type .\n" "set its initial value to if given and the option did not exist\n" "Available types:\n" " int: integer\n" " bool: boolean (true/false or yes/no)\n" " str: character string\n" " regex: regular expression\n" " int-list: list of integers\n" " str-list: list of character strings\n" " line-flag-list: list of line flags\n", ParameterDesc{ { { "hidden", { false, "do not display option name when completing" } }, { "docstring", { true, "specify option description" } } }, ParameterDesc::Flags::SwitchesOnlyAtStart, 2, 3 }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { Option* opt = nullptr; OptionFlags flags = OptionFlags::None; if (parser.get_switch("hidden")) flags = OptionFlags::Hidden; auto docstring = parser.get_switch("docstring").value_or(StringView{}).str(); OptionsRegistry& reg = GlobalScope::instance().option_registry(); if (parser[0] == "int") opt = ®.declare_option(parser[1], docstring, 0, flags); else if (parser[0] == "bool") opt = ®.declare_option(parser[1], docstring, 0, flags); else if (parser[0] == "str") opt = ®.declare_option(parser[1], docstring, "", flags); else if (parser[0] == "regex") opt = ®.declare_option(parser[1], docstring, Regex{}, flags); else if (parser[0] == "int-list") opt = ®.declare_option>(parser[1], docstring, {}, flags); else if (parser[0] == "str-list") opt = ®.declare_option>(parser[1], docstring, {}, flags); else if (parser[0] == "line-flag-list") opt = ®.declare_option>(parser[1], docstring, {}, flags); else throw runtime_error(format("unknown type {}", parser[0])); if (parser.positional_count() == 3) opt->set_from_string(parser[2]); } }; KeymapMode parse_keymap_mode(const String& str) { if (prefix_match("normal", str)) return KeymapMode::Normal; if (prefix_match("insert", str)) return KeymapMode::Insert; if (prefix_match("menu", str)) return KeymapMode::Menu; if (prefix_match("prompt", str)) return KeymapMode::Prompt; if (prefix_match("goto", str)) return KeymapMode::Goto; if (prefix_match("view", str)) return KeymapMode::View; if (prefix_match("user", str)) return KeymapMode::User; throw runtime_error(format("unknown keymap mode '{}'", str)); } const CommandDesc map_key_cmd = { "map", nullptr, "map : map to in given mode in .\n" " can be:\n" " normal\n" " insert\n" " menu\n" " prompt\n" " goto\n" " view\n" " user\n", ParameterDesc{{}, ParameterDesc::Flags::None, 4, 4}, CommandFlags::None, CommandHelper{}, [](const Context& context, CompletionFlags flags, CommandParameters params, size_t token_to_complete, ByteCount pos_in_token) -> Completions { if (token_to_complete == 0) return { 0_byte, params[0].length(), complete(params[0], pos_in_token, scopes) }; if (token_to_complete == 1) { constexpr const char* modes[] = { "normal", "insert", "menu", "prompt", "goto", "view", "user" }; return { 0_byte, params[1].length(), complete(params[1], pos_in_token, modes) }; } return {}; }, [](const ParametersParser& parser, Context& context) { KeymapManager& keymaps = get_scope(parser[0], context).keymaps(); KeymapMode keymap_mode = parse_keymap_mode(parser[1]); KeyList key = parse_keys(parser[2]); if (key.size() != 1) throw runtime_error("only a single key can be mapped"); KeyList mapping = parse_keys(parser[3]); keymaps.map_key(key[0], keymap_mode, std::move(mapping)); } }; const ParameterDesc context_wrap_params = { { { "client", { true, "run in given client context" } }, { "try-client", { true, "run in given client context if it exists, or else in the current one" } }, { "buffer", { true, "run in a disposable context for each given buffer in the comma separated list argument" } }, { "draft", { false, "run in a disposable context" } }, { "no-hooks", { false, "disable hooks" } }, { "with-maps", { false, "use user defined key mapping when executing keys" } }, { "itersel", { false, "run once for each selection with that selection as the only one" } } }, ParameterDesc::Flags::SwitchesOnlyAtStart, 1 }; template struct DisableOption { DisableOption(Context& context, const char* name) : m_option(context.options()[name]), m_prev_value(m_option.get()) { m_option.set(T{}); } ~DisableOption() { m_option.set(m_prev_value); } private: Option& m_option; T m_prev_value; }; template void context_wrap(const ParametersParser& parser, Context& context, Func func) { // Disable these options to avoid costly code paths (and potential screen // redraws) That are useful only in interactive contexts. DisableOption disable_autoinfo(context, "autoinfo"); DisableOption disable_autoshowcompl(context, "autoshowcompl"); DisableOption disable_incsearch(context, "incsearch"); const bool disable_hooks = parser.get_switch("no-hooks") or context.user_hooks_support().is_disabled(); const bool disable_keymaps = not parser.get_switch("with-maps"); ClientManager& cm = ClientManager::instance(); if (auto bufnames = parser.get_switch("buffer")) { auto context_wrap_for_buffer = [&](Buffer& buffer) { InputHandler input_handler{{ buffer, Selection{} }, Context::Flags::Transient}; Context& c = input_handler.context(); // Propagate user hooks disabled status to the temporary context ScopedDisable hook_disable(c.user_hooks_support(), disable_hooks); ScopedDisable keymaps_disable(c.keymaps_support(), disable_keymaps); ScopedDisable disable_history{c.history_support()}; func(parser, c); }; if (*bufnames == "*") for (auto buffer : BufferManager::instance()) context_wrap_for_buffer(*buffer); else for (auto& name : split(*bufnames, ',')) context_wrap_for_buffer(BufferManager::instance().get_buffer(name)); return; } Context* real_context = &context; if (auto client_name = parser.get_switch("client")) real_context = &cm.get_client(*client_name).context(); else if (auto client_name = parser.get_switch("try-client")) { if (Client* client = cm.get_client_ifp(*client_name)) real_context = &client->context(); } if (parser.get_switch("draft")) { InputHandler input_handler(real_context->selections(), Context::Flags::Transient, real_context->name()); Context& c = input_handler.context(); // We do not want this draft context to commit undo groups if the real one is // going to commit the whole thing later if (real_context->is_editing()) c.disable_undo_handling(); ScopedDisable hook_disable(c.user_hooks_support(), disable_hooks); ScopedDisable keymaps_disable(c.keymaps_support(), disable_keymaps); ScopedDisable disable_history{c.history_support()}; if (parser.get_switch("itersel")) { SelectionList sels{real_context->selections()}; ScopedEdition edition{c}; for (auto& sel : sels) { c.selections_write_only() = SelectionList{ sels.buffer(), sel, sels.timestamp() }; c.selections().update(); func(parser, c); if (&sels.buffer() != &c.buffer()) throw runtime_error("the buffer has changed while iterating on selections"); } } else func(parser, c); } else { if (parser.get_switch("itersel")) throw runtime_error("-itersel makes no sense without -draft"); ScopedDisable hook_disable(real_context->user_hooks_support(), disable_hooks); ScopedDisable keymaps_disable(real_context->keymaps_support(), disable_keymaps); ScopedDisable disable_history{real_context->history_support()}; func(parser, *real_context); } } const CommandDesc exec_string_cmd = { "exec", nullptr, "exec : execute given keys as if entered by user", context_wrap_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { context_wrap(parser, context, [](const ParametersParser& parser, Context& context) { KeyList keys; for (auto& param : parser) { KeyList param_keys = parse_keys(param); keys.insert(keys.end(), param_keys.begin(), param_keys.end()); } exec_keys(keys, context); }); } }; const CommandDesc eval_string_cmd = { "eval", nullptr, "eval ...: execute commands as if entered by user", context_wrap_params, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { context_wrap(parser, context, [](const ParametersParser& parser, Context& context) { String command = join(parser, ' ', false); CommandManager::instance().execute(command, context); }); } }; const CommandDesc prompt_cmd = { "prompt", nullptr, "prompt : prompt the use to enter a text string " "stores it in and then executes ", ParameterDesc{ { { "init", { true, "set initial prompt content" } } }, ParameterDesc::Flags::None, 3, 3 }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& params, Context& context) { if (params[1].length() != 1) throw runtime_error("register name should be a single character"); const char reg = params[1][0_byte]; const String& command = params[2]; auto initstr = params.get_switch("init").value_or(StringView{}); context.input_handler().prompt( params[0], initstr.str(), get_face("Prompt"), Completer{}, [=](StringView str, PromptEvent event, Context& context) { if (event != PromptEvent::Validate) return; RegisterManager::instance()[reg] = ConstArrayView(str.str()); CommandManager::instance().execute(command, context); }); } }; const CommandDesc menu_cmd = { "menu", nullptr, "menu ...: display a " "menu and execute commands for the selected item", ParameterDesc{ { { "auto-single", { false, "instantly validate if only one item is available" } }, { "select-cmds", { false, "each item specify an additional command to run when selected" } } } }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { const bool with_select_cmds = (bool)parser.get_switch("select-cmds"); const size_t modulo = with_select_cmds ? 3 : 2; const size_t count = parser.positional_count(); if (count == 0 or (count % modulo) != 0) throw wrong_argument_count(); if (count == modulo and parser.get_switch("auto-single")) { CommandManager::instance().execute(parser[1], context); return; } Vector choices; Vector commands; Vector select_cmds; for (int i = 0; i < count; i += modulo) { choices.push_back(parser[i]); commands.push_back(parser[i+1]); if (with_select_cmds) select_cmds.push_back(parser[i+2]); } context.input_handler().menu(choices, [=](int choice, MenuEvent event, Context& context) { if (event == MenuEvent::Validate and choice >= 0 and choice < commands.size()) CommandManager::instance().execute(commands[choice], context); if (event == MenuEvent::Select and choice >= 0 and choice < select_cmds.size()) CommandManager::instance().execute(select_cmds[choice], context); }); } }; const CommandDesc onkey_cmd = { "onkey", nullptr, "onkey : wait for next user key, store it in and execute ", ParameterDesc{ {}, ParameterDesc::Flags::None, 2, 2 }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { String reg = parser[0]; String command = parser[1]; context.input_handler().on_next_key(KeymapMode::None, [=](Key key, Context& context) { RegisterManager::instance()[reg] = key_to_str(key); CommandManager::instance().execute(command, context); }); } }; const CommandDesc info_cmd = { "info", nullptr, "info ...: display an info box with the params as content", ParameterDesc{ { { "anchor", { true, "set info anchoring ." } }, { "placement", { true, "set placement relative to anchor (above, below)" } }, { "title", { true, "set info title" } } }, ParameterDesc::Flags::None, 0, 1 }, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { context.ui().info_hide(); if (parser.positional_count() > 0) { InfoStyle style = InfoStyle::Prompt; CharCoord pos; if (auto anchor = parser.get_switch("anchor")) { auto dot = find(*anchor, '.'); if (dot == anchor->end()) throw runtime_error("expected . for anchor"); ByteCoord coord{str_to_int({anchor->begin(), dot})-1, str_to_int({dot+1, anchor->end()})-1}; pos = context.window().display_position(coord); style = InfoStyle::Inline; if (auto placement = parser.get_switch("placement")) { if (*placement == "above") style = InfoStyle::InlineAbove; else if (*placement == "below") style = InfoStyle::InlineBelow; else throw runtime_error(format("invalid placement '{}'", *placement)); } } auto title = parser.get_switch("title").value_or(StringView{}); context.ui().info_show(title, parser[0], pos, get_face("Information"), style); } } }; const CommandDesc try_catch_cmd = { "try", nullptr, "try [catch ]: execute in current context.\n" "if an error is raised and is specified, execute it; " "The error is not propagated further.", ParameterDesc{{}, ParameterDesc::Flags::None, 1, 3}, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { if (parser.positional_count() == 2) throw wrong_argument_count(); const bool do_catch = parser.positional_count() == 3; if (do_catch and parser[1] != "catch") throw runtime_error("usage: try [catch ]"); CommandManager& command_manager = CommandManager::instance(); try { command_manager.execute(parser[0], context); } catch (Kakoune::runtime_error& e) { if (do_catch) command_manager.execute(parser[2], context); } } }; static Completions complete_face(const Context&, CompletionFlags flags, const String& prefix, ByteCount cursor_pos) { return {0_byte, cursor_pos, FaceRegistry::instance().complete_alias_name(prefix, cursor_pos)}; } const CommandDesc face_cmd = { "face", nullptr, "face : set face to refer to \n" "\n" "facespec format is [,][+]\n" "colors are either a color name, or rgb:###### values.\n" "attributes is a combination of:\n" " u: underline, r: reverse, b: bold, B: blink, d: dim\n" "facespec can as well just be the name of another face" , ParameterDesc{{}, ParameterDesc::Flags::None, 2, 2}, CommandFlags::None, CommandHelper{}, PerArgumentCommandCompleter({ complete_face, complete_face }), [](const ParametersParser& parser, Context& context) { FaceRegistry::instance().register_alias(parser[0], parser[1], true); } }; const CommandDesc set_client_name_cmd = { "nameclient", "nc", "nameclient : set current client name to ", single_name_param, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { if (ClientManager::instance().validate_client_name(parser[0])) context.set_name(parser[0]); else if (context.name() != parser[0]) throw runtime_error(format("client name '{}' is not unique", parser[0])); } }; const CommandDesc set_register_cmd = { "reg", nullptr, "reg : set register to ", ParameterDesc{{}, ParameterDesc::Flags::None, 2, 2}, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { RegisterManager::instance()[parser[0]] = ConstArrayView(parser[1]); } }; const CommandDesc select_cmd = { "select", nullptr, "select : select given selections", ParameterDesc{{}, ParameterDesc::Flags::None, 1, 1}, CommandFlags::None, CommandHelper{}, CommandCompleter{}, [](const ParametersParser& parser, Context& context) { context.selections_write_only() = selection_list_from_string(context.buffer(), parser[0]); } }; const CommandDesc change_working_directory_cmd = { "cd", nullptr, "cd : change server working directory to ", single_name_param, CommandFlags::None, CommandHelper{}, filename_completer, [](const ParametersParser& parser, Context&) { if (chdir(parse_filename(parser[0]).c_str()) != 0) throw runtime_error(format("cannot change to directory '{}'", parser[0])); } }; class RegisterRestorer { public: RegisterRestorer(char name, const Context& context) : m_name(name) { ConstArrayView save = RegisterManager::instance()[name].values(context); m_save = Vector(save.begin(), save.end()); } ~RegisterRestorer() { RegisterManager::instance()[m_name] = m_save; } private: Vector m_save; char m_name; }; } void exec_keys(ConstArrayView keys, Context& context) { RegisterRestorer quote('"', context); RegisterRestorer slash('/', context); ScopedEdition edition(context); for (auto& key : keys) context.input_handler().handle_key(key); } void register_commands() { CommandManager& cm = CommandManager::instance(); cm.register_command("nop", [](const ParametersParser&, Context&){}, "do nothing", {}); auto register_command = [&](const CommandDesc& c) { cm.register_command(c.name, c.func, c.docstring, c.params, c.flags, c.helper, c.completer); if (c.alias) GlobalScope::instance().aliases().add_alias(c.alias, c.name); }; register_command(edit_cmd); register_command(force_edit_cmd); register_command(write_cmd); register_command(writeall_cmd); register_command(writeall_quit_cmd); register_command(quit_cmd); register_command(force_quit_cmd); register_command(write_quit_cmd); register_command(force_write_quit_cmd); register_command(buffer_cmd); register_command(buffernext_cmd); register_command(bufferprev_cmd); register_command(delbuf_cmd); register_command(force_delbuf_cmd); register_command(namebuf_cmd); register_command(add_highlighter_cmd); register_command(rm_highlighter_cmd); register_command(add_hook_cmd); register_command(rm_hook_cmd); register_command(define_command_cmd); register_command(alias_cmd); register_command(unalias_cmd); register_command(echo_cmd); register_command(debug_cmd); register_command(source_cmd); register_command(set_option_cmd); register_command(declare_option_cmd); register_command(map_key_cmd); register_command(exec_string_cmd); register_command(eval_string_cmd); register_command(prompt_cmd); register_command(menu_cmd); register_command(onkey_cmd); register_command(info_cmd); register_command(try_catch_cmd); register_command(face_cmd); register_command(set_client_name_cmd); register_command(set_register_cmd); register_command(select_cmd); register_command(change_working_directory_cmd); } }