1 files changed, 321 insertions, 0 deletions

diff --git a/examples/dte/editor.c b/examples/dte/editor.c
new file mode 100644
index 0000000..aa88d6e
--- /dev/null
+++ b/examples/dte/editor.c
@@ -0,0 +1,321 @@
+#include "compat.h"
+#include <errno.h>
+#include <langinfo.h>
+#include <locale.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include "editor.h"
+#include "bind.h"
+#include "bookmark.h"
+#include "command/macro.h"
+#include "commands.h"
+#include "compiler.h"
+#include "encoding.h"
+#include "error.h"
+#include "file-option.h"
+#include "filetype.h"
+#include "lock.h"
+#include "mode.h"
+#include "regexp.h"
+#include "screen.h"
+#include "search.h"
+#include "signals.h"
+#include "syntax/syntax.h"
+#include "tag.h"
+#include "terminal/input.h"
+#include "terminal/mode.h"
+#include "terminal/output.h"
+#include "terminal/style.h"
+#include "util/ascii.h"
+#include "util/debug.h"
+#include "util/exitcode.h"
+#include "util/intern.h"
+#include "util/log.h"
+#include "util/utf8.h"
+#include "util/xmalloc.h"
+#include "util/xstdio.h"
+#include "window.h"
+#include "../build/version.h"
+
+static void set_and_check_locale(void)
+{
+    const char *default_locale = setlocale(LC_CTYPE, "");
+    if (likely(default_locale)) {
+        const char *codeset = nl_langinfo(CODESET);
+        LOG_INFO("locale: %s (codeset: %s)", default_locale, codeset);
+        if (likely(lookup_encoding(codeset) == UTF8)) {
+            return;
+        }
+    } else {
+        LOG_ERROR("failed to set default locale");
+    }
+
+    static const char fallbacks[][12] = {"C.UTF-8", "en_US.UTF-8"};
+    const char *fallback = NULL;
+    for (size_t i = 0; i < ARRAYLEN(fallbacks) && !fallback; i++) {
+        fallback = setlocale(LC_CTYPE, fallbacks[i]);
+    }
+    if (fallback) {
+        LOG_INFO("using fallback locale for LC_CTYPE: %s", fallback);
+        return;
+    }
+
+    LOG_ERROR("no UTF-8 fallback locales found");
+    fputs("setlocale() failed\n", stderr);
+    exit(EX_CONFIG);
+}
+
+EditorState *init_editor_state(void)
+{
+    EditorState *e = xnew(EditorState, 1);
+    *e = (EditorState) {
+        .status = EDITOR_INITIALIZING,
+        .input_mode = INPUT_NORMAL,
+        .version = VERSION,
+        .command_history = {
+            .max_entries = 512,
+        },
+        .search_history = {
+            .max_entries = 128,
+        },
+        .cursor_styles = {
+            [CURSOR_MODE_DEFAULT] = {.type = CURSOR_DEFAULT, .color = COLOR_DEFAULT},
+            [CURSOR_MODE_INSERT] = {.type = CURSOR_KEEP, .color = COLOR_KEEP},
+            [CURSOR_MODE_OVERWRITE] = {.type = CURSOR_KEEP, .color = COLOR_KEEP},
+            [CURSOR_MODE_CMDLINE] = {.type = CURSOR_KEEP, .color = COLOR_KEEP},
+        },
+        .modes = {
+            [INPUT_NORMAL] = {.cmds = &normal_commands},
+            [INPUT_COMMAND] = {.cmds = &cmd_mode_commands},
+            [INPUT_SEARCH] = {.cmds = &search_mode_commands},
+        },
+        .options = {
+            .auto_indent = true,
+            .detect_indent = 0,
+            .editorconfig = false,
+            .emulate_tab = false,
+            .expand_tab = false,
+            .file_history = true,
+            .indent_width = 8,
+            .overwrite = false,
+            .save_unmodified = SAVE_FULL,
+            .syntax = true,
+            .tab_width = 8,
+            .text_width = 72,
+            .ws_error = WSE_SPECIAL,
+
+            // Global-only options
+            .case_sensitive_search = CSS_TRUE,
+            .crlf_newlines = false,
+            .display_special = false,
+            .esc_timeout = 100,
+            .filesize_limit = 250,
+            .lock_files = true,
+            .optimize_true_color = true,
+            .scroll_margin = 0,
+            .select_cursor_char = true,
+            .set_window_title = false,
+            .show_line_numbers = false,
+            .statusline_left = str_intern(" %f%s%m%s%r%s%M"),
+            .statusline_right = str_intern(" %y,%X  %u  %o  %E%s%b%s%n %t   %p "),
+            .tab_bar = true,
+            .utf8_bom = false,
+        }
+    };
+
+    sanity_check_global_options(&e->options);
+
+    for (size_t i = 0; i < ARRAYLEN(e->modes); i++) {
+        const CommandSet *cmds = e->modes[i].cmds;
+        BUG_ON(!cmds);
+        BUG_ON(!cmds->lookup);
+    }
+
+    const char *home = getenv("HOME");
+    const char *dte_home = getenv("DTE_HOME");
+    e->home_dir = strview_intern(home ? home : "");
+    if (dte_home) {
+        e->user_config_dir = xstrdup(dte_home);
+    } else {
+        e->user_config_dir = xasprintf("%s/.dte", e->home_dir.data);
+    }
+
+    LOG_INFO("dte version: " VERSION);
+    LOG_INFO("features:%s", feature_string);
+
+    pid_t pid = getpid();
+    bool leader = pid == getsid(0);
+    e->session_leader = leader;
+    LOG_INFO("pid: %jd%s", (intmax_t)pid, leader ? " (session leader)" : "");
+
+    pid_t pgid = getpgrp();
+    if (pgid != pid) {
+        LOG_INFO("pgid: %jd", (intmax_t)pgid);
+    }
+
+    set_and_check_locale();
+    init_file_locks_context(e->user_config_dir, pid);
+
+    // Allow child processes to detect that they're running under dte
+    if (unlikely(setenv("DTE_VERSION", VERSION, true) != 0)) {
+        fatal_error("setenv", errno);
+    }
+
+    RegexpWordBoundaryTokens *wb = &e->regexp_word_tokens;
+    if (regexp_init_word_boundary_tokens(wb)) {
+        LOG_INFO("regex word boundary tokens detected: %s %s", wb->start, wb->end);
+    } else {
+        LOG_WARNING("no regex word boundary tokens detected");
+    }
+
+    term_input_init(&e->terminal.ibuf);
+    term_output_init(&e->terminal.obuf);
+    hashmap_init(&e->aliases, 32);
+    intmap_init(&e->modes[INPUT_NORMAL].key_bindings, 150);
+    intmap_init(&e->modes[INPUT_COMMAND].key_bindings, 40);
+    intmap_init(&e->modes[INPUT_SEARCH].key_bindings, 40);
+    return e;
+}
+
+void free_editor_state(EditorState *e)
+{
+    free(e->clipboard.buf);
+    free_file_options(&e->file_options);
+    free_filetypes(&e->filetypes);
+    free_syntaxes(&e->syntaxes);
+    file_history_free(&e->file_history);
+    history_free(&e->command_history);
+    history_free(&e->search_history);
+    search_free_regexp(&e->search);
+    term_output_free(&e->terminal.obuf);
+    term_input_free(&e->terminal.ibuf);
+    cmdline_free(&e->cmdline);
+    clear_messages(&e->messages);
+    free_macro(&e->macro);
+    tag_file_free(&e->tagfile);
+
+    ptr_array_free_cb(&e->bookmarks, FREE_FUNC(file_location_free));
+    ptr_array_free_cb(&e->buffers, FREE_FUNC(free_buffer));
+    hashmap_free(&e->compilers, FREE_FUNC(free_compiler));
+    hashmap_free(&e->colors.other, free);
+    hashmap_free(&e->aliases, free);
+
+    for (size_t i = 0; i < ARRAYLEN(e->modes); i++) {
+        free_bindings(&e->modes[i].key_bindings);
+    }
+
+    free_interned_strings();
+    free_interned_regexps();
+
+    // TODO: intern this (so that it's freed by free_intern_pool())
+    free((void*)e->user_config_dir);
+
+    free(e);
+}
+
+static void sanity_check(const View *view)
+{
+#if DEBUG >= 1
+    const Block *blk;
+    block_for_each(blk, &view->buffer->blocks) {
+        if (blk == view->cursor.blk) {
+            BUG_ON(view->cursor.offset > view->cursor.blk->size);
+            return;
+        }
+    }
+    BUG("cursor not seen");
+#else
+    (void)view;
+#endif
+}
+
+void any_key(Terminal *term, unsigned int esc_timeout)
+{
+    KeyCode key;
+    xfputs("Press any key to continue\r\n", stderr);
+    while ((key = term_read_key(term, esc_timeout)) == KEY_NONE) {
+        ;
+    }
+    bool bracketed_paste = key == KEY_BRACKETED_PASTE;
+    if (bracketed_paste || key == KEY_DETECTED_PASTE) {
+        term_discard_paste(&term->ibuf, bracketed_paste);
+    }
+}
+
+NOINLINE
+void ui_resize(EditorState *e)
+{
+    if (e->status == EDITOR_INITIALIZING) {
+        return;
+    }
+    resized = 0;
+    update_screen_size(&e->terminal, e->root_frame);
+    normal_update(e);
+}
+
+void ui_start(EditorState *e)
+{
+    if (e->status == EDITOR_INITIALIZING) {
+        return;
+    }
+
+    // Note: the order of these calls is important - Kitty saves/restores
+    // some terminal state when switching buffers, so switching to the
+    // alternate screen buffer needs to happen before modes are enabled
+    term_use_alt_screen_buffer(&e->terminal);
+    term_enable_private_modes(&e->terminal);
+
+    ui_resize(e);
+}
+
+void ui_end(EditorState *e)
+{
+    if (e->status == EDITOR_INITIALIZING) {
+        return;
+    }
+    Terminal *term = &e->terminal;
+    TermOutputBuffer *obuf = &term->obuf;
+    term_clear_screen(obuf);
+    term_move_cursor(obuf, 0, term->height - 1);
+    term_restore_cursor_style(term);
+    term_show_cursor(term);
+    term_restore_private_modes(term);
+    term_use_normal_screen_buffer(term);
+    term_end_sync_update(term);
+    term_output_flush(obuf);
+    term_cooked();
+}
+
+int main_loop(EditorState *e)
+{
+    while (e->status == EDITOR_RUNNING) {
+        if (unlikely(resized)) {
+            LOG_INFO("SIGWINCH received");
+            ui_resize(e);
+        }
+
+        KeyCode key = term_read_key(&e->terminal, e->options.esc_timeout);
+        if (unlikely(key == KEY_NONE)) {
+            continue;
+        }
+
+        const ScreenState s = {
+            .is_modified = buffer_modified(e->buffer),
+            .id = e->buffer->id,
+            .cy = e->view->cy,
+            .vx = e->view->vx,
+            .vy = e->view->vy
+        };
+
+        clear_error();
+        handle_input(e, key);
+        sanity_check(e->view);
+        update_screen(e, &s);
+    }
+
+    BUG_ON(e->status < 0 || e->status > EDITOR_EXIT_MAX);
+    return e->status;
+}