#include "window.hh" #include "buffer.hh" #include "shell_manager.hh" #include "commands.hh" #include "command_manager.hh" #include "buffer_manager.hh" #include "register_manager.hh" #include "selectors.hh" #include "assert.hh" #include "debug.hh" #include "highlighters.hh" #include "highlighter_registry.hh" #include "filters.hh" #include "filter_registry.hh" #include "hook_manager.hh" #include "option_manager.hh" #include "event_manager.hh" #include "context.hh" #include "ncurses.hh" #include "string.hh" #include "file.hh" #include "color_registry.hh" #include "remote.hh" #if defined(__APPLE__) #include #endif #include #include #include #include #include using namespace Kakoune; using namespace std::placeholders; namespace Kakoune { bool quit_requested = false; template void do_insert(Context& context) { context.input_handler().insert(context, mode); } void do_repeat_insert(Context& context) { context.input_handler().repeat_last_insert(context); } template void do_go(Context& context) { int count = context.numeric_param(); if (count != 0) { BufferIterator target = context.editor().buffer().iterator_at_line_begin(count-1); context.editor().select(target); if (context.has_window()) context.window().center_selection(); } else context.input_handler().on_next_key([](const Key& key, Context& context) { if (key.modifiers != Key::Modifiers::None) return; Editor& editor = context.editor(); switch (key.key) { case 'g': case 't': editor.select(editor.buffer().begin()); break; case 'l': case 'L': editor.select(select_to_eol, mode); break; case 'h': case 'H': editor.select(select_to_eol_reverse, mode); break; case 'b': { const Buffer& buf = editor.buffer(); editor.select(buf.iterator_at_line_begin(buf.line_count() - 1)); break; } } }); } void do_replace_with_char(Context& context) { context.input_handler().on_next_key([](const Key& key, Context& context) { context.editor().insert(String() + key.key, InsertMode::Replace); }); } void do_command(Context& context) { context.input_handler().prompt( ":", std::bind(&CommandManager::complete, &CommandManager::instance(), _1, _2, _3), [](const String& cmdline, Context& context) { CommandManager::instance().execute(cmdline, context); }, context); } void do_pipe(Context& context) { context.input_handler().prompt("|", complete_nothing, [](const String& cmdline, Context& context) { Editor& editor = context.editor(); std::vector strings; for (auto& sel : const_cast(context.editor()).selections()) strings.push_back(ShellManager::instance().pipe(String(sel.begin(), sel.end()), cmdline, context, {}, {})); editor.insert(strings, InsertMode::Replace); }, context); } template void do_search(Context& context) { context.input_handler().prompt("/", complete_nothing, [](const String& str, Context& context) { String ex = str; if (ex.empty()) ex = RegisterManager::instance()['/'].values(context)[0]; else RegisterManager::instance()['/'] = ex; context.editor().select(std::bind(select_next_match, _1, ex), mode); }, context); } template void do_search_next(Context& context) { const String& ex = RegisterManager::instance()['/'].values(context)[0]; if (not ex.empty()) context.editor().select(std::bind(select_next_match, _1, ex), mode); else context.print_status("no search pattern"); } void do_yank(Context& context) { RegisterManager::instance()['"'] = context.editor().selections_content(); } void do_erase(Context& context) { RegisterManager::instance()['"'] = context.editor().selections_content(); context.editor().erase(); } void do_change(Context& context) { RegisterManager::instance()['"'] = context.editor().selections_content(); do_insert(context); } static InsertMode adapt_for_linewise(InsertMode mode) { if (mode == InsertMode::Append) return InsertMode::AppendAtLineEnd; if (mode == InsertMode::Insert) return InsertMode::InsertAtLineBegin; assert(false); return InsertMode::Insert; } template void do_paste(Context& context) { Editor& editor = context.editor(); int count = context.numeric_param(); auto strings = RegisterManager::instance()['"'].values(context); InsertMode mode = insert_mode; if (count == 0) { for (auto& str : strings) { if (not str.empty() and str.back() == '\n') { mode = adapt_for_linewise(mode); break; } } editor.insert(strings, mode); } else if (count <= strings.size()) { auto& str = strings[count-1]; if (not str.empty() and str.back() == '\n') mode = adapt_for_linewise(mode); editor.insert(str, mode); } } void do_select_regex(Context& context) { context.input_handler().prompt("select: ", complete_nothing, [](const String& ex, Context& context) { context.editor().multi_select(std::bind(select_all_matches, _1, ex)); }, context); } void do_split_regex(Context& context) { context.input_handler().prompt("select: ", complete_nothing, [](const String& ex, Context& context) { context.editor().multi_select(std::bind(split_selection, _1, ex)); }, context); } void do_join(Context& context) { Editor& editor = context.editor(); editor.select(select_whole_lines); editor.select(select_to_eol, SelectMode::Extend); editor.multi_select(std::bind(select_all_matches, _1, "\n\\h*")); editor.insert(" ", InsertMode::Replace); editor.clear_selections(); editor.move_selections(-1_char); } template void do_select_object(Context& context) { context.input_handler().on_next_key( [](const Key& key, Context& context) { typedef std::function Selector; static const std::unordered_map key_to_selector = { { { Key::Modifiers::None, '(' }, std::bind(select_surrounding, _1, std::pair{ '(', ')' }, inner) }, { { Key::Modifiers::None, ')' }, std::bind(select_surrounding, _1, std::pair{ '(', ')' }, inner) }, { { Key::Modifiers::None, 'b' }, std::bind(select_surrounding, _1, std::pair{ '(', ')' }, inner) }, { { Key::Modifiers::None, '{' }, std::bind(select_surrounding, _1, std::pair{ '{', '}' }, inner) }, { { Key::Modifiers::None, '}' }, std::bind(select_surrounding, _1, std::pair{ '{', '}' }, inner) }, { { Key::Modifiers::None, 'B' }, std::bind(select_surrounding, _1, std::pair{ '{', '}' }, inner) }, { { Key::Modifiers::None, '[' }, std::bind(select_surrounding, _1, std::pair{ '[', ']' }, inner) }, { { Key::Modifiers::None, ']' }, std::bind(select_surrounding, _1, std::pair{ '[', ']' }, inner) }, { { Key::Modifiers::None, '<' }, std::bind(select_surrounding, _1, std::pair{ '<', '>' }, inner) }, { { Key::Modifiers::None, '>' }, std::bind(select_surrounding, _1, std::pair{ '<', '>' }, inner) }, { { Key::Modifiers::None, 'w' }, std::bind(select_whole_word, _1, inner) }, { { Key::Modifiers::None, 'W' }, std::bind(select_whole_word, _1, inner) }, }; auto it = key_to_selector.find(key); if (it != key_to_selector.end()) context.editor().select(it->second); }); } template void do_scroll(Context& context) { static_assert(key == Key::PageUp or key == Key::PageDown, "do_scrool only implements PageUp and PageDown"); Window& window = context.window(); Buffer& buffer = context.buffer(); DisplayCoord 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_line_begin(position.line), buffer.iterator_at_line_end(position.line), position.column); window.select(cursor_pos); window.set_position(position); } template class Repeated { public: Repeated(T t) : m_func(t) {} void operator() (Context& context) { int count = context.numeric_param(); do { m_func(context); } while(--count > 0); } private: T m_func; }; template Repeated repeated(T func) { return Repeated(func); } namespace SelectFlags { enum Type { None = 0, Reverse = 1, Inclusive = 2, Extend = 4 }; } template void select_to_next_char(Context& context) { int param = context.numeric_param(); context.input_handler().on_next_key([param](const Key& key, Context& context) { context.editor().select( std::bind(flags & SelectFlags::Reverse ? select_to_reverse : select_to, _1, key.key, param, flags & SelectFlags::Inclusive), flags & SelectFlags::Extend ? SelectMode::Extend : SelectMode::Replace); }); } String runtime_directory() { char buffer[2048]; #if defined(__linux__) ssize_t res = readlink("/proc/self/exe", buffer, 2048); assert(res != -1); buffer[res] = '\0'; #elif defined(__APPLE__) uint32_t bufsize = 2048; _NSGetExecutablePath(buffer, &bufsize); char* canonical_path = realpath(buffer, NULL); strncpy(buffer, canonical_path, 2048); free(canonical_path); #else # error "finding executable path is not implemented on this platform" #endif char* ptr = strrchr(buffer, '/'); if (not ptr) throw runtime_error("unable do determine runtime directory"); return String(buffer, ptr); } std::unordered_map> keymap = { { { Key::Modifiers::None, 'h' }, [](Context& context) { context.editor().move_selections(-CharCount(std::max(context.numeric_param(),1))); } }, { { Key::Modifiers::None, 'j' }, [](Context& context) { context.editor().move_selections( LineCount(std::max(context.numeric_param(),1))); } }, { { Key::Modifiers::None, 'k' }, [](Context& context) { context.editor().move_selections(-LineCount(std::max(context.numeric_param(),1))); } }, { { Key::Modifiers::None, 'l' }, [](Context& context) { context.editor().move_selections( CharCount(std::max(context.numeric_param(),1))); } }, { { Key::Modifiers::None, 'H' }, [](Context& context) { context.editor().move_selections(-CharCount(std::max(context.numeric_param(),1)), SelectMode::Extend); } }, { { Key::Modifiers::None, 'J' }, [](Context& context) { context.editor().move_selections( LineCount(std::max(context.numeric_param(),1)), SelectMode::Extend); } }, { { Key::Modifiers::None, 'K' }, [](Context& context) { context.editor().move_selections(-LineCount(std::max(context.numeric_param(),1)), SelectMode::Extend); } }, { { Key::Modifiers::None, 'L' }, [](Context& context) { context.editor().move_selections( CharCount(std::max(context.numeric_param(),1)), SelectMode::Extend); } }, { { Key::Modifiers::None, 't' }, select_to_next_char }, { { Key::Modifiers::None, 'f' }, select_to_next_char }, { { Key::Modifiers::None, 'T' }, select_to_next_char }, { { Key::Modifiers::None, 'F' }, select_to_next_char }, { { Key::Modifiers::Alt, 't' }, select_to_next_char }, { { Key::Modifiers::Alt, 'f' }, select_to_next_char }, { { Key::Modifiers::Alt, 'T' }, select_to_next_char }, { { Key::Modifiers::Alt, 'F' }, select_to_next_char }, { { Key::Modifiers::None, 'd' }, do_erase }, { { Key::Modifiers::None, 'c' }, do_change }, { { Key::Modifiers::None, 'i' }, do_insert }, { { Key::Modifiers::None, 'I' }, do_insert }, { { Key::Modifiers::None, 'a' }, do_insert }, { { Key::Modifiers::None, 'A' }, do_insert }, { { Key::Modifiers::None, 'o' }, do_insert }, { { Key::Modifiers::None, 'O' }, do_insert }, { { Key::Modifiers::None, 'r' }, do_replace_with_char }, { { Key::Modifiers::None, 'g' }, do_go }, { { Key::Modifiers::None, 'G' }, do_go }, { { Key::Modifiers::None, 'y' }, do_yank }, { { Key::Modifiers::None, 'p' }, do_paste }, { { Key::Modifiers::None, 'P' }, do_paste }, { { Key::Modifiers::Alt, 'p' }, do_paste }, { { Key::Modifiers::None, 's' }, do_select_regex }, { { Key::Modifiers::None, '.' }, do_repeat_insert }, { { Key::Modifiers::None, '%' }, [](Context& context) { context.editor().clear_selections(); context.editor().select(select_whole_buffer); } }, { { Key::Modifiers::None, ':' }, do_command }, { { Key::Modifiers::None, '|' }, do_pipe }, { { Key::Modifiers::None, ' ' }, [](Context& context) { int count = context.numeric_param(); if (count == 0) context.editor().clear_selections(); else context.editor().keep_selection(count-1); } }, { { Key::Modifiers::Alt, ' ' }, [](Context& context) { int count = context.numeric_param(); if (count == 0) context.editor().clear_selections(); else context.editor().remove_selection(count-1); } }, { { Key::Modifiers::None, 'w' }, repeated([](Context& context) { context.editor().select(select_to_next_word); }) }, { { Key::Modifiers::None, 'e' }, repeated([](Context& context) { context.editor().select(select_to_next_word_end); }) }, { { Key::Modifiers::None, 'b' }, repeated([](Context& context) { context.editor().select(select_to_previous_word); }) }, { { Key::Modifiers::None, 'W' }, repeated([](Context& context) { context.editor().select(select_to_next_word, SelectMode::Extend); }) }, { { Key::Modifiers::None, 'E' }, repeated([](Context& context) { context.editor().select(select_to_next_word_end, SelectMode::Extend); }) }, { { Key::Modifiers::None, 'B' }, repeated([](Context& context) { context.editor().select(select_to_previous_word, SelectMode::Extend); }) }, { { Key::Modifiers::None, 'x' }, repeated([](Context& context) { context.editor().select(select_line, SelectMode::Replace); }) }, { { Key::Modifiers::None, 'X' }, repeated([](Context& context) { context.editor().select(select_line, SelectMode::Extend); }) }, { { Key::Modifiers::None, 'm' }, [](Context& context) { context.editor().select(select_matching); } }, { { Key::Modifiers::None, 'M' }, [](Context& context) { context.editor().select(select_matching, SelectMode::Extend); } }, { { Key::Modifiers::None, '/' }, do_search }, { { Key::Modifiers::None, '?' }, do_search }, { { Key::Modifiers::None, 'n' }, do_search_next }, { { Key::Modifiers::None, 'N' }, do_search_next }, { { Key::Modifiers::None, 'u' }, repeated([](Context& context) { if (not context.editor().undo()) { context.print_status("nothing left to undo"); } }) }, { { Key::Modifiers::None, 'U' }, repeated([](Context& context) { if (not context.editor().redo()) { context.print_status("nothing left to redo"); } }) }, { { Key::Modifiers::Alt, 'i' }, do_select_object }, { { Key::Modifiers::Alt, 'a' }, do_select_object }, { { Key::Modifiers::Alt, 'w' }, repeated([](Context& context) { context.editor().select(select_to_next_word); }) }, { { Key::Modifiers::Alt, 'e' }, repeated([](Context& context) { context.editor().select(select_to_next_word_end); }) }, { { Key::Modifiers::Alt, 'b' }, repeated([](Context& context) { context.editor().select(select_to_previous_word); }) }, { { Key::Modifiers::Alt, 'W' }, repeated([](Context& context) { context.editor().select(select_to_next_word, SelectMode::Extend); }) }, { { Key::Modifiers::Alt, 'E' }, repeated([](Context& context) { context.editor().select(select_to_next_word_end, SelectMode::Extend); }) }, { { Key::Modifiers::Alt, 'B' }, repeated([](Context& context) { context.editor().select(select_to_previous_word, SelectMode::Extend); }) }, { { Key::Modifiers::Alt, 'l' }, repeated([](Context& context) { context.editor().select(select_to_eol, SelectMode::Replace); }) }, { { Key::Modifiers::Alt, 'L' }, repeated([](Context& context) { context.editor().select(select_to_eol, SelectMode::Extend); }) }, { { Key::Modifiers::Alt, 'h' }, repeated([](Context& context) { context.editor().select(select_to_eol_reverse, SelectMode::Replace); }) }, { { Key::Modifiers::Alt, 'H' }, repeated([](Context& context) { context.editor().select(select_to_eol_reverse, SelectMode::Extend); }) }, { { Key::Modifiers::Alt, 's' }, do_split_regex }, { { Key::Modifiers::Alt, 'j' }, do_join }, { { Key::Modifiers::Alt, 'x' }, [](Context& context) { context.editor().select(select_whole_lines); } }, { { Key::Modifiers::Alt, 'c' }, [](Context& context) { if (context.has_window()) context.window().center_selection(); } }, { { Key::Modifiers::None, Key::PageUp }, do_scroll }, { { Key::Modifiers::None, Key::PageDown }, do_scroll }, }; } void run_unit_tests(); void register_env_vars() { ShellManager& shell_manager = ShellManager::instance(); shell_manager.register_env_var("bufname", [](const String& name, const Context& context) { return context.buffer().name(); }); shell_manager.register_env_var("selection", [](const String& name, const Context& context) { return context.editor().selections_content().back(); }); shell_manager.register_env_var("runtime", [](const String& name, const Context& context) { return runtime_directory(); }); shell_manager.register_env_var("opt_.+", [](const String& name, const Context& context) { return context.option_manager()[name.substr(4_byte)].as_string(); }); shell_manager.register_env_var("reg_.+", [](const String& name, const Context& context) { return RegisterManager::instance()[name[4]].values(context)[0]; }); } void register_registers() { RegisterManager& register_manager = RegisterManager::instance(); register_manager.register_dynamic_register('%', [](const Context& context) { return std::vector(1, context.buffer().name()); }); register_manager.register_dynamic_register('.', [](const Context& context) { return context.editor().selections_content(); }); for (size_t i = 0; i < 10; ++i) { register_manager.register_dynamic_register('0'+i, [i](const Context& context) { std::vector result; for (auto& sel_and_cap : context.editor().selections()) result.emplace_back(i < sel_and_cap.captures.size() ? sel_and_cap.captures[i] : ""); return result; }); } } struct Client { std::unique_ptr ui; std::unique_ptr input_handler; std::unique_ptr context; Client(UserInterface* ui, Window& window) : ui(ui), input_handler(new InputHandler{}), context(new Context(*input_handler, window, *ui)) {} Client() {} }; Client create_local_client(const String& file) { Buffer* buffer = nullptr; UserInterface* ui = new NCursesUI{}; if (not file.empty()) { buffer = create_buffer_from_file(file); if (not buffer) { ui->print_status("new file " + file, -1); buffer = new Buffer(file, Buffer::Type::NewFile); } } else buffer = new Buffer("*scratch*", Buffer::Type::Scratch); Client client{ui, *buffer->get_or_create_window()}; InputHandler* input_handler = client.input_handler.get(); Context* context = client.context.get(); EventManager::instance().watch(0, [=](int) { try { input_handler->handle_next_input(*context); } catch (Kakoune::runtime_error& error) { ui->print_status(error.description(), -1); } }); context->draw_ifn(); return client; } std::vector clients; void setup_server() { auto filename = "/tmp/kak-" + int_to_str(getpid()); int listen_sock = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0); sockaddr_un addr; addr.sun_family = AF_UNIX; strncpy(addr.sun_path, filename.c_str(), sizeof(addr.sun_path) - 1); if (bind(listen_sock, (sockaddr*) &addr, sizeof(sockaddr_un)) == -1) throw runtime_error("unable to bind listen socket " + filename); if (listen(listen_sock, 4) == -1) throw runtime_error("unable to listen on socket " + filename); auto accepter = [=](int socket) { sockaddr_un client_addr; socklen_t client_addr_len = sizeof(sockaddr_un); int sock = accept(socket, (sockaddr*) &client_addr, &client_addr_len); if (sock == -1) throw runtime_error("accept failed"); auto& buffer = *BufferManager::instance().begin(); RemoteUI* ui = new RemoteUI{sock}; Client client{ui, *buffer->get_or_create_window()}; InputHandler* input_handler = client.input_handler.get(); Context* context = client.context.get(); EventManager::instance().watch(sock, [=](int) { try { input_handler->handle_next_input(*context); } catch (Kakoune::runtime_error& error) { ui->print_status(error.description(), -1); } }); clients.push_back(std::move(client)); }; EventManager::instance().watch(listen_sock, accepter); } RemoteClient* connect_to(const String& pid) { auto filename = "/tmp/kak-" + pid; int sock = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0); sockaddr_un addr; addr.sun_family = AF_UNIX; strncpy(addr.sun_path, filename.c_str(), sizeof(addr.sun_path) - 1); if (connect(sock, (sockaddr*)&addr, sizeof(addr.sun_path)) == -1) throw runtime_error("connect to " + filename + " failed"); NCursesUI* ui = new NCursesUI{}; RemoteClient* remote_client = new RemoteClient{sock, ui}; EventManager::instance().watch(sock, [=](int) { try { remote_client->process_next_message(); } catch (Kakoune::runtime_error& error) { ui->print_status(error.description(), -1); } }); EventManager::instance().watch(0, [=](int) { try { remote_client->write_next_key(); } catch (Kakoune::runtime_error& error) { ui->print_status(error.description(), -1); } }); return remote_client; } int main(int argc, char* argv[]) { try { EventManager event_manager; if (argc == 3 and String("-c") == argv[1]) { std::unique_ptr client(connect_to(argv[2])); while(not quit_requested) event_manager.handle_next_events(); return 0; } GlobalOptionManager option_manager; GlobalHookManager hook_manager; ShellManager shell_manager; CommandManager command_manager; BufferManager buffer_manager; RegisterManager register_manager; HighlighterRegistry highlighter_registry; FilterRegistry filter_registry; ColorRegistry color_registry; run_unit_tests(); register_env_vars(); register_registers(); register_commands(); register_highlighters(); register_filters(); write_debug("*** This is the debug buffer, where debug info will be written ***\n"); write_debug("pid: " + int_to_str(getpid()) + "\n"); write_debug("utf-8 test: é á ï"); setup_server(); Client local_client; try { Context initialisation_context; command_manager.execute("source " + runtime_directory() + "/kakrc", initialisation_context); } catch (Kakoune::runtime_error& error) { write_debug("error while parsing kakrc: " + error.description()); } local_client = create_local_client(argc > 1 ? argv[1] : ""); while(not quit_requested) event_manager.handle_next_events(); } catch (Kakoune::exception& error) { puts("uncaught exception:\n"); puts(error.description().c_str()); return -1; } return 0; }