poe

Уређивач .po фајлова
Дневник | Датотеке | Референце | ПРОЧИТАЈМЕ | ЛИЦЕНЦА

чување 1494b2807f67a26127739ba46154a9e1160ac6d6
родитељ 4eee4739e9db4d25351e45b803aa0d7edf606157
Аутор: Страхиња Радић <contact@strahinja.org>
Датум:   Thu, 24 Feb 2022 21:45:21 +0100

Switch to termbox2; convert key handler to be similar to sled; convert config.def.h to be similar to sled

Signed-off-by: Страхиња Радић <contact@strahinja.org>

Diffstat:
MLICENSE.termbox | 10+++++-----
MREADME | 8++++----
MTODO | 4++--
Dbytebuffer.inl | 70----------------------------------------------------------------------
Mconfig.def.h | 217+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Mconfig.h | 219++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
Mdraw.c | 65++++++++++++++++++++++++++++++++++++++++-------------------------
Mdraw.h | 9+++++++++
Dinput.inl | 228-------------------------------------------------------------------------------
Mpo.c | 45+++++++++++++++++++++++++++------------------
Mpoe.1.in | 2+-
Mpoe.c | 804+++++++++++++++++++++++++++++++++++--------------------------------------------
Mpoe.do | 2+-
Dterm.inl | 311-------------------------------------------------------------------------------
Dtermbox.c | 681-------------------------------------------------------------------------------
Mtermbox.h | 3307+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++------
Dutf8.c | 79-------------------------------------------------------------------------------
Mutil.c | 4++--
измењених датотека: 18, додавања: 3961(+), брисања: 2104(-)

diff --git a/LICENSE.termbox b/LICENSE.termbox @@ -1,4 +1,4 @@ -Copyright (C) 2010-2013 nsf <no.smile.face@gmail.com> +Copyright (c) 2021 termbox developers Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal @@ -7,13 +7,13 @@ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: -The above copyright notice and this permission notice shall be included in -all copies or substantial portions of the Software. +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, -OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN -THE SOFTWARE. +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/README b/README @@ -29,7 +29,7 @@ License ------- poe - .po file editor -Copyright (C) 2021 Страхиња Радић +Copyright (C) 2021, 2022 Страхиња Радић This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software @@ -38,9 +38,9 @@ version. --- -This program includes code from the termbox TUI library, +This program includes code from the termbox2 TUI library, - https://github.com/termbox/termbox + https://github.com/termbox/termbox2 The file LICENSE.termbox applies to the portion of this program which is from -the termbox library. +the termbox2 library. Currently, this includes only the file termbox.h. diff --git a/TODO b/TODO @@ -3,9 +3,9 @@ [~] Tidy up basic logic in poe.c - [ ] Convert key handler to be similar to sled + [x] Convert key handler to be similar to sled - [ ] Convert to termbox2? Test. + [x] Convert to termbox2? Test. < > Wrap long lines? Not if it would complicate the code too much. diff --git a/bytebuffer.inl b/bytebuffer.inl @@ -1,70 +0,0 @@ -struct bytebuffer { - char *buf; - int len; - int cap; -}; - -static void bytebuffer_reserve(struct bytebuffer *b, int cap) { - if (b->cap >= cap) { - return; - } - - // prefer doubling capacity - if (b->cap * 2 >= cap) { - cap = b->cap * 2; - } - - char *newbuf = realloc(b->buf, cap); - b->buf = newbuf; - b->cap = cap; -} - -static void bytebuffer_init(struct bytebuffer *b, int cap) { - b->cap = 0; - b->len = 0; - b->buf = 0; - - if (cap > 0) { - b->cap = cap; - b->buf = malloc(cap); // just assume malloc works always - } -} - -static void bytebuffer_free(struct bytebuffer *b) { - if (b->buf) - free(b->buf); -} - -static void bytebuffer_clear(struct bytebuffer *b) { - b->len = 0; -} - -static void bytebuffer_append(struct bytebuffer *b, const char *data, int len) { - bytebuffer_reserve(b, b->len + len); - memcpy(b->buf + b->len, data, len); - b->len += len; -} - -static void bytebuffer_puts(struct bytebuffer *b, const char *str) { - bytebuffer_append(b, str, strlen(str)); -} - -static void bytebuffer_resize(struct bytebuffer *b, int len) { - bytebuffer_reserve(b, len); - b->len = len; -} - -static void bytebuffer_flush(struct bytebuffer *b, int fd) { - write(fd, b->buf, b->len); - bytebuffer_clear(b); -} - -static void bytebuffer_truncate(struct bytebuffer *b, int n) { - if (n <= 0) - return; - if (n > b->len) - n = b->len; - const int nmove = b->len - n; - memmove(b->buf, b->buf+n, nmove); - b->len -= n; -} diff --git a/config.def.h b/config.def.h @@ -1,5 +1,12 @@ /* See the file LICENSE for copyright and license details. */ +/* Comment out to not create backups */ +#define CREATE_BACKUPS + +#ifdef PO_C_GLOBALS +static const char* backup_suffix = "~"; +#endif /* PO_C_GLOBALS */ + #define SAVE_WRAP_WIDTH 70 #define WRAP_FIRST_MSGSTR 0 #define WARN_COUNT_DOTS 0 @@ -59,6 +66,216 @@ #define TITLE_FG TB_BLACK #define TITLE_BG TB_WHITE +#ifdef POE_C_GLOBALS +int cancel_close(struct DrawState* state); +int cancel_msgstr(struct DrawState* state); +int cancel_search(struct DrawState* state); +int copy_msgid_to_input(struct DrawState* state); +int edit_page_up(struct DrawState* state); +int edit_page_down(struct DrawState* state); +int erase_backwards(struct DrawState* state); +int erase_forward(struct DrawState* state); +int erase_prev_word(struct DrawState* state); +int erase_search_backwards(struct DrawState* state); +int erase_search_forward(struct DrawState* state); +int erase_search_prev_word(struct DrawState* state); +int erase_search_to_start(struct DrawState* state); +int erase_search_to_end(struct DrawState* state); +int erase_to_end(struct DrawState* state); +int erase_to_start(struct DrawState* state); +int exit_program(struct DrawState* state); +int hide_search(struct DrawState* state); +int insert_line(struct DrawState* state); +int insert_search_space(struct DrawState* state); +int insert_search_tab(struct DrawState* state); +int insert_space(struct DrawState* state); +int insert_tab(struct DrawState* state); +int move_end(struct DrawState* state); +int move_down(struct DrawState* state); +int move_left(struct DrawState* state); +int move_list_down(struct DrawState* state); +int move_list_end(struct DrawState* state); +int move_list_page_down(struct DrawState* state); +int move_list_page_up(struct DrawState* state); +int move_list_start(struct DrawState* state); +int move_list_up(struct DrawState* state); +int move_next_word(struct DrawState* state); +int move_prev_word(struct DrawState* state); +int move_right(struct DrawState* state); +int move_search_end(struct DrawState* state); +int move_search_left(struct DrawState* state); +int move_search_next_word(struct DrawState* state); +int move_search_prev_word(struct DrawState* state); +int move_search_right(struct DrawState* state); +int move_search_start(struct DrawState* state); +int move_start(struct DrawState* state); +int move_up(struct DrawState* state); +int next_fuzzy(struct DrawState* state); +int next_match(struct DrawState* state); +int next_plural(struct DrawState* state); +int next_untranslated(struct DrawState* state); +int paste_paste_buffer_to_input(struct DrawState* state); +int prev_match(struct DrawState* state); +int prev_plural(struct DrawState* state); +int save_msgstr(struct DrawState* state); +int show_edit(struct DrawState* state); +int show_help(struct DrawState* state); +int show_search(struct DrawState* state); +int toggle_fuzzy(struct DrawState* state); +int toggle_info_focused(struct DrawState* state); +int yank_msgstr_to_paste_buffer(struct DrawState* state); +int write_file(struct DrawState* state); + +/* clang-format off */ + +/* Always processed */ +static const struct Key top_level_keys[] = { + { .ev = { .key = TB_KEY_ESC }, .callback = cancel_close }, + { .ev = { .key = TB_KEY_CTRL_Q }, + .callback = exit_program }, +}; + +/* Edit msgstr box open */ +static const struct Key edit_keys[] = { + { .ev = { .key = TB_KEY_ARROW_DOWN }, + .callback = move_down }, + { .ev = { .key = TB_KEY_ARROW_LEFT }, + .callback = move_left }, + { .ev = { .key = TB_KEY_ARROW_RIGHT }, + .callback = move_right }, + { .ev = { .key = TB_KEY_ARROW_UP }, + .callback = move_up }, + { .ev = { .key = TB_KEY_BACKSPACE }, + .callback = erase_backwards }, + { .ev = { .key = TB_KEY_BACKSPACE2 }, + .callback = erase_backwards }, + { .ev = { .key = TB_KEY_CTRL_A }, + .callback = move_start }, + { .ev = { .key = TB_KEY_CTRL_C }, + .callback = yank_msgstr_to_paste_buffer }, + { .ev = { .key = TB_KEY_CTRL_E }, + .callback = move_end }, + { .ev = { .key = TB_KEY_CTRL_K }, + .callback = erase_to_end }, + { .ev = { .key = TB_KEY_CTRL_N }, + .callback = next_plural }, + { .ev = { .key = TB_KEY_CTRL_P }, + .callback = prev_plural }, + { .ev = { .key = TB_KEY_CTRL_R }, + .callback = move_prev_word }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = save_msgstr }, + { .ev = { .key = TB_KEY_CTRL_T }, + .callback = move_next_word }, + { .ev = { .key = TB_KEY_CTRL_U }, + .callback = erase_to_start }, + { .ev = { .key = TB_KEY_CTRL_V }, + .callback = paste_paste_buffer_to_input }, + { .ev = { .key = TB_KEY_CTRL_W }, + .callback = erase_prev_word }, + { .ev = { .key = TB_KEY_CTRL_Z }, + .callback = toggle_fuzzy }, + { .ev = { .key = TB_KEY_DELETE }, + .callback = erase_forward }, + { .ev = { .key = TB_KEY_END }, .callback = move_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = insert_line }, + { .ev = { .key = TB_KEY_F5 }, .callback = copy_msgid_to_input }, + { .ev = { .key = TB_KEY_F6 }, .callback = toggle_info_focused }, + { .ev = { .key = TB_KEY_F8 }, .callback = next_untranslated }, + { .ev = { .key = TB_KEY_F9 }, .callback = next_fuzzy }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_start }, + { .ev = { .key = TB_KEY_PGDN }, .callback = edit_page_down }, + { .ev = { .key = TB_KEY_PGUP }, .callback = edit_page_up }, + { .ev = { .key = TB_KEY_SPACE }, + .callback = insert_space }, + { .ev = { .key = TB_KEY_TAB }, .callback = insert_tab }, +}; + +/* Search box open */ +static const struct Key search_keys[] = { + { .ev = { .key = TB_KEY_ARROW_LEFT }, + .callback = move_search_left }, + { .ev = { .key = TB_KEY_ARROW_RIGHT }, + .callback = move_search_right }, + { .ev = { .key = TB_KEY_BACKSPACE }, + .callback = erase_search_backwards }, + { .ev = { .key = TB_KEY_BACKSPACE2 }, + .callback = erase_search_backwards }, + { .ev = { .key = TB_KEY_CTRL_A }, + .callback = move_search_start }, + { .ev = { .key = TB_KEY_CTRL_E }, + .callback = move_search_end }, + { .ev = { .key = TB_KEY_CTRL_K }, + .callback = erase_search_to_end }, + { .ev = { .key = TB_KEY_CTRL_R }, + .callback = move_search_prev_word }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = write_file }, + { .ev = { .key = TB_KEY_CTRL_T }, + .callback = move_search_next_word }, + { .ev = { .key = TB_KEY_CTRL_U }, + .callback = erase_search_to_start }, + { .ev = { .key = TB_KEY_CTRL_W }, + .callback = erase_search_prev_word }, + { .ev = { .key = TB_KEY_DELETE }, + .callback = erase_search_forward }, + { .ev = { .key = TB_KEY_END }, .callback = move_search_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = hide_search }, + { .ev = { .key = TB_KEY_F3 }, .callback = next_match }, + { .ev = { .key = TB_KEY_F4 }, .callback = prev_match }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_search_start }, + { .ev = { .key = TB_KEY_SPACE }, + .callback = insert_search_space }, + { .ev = { .key = TB_KEY_TAB }, .callback = insert_search_tab }, +}; + +/* Main window */ +static const struct Key main_keys[] = { + { .ev = { .key = TB_KEY_ARROW_DOWN }, + .callback = move_list_down }, + { .ev = { .key = TB_KEY_ARROW_UP }, + .callback = move_list_up }, + { .ev = { .key = TB_KEY_END }, + .callback = move_list_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = show_edit }, + { .ev = { .key = TB_KEY_F1 }, .callback = show_help }, + { .ev = { .key = TB_KEY_F3 }, .callback = next_match }, + { .ev = { .key = TB_KEY_F4 }, .callback = prev_match }, + { .ev = { .key = TB_KEY_F7 }, .callback = show_search }, + { .ev = { .key = TB_KEY_F8 }, .callback = next_untranslated }, + { .ev = { .key = TB_KEY_F9 }, .callback = next_fuzzy }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_list_start }, + { .ev = { .key = TB_KEY_CTRL_B }, + .callback = move_list_page_up }, + { .ev = { .key = TB_KEY_CTRL_F }, + .callback = move_list_page_down }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = write_file }, + { .ev = { .key = TB_KEY_CTRL_Z }, + .callback = toggle_fuzzy }, + { .ev = { .key = TB_KEY_PGDN }, + .callback = move_list_page_down }, + { .ev = { .key = TB_KEY_PGUP }, .callback = move_list_page_up }, + { .ev = { .ch = '/' }, .callback = show_search }, + { .ev = { .ch = 'G' }, .callback = move_list_end }, + { .ev = { .ch = 'H' }, .callback = show_help }, + { .ev = { .ch = 'f' }, .callback = next_fuzzy }, + { .ev = { .ch = 'g' }, .callback = move_list_start }, + { .ev = { .ch = 'j' }, .callback = move_list_down }, + { .ev = { .ch = 'k' }, .callback = move_list_up }, + { .ev = { .ch = 'N' }, .callback = prev_match }, + { .ev = { .ch = 'n' }, .callback = next_match }, + { .ev = { .ch = 'q' }, .callback = exit_program }, + { .ev = { .ch = 'u' }, .callback = next_untranslated }, + { .ev = { .ch = 'w' }, .callback = write_file }, + { .ev = { .ch = 'z' }, .callback = toggle_fuzzy }, +}; + +#endif /* POE_C_GLOBALS */ + void format_filename(char* result, size_t result_size, const char* format, const struct DrawState* state); void format_msgs(char* result, size_t result_size, const char* format, diff --git a/config.h b/config.h @@ -1,6 +1,13 @@ /* See the file LICENSE for copyright and license details. */ -#define SAVE_WRAP_WIDTH 0 +/* Comment out to not create backups */ +#define CREATE_BACKUPS + +#ifdef PO_C_GLOBALS +static const char* backup_suffix = "~"; +#endif /* PO_C_GLOBALS */ + +#define SAVE_WRAP_WIDTH 70 #define WRAP_FIRST_MSGSTR 0 #define WARN_COUNT_DOTS 0 @@ -59,6 +66,216 @@ #define TITLE_FG TB_BLACK #define TITLE_BG TB_WHITE +#ifdef POE_C_GLOBALS +int cancel_close(struct DrawState* state); +int cancel_msgstr(struct DrawState* state); +int cancel_search(struct DrawState* state); +int copy_msgid_to_input(struct DrawState* state); +int edit_page_up(struct DrawState* state); +int edit_page_down(struct DrawState* state); +int erase_backwards(struct DrawState* state); +int erase_forward(struct DrawState* state); +int erase_prev_word(struct DrawState* state); +int erase_search_backwards(struct DrawState* state); +int erase_search_forward(struct DrawState* state); +int erase_search_prev_word(struct DrawState* state); +int erase_search_to_start(struct DrawState* state); +int erase_search_to_end(struct DrawState* state); +int erase_to_end(struct DrawState* state); +int erase_to_start(struct DrawState* state); +int exit_program(struct DrawState* state); +int hide_search(struct DrawState* state); +int insert_line(struct DrawState* state); +int insert_search_space(struct DrawState* state); +int insert_search_tab(struct DrawState* state); +int insert_space(struct DrawState* state); +int insert_tab(struct DrawState* state); +int move_end(struct DrawState* state); +int move_down(struct DrawState* state); +int move_left(struct DrawState* state); +int move_list_down(struct DrawState* state); +int move_list_end(struct DrawState* state); +int move_list_page_down(struct DrawState* state); +int move_list_page_up(struct DrawState* state); +int move_list_start(struct DrawState* state); +int move_list_up(struct DrawState* state); +int move_next_word(struct DrawState* state); +int move_prev_word(struct DrawState* state); +int move_right(struct DrawState* state); +int move_search_end(struct DrawState* state); +int move_search_left(struct DrawState* state); +int move_search_next_word(struct DrawState* state); +int move_search_prev_word(struct DrawState* state); +int move_search_right(struct DrawState* state); +int move_search_start(struct DrawState* state); +int move_start(struct DrawState* state); +int move_up(struct DrawState* state); +int next_fuzzy(struct DrawState* state); +int next_match(struct DrawState* state); +int next_plural(struct DrawState* state); +int next_untranslated(struct DrawState* state); +int paste_paste_buffer_to_input(struct DrawState* state); +int prev_match(struct DrawState* state); +int prev_plural(struct DrawState* state); +int save_msgstr(struct DrawState* state); +int show_edit(struct DrawState* state); +int show_help(struct DrawState* state); +int show_search(struct DrawState* state); +int toggle_fuzzy(struct DrawState* state); +int toggle_info_focused(struct DrawState* state); +int yank_msgstr_to_paste_buffer(struct DrawState* state); +int write_file(struct DrawState* state); + +/* clang-format off */ + +/* Always processed */ +static const struct Key top_level_keys[] = { + { .ev = { .key = TB_KEY_ESC }, .callback = cancel_close }, + { .ev = { .key = TB_KEY_CTRL_Q }, + .callback = exit_program }, +}; + +/* Edit msgstr box open */ +static const struct Key edit_keys[] = { + { .ev = { .key = TB_KEY_ARROW_DOWN }, + .callback = move_down }, + { .ev = { .key = TB_KEY_ARROW_LEFT }, + .callback = move_left }, + { .ev = { .key = TB_KEY_ARROW_RIGHT }, + .callback = move_right }, + { .ev = { .key = TB_KEY_ARROW_UP }, + .callback = move_up }, + { .ev = { .key = TB_KEY_BACKSPACE }, + .callback = erase_backwards }, + { .ev = { .key = TB_KEY_BACKSPACE2 }, + .callback = erase_backwards }, + { .ev = { .key = TB_KEY_CTRL_A }, + .callback = move_start }, + { .ev = { .key = TB_KEY_CTRL_C }, + .callback = yank_msgstr_to_paste_buffer }, + { .ev = { .key = TB_KEY_CTRL_E }, + .callback = move_end }, + { .ev = { .key = TB_KEY_CTRL_K }, + .callback = erase_to_end }, + { .ev = { .key = TB_KEY_CTRL_N }, + .callback = next_plural }, + { .ev = { .key = TB_KEY_CTRL_P }, + .callback = prev_plural }, + { .ev = { .key = TB_KEY_CTRL_R }, + .callback = move_prev_word }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = save_msgstr }, + { .ev = { .key = TB_KEY_CTRL_T }, + .callback = move_next_word }, + { .ev = { .key = TB_KEY_CTRL_U }, + .callback = erase_to_start }, + { .ev = { .key = TB_KEY_CTRL_V }, + .callback = paste_paste_buffer_to_input }, + { .ev = { .key = TB_KEY_CTRL_W }, + .callback = erase_prev_word }, + { .ev = { .key = TB_KEY_CTRL_Z }, + .callback = toggle_fuzzy }, + { .ev = { .key = TB_KEY_DELETE }, + .callback = erase_forward }, + { .ev = { .key = TB_KEY_END }, .callback = move_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = insert_line }, + { .ev = { .key = TB_KEY_F5 }, .callback = copy_msgid_to_input }, + { .ev = { .key = TB_KEY_F6 }, .callback = toggle_info_focused }, + { .ev = { .key = TB_KEY_F8 }, .callback = next_untranslated }, + { .ev = { .key = TB_KEY_F9 }, .callback = next_fuzzy }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_start }, + { .ev = { .key = TB_KEY_PGDN }, .callback = edit_page_down }, + { .ev = { .key = TB_KEY_PGUP }, .callback = edit_page_up }, + { .ev = { .key = TB_KEY_SPACE }, + .callback = insert_space }, + { .ev = { .key = TB_KEY_TAB }, .callback = insert_tab }, +}; + +/* Search box open */ +static const struct Key search_keys[] = { + { .ev = { .key = TB_KEY_ARROW_LEFT }, + .callback = move_search_left }, + { .ev = { .key = TB_KEY_ARROW_RIGHT }, + .callback = move_search_right }, + { .ev = { .key = TB_KEY_BACKSPACE }, + .callback = erase_search_backwards }, + { .ev = { .key = TB_KEY_BACKSPACE2 }, + .callback = erase_search_backwards }, + { .ev = { .key = TB_KEY_CTRL_A }, + .callback = move_search_start }, + { .ev = { .key = TB_KEY_CTRL_E }, + .callback = move_search_end }, + { .ev = { .key = TB_KEY_CTRL_K }, + .callback = erase_search_to_end }, + { .ev = { .key = TB_KEY_CTRL_R }, + .callback = move_search_prev_word }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = write_file }, + { .ev = { .key = TB_KEY_CTRL_T }, + .callback = move_search_next_word }, + { .ev = { .key = TB_KEY_CTRL_U }, + .callback = erase_search_to_start }, + { .ev = { .key = TB_KEY_CTRL_W }, + .callback = erase_search_prev_word }, + { .ev = { .key = TB_KEY_DELETE }, + .callback = erase_search_forward }, + { .ev = { .key = TB_KEY_END }, .callback = move_search_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = hide_search }, + { .ev = { .key = TB_KEY_F3 }, .callback = next_match }, + { .ev = { .key = TB_KEY_F4 }, .callback = prev_match }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_search_start }, + { .ev = { .key = TB_KEY_SPACE }, + .callback = insert_search_space }, + { .ev = { .key = TB_KEY_TAB }, .callback = insert_search_tab }, +}; + +/* Main window */ +static const struct Key main_keys[] = { + { .ev = { .key = TB_KEY_ARROW_DOWN }, + .callback = move_list_down }, + { .ev = { .key = TB_KEY_ARROW_UP }, + .callback = move_list_up }, + { .ev = { .key = TB_KEY_END }, + .callback = move_list_end }, + { .ev = { .key = TB_KEY_ENTER }, + .callback = show_edit }, + { .ev = { .key = TB_KEY_F1 }, .callback = show_help }, + { .ev = { .key = TB_KEY_F3 }, .callback = next_match }, + { .ev = { .key = TB_KEY_F4 }, .callback = prev_match }, + { .ev = { .key = TB_KEY_F7 }, .callback = show_search }, + { .ev = { .key = TB_KEY_F8 }, .callback = next_untranslated }, + { .ev = { .key = TB_KEY_F9 }, .callback = next_fuzzy }, + { .ev = { .key = TB_KEY_HOME }, .callback = move_list_start }, + { .ev = { .key = TB_KEY_CTRL_B }, + .callback = move_list_page_up }, + { .ev = { .key = TB_KEY_CTRL_F }, + .callback = move_list_page_down }, + { .ev = { .key = TB_KEY_CTRL_S }, + .callback = write_file }, + { .ev = { .key = TB_KEY_CTRL_Z }, + .callback = toggle_fuzzy }, + { .ev = { .key = TB_KEY_PGDN }, + .callback = move_list_page_down }, + { .ev = { .key = TB_KEY_PGUP }, .callback = move_list_page_up }, + { .ev = { .ch = '/' }, .callback = show_search }, + { .ev = { .ch = 'G' }, .callback = move_list_end }, + { .ev = { .ch = 'H' }, .callback = show_help }, + { .ev = { .ch = 'f' }, .callback = next_fuzzy }, + { .ev = { .ch = 'g' }, .callback = move_list_start }, + { .ev = { .ch = 'j' }, .callback = move_list_down }, + { .ev = { .ch = 'k' }, .callback = move_list_up }, + { .ev = { .ch = 'N' }, .callback = prev_match }, + { .ev = { .ch = 'n' }, .callback = next_match }, + { .ev = { .ch = 'q' }, .callback = exit_program }, + { .ev = { .ch = 'u' }, .callback = next_untranslated }, + { .ev = { .ch = 'w' }, .callback = write_file }, + { .ev = { .ch = 'z' }, .callback = toggle_fuzzy }, +}; + +#endif /* POE_C_GLOBALS */ + void format_filename(char* result, size_t result_size, const char* format, const struct DrawState* state); void format_msgs(char* result, size_t result_size, const char* format, diff --git a/draw.c b/draw.c @@ -1,15 +1,19 @@ /* See the file LICENSE for copyright and license details. */ + +#define _POSIX_C_SOURCE 200809L +#define _DEFAULT_SOURCE +#define _XOPEN_SOURCE + #include <errno.h> #include <stdio.h> #include <stdlib.h> #include <string.h> -#include "termbox.h" +#include "version.h" #include "util.h" #include "po.h" #include "draw.h" #include "config.h" -#include "version.h" /* clang-format off */ static const char* help_title = "Main view help"; @@ -130,38 +134,49 @@ init_drawstate(struct DrawState* state, char* error, char* filename, void free_drawstate(struct DrawState* state) { - struct BufferLine* current = NULL; - struct PoEntry* current_entry = NULL; size_t i = 0; - free(state->search); - for (i = 0; i < state->input_rows_count; i++) - { - current = state->input_buffer + i; - free_bufferline(current); - } + free_bufferline(state->input_buffer + i); free(state->input_buffer); for (i = 0; i < state->paste_rows_count; i++) - { - current = state->paste_buffer + i; - free_bufferline(current); - } + free_bufferline(state->paste_buffer + i); free(state->paste_buffer); for (i = 0; i < state->info_rows_count; i++) - { - current = state->info_buffer + i; - free_bufferline(current); - } + free_bufferline(state->info_buffer + i); free(state->info_buffer); for (i = 0; i < state->real_msgid_count; i++) - { - current_entry = state->entries + i; - free_po_entry(current_entry); - } + free_po_entry(state->entries + i); + free(state->search); free(state->entries); } +void +format_filename(char* result, size_t result_size, const char* format, + const struct DrawState* state) +{ + if (!result) + return; + snprintf(result, result_size, format, u8_basename(state->filename), + state->show_edit ? ( + (state->dirty_before_edit || state->dirty) ? "*" : "") + : (state->dirty ? "*" : "")); +} + +void +format_msgs(char* result, size_t result_size, const char* format, + const struct DrawState* state) +{ + if (!result) + return; + + snprintf(result, result_size, format, state ? state->msgid_number : 0, + state ? state->msgid_count : 0, + state ? state->untranslated_count : 0, + state ? state->fuzzy_count : 0, + state ? state->obsolete_count : 0); +} + struct DrawState* set_nplurals(struct DrawState* state, struct PoEntry* entry) { @@ -565,7 +580,7 @@ draw_help(const struct DrawState* state) draw_box(startx, endx, starty, endy, DLG_FG, DLG_BG); if (!draw_string(startx, starty, TITLE_FG, TITLE_BG, help_title, - strlen(help_title) + 2, w - 2, 1, 1, CENTER, 0, 0, NULL, 0, + strlen(help_title) + 2, w, 1, 1, CENTER, 0, 0, NULL, 0, 0)) return NULL; while (row < LEN(help)) @@ -591,7 +606,7 @@ draw_editbox(const struct DrawState* state) int maxx = state->maxx; int maxy = state->maxy; char flags_buf[MAXFLAGSBUF + 4]; - struct PoEntry* entry = &state->entries[state->msgid_number - 1]; + struct PoEntry* entry = state->entries + state->msgid_number - 1; size_t dlen = 0; strcpy(flags_buf, "[ "); @@ -662,7 +677,7 @@ draw_editbox(const struct DrawState* state) } if (state->edit_info_focused) - tb_set_cursor(TB_HIDE_CURSOR, TB_HIDE_CURSOR); + tb_hide_cursor(); else tb_set_cursor(startx + 2 + state->input_display_column - state->input_first_shown_column, diff --git a/draw.h b/draw.h @@ -1,6 +1,10 @@ /* See the file LICENSE for copyright and license details. */ #include <stddef.h> +#include <termios.h> +#include <unistd.h> +#include <wchar.h> +#define TB_OPT_V1_COMPAT #include "termbox.h" #define BORDER_NW ((uint32_t)L'\u250c') @@ -83,6 +87,11 @@ struct DrawState { size_t first_shown_msgid; }; +struct Key { + struct tb_event ev; + int (*callback)(struct DrawState*); +}; + struct StatusSegment { const char* format; void (*callback)(char*, size_t, const char*, const struct DrawState*); diff --git a/input.inl b/input.inl @@ -1,228 +0,0 @@ -// if s1 starts with s2 returns true, else false -// len is the length of s1 -// s2 should be null-terminated -static bool starts_with(const char *s1, int len, const char *s2) -{ - int n = 0; - while (*s2 && n < len) { - if (*s1++ != *s2++) - return false; - n++; - } - return *s2 == 0; -} - -static int parse_mouse_event(struct tb_event *event, const char *buf, int len) { - if (len >= 6 && starts_with(buf, len, "\033[M")) { - // X10 mouse encoding, the simplest one - // \033 [ M Cb Cx Cy - int b = buf[3] - 32; - switch (b & 3) { - case 0: - if ((b & 64) != 0) - event->key = TB_KEY_MOUSE_WHEEL_UP; - else - event->key = TB_KEY_MOUSE_LEFT; - break; - case 1: - if ((b & 64) != 0) - event->key = TB_KEY_MOUSE_WHEEL_DOWN; - else - event->key = TB_KEY_MOUSE_MIDDLE; - break; - case 2: - event->key = TB_KEY_MOUSE_RIGHT; - break; - case 3: - event->key = TB_KEY_MOUSE_RELEASE; - break; - default: - return -6; - } - event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default - if ((b & 32) != 0) - event->mod |= TB_MOD_MOTION; - - // the coord is 1,1 for upper left - event->x = (uint8_t)buf[4] - 1 - 32; - event->y = (uint8_t)buf[5] - 1 - 32; - - return 6; - } else if (starts_with(buf, len, "\033[<") || starts_with(buf, len, "\033[")) { - // xterm 1006 extended mode or urxvt 1015 extended mode - // xterm: \033 [ < Cb ; Cx ; Cy (M or m) - // urxvt: \033 [ Cb ; Cx ; Cy M - int i, mi = -1, starti = -1; - int isM, isU, s1 = -1, s2 = -1; - int n1 = 0, n2 = 0, n3 = 0; - - for (i = 0; i < len; i++) { - // We search the first (s1) and the last (s2) ';' - if (buf[i] == ';') { - if (s1 == -1) - s1 = i; - s2 = i; - } - - // We search for the first 'm' or 'M' - if ((buf[i] == 'm' || buf[i] == 'M') && mi == -1) { - mi = i; - break; - } - } - if (mi == -1) - return 0; - - // whether it's a capital M or not - isM = (buf[mi] == 'M'); - - if (buf[2] == '<') { - isU = 0; - starti = 3; - } else { - isU = 1; - starti = 2; - } - - if (s1 == -1 || s2 == -1 || s1 == s2) - return 0; - - n1 = strtoul(&buf[starti], NULL, 10); - n2 = strtoul(&buf[s1 + 1], NULL, 10); - n3 = strtoul(&buf[s2 + 1], NULL, 10); - - if (isU) - n1 -= 32; - - switch (n1 & 3) { - case 0: - if ((n1&64) != 0) { - event->key = TB_KEY_MOUSE_WHEEL_UP; - } else { - event->key = TB_KEY_MOUSE_LEFT; - } - break; - case 1: - if ((n1&64) != 0) { - event->key = TB_KEY_MOUSE_WHEEL_DOWN; - } else { - event->key = TB_KEY_MOUSE_MIDDLE; - } - break; - case 2: - event->key = TB_KEY_MOUSE_RIGHT; - break; - case 3: - event->key = TB_KEY_MOUSE_RELEASE; - break; - default: - return mi + 1; - } - - if (!isM) { - // on xterm mouse release is signaled by lowercase m - event->key = TB_KEY_MOUSE_RELEASE; - } - - event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default - if ((n1&32) != 0) - event->mod |= TB_MOD_MOTION; - - event->x = (uint8_t)n2 - 1; - event->y = (uint8_t)n3 - 1; - - return mi + 1; - } - - return 0; -} - -// convert escape sequence to event, and return consumed bytes on success (failure == 0) -static int parse_escape_seq(struct tb_event *event, const char *buf, int len) -{ - int mouse_parsed = parse_mouse_event(event, buf, len); - - if (mouse_parsed != 0) - return mouse_parsed; - - // it's pretty simple here, find 'starts_with' match and return - // success, else return failure - int i; - for (i = 0; keys[i]; i++) { - if (starts_with(buf, len, keys[i])) { - event->ch = 0; - event->key = 0xFFFF-i; - return strlen(keys[i]); - } - } - return 0; -} - -static bool extract_event(struct tb_event *event, struct bytebuffer *inbuf, int inputmode) -{ - const char *buf = inbuf->buf; - const int len = inbuf->len; - if (len == 0) - return false; - - if (buf[0] == '\033') { - int n = parse_escape_seq(event, buf, len); - if (n != 0) { - bool success = true; - if (n < 0) { - success = false; - n = -n; - } - bytebuffer_truncate(inbuf, n); - return success; - } else { - // it's not escape sequence, then it's ALT or ESC, - // check inputmode - if (inputmode&TB_INPUT_ESC) { - // if we're in escape mode, fill ESC event, pop - // buffer, return success - event->ch = 0; - event->key = TB_KEY_ESC; - event->mod = 0; - bytebuffer_truncate(inbuf, 1); - return true; - } else if (inputmode&TB_INPUT_ALT) { - // if we're in alt mode, set ALT modifier to - // event and redo parsing - event->mod = TB_MOD_ALT; - bytebuffer_truncate(inbuf, 1); - return extract_event(event, inbuf, inputmode); - } - assert(!"never got here"); - } - } - - // if we're here, this is not an escape sequence and not an alt sequence - // so, it's a FUNCTIONAL KEY or a UNICODE character - - // first of all check if it's a functional key - if ((unsigned char)buf[0] <= TB_KEY_SPACE || - (unsigned char)buf[0] == TB_KEY_BACKSPACE2) - { - // fill event, pop buffer, return success */ - event->ch = 0; - event->key = (uint16_t)buf[0]; - bytebuffer_truncate(inbuf, 1); - return true; - } - - // feh... we got utf8 here - - // check if there is all bytes - if (len >= tb_utf8_char_length(buf[0])) { - /* everything ok, fill event, pop buffer, return success */ - tb_utf8_char_to_unicode(&event->ch, buf); - event->key = 0; - bytebuffer_truncate(inbuf, tb_utf8_char_length(buf[0])); - return true; - } - - // event isn't recognized, perhaps there is not enough bytes in utf8 - // sequence - return false; -} diff --git a/po.c b/po.c @@ -2,13 +2,16 @@ #include <stdio.h> #include <stdlib.h> #include <string.h> +#include <sys/stat.h> #include <time.h> +#define PO_C_GLOBALS + +#include "version.h" +#include "util.h" #include "po.h" #include "draw.h" #include "config.h" -#include "util.h" -#include "version.h" extern const char* errors[]; extern const char* flag_strings[]; @@ -498,19 +501,28 @@ save_file(const struct PoEntry* entries, const size_t real_msgid_count, const char* filename, const int nplurals) { const struct PoEntry* current = entries; - size_t len = strlen(filename); FILE* output = NULL; - char* backup_filename = calloc(len + 2, 1); - char tmp_filename[2][MAXPATH]; - strcpy(tmp_filename[0], filename); - strcpy(tmp_filename[1], dirname(tmp_filename[0])); - strncat(tmp_filename[1], "/poe.tmp", MAXPATH - 1); - - if (!(output = fopen(tmp_filename[1], "wt"))) +#ifdef CREATE_BACKUPS + char* backup_filename = NULL; +#endif + struct stat st; + +#ifdef CREATE_BACKUPS + if (!(backup_filename = calloc(MAXPATH, 1))) + return SAVE_ERR_CANT_ALLOC; +#endif + +#ifdef CREATE_BACKUPS + if (stat(filename, &st) != -1) + { + strcpy(backup_filename, filename); + strcat(backup_filename, backup_suffix); + if (rename(filename, backup_filename) == -1) + return SAVE_ERR_CANT_MOVE_FILE; + } +#endif + if (!(output = fopen(filename, "wt"))) return SAVE_ERR_CANT_OPEN_FILE; - strcpy(backup_filename, filename); - backup_filename[len] = '~'; - backup_filename[len + 1] = 0; save_loop_entries_start: if (current >= entries + real_msgid_count) @@ -839,12 +851,9 @@ save_loop_entries_start: goto save_loop_entries_start; save_loop_entries_done: fclose(output); - - if (rename(filename, backup_filename) == -1) - return SAVE_ERR_CANT_MOVE_FILE; - if (rename(tmp_filename[1], filename) == -1) - return SAVE_ERR_CANT_MOVE_FILE; +#ifdef CREATE_BACKUPS free(backup_filename); +#endif return SAVE_ERR_NONE; } diff --git a/poe.1.in b/poe.1.in @@ -48,7 +48,7 @@ poe Copyright \(co 2021, 2022 Strahinya Radich. This program is licensed under GNU GPL v3 or later. See the file .I LICENSE in the poe repository for details. For licensing details about the portion of -the code taken from the termbox library, see the file +the code taken from the termbox2 library, see the file .IR "LICENSE.termbox" . . .SH DESCRIPTION diff --git a/poe.c b/poe.c @@ -1,6 +1,8 @@ /* See the file LICENSE for copyright and license details. */ #define _POSIX_C_SOURCE 200809L +#define _DEFAULT_SOURCE +#define _XOPEN_SOURCE #include <errno.h> #include <fcntl.h> @@ -13,13 +15,18 @@ #include <time.h> #include <unistd.h> +#define TB_OPT_V1_COMPAT +#define TB_IMPL + +#define POE_C_GLOBALS + #include "version.h" -#include "termbox.h" #include "util.h" #include "po.h" #include "draw.h" #include "config.h" +extern const char* backup_suffix; extern const char* errors[]; extern const char* prompt_dirty; extern const char* prompt_cancel_dirty; @@ -27,70 +34,15 @@ extern const char* prompt_overwrite; extern const char* program_name; int print_error(const int code, const char* msg, ...); -int version(); -int usage(); -void format_filename(char* result, size_t result_size, const char* format, - const struct DrawState* state); -void format_msgs(char* result, size_t result_size, const char* format, - const struct DrawState* state); void overwrite_copy_msgid_callback(struct DrawState* state, struct tb_event* ev); void overwrite_paste_callback(struct DrawState* state, struct tb_event* ev); void cancel_callback(struct DrawState* state, struct tb_event* ev); void quit_callback(struct DrawState* state, struct tb_event* ev); struct BufferLine* load_info(struct DrawState* state); struct BufferLine* load_msgstr(struct DrawState* state); -void save_msgstr(struct DrawState* state); -void cancel_msgstr(struct DrawState* state); -void show_search(struct DrawState* state); -void hide_search(struct DrawState* state); -void cancel_search(struct DrawState* state); -void move_search_start(struct DrawState* state); -void move_search_left(struct DrawState* state); -void move_search_right(struct DrawState* state); -void move_search_end(struct DrawState* state); void insert_search_char(struct DrawState* state, uint32_t ch); -void erase_search_backwards(struct DrawState* state); -void erase_search_forward(struct DrawState* state); -void erase_search_to_start(struct DrawState* state); -void erase_search_to_end(struct DrawState* state); -void move_search_prev_word(struct DrawState* state); -void move_search_next_word(struct DrawState* state); -void erase_search_prev_word(struct DrawState* state); -struct DrawState* copy_msgid_to_input(struct DrawState* state); -struct DrawState* yank_msgstr_to_paste_buffer(struct DrawState* state); -struct DrawState* paste_paste_buffer_to_input(struct DrawState* state); void goto_msgid(struct DrawState* state, const size_t msgid_number); -void next_match(struct DrawState* state); -void prev_match(struct DrawState* state); -void next_untranslated(struct DrawState* state); -void next_fuzzy(struct DrawState* state); -void show_edit(struct DrawState* state); -void move_up(struct DrawState* state); -void move_down(struct DrawState* state); -void edit_page_up(struct DrawState* state); -void edit_page_down(struct DrawState* state); -void move_start(struct DrawState* state); -void move_left(struct DrawState* state); -void move_right(struct DrawState* state); -void move_prev_word(struct DrawState* state); -void move_next_word(struct DrawState* state); -void erase_prev_word(struct DrawState* state); -void move_end(struct DrawState* state); -void move_list_up(struct DrawState* state); -void move_list_down(struct DrawState* state); -void move_list_start(struct DrawState* state); -void move_list_end(struct DrawState* state); -void move_list_page_up(struct DrawState* state); -void move_list_page_down(struct DrawState* state); -void next_plural(struct DrawState* state); -void prev_plural(struct DrawState* state); -void toggle_fuzzy(struct DrawState* state); void insert_char(struct DrawState* state, uint32_t ch); -void erase_backwards(struct DrawState* state); -void erase_forward(struct DrawState* state); -void erase_to_start(struct DrawState* state); -void erase_to_end(struct DrawState* state); -struct DrawState* insert_line(struct DrawState* state); struct DrawState* join_lines(struct DrawState* state, size_t first_line, int forward); void handle_editbox_key_event(struct tb_event* ev, struct DrawState* state); @@ -110,47 +62,35 @@ print_error(const int code, const char* msg, ...) } int -version() +write_file(struct DrawState* state) { - printf("%s %s, built on %s\n", program_name, VERSION, DATE); - return 0; -} + if (!state) + return 1; -int -usage() -{ - printf("Usage: %s [-h|--help] [-v|--version] [filename.po]\n", - program_name); + int result = save_file(state->entries, state->real_msgid_count, + state->filename, state->nplurals); + switch (result) + { + case SAVE_ERR_CANT_ALLOC: + strcpy(state->error, errors[ERR_CANT_ALLOC]); + state->running = 0; + break; + case SAVE_ERR_CANT_OPEN_FILE: + strcpy(state->error, errors[ERR_CANT_SAVE]); + state->running = 0; + break; + case SAVE_ERR_CANT_MOVE_FILE: + strcpy(state->error, strerror(errno)); + state->running = 0; + break; + default: + *state->error = 0; + state->dirty = 0; + } return 0; } void -format_filename(char* result, size_t result_size, const char* format, - const struct DrawState* state) -{ - if (!result) - return; - snprintf(result, result_size, format, u8_basename(state->filename), - state->show_edit ? ( - (state->dirty_before_edit || state->dirty) ? "*" : "") - : (state->dirty ? "*" : "")); -} - -void -format_msgs(char* result, size_t result_size, const char* format, - const struct DrawState* state) -{ - if (!result) - return; - - snprintf(result, result_size, format, state ? state->msgid_number : 0, - state ? state->msgid_count : 0, - state ? state->untranslated_count : 0, - state ? state->fuzzy_count : 0, - state ? state->obsolete_count : 0); -} - -void overwrite_copy_msgid_callback(struct DrawState* state, struct tb_event* ev) { if (!state) @@ -192,26 +132,55 @@ quit_callback(struct DrawState* state, struct tb_event* ev) return; if (ev->ch == 'y') - { - int result = save_file(state->entries, state->real_msgid_count, - state->filename, state->nplurals); - switch (result) - { - case SAVE_ERR_CANT_ALLOC: - strcpy(state->error, errors[ERR_CANT_ALLOC]); - break; - case SAVE_ERR_CANT_OPEN_FILE: - strcpy(state->error, errors[ERR_CANT_SAVE]); - break; - default: - *state->error = 0; - } - } - state->dirty = 0; + write_file(state); state->prompt_callback = NULL; state->running = 0; } +int +cancel_close(struct DrawState* state) +{ + if (!state) + return 1; + + *state->error = 0; + if (state->in_prompt) + { + state->in_prompt = 0; + *state->prompt = 0; + } + else if (state->show_help) + state->show_help = 0; + else if (state->show_edit) + cancel_msgstr(state); + else if (state->show_search) + cancel_search(state); + else + strcpy(state->error, errors[ERR_EXIT_KEY]); + + return 0; +} + +int +exit_program(struct DrawState* state) +{ + if (state->show_edit) + { + strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); + return 0; + } + if (state->dirty) + { + state->in_prompt = 1; + strcpy(state->prompt, prompt_dirty); + state->prompt_callback = quit_callback; + } + else + state->running = 0; + *state->error = 0; + return 0; +} + struct BufferLine* load_info(struct DrawState* state) { @@ -468,13 +437,16 @@ load_msgstr(struct DrawState* state) return state->input_buffer; } -void +int save_msgstr(struct DrawState* state) { uint32_t* to_save = NULL; size_t to_save_size = 0; struct PoEntry* current = NULL; + if (!state) + return 1; + state->show_edit = 0; state->edit_info_focused = 0; @@ -485,7 +457,7 @@ save_msgstr(struct DrawState* state) if (!to_save) { strcpy(state->error, errors[ERR_CANT_ALLOC]); - return; + return 0; } struct BufferLine* pinput_buffer = NULL; for (size_t i = 0; i < state->input_rows_count; i++) @@ -521,11 +493,12 @@ save_msgstr(struct DrawState* state) update_warning(current->msgid, to_save, &current->warning); state->input_changed_fuzzy = UNCHANGED; - tb_set_cursor(TB_HIDE_CURSOR, TB_HIDE_CURSOR); + tb_hide_cursor(); free(to_save); + return 0; } -void +int cancel_msgstr(struct DrawState* state) { if (!state->in_prompt && state->dirty) @@ -533,7 +506,7 @@ cancel_msgstr(struct DrawState* state) state->in_prompt = 1; strcpy(state->prompt, prompt_cancel_dirty); state->prompt_callback = cancel_callback; - return; + return 0; } state->show_edit = 0; @@ -563,11 +536,19 @@ cancel_msgstr(struct DrawState* state) default:; } state->input_changed_fuzzy = UNCHANGED; + tb_hide_cursor(); + return 0; +} - tb_set_cursor(TB_HIDE_CURSOR, TB_HIDE_CURSOR); +int +show_help(struct DrawState* state) +{ + state->show_help = 1; + *state->error = 0; + return 0; } -void +int show_search(struct DrawState* state) { state->show_search = 1; @@ -579,36 +560,40 @@ show_search(struct DrawState* state) *state->search = 0; tb_set_cursor((state->maxx - state->search_width + 2) / 2, state->maxy - 1 - SEARCH_HEIGHT + 1); + return 0; } -void +int hide_search(struct DrawState* state) { state->show_search = 0; - tb_set_cursor(TB_HIDE_CURSOR, TB_HIDE_CURSOR); + tb_hide_cursor(); + return 0; } -void +int cancel_search(struct DrawState* state) { state->show_search = 0; *state->search = 0; - tb_set_cursor(TB_HIDE_CURSOR, TB_HIDE_CURSOR); + tb_hide_cursor(); + return 0; } -void +int move_search_start(struct DrawState* state) { state->search_column = 0; state->search_display_column = 0; state->search_first_shown_column = 0; + return 0; } -void +int move_search_left(struct DrawState* state) { if (state->search_column == 0) - return; + return 0; if (state->search[state->search_column - 1] == '\t') state->search_display_column = display_length(state, @@ -619,14 +604,15 @@ move_search_left(struct DrawState* state) if (state->search_display_column < state->search_first_shown_column) state->search_first_shown_column = state->search_display_column; + return 0; } -void +int move_search_right(struct DrawState* state) { size_t len = u32_strlen(state->search); if (state->search_column == len) - return; + return 0; if (state->search[state->search_column] == '\t') state->search_display_column = display_length(state, @@ -639,9 +625,10 @@ move_search_right(struct DrawState* state) > state->search_width - 4) state->search_first_shown_column = state->search_display_column - state->search_width + 4; + return 0; } -void +int move_search_end(struct DrawState* state) { size_t len = u32_strlen(state->search); @@ -654,6 +641,7 @@ move_search_end(struct DrawState* state) = dlen - state->search_width + 4; else state->search_first_shown_column = 0; + return 0; } void @@ -691,7 +679,7 @@ insert_search_char(struct DrawState* state, uint32_t ch) } } -void +int erase_search_backwards(struct DrawState* state) { if (state->search_column > 0) @@ -716,11 +704,12 @@ erase_search_backwards(struct DrawState* state) state->search_display_column = display_length(state, state->search, state->search_column); - next_match(state); + return next_match(state); } + return 0; } -void +int erase_search_forward(struct DrawState* state) { if (state->search[state->search_column]) @@ -733,11 +722,12 @@ erase_search_forward(struct DrawState* state) } state->search[i] = 0; - next_match(state); + return next_match(state); } + return 0; } -void +int erase_search_to_start(struct DrawState* state) { size_t i = state->search_column; @@ -749,18 +739,18 @@ erase_search_to_start(struct DrawState* state) } while (state->search[i]); state->search[len - state->search_column] = 0; move_search_start(state); - next_match(state); + return next_match(state); } -void +int erase_search_to_end(struct DrawState* state) { state->search[state->search_column] = 0; move_search_end(state); - next_match(state); + return next_match(state); } -void +int move_search_prev_word(struct DrawState* state) { int i = state->search_column; @@ -788,9 +778,10 @@ move_search_prev_word(struct DrawState* state) } state->search_display_column = display_length(state, state->search, state->search_column); + return 0; } -void +int move_search_next_word(struct DrawState* state) { int i = state->search_column; @@ -813,9 +804,10 @@ move_search_next_word(struct DrawState* state) state->search_display_column = display_length(state, state->search, state->search_column); } + return 0; } -void +int erase_search_prev_word(struct DrawState* state) { int i = state->search_column; @@ -853,27 +845,27 @@ erase_search_prev_word(struct DrawState* state) state->search_display_column = display_length(state, state->search, state->search_column); - next_match(state); + return next_match(state); } -struct DrawState* +int copy_msgid_to_input(struct DrawState* state) { if (state->edit_info_focused) - return state; + return 0; if (!state->in_prompt && *state->input_buffer->text) { state->in_prompt = 1; strcpy(state->prompt, prompt_overwrite); state->prompt_callback = overwrite_copy_msgid_callback; - return state; + return 0; } if (state->msgid_number == 1) { strcpy(state->error, errors[ERR_ILLEGAL_ON_FIRST]); - return NULL; + return 0; } struct PoEntry* current = &state->entries[state->msgid_number - 1]; @@ -892,7 +884,7 @@ copy_msgid_to_input(struct DrawState* state) const uint32_t* pmsgid = current->msgid; if (!newbuf) - return NULL; + return 0; struct BufferLine* pbuffer = NULL; for (size_t i = 0; i < lines_to_init; i++) @@ -901,7 +893,7 @@ copy_msgid_to_input(struct DrawState* state) init_bufferline(pbuffer); uint32_t* newtext = calloc(MAXBUFLINE, sizeof(uint32_t)); if (!newtext) - return NULL; + return 0; pbuffer->text = newtext; } @@ -922,19 +914,19 @@ copy_msgid_to_input(struct DrawState* state) state->input_buffer = newbuf; state->dirty = 1; - return state; + return 0; } -struct DrawState* +int yank_msgstr_to_paste_buffer(struct DrawState* state) { if (state->edit_info_focused) - return state; + return 0; if (state->msgid_number == 1) { strcpy(state->error, errors[ERR_ILLEGAL_ON_FIRST]); - return NULL; + return 0; } struct PoEntry* current = &state->entries[state->msgid_number - 1]; @@ -952,7 +944,7 @@ yank_msgstr_to_paste_buffer(struct DrawState* state) free(state->paste_buffer); state->paste_buffer = NULL; state->paste_rows_count = 0; - return state; + return 0; } msgstr_num_lines @@ -976,7 +968,7 @@ yank_msgstr_to_paste_buffer(struct DrawState* state) pmsgstr = *(current->msgstr + state->msgstr_index); if (!newbuf) - return NULL; + return 0; for (size_t i = 0; i < lines_to_init; i++) { @@ -984,7 +976,7 @@ yank_msgstr_to_paste_buffer(struct DrawState* state) init_bufferline(pbuffer); uint32_t* newtext = calloc(MAXBUFLINE, sizeof(uint32_t)); if (!newtext) - return NULL; + return 0; pbuffer->text = newtext; } @@ -1007,27 +999,27 @@ yank_msgstr_to_paste_buffer(struct DrawState* state) state->paste_rows_count = msgstr_num_lines; state->paste_buffer = newbuf; - return state; + return 0; } -struct DrawState* +int paste_paste_buffer_to_input(struct DrawState* state) { if (state->edit_info_focused) - return state; + return 0; if (!state->in_prompt && *state->input_buffer->text) { state->in_prompt = 1; strcpy(state->prompt, prompt_overwrite); state->prompt_callback = overwrite_paste_callback; - return state; + return 0; } if (state->msgid_number == 1) { strcpy(state->error, errors[ERR_ILLEGAL_ON_FIRST]); - return NULL; + return 0; } size_t lines_to_init = 0; @@ -1052,7 +1044,7 @@ paste_paste_buffer_to_input(struct DrawState* state) state->input_row = 0; state->input_rows_count = 1; state->dirty = 1; - return state; + return 0; } lines_to_init = state->input_rows_count < state->paste_rows_count @@ -1070,7 +1062,7 @@ paste_paste_buffer_to_input(struct DrawState* state) ppaste = state->paste_buffer; if (!newbuf) - return NULL; + return 0; for (size_t i = 0; i < lines_to_init; i++) { @@ -1078,7 +1070,7 @@ paste_paste_buffer_to_input(struct DrawState* state) init_bufferline(pinput); uint32_t* newtext = calloc(MAXBUFLINE, sizeof(uint32_t)); if (!newtext) - return NULL; + return 0; pinput->text = newtext; } @@ -1096,7 +1088,7 @@ paste_paste_buffer_to_input(struct DrawState* state) state->input_buffer = newbuf; state->dirty = 1; - return state; + return 0; } void @@ -1114,11 +1106,11 @@ goto_msgid(struct DrawState* state, const size_t msgid_number) = MAX(1, state->msgid_number - (state->maxy - 1) / 2); } -void +int next_match(struct DrawState* state) { if (state->msgid_count == 0 || !state->search || !*state->search) - return; + return 0; *state->error = 0; struct PoEntry* current = NULL; @@ -1132,16 +1124,17 @@ next_match(struct DrawState* state) && u32_strstr(*current->msgstr, state->search))) { goto_msgid(state, e + 1); - return; + return 0; } } + return 0; } -void +int prev_match(struct DrawState* state) { if (state->msgid_count == 0 || !state->search || !*state->search) - return; + return 0; *state->error = 0; struct PoEntry* current = NULL; @@ -1155,16 +1148,17 @@ prev_match(struct DrawState* state) && u32_strstr(*current->msgstr, state->search))) { goto_msgid(state, e); - return; + return 0; } } + return 0; } -void +int next_untranslated(struct DrawState* state) { if (state->msgid_count == 0) - return; + return 0; *state->error = 0; @@ -1192,16 +1186,17 @@ next_untranslated(struct DrawState* state) show_edit(state); } - return; + return 0; } } + return 0; } -void +int next_fuzzy(struct DrawState* state) { if (state->msgid_count == 0) - return; + return 0; *state->error = 0; @@ -1228,19 +1223,20 @@ next_fuzzy(struct DrawState* state) show_edit(state); } - return; + return 0; } } + return 0; } -void +int show_edit(struct DrawState* state) { if (!load_msgstr(state) || !load_info(state)) { strcpy(state->error, errors[ERR_CANT_ALLOC]); state->running = 0; - return; + return 0; } state->dirty_before_edit = state->dirty; @@ -1255,16 +1251,17 @@ show_edit(struct DrawState* state) state->info_first_shown_column = 0; move_start(state); *state->error = 0; + return 0; } -void +int move_up(struct DrawState* state) { if (state->edit_info_focused) { if (state->info_first_shown_row > 0) state->info_first_shown_row--; - return; + return 0; } if (state->input_row > 0) @@ -1292,9 +1289,10 @@ move_up(struct DrawState* state) state->input_display_column = display_length(state, state->input_buffer[state->input_row].text, state->input_column); + return 0; } -void +int move_down(struct DrawState* state) { if (state->edit_info_focused) @@ -1302,7 +1300,7 @@ move_down(struct DrawState* state) if (state->info_first_shown_row + (EDIT_HEIGHT - 3) / 2 < state->info_rows_count) state->info_first_shown_row++; - return; + return 0; } if (state->input_row < state->input_rows_count - 1) @@ -1332,9 +1330,10 @@ move_down(struct DrawState* state) state->input_display_column = display_length(state, state->input_buffer[state->input_row].text, state->input_column); + return 0; } -void +int edit_page_up(struct DrawState* state) { if (state->edit_info_focused) @@ -1343,7 +1342,7 @@ edit_page_up(struct DrawState* state) state->info_first_shown_row -= (EDIT_HEIGHT - 3) / 2; else move_start(state); - return; + return 0; } if (state->dirty) @@ -1352,10 +1351,10 @@ edit_page_up(struct DrawState* state) state->dirty = state->dirty_before_edit; move_list_up(state); state->msgstr_index = 0; - show_edit(state); + return show_edit(state); } -void +int edit_page_down(struct DrawState* state) { if (state->edit_info_focused) @@ -1363,7 +1362,7 @@ edit_page_down(struct DrawState* state) if (state->info_first_shown_row + (EDIT_HEIGHT - 3) / 2 < state->info_rows_count - 1) state->info_first_shown_row += (EDIT_HEIGHT - 3) / 2; - return; + return 0; } if (state->dirty) @@ -1372,36 +1371,37 @@ edit_page_down(struct DrawState* state) state->dirty = state->dirty_before_edit; move_list_down(state); state->msgstr_index = 0; - show_edit(state); + return show_edit(state); } -void +int move_start(struct DrawState* state) { if (state->edit_info_focused) { state->info_first_shown_row = 0; state->info_first_shown_column = 0; - return; + return 0; } state->input_column = 0; state->input_display_column = 0; state->input_first_shown_column = 0; + return 0; } -void +int move_left(struct DrawState* state) { if (state->edit_info_focused) { if (state->info_first_shown_column > 0) state->info_first_shown_column--; - return; + return 0; } if (state->input_column == 0) - return; + return 0; if (state->input_buffer[state->input_row].text[state->input_column - 1] == '\t') @@ -1414,9 +1414,10 @@ move_left(struct DrawState* state) if (state->input_display_column < state->input_first_shown_column) state->input_first_shown_column = state->input_display_column; + return 0; } -void +int move_right(struct DrawState* state) { if (state->edit_info_focused) @@ -1424,12 +1425,12 @@ move_right(struct DrawState* state) if (state->info_first_shown_column + state->edit_width - 3 < state->info_rows_maxlen + 1) state->info_first_shown_column++; - return; + return 0; } size_t len = state->input_buffer[state->input_row].length; if (state->input_column == len) - return; + return 0; if (state->input_buffer[state->input_row].text[state->input_column] == '\t') @@ -1444,15 +1445,16 @@ move_right(struct DrawState* state) > state->edit_width - 4) state->input_first_shown_column = state->input_display_column - state->edit_width + 4; + return 0; } -void +int move_prev_word(struct DrawState* state) { if (state->edit_info_focused) { strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); - return; + return 0; } int i = state->input_column; @@ -1472,15 +1474,16 @@ move_prev_word(struct DrawState* state) state->input_column); if (state->input_display_column < state->input_first_shown_column) state->input_first_shown_column = state->input_display_column; + return 0; } -void +int move_next_word(struct DrawState* state) { if (state->edit_info_focused) { strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); - return; + return 0; } int i = state->input_column; @@ -1499,15 +1502,16 @@ move_next_word(struct DrawState* state) > state->edit_width - 4) state->input_first_shown_column = state->input_display_column - state->edit_width + 4; + return 0; } -void +int erase_prev_word(struct DrawState* state) { if (state->edit_info_focused) { strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); - return; + return 0; } int i = state->input_column; @@ -1536,9 +1540,38 @@ erase_prev_word(struct DrawState* state) state->input_buffer[state->input_row].text, state->input_column); state->dirty = 1; + return 0; } -void +int +insert_space(struct DrawState* state) +{ + insert_char(state, (uint32_t)L' '); + return 0; +} + +int +insert_tab(struct DrawState* state) +{ + insert_char(state, (uint32_t)L'\t'); + return 0; +} + +int +insert_search_space(struct DrawState* state) +{ + insert_search_char(state, (uint32_t)L' '); + return 0; +} + +int +insert_search_tab(struct DrawState* state) +{ + insert_search_char(state, (uint32_t)L'\t'); + return 0; +} + +int move_end(struct DrawState* state) { if (state->edit_info_focused) @@ -1547,7 +1580,7 @@ move_end(struct DrawState* state) state->info_first_shown_row = state->info_rows_count - (EDIT_HEIGHT - 3) / 2; state->info_first_shown_column = 0; - return; + return 0; } size_t len = state->input_buffer[state->input_row].length; @@ -1560,9 +1593,10 @@ move_end(struct DrawState* state) state->input_first_shown_column = dlen - state->edit_width + 4; else state->input_first_shown_column = 0; + return 0; } -void +int move_list_up(struct DrawState* state) { if (state->msgid_number > 1) @@ -1572,9 +1606,10 @@ move_list_up(struct DrawState* state) if (state->msgid_number < state->first_shown_msgid) state->first_shown_msgid--; } + return 0; } -void +int move_list_down(struct DrawState* state) { if (state->msgid_number < state->msgid_count) @@ -1585,17 +1620,19 @@ move_list_down(struct DrawState* state) > state->maxy - 2) state->first_shown_msgid++; } + return 0; } -void +int move_list_start(struct DrawState* state) { *state->error = 0; state->msgid_number = state->msgid_count > 0 ? 1 : 0; state->first_shown_msgid = state->msgid_number; + return 0; } -void +int move_list_end(struct DrawState* state) { *state->error = 0; @@ -1603,9 +1640,10 @@ move_list_end(struct DrawState* state) state->first_shown_msgid = state->msgid_count > 0 ? MAX(1, state->msgid_number - (state->maxy - 2)) : 0; + return 0; } -void +int move_list_page_up(struct DrawState* state) { *state->error = 0; @@ -1616,10 +1654,11 @@ move_list_page_up(struct DrawState* state) state->first_shown_msgid = state->msgid_number; } else - move_list_start(state); + return move_list_start(state); + return 0; } -void +int move_list_page_down(struct DrawState* state) { *state->error = 0; @@ -1632,39 +1671,48 @@ move_list_page_down(struct DrawState* state) = state->msgid_number - (state->maxy - 2); } else - move_list_end(state); + return move_list_end(state); + return 0; } -void +int next_plural(struct DrawState* state) { + if (!state) + return 1; if (state->edit_info_focused) - return; + return 0; struct PoEntry* current = &state->entries[state->msgid_number - 1]; save_msgstr(state); if (state->msgstr_index + 1 < current->msgstr_count) state->msgstr_index++; show_edit(state); + return 0; } -void +int prev_plural(struct DrawState* state) { + if (!state) + return 1; if (state->edit_info_focused) - return; + return 0; save_msgstr(state); if (state->msgstr_index > 0) state->msgstr_index--; show_edit(state); + return 0; } -void +int toggle_fuzzy(struct DrawState* state) { + if (!state) + return 1; if (state->msgid_number == 1) { strcpy(state->error, errors[ERR_ILLEGAL_ON_FIRST]); - return; + return 0; } if (state->msgid_count > 0) { @@ -1687,6 +1735,14 @@ toggle_fuzzy(struct DrawState* state) } state->dirty = 1; } + return 0; +} + +int +toggle_info_focused(struct DrawState* state) +{ + state->edit_info_focused = !state->edit_info_focused; + return 0; } void @@ -1728,10 +1784,12 @@ insert_char(struct DrawState* state, uint32_t ch) state->dirty = 1; } -void +int erase_backwards(struct DrawState* state) { struct BufferLine* buffer_line = &state->input_buffer[state->input_row]; + if (state->edit_info_focused) + return 0; if (state->input_column > 0) { @@ -1768,12 +1826,15 @@ erase_backwards(struct DrawState* state) move_end(state); erase_forward(state); } + return 0; } -void +int erase_forward(struct DrawState* state) { struct BufferLine* buffer_line = &state->input_buffer[state->input_row]; + if (state->edit_info_focused) + return 0; if (buffer_line->text[state->input_column]) { @@ -1795,15 +1856,16 @@ erase_forward(struct DrawState* state) state->running = 0; } } + return 0; } -void +int erase_to_start(struct DrawState* state) { if (state->edit_info_focused) { strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); - return; + return 0; } size_t i = state->input_column; @@ -1819,36 +1881,46 @@ erase_to_start(struct DrawState* state) buffer_line->length = len - state->input_column; move_start(state); state->dirty = 1; + return 0; } -void +int erase_to_end(struct DrawState* state) { if (state->edit_info_focused) - return; + return 0; struct BufferLine* buffer_line = &state->input_buffer[state->input_row]; buffer_line->text[state->input_column] = 0; buffer_line->length = state->input_column; move_end(state); state->dirty = 1; + return 0; } -struct DrawState* +int insert_line(struct DrawState* state) { if (state->edit_info_focused) - return state; + return 0; state->input_rows_count++; struct BufferLine* newbuf = realloc(state->input_buffer, state->input_rows_count * sizeof(struct BufferLine)); if (!newbuf) - return NULL; + { + strcpy(state->error, errors[ERR_CANT_ALLOC]); + state->running = 0; + return 0; + } state->input_buffer = newbuf; init_bufferline(&state->input_buffer[state->input_rows_count - 1]); uint32_t* newtext = calloc(MAXBUFLINE, sizeof(uint32_t)); if (!newtext) - return NULL; + { + strcpy(state->error, errors[ERR_CANT_ALLOC]); + state->running = 0; + return 0; + } state->input_buffer[state->input_rows_count - 1].text = newtext; for (size_t i = state->input_rows_count - 1; i > state->input_row + 1; i--) @@ -1875,7 +1947,7 @@ insert_line(struct DrawState* state) move_start(state); state->dirty = 1; - return state; + return 0; } struct DrawState* @@ -1918,9 +1990,10 @@ join_lines(struct DrawState* state, size_t first_line, int forward) } void -handle_editbox_key_event(struct tb_event* ev, struct DrawState* state) +handle_key_event(struct tb_event* ev, struct DrawState* state) { - if (state->msgid_count < 1) + const struct Key* current = NULL; + if (!state) return; if (state->in_prompt) @@ -1933,249 +2006,83 @@ handle_editbox_key_event(struct tb_event* ev, struct DrawState* state) *state->prompt = 0; } else if (ev->key == TB_KEY_ESC) - { - state->in_prompt = 0; - *state->prompt = 0; - } + cancel_close(state); return; } - if (*state->error) - *state->error = 0; - - if (ev->key != TB_KEY_ARROW_DOWN && ev->key != TB_KEY_ARROW_UP) + /* Remember to also edit the following condition if you reassign motion + * keys */ + if (ev->key != TB_KEY_ARROW_DOWN && ev->key != TB_KEY_ARROW_UP + && state->show_edit) state->saved_column = -1; - if (ev->key == TB_KEY_ARROW_DOWN) - move_down(state); - else if (ev->key == TB_KEY_ARROW_LEFT) - move_left(state); - else if (ev->key == TB_KEY_ARROW_RIGHT) - move_right(state); - else if (ev->key == TB_KEY_ARROW_UP) - move_up(state); - else if ((ev->key == TB_KEY_BACKSPACE || ev->key == TB_KEY_BACKSPACE2) - && !state->edit_info_focused) - erase_backwards(state); - else if (ev->key == TB_KEY_CTRL_C) - yank_msgstr_to_paste_buffer(state); - else if (ev->key == TB_KEY_CTRL_K) - erase_to_end(state); - else if (ev->key == TB_KEY_CTRL_N) - next_plural(state); - else if (ev->key == TB_KEY_CTRL_P) - prev_plural(state); - else if (ev->key == TB_KEY_CTRL_Q) - strcpy(state->error, errors[ERR_DLG_OPEN_QUIT]); - else if (ev->key == TB_KEY_CTRL_R) - move_prev_word(state); - else if (ev->key == TB_KEY_CTRL_S) - save_msgstr(state); - else if (ev->key == TB_KEY_CTRL_T) - move_next_word(state); - else if (ev->key == TB_KEY_CTRL_U) - erase_to_start(state); - else if (ev->key == TB_KEY_CTRL_V) - paste_paste_buffer_to_input(state); - else if (ev->key == TB_KEY_CTRL_W) - erase_prev_word(state); - else if (ev->key == TB_KEY_CTRL_Z) - toggle_fuzzy(state); - else if (ev->key == TB_KEY_DELETE && !state->edit_info_focused) - erase_forward(state); - else if (ev->key == TB_KEY_END || ev->key == TB_KEY_CTRL_E) - move_end(state); - else if (ev->key == TB_KEY_ENTER) + for (int key = 0; key < LEN(top_level_keys); key++) { - if (!insert_line(state)) - { - strcpy(state->error, errors[ERR_CANT_ALLOC]); - state->running = 0; - } + current = top_level_keys + key; + if (((current->ev.key == ev->key) && !ev->ch) + || (current->ev.ch + && (current->ev.ch == ev->ch))) + if (current->callback + && !current->callback(state)) + return; } - else if (ev->key == TB_KEY_ESC) - cancel_msgstr(state); - else if (ev->key == TB_KEY_F5) - copy_msgid_to_input(state); - else if (ev->key == TB_KEY_F6) - state->edit_info_focused = !state->edit_info_focused; - else if (ev->key == TB_KEY_F8) - next_untranslated(state); - else if (ev->key == TB_KEY_F9) - next_fuzzy(state); - else if (ev->key == TB_KEY_HOME || ev->key == TB_KEY_CTRL_A) - move_start(state); - else if (ev->key == TB_KEY_PGDN || ev->key == TB_KEY_CTRL_F) - edit_page_down(state); - else if (ev->key == TB_KEY_PGUP || ev->key == TB_KEY_CTRL_B) - edit_page_up(state); - else if (ev->key == TB_KEY_SPACE) - insert_char(state, ' '); - else if (ev->key == TB_KEY_TAB) - insert_char(state, '\t'); - else if (ev->ch != 0) - insert_char(state, ev->ch); -} -void -handle_key_event(struct tb_event* ev, struct DrawState* state) -{ - if (!state) + if (state->show_help) + { + strcpy(state->error, errors[ERR_DLG_OPEN]); return; - - if (state->show_edit) - handle_editbox_key_event(ev, state); - else if (state->show_search) - handle_search_key_event(ev, state); - else if (ev->key == TB_KEY_ESC) + } + else if (state->show_edit) { - if (*state->error) - *state->error = 0; - else if (state->in_prompt) + for (int key = 0; key < LEN(edit_keys); key++) { - state->in_prompt = 0; - *state->prompt = 0; + current = edit_keys + key; + if (((current->ev.key == ev->key) && !ev->ch) + || (current->ev.ch + && (current->ev.ch == ev->ch))) + if (current->callback + && !current->callback(state)) + return; + } + if (ev->ch) + { + insert_char(state, ev->ch); return; } - else if (state->show_help) - state->show_help = 0; - else - strcpy(state->error, errors[ERR_EXIT_KEY]); } - else if ((ev->ch == 'y' || ev->ch == 'n') && state->in_prompt) - { - if (state->prompt_callback) - state->prompt_callback(state, ev); - state->in_prompt = 0; - *state->prompt = 0; - } - else if (state->in_prompt) - return; - else if (ev->ch == 'q' || ev->key == TB_KEY_CTRL_Q) + else if (state->show_search) { - if (state->dirty) + for (int key = 0; key < LEN(search_keys); key++) { - state->in_prompt = 1; - strcpy(state->prompt, prompt_dirty); - state->prompt_callback = quit_callback; + current = search_keys + key; + if (((current->ev.key == ev->key) && !ev->ch) + || (current->ev.ch + && (current->ev.ch == ev->ch))) + if (current->callback + && !current->callback(state)) + return; } - else - state->running = 0; - *state->error = 0; - } - else if (state->show_help) - strcpy(state->error, errors[ERR_DLG_OPEN]); - else if (ev->key == TB_KEY_ARROW_DOWN || ev->ch == 'j') - move_list_down(state); - else if (ev->key == TB_KEY_ARROW_UP || ev->ch == 'k') - move_list_up(state); - else if (ev->key == TB_KEY_CTRL_B || ev->key == TB_KEY_PGUP) - move_list_page_up(state); - else if (ev->key == TB_KEY_CTRL_F || ev->key == TB_KEY_PGDN) - move_list_page_down(state); - else if (ev->key == TB_KEY_CTRL_S || ev->ch == 'w') - { - if (!state->dirty) - return; - int result = save_file(state->entries, state->real_msgid_count, - state->filename, state->nplurals); - switch (result) + if (ev->ch) { - case SAVE_ERR_CANT_ALLOC: - strcpy(state->error, errors[ERR_CANT_ALLOC]); - state->running = 0; - break; - case SAVE_ERR_CANT_OPEN_FILE: - strcpy(state->error, errors[ERR_CANT_SAVE]); - state->running = 0; - break; - case SAVE_ERR_CANT_MOVE_FILE: - strcpy(state->error, strerror(errno)); - state->running = 0; - break; - case SAVE_ERR_NONE: - default: - *state->error = 0; + insert_search_char(state, ev->ch); + return; } - state->dirty = 0; } - else if (ev->key == TB_KEY_CTRL_Z || ev->ch == 'z') - toggle_fuzzy(state); - else if (ev->key == TB_KEY_END || ev->ch == 'G') - move_list_end(state); - else if (ev->key == TB_KEY_ENTER) + else { - if (state->msgid_count == 0) - strcpy(state->error, errors[ERR_EMPTY_FILE]); - else + for (int key = 0; key < LEN(main_keys); key++) { - state->msgstr_index = 0; - show_edit(state); + current = main_keys + key; + if (((current->ev.key == ev->key) && !ev->ch) + || (current->ev.ch + && (current->ev.ch == ev->ch))) + if (current->callback + && !current->callback(state)) + return; } } - else if (ev->key == TB_KEY_F1 || ev->ch == 'H') - { - state->show_help = 1; - *state->error = 0; - } - else if (ev->key == TB_KEY_F3 || ev->ch == 'n') - next_match(state); - else if (ev->key == TB_KEY_F4 || ev->ch == 'N') - prev_match(state); - else if (ev->key == TB_KEY_F7 || ev->ch == '/') - show_search(state); - else if (ev->key == TB_KEY_F8 || ev->ch == 'u') - next_untranslated(state); - else if (ev->key == TB_KEY_F9 || ev->ch == 'f') - next_fuzzy(state); - else if (ev->key == TB_KEY_HOME || ev->ch == 'g') - move_list_start(state); - else - strcpy(state->error, errors[ERR_UNKNOWN_KEY]); -} -void -handle_search_key_event(struct tb_event* ev, struct DrawState* state) -{ - if (!state) - return; - - if (ev->key == TB_KEY_ARROW_LEFT) - move_search_left(state); - else if (ev->key == TB_KEY_ARROW_RIGHT) - move_search_right(state); - else if ((ev->key == TB_KEY_BACKSPACE || ev->key == TB_KEY_BACKSPACE2)) - erase_search_backwards(state); - else if (ev->key == TB_KEY_CTRL_K) - erase_search_to_end(state); - else if (ev->key == TB_KEY_CTRL_R) - move_search_prev_word(state); - else if (ev->key == TB_KEY_CTRL_T) - move_search_next_word(state); - else if (ev->key == TB_KEY_CTRL_U) - erase_search_to_start(state); - else if (ev->key == TB_KEY_CTRL_W) - erase_search_prev_word(state); - else if (ev->key == TB_KEY_DELETE) - erase_search_forward(state); - else if (ev->key == TB_KEY_END || ev->key == TB_KEY_CTRL_E) - move_search_end(state); - else if (ev->key == TB_KEY_ENTER) - hide_search(state); - else if (ev->key == TB_KEY_ESC) - cancel_search(state); - else if (ev->key == TB_KEY_F3) - next_match(state); - else if (ev->key == TB_KEY_F4) - prev_match(state); - else if (ev->key == TB_KEY_HOME || ev->key == TB_KEY_CTRL_A) - move_search_start(state); - else if (ev->key == TB_KEY_SPACE) - insert_search_char(state, ' '); - else if (ev->key == TB_KEY_TAB) - insert_search_char(state, '\t'); - else if (ev->ch != 0) - insert_search_char(state, ev->ch); + strcpy(state->error, errors[ERR_UNKNOWN_KEY]); } int @@ -2197,9 +2104,16 @@ main(int argc, char** argv) return print_error(1, "No filename given"); if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help")) - return usage(); + { + printf("Usage: %s [-h|--help] [-v|--version] [filename.po]\n", + program_name); + return 0; + } else if (!strcmp(argv[1], "-v") || !strcmp(argv[1], "--version")) - return version(); + { + printf("%s %s, built on %s\n", program_name, VERSION, DATE); + return 0; + } else { strncpy(filename, argv[1], MAXPATH - 1); diff --git a/poe.do b/poe.do @@ -5,4 +5,4 @@ for f in *.h *.c; do echo $f echo $f | sed -e's/\.c$/.o/g' done | xargs redo-ifchange -${POE_CC:-gcc -g} -Wall -std=c99 -o $3 po.o poe.o draw.o termbox.o utf8.o util.o +${POE_CC:-gcc -g} -Wall -std=c99 -o $3 po.o poe.o draw.o util.o diff --git a/term.inl b/term.inl @@ -1,311 +0,0 @@ -enum { - T_ENTER_CA, - T_EXIT_CA, - T_SHOW_CURSOR, - T_HIDE_CURSOR, - T_CLEAR_SCREEN, - T_SGR0, - T_UNDERLINE, - T_BOLD, - T_BLINK, - T_REVERSE, - T_ENTER_KEYPAD, - T_EXIT_KEYPAD, - T_ENTER_MOUSE, - T_EXIT_MOUSE, - T_FUNCS_NUM, -}; - -#define ENTER_MOUSE_SEQ "\x1b[?1000h\x1b[?1002h\x1b[?1015h\x1b[?1006h" -#define EXIT_MOUSE_SEQ "\x1b[?1006l\x1b[?1015l\x1b[?1002l\x1b[?1000l" - -#define EUNSUPPORTED_TERM -1 - -// rxvt-256color -static const char *rxvt_256color_keys[] = { - "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 -}; -static const char *rxvt_256color_funcs[] = { - "\0337\033[?47h", "\033[2J\033[?47l\0338", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033=", "\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, -}; - -// Eterm -static const char *eterm_keys[] = { - "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 -}; -static const char *eterm_funcs[] = { - "\0337\033[?47h", "\033[2J\033[?47l\0338", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "", "", "", "", -}; - -// screen -static const char *screen_keys[] = { - "\033OP","\033OQ","\033OR","\033OS","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[1~","\033[4~","\033[5~","\033[6~","\033OA","\033OB","\033OD","\033OC", 0 -}; -static const char *screen_funcs[] = { - "\033[?1049h", "\033[?1049l", "\033[34h\033[?25h", "\033[?25l", "\033[H\033[J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033[?1h\033=", "\033[?1l\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, -}; - -// rxvt-unicode -static const char *rxvt_unicode_keys[] = { - "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 -}; -static const char *rxvt_unicode_funcs[] = { - "\033[?1049h", "\033[r\033[?1049l", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m\033(B", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033=", "\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, -}; - -// linux -static const char *linux_keys[] = { - "\033[[A","\033[[B","\033[[C","\033[[D","\033[[E","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[1~","\033[4~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 -}; -static const char *linux_funcs[] = { - "", "", "\033[?25h\033[?0c", "\033[?25l\033[?1c", "\033[H\033[J", "\033[0;10m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "", "", "", "", -}; - -// xterm -static const char *xterm_keys[] = { - "\033OP","\033OQ","\033OR","\033OS","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033OH","\033OF","\033[5~","\033[6~","\033OA","\033OB","\033OD","\033OC", 0 -}; -static const char *xterm_funcs[] = { - "\033[?1049h", "\033[?1049l", "\033[?12l\033[?25h", "\033[?25l", "\033[H\033[2J", "\033(B\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033[?1h\033=", "\033[?1l\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, -}; - -static struct term { - const char *name; - const char **keys; - const char **funcs; -} terms[] = { - {"rxvt-256color", rxvt_256color_keys, rxvt_256color_funcs}, - {"Eterm", eterm_keys, eterm_funcs}, - {"screen", screen_keys, screen_funcs}, - {"rxvt-unicode", rxvt_unicode_keys, rxvt_unicode_funcs}, - {"linux", linux_keys, linux_funcs}, - {"xterm", xterm_keys, xterm_funcs}, - {0, 0, 0}, -}; - -static bool init_from_terminfo = false; -static const char **keys; -static const char **funcs; - -static int try_compatible(const char *term, const char *name, - const char **tkeys, const char **tfuncs) -{ - if (strstr(term, name)) { - keys = tkeys; - funcs = tfuncs; - return 0; - } - - return EUNSUPPORTED_TERM; -} - -static int init_term_builtin(void) -{ - int i; - const char *term = getenv("TERM"); - - if (term) { - for (i = 0; terms[i].name; i++) { - if (!strcmp(terms[i].name, term)) { - keys = terms[i].keys; - funcs = terms[i].funcs; - return 0; - } - } - - /* let's do some heuristic, maybe it's a compatible terminal */ - if (try_compatible(term, "xterm", xterm_keys, xterm_funcs) == 0) - return 0; - if (try_compatible(term, "rxvt", rxvt_unicode_keys, rxvt_unicode_funcs) == 0) - return 0; - if (try_compatible(term, "linux", linux_keys, linux_funcs) == 0) - return 0; - if (try_compatible(term, "Eterm", eterm_keys, eterm_funcs) == 0) - return 0; - if (try_compatible(term, "screen", screen_keys, screen_funcs) == 0) - return 0; - if (try_compatible(term, "tmux", screen_keys, screen_funcs) == 0) - return 0; - /* let's assume that 'cygwin' is xterm compatible */ - if (try_compatible(term, "cygwin", xterm_keys, xterm_funcs) == 0) - return 0; - } - - return EUNSUPPORTED_TERM; -} - -//---------------------------------------------------------------------- -// terminfo -//---------------------------------------------------------------------- - -static char *read_file(const char *file) { - FILE *f = fopen(file, "rb"); - if (!f) - return 0; - - struct stat st; - if (fstat(fileno(f), &st) != 0) { - fclose(f); - return 0; - } - - char *data = malloc(st.st_size); - if (!data) { - fclose(f); - return 0; - } - - if (fread(data, 1, st.st_size, f) != (size_t)st.st_size) { - fclose(f); - free(data); - return 0; - } - - fclose(f); - return data; -} - -static char *terminfo_try_path(const char *path, const char *term) { - char tmp[4096]; - snprintf(tmp, sizeof(tmp), "%s/%c/%s", path, term[0], term); - tmp[sizeof(tmp)-1] = '\0'; - char *data = read_file(tmp); - if (data) { - return data; - } - - // fallback to darwin specific dirs structure - snprintf(tmp, sizeof(tmp), "%s/%x/%s", path, term[0], term); - tmp[sizeof(tmp)-1] = '\0'; - return read_file(tmp); -} - -static char *load_terminfo(void) { - char tmp[4096]; - const char *term = getenv("TERM"); - if (!term) { - return 0; - } - - // if TERMINFO is set, no other directory should be searched - const char *terminfo = getenv("TERMINFO"); - if (terminfo) { - return terminfo_try_path(terminfo, term); - } - - // next, consider ~/.terminfo - const char *home = getenv("HOME"); - if (home) { - snprintf(tmp, sizeof(tmp), "%s/.terminfo", home); - tmp[sizeof(tmp)-1] = '\0'; - char *data = terminfo_try_path(tmp, term); - if (data) - return data; - } - - // next, TERMINFO_DIRS - const char *dirs = getenv("TERMINFO_DIRS"); - if (dirs) { - snprintf(tmp, sizeof(tmp), "%s", dirs); - tmp[sizeof(tmp)-1] = '\0'; - char *dir = strtok(tmp, ":"); - while (dir) { - const char *cdir = dir; - if (strcmp(cdir, "") == 0) { - cdir = "/usr/share/terminfo"; - } - char *data = terminfo_try_path(cdir, term); - if (data) - return data; - dir = strtok(0, ":"); - } - } - - // fallback to /usr/share/terminfo - return terminfo_try_path("/usr/share/terminfo", term); -} - -#define TI_MAGIC 0432 -#define TI_ALT_MAGIC 542 -#define TI_HEADER_LENGTH 12 -#define TB_KEYS_NUM 22 - -static const char *terminfo_copy_string(char *data, int str, int table) { - const int16_t off = *(int16_t*)(data + str); - const char *src = data + table + off; - int len = strlen(src); - char *dst = malloc(len+1); - strcpy(dst, src); - return dst; -} - -static const int16_t ti_funcs[] = { - 28, 40, 16, 13, 5, 39, 36, 27, 26, 34, 89, 88, -}; - -static const int16_t ti_keys[] = { - 66, 68 /* apparently not a typo; 67 is F10 for whatever reason */, 69, - 70, 71, 72, 73, 74, 75, 67, 216, 217, 77, 59, 76, 164, 82, 81, 87, 61, - 79, 83, -}; - -static int init_term(void) { - int i; - char *data = load_terminfo(); - if (!data) { - init_from_terminfo = false; - return init_term_builtin(); - } - - int16_t *header = (int16_t*)data; - - const int number_sec_len = header[0] == TI_ALT_MAGIC ? 4 : 2; - - if ((header[1] + header[2]) % 2) { - // old quirk to align everything on word boundaries - header[2] += 1; - } - - const int str_offset = TI_HEADER_LENGTH + - header[1] + header[2] + number_sec_len * header[3]; - const int table_offset = str_offset + 2 * header[4]; - - keys = malloc(sizeof(const char*) * (TB_KEYS_NUM+1)); - for (i = 0; i < TB_KEYS_NUM; i++) { - keys[i] = terminfo_copy_string(data, - str_offset + 2 * ti_keys[i], table_offset); - } - keys[TB_KEYS_NUM] = 0; - - funcs = malloc(sizeof(const char*) * T_FUNCS_NUM); - // the last two entries are reserved for mouse. because the table offset is - // not there, the two entries have to fill in manually - for (i = 0; i < T_FUNCS_NUM-2; i++) { - funcs[i] = terminfo_copy_string(data, - str_offset + 2 * ti_funcs[i], table_offset); - } - - funcs[T_FUNCS_NUM-2] = ENTER_MOUSE_SEQ; - funcs[T_FUNCS_NUM-1] = EXIT_MOUSE_SEQ; - - init_from_terminfo = true; - free(data); - return 0; -} - -static void shutdown_term(void) { - if (init_from_terminfo) { - int i; - for (i = 0; i < TB_KEYS_NUM; i++) { - free((void*)keys[i]); - } - // the last two entries are reserved for mouse. because the table offset - // is not there, the two entries have to fill in manually and do not - // need to be freed. - for (i = 0; i < T_FUNCS_NUM-2; i++) { - free((void*)funcs[i]); - } - free(keys); - free(funcs); - } -} diff --git a/termbox.c b/termbox.c @@ -1,681 +0,0 @@ -#define _POSIX_C_SOURCE 200809L -#define _XOPEN_SOURCE - -#include <assert.h> -#include <stdlib.h> -#include <string.h> -#include <errno.h> -#include <fcntl.h> -#include <signal.h> -#include <stdio.h> -#include <stdbool.h> -#include <sys/select.h> -#include <sys/ioctl.h> -#include <sys/time.h> -#include <sys/stat.h> -#include <termios.h> -#include <unistd.h> -#include <wchar.h> - -#include "termbox.h" - -#include "bytebuffer.inl" -#include "term.inl" -#include "input.inl" - -struct cellbuf { - int width; - int height; - struct tb_cell *cells; -}; - -#define CELL(buf, x, y) (buf)->cells[(y) * (buf)->width + (x)] -#define IS_CURSOR_HIDDEN(cx, cy) (cx == -1 || cy == -1) -#define LAST_COORD_INIT -1 - -static struct termios orig_tios; - -static struct cellbuf back_buffer; -static struct cellbuf front_buffer; -static struct bytebuffer output_buffer; -static struct bytebuffer input_buffer; - -static int termw = -1; -static int termh = -1; - -static int inputmode = TB_INPUT_ESC; -static int outputmode = TB_OUTPUT_NORMAL; - -static int inout; -static int winch_fds[2]; - -static int lastx = LAST_COORD_INIT; -static int lasty = LAST_COORD_INIT; -static int cursor_x = -1; -static int cursor_y = -1; - -static uint16_t background = TB_DEFAULT; -static uint16_t foreground = TB_DEFAULT; - -static void write_cursor(int x, int y); -static void write_sgr(uint16_t fg, uint16_t bg); - -static void cellbuf_init(struct cellbuf *buf, int width, int height); -static void cellbuf_resize(struct cellbuf *buf, int width, int height); -static void cellbuf_clear(struct cellbuf *buf); -static void cellbuf_free(struct cellbuf *buf); - -static void update_size(void); -static void update_term_size(void); -static void send_attr(uint16_t fg, uint16_t bg); -static void send_char(int x, int y, uint32_t c); -static void send_clear(void); -static void sigwinch_handler(int xxx); -static int wait_fill_event(struct tb_event *event, struct timeval *timeout); - -/* may happen in a different thread */ -static volatile int buffer_size_change_request; - -/* -------------------------------------------------------- */ - -int tb_init_fd(int inout_) -{ - inout = inout_; - if (inout == -1) { - return TB_EFAILED_TO_OPEN_TTY; - } - - if (init_term() < 0) { - close(inout); - return TB_EUNSUPPORTED_TERMINAL; - } - - if (pipe(winch_fds) < 0) { - close(inout); - return TB_EPIPE_TRAP_ERROR; - } - - struct sigaction sa; - memset(&sa, 0, sizeof(sa)); - sa.sa_handler = sigwinch_handler; - sa.sa_flags = 0; - sigaction(SIGWINCH, &sa, 0); - - tcgetattr(inout, &orig_tios); - - struct termios tios; - memcpy(&tios, &orig_tios, sizeof(tios)); - - tios.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP - | INLCR | IGNCR | ICRNL | IXON); - tios.c_oflag &= ~OPOST; - tios.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN); - tios.c_cflag &= ~(CSIZE | PARENB); - tios.c_cflag |= CS8; - tios.c_cc[VMIN] = 0; - tios.c_cc[VTIME] = 0; - tcsetattr(inout, TCSAFLUSH, &tios); - - bytebuffer_init(&input_buffer, 128); - bytebuffer_init(&output_buffer, 32 * 1024); - - bytebuffer_puts(&output_buffer, funcs[T_ENTER_CA]); - bytebuffer_puts(&output_buffer, funcs[T_ENTER_KEYPAD]); - bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); - send_clear(); - - update_term_size(); - cellbuf_init(&back_buffer, termw, termh); - cellbuf_init(&front_buffer, termw, termh); - cellbuf_clear(&back_buffer); - cellbuf_clear(&front_buffer); - - return 0; -} - -int tb_init_file(const char* name){ - return tb_init_fd(open(name, O_RDWR)); -} - -int tb_init(void) -{ - return tb_init_file("/dev/tty"); -} - -void tb_shutdown(void) -{ - if (termw == -1) { - fputs("tb_shutdown() should not be called twice.", stderr); - abort(); - } - - bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); - bytebuffer_puts(&output_buffer, funcs[T_SGR0]); - bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_CA]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_KEYPAD]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - tcsetattr(inout, TCSAFLUSH, &orig_tios); - - shutdown_term(); - close(inout); - close(winch_fds[0]); - close(winch_fds[1]); - - cellbuf_free(&back_buffer); - cellbuf_free(&front_buffer); - bytebuffer_free(&output_buffer); - bytebuffer_free(&input_buffer); - termw = termh = -1; -} - -void tb_present(void) -{ - int x,y,w,i; - struct tb_cell *back, *front; - - /* invalidate cursor position */ - lastx = LAST_COORD_INIT; - lasty = LAST_COORD_INIT; - - if (buffer_size_change_request) { - update_size(); - buffer_size_change_request = 0; - } - - for (y = 0; y < front_buffer.height; ++y) { - for (x = 0; x < front_buffer.width; ) { - back = &CELL(&back_buffer, x, y); - front = &CELL(&front_buffer, x, y); - w = wcwidth(back->ch); - if (w < 1) w = 1; - if (memcmp(back, front, sizeof(struct tb_cell)) == 0) { - x += w; - continue; - } - memcpy(front, back, sizeof(struct tb_cell)); - send_attr(back->fg, back->bg); - if (w > 1 && x >= front_buffer.width - (w - 1)) { - // Not enough room for wide ch, so send spaces - for (i = x; i < front_buffer.width; ++i) { - send_char(i, y, ' '); - } - } else { - send_char(x, y, back->ch); - for (i = 1; i < w; ++i) { - front = &CELL(&front_buffer, x + i, y); - front->ch = 0; - front->fg = back->fg; - front->bg = back->bg; - } - } - x += w; - } - } - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); - bytebuffer_flush(&output_buffer, inout); -} - -void tb_set_cursor(int cx, int cy) -{ - if (IS_CURSOR_HIDDEN(cursor_x, cursor_y) && !IS_CURSOR_HIDDEN(cx, cy)) - bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); - - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y) && IS_CURSOR_HIDDEN(cx, cy)) - bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); - - cursor_x = cx; - cursor_y = cy; - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); -} - -void tb_put_cell(int x, int y, const struct tb_cell *cell) -{ - if ((unsigned)x >= (unsigned)back_buffer.width) - return; - if ((unsigned)y >= (unsigned)back_buffer.height) - return; - CELL(&back_buffer, x, y) = *cell; -} - -void tb_change_cell(int x, int y, uint32_t ch, uint16_t fg, uint16_t bg) -{ - struct tb_cell c = {ch, fg, bg}; - tb_put_cell(x, y, &c); -} - -void tb_blit(int x, int y, int w, int h, const struct tb_cell *cells) -{ - if (x + w < 0 || x >= back_buffer.width) - return; - if (y + h < 0 || y >= back_buffer.height) - return; - int xo = 0, yo = 0, ww = w, hh = h; - if (x < 0) { - xo = -x; - ww -= xo; - x = 0; - } - if (y < 0) { - yo = -y; - hh -= yo; - y = 0; - } - if (ww > back_buffer.width - x) - ww = back_buffer.width - x; - if (hh > back_buffer.height - y) - hh = back_buffer.height - y; - - int sy; - struct tb_cell *dst = &CELL(&back_buffer, x, y); - const struct tb_cell *src = cells + yo * w + xo; - size_t size = sizeof(struct tb_cell) * ww; - - for (sy = 0; sy < hh; ++sy) { - memcpy(dst, src, size); - dst += back_buffer.width; - src += w; - } -} - -struct tb_cell *tb_cell_buffer(void) -{ - return back_buffer.cells; -} - -int tb_poll_event(struct tb_event *event) -{ - return wait_fill_event(event, 0); -} - -int tb_peek_event(struct tb_event *event, int timeout) -{ - struct timeval tv; - tv.tv_sec = timeout / 1000; - tv.tv_usec = (timeout - (tv.tv_sec * 1000)) * 1000; - return wait_fill_event(event, &tv); -} - -int tb_width(void) -{ - return termw; -} - -int tb_height(void) -{ - return termh; -} - -void tb_clear(void) -{ - if (buffer_size_change_request) { - update_size(); - buffer_size_change_request = 0; - } - cellbuf_clear(&back_buffer); -} - -int tb_select_input_mode(int mode) -{ - if (mode) { - if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == 0) - mode |= TB_INPUT_ESC; - - /* technically termbox can handle that, but let's be nice and show here - what mode is actually used */ - if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == (TB_INPUT_ESC | TB_INPUT_ALT)) - mode &= ~TB_INPUT_ALT; - - inputmode = mode; - if (mode&TB_INPUT_MOUSE) { - bytebuffer_puts(&output_buffer, funcs[T_ENTER_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - } else { - bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - } - } - return inputmode; -} - -int tb_select_output_mode(int mode) -{ - if (mode) - outputmode = mode; - return outputmode; -} - -void tb_set_clear_attributes(uint16_t fg, uint16_t bg) -{ - foreground = fg; - background = bg; -} - -/* -------------------------------------------------------- */ - -static int convertnum(uint32_t num, char* buf) { - int i, l = 0; - int ch; - do { - buf[l++] = '0' + (num % 10); - num /= 10; - } while (num); - for(i = 0; i < l / 2; i++) { - ch = buf[i]; - buf[i] = buf[l - 1 - i]; - buf[l - 1 - i] = ch; - } - return l; -} - -#define WRITE_LITERAL(X) bytebuffer_append(&output_buffer, (X), sizeof(X)-1) -#define WRITE_INT(X) bytebuffer_append(&output_buffer, buf, convertnum((X), buf)) - -static void write_cursor(int x, int y) { - char buf[32]; - WRITE_LITERAL("\033["); - WRITE_INT(y+1); - WRITE_LITERAL(";"); - WRITE_INT(x+1); - WRITE_LITERAL("H"); -} - -static void write_sgr(uint16_t fg, uint16_t bg) { - char buf[32]; - - if (fg == TB_DEFAULT && bg == TB_DEFAULT) - return; - - switch (outputmode) { - case TB_OUTPUT_256: - case TB_OUTPUT_216: - case TB_OUTPUT_GRAYSCALE: - WRITE_LITERAL("\033["); - if (fg != TB_DEFAULT) { - WRITE_LITERAL("38;5;"); - WRITE_INT(fg); - if (bg != TB_DEFAULT) { - WRITE_LITERAL(";"); - } - } - if (bg != TB_DEFAULT) { - WRITE_LITERAL("48;5;"); - WRITE_INT(bg); - } - WRITE_LITERAL("m"); - break; - case TB_OUTPUT_NORMAL: - default: - WRITE_LITERAL("\033["); - if (fg != TB_DEFAULT) { - WRITE_LITERAL("3"); - WRITE_INT(fg - 1); - if (bg != TB_DEFAULT) { - WRITE_LITERAL(";"); - } - } - if (bg != TB_DEFAULT) { - WRITE_LITERAL("4"); - WRITE_INT(bg - 1); - } - WRITE_LITERAL("m"); - break; - } -} - -static void cellbuf_init(struct cellbuf *buf, int width, int height) -{ - buf->cells = (struct tb_cell*)malloc(sizeof(struct tb_cell) * width * height); - assert(buf->cells); - buf->width = width; - buf->height = height; -} - -static void cellbuf_resize(struct cellbuf *buf, int width, int height) -{ - if (buf->width == width && buf->height == height) - return; - - int oldw = buf->width; - int oldh = buf->height; - struct tb_cell *oldcells = buf->cells; - - cellbuf_init(buf, width, height); - cellbuf_clear(buf); - - int minw = (width < oldw) ? width : oldw; - int minh = (height < oldh) ? height : oldh; - int i; - - for (i = 0; i < minh; ++i) { - struct tb_cell *csrc = oldcells + (i * oldw); - struct tb_cell *cdst = buf->cells + (i * width); - memcpy(cdst, csrc, sizeof(struct tb_cell) * minw); - } - - free(oldcells); -} - -static void cellbuf_clear(struct cellbuf *buf) -{ - int i; - int ncells = buf->width * buf->height; - - for (i = 0; i < ncells; ++i) { - buf->cells[i].ch = ' '; - buf->cells[i].fg = foreground; - buf->cells[i].bg = background; - } -} - -static void cellbuf_free(struct cellbuf *buf) -{ - free(buf->cells); -} - -static void get_term_size(int *w, int *h) -{ - struct winsize sz; - memset(&sz, 0, sizeof(sz)); - - ioctl(inout, TIOCGWINSZ, &sz); - - if (w) *w = sz.ws_col; - if (h) *h = sz.ws_row; -} - -static void update_term_size(void) -{ - struct winsize sz; - memset(&sz, 0, sizeof(sz)); - - ioctl(inout, TIOCGWINSZ, &sz); - - termw = sz.ws_col; - termh = sz.ws_row; -} - -static void send_attr(uint16_t fg, uint16_t bg) -{ -#define LAST_ATTR_INIT 0xFFFF - static uint16_t lastfg = LAST_ATTR_INIT, lastbg = LAST_ATTR_INIT; - if (fg != lastfg || bg != lastbg) { - bytebuffer_puts(&output_buffer, funcs[T_SGR0]); - - uint16_t fgcol; - uint16_t bgcol; - - switch (outputmode) { - case TB_OUTPUT_256: - fgcol = fg & 0xFF; - bgcol = bg & 0xFF; - break; - - case TB_OUTPUT_216: - fgcol = fg & 0xFF; if (fgcol > 215) fgcol = 7; - bgcol = bg & 0xFF; if (bgcol > 215) bgcol = 0; - fgcol += 0x10; - bgcol += 0x10; - break; - - case TB_OUTPUT_GRAYSCALE: - fgcol = fg & 0xFF; if (fgcol > 23) fgcol = 23; - bgcol = bg & 0xFF; if (bgcol > 23) bgcol = 0; - fgcol += 0xe8; - bgcol += 0xe8; - break; - - case TB_OUTPUT_NORMAL: - default: - fgcol = fg & 0x0F; - bgcol = bg & 0x0F; - } - - if (fg & TB_BOLD) - bytebuffer_puts(&output_buffer, funcs[T_BOLD]); - if (bg & TB_BOLD) - bytebuffer_puts(&output_buffer, funcs[T_BLINK]); - if (fg & TB_UNDERLINE) - bytebuffer_puts(&output_buffer, funcs[T_UNDERLINE]); - if ((fg & TB_REVERSE) || (bg & TB_REVERSE)) - bytebuffer_puts(&output_buffer, funcs[T_REVERSE]); - - write_sgr(fgcol, bgcol); - - lastfg = fg; - lastbg = bg; - } -} - -static void send_char(int x, int y, uint32_t c) -{ - char buf[7]; - int bw = tb_utf8_unicode_to_char(buf, c); - if (x-1 != lastx || y != lasty) - write_cursor(x, y); - lastx = x; lasty = y; - if(!c) buf[0] = ' '; // replace 0 with whitespace - bytebuffer_append(&output_buffer, buf, bw); -} - -static void send_clear(void) -{ - send_attr(foreground, background); - bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); - bytebuffer_flush(&output_buffer, inout); - - /* we need to invalidate cursor position too and these two vars are - * used only for simple cursor positioning optimization, cursor - * actually may be in the correct place, but we simply discard - * optimization once and it gives us simple solution for the case when - * cursor moved */ - lastx = LAST_COORD_INIT; - lasty = LAST_COORD_INIT; -} - -static void sigwinch_handler(int xxx) -{ - (void) xxx; - const int zzz = 1; - write(winch_fds[1], &zzz, sizeof(int)); -} - -static void update_size(void) -{ - update_term_size(); - cellbuf_resize(&back_buffer, termw, termh); - cellbuf_resize(&front_buffer, termw, termh); - cellbuf_clear(&front_buffer); - send_clear(); -} - -static int read_up_to(int n) { - assert(n > 0); - const int prevlen = input_buffer.len; - bytebuffer_resize(&input_buffer, prevlen + n); - - int read_n = 0; - while (read_n <= n) { - ssize_t r = 0; - if (read_n < n) { - r = read(inout, input_buffer.buf + prevlen + read_n, n - read_n); - } -#ifdef __CYGWIN__ - // While linux man for tty says when VMIN == 0 && VTIME == 0, read - // should return 0 when there is nothing to read, cygwin's read returns - // -1. Not sure why and if it's correct to ignore it, but let's pretend - // it's zero. - if (r < 0) r = 0; -#endif - if (r < 0) { - // EAGAIN / EWOULDBLOCK shouldn't occur here - assert(errno != EAGAIN && errno != EWOULDBLOCK); - return -1; - } else if (r > 0) { - read_n += r; - } else { - bytebuffer_resize(&input_buffer, prevlen + read_n); - return read_n; - } - } - assert(!"unreachable"); - return 0; -} - -static int wait_fill_event(struct tb_event *event, struct timeval *timeout) -{ - // ;-) -#define ENOUGH_DATA_FOR_PARSING 64 - fd_set events; - memset(event, 0, sizeof(struct tb_event)); - - // try to extract event from input buffer, return on success - event->type = TB_EVENT_KEY; - if (extract_event(event, &input_buffer, inputmode)) - return event->type; - - // it looks like input buffer is incomplete, let's try the short path, - // but first make sure there is enough space - int n = read_up_to(ENOUGH_DATA_FOR_PARSING); - if (n < 0) - return -1; - if (n > 0 && extract_event(event, &input_buffer, inputmode)) - return event->type; - - // n == 0, or not enough data, let's go to select - while (1) { - FD_ZERO(&events); - FD_SET(inout, &events); - FD_SET(winch_fds[0], &events); - int maxfd = (winch_fds[0] > inout) ? winch_fds[0] : inout; - int result = select(maxfd+1, &events, 0, 0, timeout); - if (!result) - return 0; - - if (FD_ISSET(inout, &events)) { - event->type = TB_EVENT_KEY; - n = read_up_to(ENOUGH_DATA_FOR_PARSING); - if (n < 0) - return -1; - - if (n == 0) - continue; - - if (extract_event(event, &input_buffer, inputmode)) - return event->type; - } - if (FD_ISSET(winch_fds[0], &events)) { - event->type = TB_EVENT_RESIZE; - int zzz = 0; - read(winch_fds[0], &zzz, sizeof(int)); - buffer_size_change_request = 1; - get_term_size(&event->w, &event->h); - return TB_EVENT_RESIZE; - } - } -} diff --git a/termbox.h b/termbox.h @@ -1,55 +1,60 @@ -#pragma once +/* +MIT License -#include <stdint.h> +Copyright (c) 2010-2020 nsf <no.smile.face@gmail.com> + 2015-2021 Adam Saponara <as@php.net> + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. -/* for shared objects */ -#if __GNUC__ >= 4 - #define SO_IMPORT __attribute__((visibility("default"))) +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. +*/ + +#ifndef __TERMBOX_H +#define __TERMBOX_H + +#include <errno.h> +#include <fcntl.h> +#include <stdarg.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/ioctl.h> +#include <sys/stat.h> +#include <sys/time.h> +#include <sys/types.h> +#if !TB_OPT_SELECT +#include <poll.h> #else - #define SO_IMPORT +#include <sys/select.h> #endif +#include <limits.h> +#include <signal.h> +#include <termios.h> +#include <unistd.h> +#include <wchar.h> #ifdef __cplusplus -extern "C" { +extern "C" { // __ffi_strip #endif -/* Key constants. See also struct tb_event's key field. - * - * These are a safe subset of terminfo keys, which exist on all popular - * terminals. Termbox uses only them to stay truly portable. - */ -#define TB_KEY_F1 (0xFFFF-0) -#define TB_KEY_F2 (0xFFFF-1) -#define TB_KEY_F3 (0xFFFF-2) -#define TB_KEY_F4 (0xFFFF-3) -#define TB_KEY_F5 (0xFFFF-4) -#define TB_KEY_F6 (0xFFFF-5) -#define TB_KEY_F7 (0xFFFF-6) -#define TB_KEY_F8 (0xFFFF-7) -#define TB_KEY_F9 (0xFFFF-8) -#define TB_KEY_F10 (0xFFFF-9) -#define TB_KEY_F11 (0xFFFF-10) -#define TB_KEY_F12 (0xFFFF-11) -#define TB_KEY_INSERT (0xFFFF-12) -#define TB_KEY_DELETE (0xFFFF-13) -#define TB_KEY_HOME (0xFFFF-14) -#define TB_KEY_END (0xFFFF-15) -#define TB_KEY_PGUP (0xFFFF-16) -#define TB_KEY_PGDN (0xFFFF-17) -#define TB_KEY_ARROW_UP (0xFFFF-18) -#define TB_KEY_ARROW_DOWN (0xFFFF-19) -#define TB_KEY_ARROW_LEFT (0xFFFF-20) -#define TB_KEY_ARROW_RIGHT (0xFFFF-21) -#define TB_KEY_MOUSE_LEFT (0xFFFF-22) -#define TB_KEY_MOUSE_RIGHT (0xFFFF-23) -#define TB_KEY_MOUSE_MIDDLE (0xFFFF-24) -#define TB_KEY_MOUSE_RELEASE (0xFFFF-25) -#define TB_KEY_MOUSE_WHEEL_UP (0xFFFF-26) -#define TB_KEY_MOUSE_WHEEL_DOWN (0xFFFF-27) - -/* These are all ASCII code points below SPACE character and a BACKSPACE key. */ +/* ASCII key constants (tb_event.key) */ #define TB_KEY_CTRL_TILDE 0x00 -#define TB_KEY_CTRL_2 0x00 /* clash with 'CTRL_TILDE' */ +#define TB_KEY_CTRL_2 0x00 /* clash with 'CTRL_TILDE' */ #define TB_KEY_CTRL_A 0x01 #define TB_KEY_CTRL_B 0x02 #define TB_KEY_CTRL_C 0x03 @@ -60,14 +65,14 @@ extern "C" { #define TB_KEY_BACKSPACE 0x08 #define TB_KEY_CTRL_H 0x08 /* clash with 'CTRL_BACKSPACE' */ #define TB_KEY_TAB 0x09 -#define TB_KEY_CTRL_I 0x09 /* clash with 'TAB' */ -#define TB_KEY_CTRL_J 0x0A -#define TB_KEY_CTRL_K 0x0B -#define TB_KEY_CTRL_L 0x0C -#define TB_KEY_ENTER 0x0D -#define TB_KEY_CTRL_M 0x0D /* clash with 'ENTER' */ -#define TB_KEY_CTRL_N 0x0E -#define TB_KEY_CTRL_O 0x0F +#define TB_KEY_CTRL_I 0x09 /* clash with 'TAB' */ +#define TB_KEY_CTRL_J 0x0a +#define TB_KEY_CTRL_K 0x0b +#define TB_KEY_CTRL_L 0x0c +#define TB_KEY_ENTER 0x0d +#define TB_KEY_CTRL_M 0x0d /* clash with 'ENTER' */ +#define TB_KEY_CTRL_N 0x0e +#define TB_KEY_CTRL_O 0x0f #define TB_KEY_CTRL_P 0x10 #define TB_KEY_CTRL_Q 0x11 #define TB_KEY_CTRL_R 0x12 @@ -78,168 +83,317 @@ extern "C" { #define TB_KEY_CTRL_W 0x17 #define TB_KEY_CTRL_X 0x18 #define TB_KEY_CTRL_Y 0x19 -#define TB_KEY_CTRL_Z 0x1A -#define TB_KEY_ESC 0x1B -#define TB_KEY_CTRL_LSQ_BRACKET 0x1B /* clash with 'ESC' */ -#define TB_KEY_CTRL_3 0x1B /* clash with 'ESC' */ -#define TB_KEY_CTRL_4 0x1C -#define TB_KEY_CTRL_BACKSLASH 0x1C /* clash with 'CTRL_4' */ -#define TB_KEY_CTRL_5 0x1D -#define TB_KEY_CTRL_RSQ_BRACKET 0x1D /* clash with 'CTRL_5' */ -#define TB_KEY_CTRL_6 0x1E -#define TB_KEY_CTRL_7 0x1F -#define TB_KEY_CTRL_SLASH 0x1F /* clash with 'CTRL_7' */ -#define TB_KEY_CTRL_UNDERSCORE 0x1F /* clash with 'CTRL_7' */ +#define TB_KEY_CTRL_Z 0x1a +#define TB_KEY_ESC 0x1b +#define TB_KEY_CTRL_LSQ_BRACKET 0x1b /* clash with 'ESC' */ +#define TB_KEY_CTRL_3 0x1b /* clash with 'ESC' */ +#define TB_KEY_CTRL_4 0x1c +#define TB_KEY_CTRL_BACKSLASH 0x1c /* clash with 'CTRL_4' */ +#define TB_KEY_CTRL_5 0x1d +#define TB_KEY_CTRL_RSQ_BRACKET 0x1d /* clash with 'CTRL_5' */ +#define TB_KEY_CTRL_6 0x1e +#define TB_KEY_CTRL_7 0x1f +#define TB_KEY_CTRL_SLASH 0x1f /* clash with 'CTRL_7' */ +#define TB_KEY_CTRL_UNDERSCORE 0x1f /* clash with 'CTRL_7' */ #define TB_KEY_SPACE 0x20 -#define TB_KEY_BACKSPACE2 0x7F -#define TB_KEY_CTRL_8 0x7F /* clash with 'BACKSPACE2' */ +#define TB_KEY_BACKSPACE2 0x7f +#define TB_KEY_CTRL_8 0x7f /* clash with 'BACKSPACE2' */ -/* These are non-existing ones. - * - * #define TB_KEY_CTRL_1 clash with '1' - * #define TB_KEY_CTRL_9 clash with '9' - * #define TB_KEY_CTRL_0 clash with '0' - */ +#define tb_key_i(i) 0xffff - (i) +/* Terminal-dependent key constants (tb_event.key) and terminfo capabilities */ +/* BEGIN codegen h */ +/* Produced by ./codegen.sh on Fri, 18 Feb 2022 13:51:38 +0000 */ +#define TB_KEY_F1 (0xffff - 0) +#define TB_KEY_F2 (0xffff - 1) +#define TB_KEY_F3 (0xffff - 2) +#define TB_KEY_F4 (0xffff - 3) +#define TB_KEY_F5 (0xffff - 4) +#define TB_KEY_F6 (0xffff - 5) +#define TB_KEY_F7 (0xffff - 6) +#define TB_KEY_F8 (0xffff - 7) +#define TB_KEY_F9 (0xffff - 8) +#define TB_KEY_F10 (0xffff - 9) +#define TB_KEY_F11 (0xffff - 10) +#define TB_KEY_F12 (0xffff - 11) +#define TB_KEY_INSERT (0xffff - 12) +#define TB_KEY_DELETE (0xffff - 13) +#define TB_KEY_HOME (0xffff - 14) +#define TB_KEY_END (0xffff - 15) +#define TB_KEY_PGUP (0xffff - 16) +#define TB_KEY_PGDN (0xffff - 17) +#define TB_KEY_ARROW_UP (0xffff - 18) +#define TB_KEY_ARROW_DOWN (0xffff - 19) +#define TB_KEY_ARROW_LEFT (0xffff - 20) +#define TB_KEY_ARROW_RIGHT (0xffff - 21) +#define TB_KEY_BACK_TAB (0xffff - 22) +#define TB_KEY_MOUSE_LEFT (0xffff - 23) +#define TB_KEY_MOUSE_RIGHT (0xffff - 24) +#define TB_KEY_MOUSE_MIDDLE (0xffff - 25) +#define TB_KEY_MOUSE_RELEASE (0xffff - 26) +#define TB_KEY_MOUSE_WHEEL_UP (0xffff - 27) +#define TB_KEY_MOUSE_WHEEL_DOWN (0xffff - 28) -/* - * Alt modifier constant, see tb_event.mod field and tb_select_input_mode function. - * Mouse-motion modifier +#define TB_CAP_F1 0 +#define TB_CAP_F2 1 +#define TB_CAP_F3 2 +#define TB_CAP_F4 3 +#define TB_CAP_F5 4 +#define TB_CAP_F6 5 +#define TB_CAP_F7 6 +#define TB_CAP_F8 7 +#define TB_CAP_F9 8 +#define TB_CAP_F10 9 +#define TB_CAP_F11 10 +#define TB_CAP_F12 11 +#define TB_CAP_INSERT 12 +#define TB_CAP_DELETE 13 +#define TB_CAP_HOME 14 +#define TB_CAP_END 15 +#define TB_CAP_PGUP 16 +#define TB_CAP_PGDN 17 +#define TB_CAP_ARROW_UP 18 +#define TB_CAP_ARROW_DOWN 19 +#define TB_CAP_ARROW_LEFT 20 +#define TB_CAP_ARROW_RIGHT 21 +#define TB_CAP_BACK_TAB 22 +#define TB_CAP__COUNT_KEYS 23 +#define TB_CAP_ENTER_CA 23 +#define TB_CAP_EXIT_CA 24 +#define TB_CAP_SHOW_CURSOR 25 +#define TB_CAP_HIDE_CURSOR 26 +#define TB_CAP_CLEAR_SCREEN 27 +#define TB_CAP_SGR0 28 +#define TB_CAP_UNDERLINE 29 +#define TB_CAP_BOLD 30 +#define TB_CAP_BLINK 31 +#define TB_CAP_ITALIC 32 +#define TB_CAP_REVERSE 33 +#define TB_CAP_ENTER_KEYPAD 34 +#define TB_CAP_EXIT_KEYPAD 35 +#define TB_CAP__COUNT 36 +/* END codegen h */ + +/* Some hard-coded caps */ +#define TB_HARDCAP_ENTER_MOUSE "\x1b[?1000h\x1b[?1002h\x1b[?1015h\x1b[?1006h" +#define TB_HARDCAP_EXIT_MOUSE "\x1b[?1006l\x1b[?1015l\x1b[?1002l\x1b[?1000l" + +/* Colors (numeric) and attributes (bitwise) (tb_cell.fg, tb_cell.bg) */ +#define TB_DEFAULT 0x0000 +#define TB_BLACK 0x0001 +#define TB_RED 0x0002 +#define TB_GREEN 0x0003 +#define TB_YELLOW 0x0004 +#define TB_BLUE 0x0005 +#define TB_MAGENTA 0x0006 +#define TB_CYAN 0x0007 +#define TB_WHITE 0x0008 +#define TB_BOLD 0x0100 +#define TB_UNDERLINE 0x0200 +#define TB_REVERSE 0x0400 +#define TB_ITALIC 0x0800 + +/* Event types (tb_event.type) */ +#define TB_EVENT_KEY 1 +#define TB_EVENT_RESIZE 2 +#define TB_EVENT_MOUSE 3 + +/* Key modifiers (bitwise) (tb_event.mod) */ +#define TB_MOD_ALT 1 +#define TB_MOD_CTRL 2 +#define TB_MOD_SHIFT 4 +#define TB_MOD_MOTION 8 + +/* Input modes (bitwise) (tb_set_input_mode) */ +#define TB_INPUT_CURRENT 0 +#define TB_INPUT_ESC 1 +#define TB_INPUT_ALT 2 +#define TB_INPUT_MOUSE 4 + +/* Output modes (tb_set_output_mode) */ +#define TB_OUTPUT_CURRENT 0 +#define TB_OUTPUT_NORMAL 1 +#define TB_OUTPUT_256 2 +#define TB_OUTPUT_216 3 +#define TB_OUTPUT_GRAYSCALE 4 +#ifdef TB_OPT_TRUECOLOR +#define TB_OUTPUT_TRUECOLOR 5 +#endif + +/* Common function return values unless otherwise noted */ +#define TB_OK 0 +#define TB_ERR -1 +#define TB_ERR_NEED_MORE -2 +#define TB_ERR_INIT_ALREADY -3 +#define TB_ERR_INIT_OPEN -4 +#define TB_ERR_MEM -5 +#define TB_ERR_NO_EVENT -6 +#define TB_ERR_NO_TERM -7 +#define TB_ERR_NOT_INIT -8 +#define TB_ERR_OUT_OF_BOUNDS -9 +#define TB_ERR_READ -10 +#define TB_ERR_RESIZE_IOCTL -11 +#define TB_ERR_RESIZE_PIPE -12 +#define TB_ERR_RESIZE_SIGACTION -13 +#define TB_ERR_POLL -14 +#define TB_ERR_TCGETATTR -15 +#define TB_ERR_TCSETATTR -16 +#define TB_ERR_UNSUPPORTED_TERM -17 +#define TB_ERR_RESIZE_WRITE -18 +#define TB_ERR_RESIZE_POLL -19 +#define TB_ERR_RESIZE_READ -20 +#define TB_ERR_RESIZE_SSCANF -21 +#define TB_ERR_CAP_COLLISION -22 + +#define TB_ERR_SELECT TB_ERR_POLL +#define TB_ERR_RESIZE_SELECT TB_ERR_RESIZE_POLL + +/* Function types to be used with tb_set_func() */ +#define TB_FUNC_EXTRACT_PRE 0 +#define TB_FUNC_EXTRACT_POST 1 + +/* Define this to set the size of the buffer used in tb_printf() + * and tb_sendf() */ -#define TB_MOD_ALT 0x01 -#define TB_MOD_MOTION 0x02 - -/* Colors (see struct tb_cell's fg and bg fields). */ -#define TB_DEFAULT 0x00 -#define TB_BLACK 0x01 -#define TB_RED 0x02 -#define TB_GREEN 0x03 -#define TB_YELLOW 0x04 -#define TB_BLUE 0x05 -#define TB_MAGENTA 0x06 -#define TB_CYAN 0x07 -#define TB_WHITE 0x08 - -/* Attributes, it is possible to use multiple attributes by combining them - * using bitwise OR ('|'). Although, colors cannot be combined. But you can - * combine attributes and a single color. See also struct tb_cell's fg and bg - * fields. +#ifndef TB_OPT_PRINTF_BUF +#define TB_OPT_PRINTF_BUF 4096 +#endif + +/* Define this to set the size of the read buffer used when reading + * from the tty */ -#define TB_BOLD 0x0100 -#define TB_UNDERLINE 0x0200 -#define TB_REVERSE 0x0400 - -/* A cell, single conceptual entity on the terminal screen. The terminal screen - * is basically a 2d array of cells. It has the following fields: - * - 'ch' is a unicode character - * - 'fg' foreground color and attributes - * - 'bg' background color and attributes +#ifndef TB_OPT_READ_BUF +#define TB_OPT_READ_BUF 64 +#endif + +/* Define this for limited back compat with termbox v1 */ +#ifdef TB_OPT_V1_COMPAT +#define tb_change_cell tb_set_cell +#define tb_put_cell(x, y, c) tb_set_cell((x), (y), (c)->ch, (c)->fg, (c)->bg) +#define tb_set_clear_attributes tb_set_clear_attrs +#define tb_select_input_mode tb_set_input_mode +#define tb_select_output_mode tb_set_output_mode +#endif + +/* Define these to swap in a different allocator */ +#ifndef tb_malloc +#define tb_malloc malloc +#define tb_realloc realloc +#define tb_free free +#endif + +#ifdef TB_OPT_TRUECOLOR +typedef uint32_t uintattr_t; +#else +typedef uint16_t uintattr_t; // __ffi_strip +#endif + +/* The terminal screen is represented as 2d array of cells. The structure is + * optimized for dealing with single-width (wcwidth()==1) Unicode code points, + * however some support for grapheme clusters (e.g., combining diacritical + * marks) and wide code points (e.g., Hiragana) is provided through ech, nech, + * cech via tb_set_cell_ex(). ech is only valid when nech>0, otherwise ch is + * used. + * + * For non-single-width code points, given N=wcwidth(ch)/wcswidth(ech): + * + * when N==0: termbox forces a single-width cell. Callers should avoid this + * if aiming to render text accurately. + * + * when N>1: termbox zeroes out the following N-1 cells and skips sending + * them to the tty. So, e.g., if the caller sets x=0,y=0 to an N==2 + * code point, the caller's next set should be at x=2,y=0. Anything + * set at x=1,y=0 will be ignored. If there are not enough columns + * remaining on the line to render N width, spaces are sent + * instead. + * + * See tb_present() for implementation. */ struct tb_cell { - uint32_t ch; - uint16_t fg; - uint16_t bg; + uint32_t ch; /* a Unicode character */ + uintattr_t fg; /* bit-wise foreground attributes */ + uintattr_t bg; /* bit-wise background attributes */ +#ifdef TB_OPT_EGC + uint32_t *ech; /* a grapheme cluster of Unicode code points */ + size_t nech; /* length in bytes of ech, 0 means use ch instead of ech */ + size_t cech; /* capacity in bytes of ech */ +#endif }; -#define TB_EVENT_KEY 1 -#define TB_EVENT_RESIZE 2 -#define TB_EVENT_MOUSE 3 - -/* An event, single interaction from the user. The 'mod' and 'ch' fields are - * valid if 'type' is TB_EVENT_KEY. The 'w' and 'h' fields are valid if 'type' - * is TB_EVENT_RESIZE. The 'x' and 'y' fields are valid if 'type' is - * TB_EVENT_MOUSE. The 'key' field is valid if 'type' is either TB_EVENT_KEY - * or TB_EVENT_MOUSE. The fields 'key' and 'ch' are mutually exclusive; only - * one of them can be non-zero at a time. +/* An incoming event from the tty. + * + * Given the event type, the following fields are relevant: + * + * when TB_EVENT_KEY: (key XOR ch, one will be zero), mod. Note there is + * overlap between TB_MOD_CTRL and TB_KEY_CTRL_*. + * TB_MOD_CTRL and TB_MOD_SHIFT are only set as + * modifiers to TB_KEY_ARROW_*. + * + * when TB_EVENT_RESIZE: w, h + * + * when TB_EVENT_MOUSE: key (TB_KEY_MOUSE_*), x, y */ struct tb_event { - uint8_t type; - uint8_t mod; /* modifiers to either 'key' or 'ch' below */ - uint16_t key; /* one of the TB_KEY_* constants */ - uint32_t ch; /* unicode character */ - int32_t w; - int32_t h; - int32_t x; - int32_t y; + uint8_t type; /* one of TB_EVENT_* constants */ + uint8_t mod; /* bit-wise TB_MOD_* constants */ + uint16_t key; /* one of TB_KEY_* constants */ + uint32_t ch; /* a Unicode code point */ + int32_t w; /* resize width */ + int32_t h; /* resize height */ + int32_t x; /* mouse x */ + int32_t y; /* mouse y */ }; -/* Error codes returned by tb_init(). All of them are self-explanatory, except - * the pipe trap error. Termbox uses unix pipes in order to deliver a message - * from a signal handler (SIGWINCH) to the main event reading loop. Honestly in - * most cases you should just check the returned code as < 0. - */ -#define TB_EUNSUPPORTED_TERMINAL -1 -#define TB_EFAILED_TO_OPEN_TTY -2 -#define TB_EPIPE_TRAP_ERROR -3 - /* Initializes the termbox library. This function should be called before any - * other functions. Function tb_init is same as tb_init_file("/dev/tty"). - * After successful initialization, the library must be - * finalized using the tb_shutdown() function. + * other functions. tb_init() is equivalent to tb_init_file("/dev/tty"). After + * successful initialization, the library must be finalized using the + * tb_shutdown() function. */ -SO_IMPORT int tb_init(void); -SO_IMPORT int tb_init_file(const char* name); -SO_IMPORT int tb_init_fd(int inout); -SO_IMPORT void tb_shutdown(void); - -/* Returns the size of the internal back buffer (which is the same as - * terminal's window size in characters). The internal buffer can be resized - * after tb_clear() or tb_present() function calls. Both dimensions have an +int tb_init(); +int tb_init_file(const char *path); +int tb_init_fd(int ttyfd); +int tb_init_rwfd(int rfd, int wfd); +int tb_shutdown(); + +/* Returns the size of the internal back buffer (which is the same as terminal's + * window size in rows and columns). The internal buffer can be resized after + * tb_clear() or tb_present() function calls. Both dimensions have an * unspecified negative value when called before tb_init() or after * tb_shutdown(). */ -SO_IMPORT int tb_width(void); -SO_IMPORT int tb_height(void); +int tb_width(); +int tb_height(); /* Clears the internal back buffer using TB_DEFAULT color or the - * color/attributes set by tb_set_clear_attributes() function. + * color/attributes set by tb_set_clear_attrs() function. */ -SO_IMPORT void tb_clear(void); -SO_IMPORT void tb_set_clear_attributes(uint16_t fg, uint16_t bg); - -/* Synchronizes the internal back buffer with the terminal. */ -SO_IMPORT void tb_present(void); +int tb_clear(); +int tb_set_clear_attrs(uintattr_t fg, uintattr_t bg); -#define TB_HIDE_CURSOR -1 +/* Synchronizes the internal back buffer with the terminal by writing to tty. */ +int tb_present(); -/* Sets the position of the cursor. Upper-left character is (0, 0). If you pass - * TB_HIDE_CURSOR as both coordinates, then the cursor will be hidden. Cursor - * is hidden by default. - */ -SO_IMPORT void tb_set_cursor(int cx, int cy); - -/* Changes cell's parameters in the internal back buffer at the specified - * position. - */ -SO_IMPORT void tb_put_cell(int x, int y, const struct tb_cell *cell); -SO_IMPORT void tb_change_cell(int x, int y, uint32_t ch, uint16_t fg, uint16_t bg); +/* Sets the position of the cursor. Upper-left character is (0, 0). */ +int tb_set_cursor(int cx, int cy); +int tb_hide_cursor(); -/* Copies the buffer from 'cells' at the specified position, assuming the - * buffer is a two-dimensional array of size ('w' x 'h'), represented as a - * one-dimensional buffer containing lines of cells starting from the top. - * - * (DEPRECATED: use tb_cell_buffer() instead and copy memory on your own) +/* Set cell contents in the internal back buffer at the specified position. Use + * tb_set_cell_ex() for rendering grapheme clusters (e.g., combining diacritical + * marks). Function tb_set_cell(x, y, ch, fg, bg) is equivalent to + * tb_set_cell_ex(x, y, &ch, 1, fg, bg). tb_extend_cell() is a shortcut for + * appending 1 code point to cell->ech. */ -SO_IMPORT void tb_blit(int x, int y, int w, int h, const struct tb_cell *cells); +int tb_set_cell(int x, int y, uint32_t ch, uintattr_t fg, uintattr_t bg); +int tb_set_cell_ex(int x, int y, uint32_t *ch, size_t nch, uintattr_t fg, + uintattr_t bg); +int tb_extend_cell(int x, int y, uint32_t ch); -/* Returns a pointer to internal cell back buffer. You can get its dimensions - * using tb_width() and tb_height() functions. The pointer stays valid as long - * as no tb_clear() and tb_present() calls are made. The buffer is - * one-dimensional buffer containing lines of cells starting from the top. - */ -SO_IMPORT struct tb_cell *tb_cell_buffer(void); - -#define TB_INPUT_CURRENT 0 /* 000 */ -#define TB_INPUT_ESC 1 /* 001 */ -#define TB_INPUT_ALT 2 /* 010 */ -#define TB_INPUT_MOUSE 4 /* 100 */ - -/* Sets the termbox input mode. Termbox has two input modes: - * 1. Esc input mode. - * When ESC sequence is in the buffer and it doesn't match any known - * ESC sequence => ESC means TB_KEY_ESC. - * 2. Alt input mode. - * When ESC sequence is in the buffer and it doesn't match any known - * sequence => ESC enables TB_MOD_ALT modifier for the next keyboard event. +/* Sets the input mode. Termbox has two input modes: + * 1. TB_INPUT_ESC + * When escape (\x1b) is in the buffer and there's no match for an escape + * sequence, a key event for TB_KEY_ESC is returned. + * 2. TB_INPUT_ALT + * When escape (\x1b) is in the buffer and there's no match for an escape + * sequence, the next keyboard event is returned with a TB_MOD_ALT modifier. * * You can also apply TB_INPUT_MOUSE via bitwise OR operation to either of the * modes (e.g. TB_INPUT_ESC | TB_INPUT_MOUSE). If none of the main two modes @@ -247,74 +401,2765 @@ SO_IMPORT struct tb_cell *tb_cell_buffer(void); * reason you've decided to use (TB_INPUT_ESC | TB_INPUT_ALT) combination, it * will behave as if only TB_INPUT_ESC was selected. * - * If 'mode' is TB_INPUT_CURRENT, it returns the current input mode. + * If mode is TB_INPUT_CURRENT, the function returns the current input mode. * - * Default termbox input mode is TB_INPUT_ESC. + * The default input mode is TB_INPUT_ESC. */ -SO_IMPORT int tb_select_input_mode(int mode); - -#define TB_OUTPUT_CURRENT 0 -#define TB_OUTPUT_NORMAL 1 -#define TB_OUTPUT_256 2 -#define TB_OUTPUT_216 3 -#define TB_OUTPUT_GRAYSCALE 4 +int tb_set_input_mode(int mode); -/* Sets the termbox output mode. Termbox has three output options: +/* Sets the termbox output mode. Termbox has three output modes: * 1. TB_OUTPUT_NORMAL => [1..8] * This mode provides 8 different colors: - * black, red, green, yellow, blue, magenta, cyan, white - * Shortcut: TB_BLACK, TB_RED, ... - * Attributes: TB_BOLD, TB_UNDERLINE, TB_REVERSE + * TB_BLACK, TB_RED, TB_GREEN, TB_YELLOW, + * TB_BLUE, TB_MAGENTA, TB_CYAN, TB_WHITE + * Colors may be bitwise OR'd with attributes: + * TB_BOLD, TB_UNDERLINE, TB_REVERSE, TB_ITALIC * * Example usage: - * tb_change_cell(x, y, '@', TB_BLACK | TB_BOLD, TB_RED); + * tb_set_cell(x, y, '@', TB_BLACK | TB_BOLD, TB_RED); * - * 2. TB_OUTPUT_256 => [0..256] - * In this mode you can leverage the 256 terminal mode: - * 0x00 - 0x07: the 8 colors as in TB_OUTPUT_NORMAL - * 0x08 - 0x0f: TB_* | TB_BOLD - * 0x10 - 0xe7: 216 different colors - * 0xe8 - 0xff: 24 different shades of grey + * 2. TB_OUTPUT_256 => [0..255] + * In this mode you get 256 distinct colors: + * 0x00 - 0x07: the 8 colors as in TB_OUTPUT_NORMAL + * 0x08 - 0x0f: TB_<color> | TB_BOLD + * 0x10 - 0xe7: 216 different colors + * 0xe8 - 0xff: 24 different shades of grey * * Example usage: - * tb_change_cell(x, y, '@', 184, 240); - * tb_change_cell(x, y, '@', 0xb8, 0xf0); + * tb_set_cell(x, y, '@', 184, 240); + * tb_set_cell(x, y, '@', 0xb8, 0xf0); * * 3. TB_OUTPUT_216 => [0..216] - * This mode supports the 3rd range of the 256 mode only. - * But you don't need to provide an offset. + * This mode supports the 3rd range of TB_OUTPUT_256 only, but you don't need + * to provide an offset. * * 4. TB_OUTPUT_GRAYSCALE => [0..23] - * This mode supports the 4th range of the 256 mode only. - * But you dont need to provide an offset. + * This mode supports the 4th range of TB_OUTPUT_256 only, but you dont need + * to provide an offset. * - * Execute build/src/demo/output to see its impact on your terminal. + * 5. TB_OUTPUT_TRUECOLOR => [0x000000..0xffffff] + * This mode provides 24-bit color on supported terminals. The format is + * 0xRRGGBB. * - * If 'mode' is TB_OUTPUT_CURRENT, it returns the current output mode. + * If mode is TB_OUTPUT_CURRENT, the function returns the current output mode. * - * Default termbox output mode is TB_OUTPUT_NORMAL. + * The default output mode is TB_OUTPUT_NORMAL. + */ +int tb_set_output_mode(int mode); + +/* Wait for an event up to timeout_ms milliseconds and fill the event structure + * with it. If no event is available within the timeout period, TB_ERR_NO_EVENT + * is returned. On a resize event, the underlying poll(2) call may be + * interrupted, yielding a return code of TB_ERR_POLL. In this case, you may + * check errno via tb_last_errno(). If it's EINTR, you can safely ignore that + * and call tb_peek_event() again. */ -SO_IMPORT int tb_select_output_mode(int mode); +int tb_peek_event(struct tb_event *event, int timeout_ms); -/* Wait for an event up to 'timeout' milliseconds and fill the 'event' - * structure with it, when the event is available. Returns the type of the - * event (one of TB_EVENT_* constants) or -1 if there was an error or 0 in case - * there were no event during 'timeout' period. +/* Same as tb_peek_event except no timeout. */ +int tb_poll_event(struct tb_event *event); + +/* Internal termbox FDs that can be used with poll() / select(). Must call + * tb_poll_event() / tb_peek_event() if activity is detected. */ +int tb_get_fds(int *ttyfd, int *resizefd); + +/* Print and printf functions. Specify param out_w to determine width of printed + * string. */ -SO_IMPORT int tb_peek_event(struct tb_event *event, int timeout); +int tb_print(int x, int y, uintattr_t fg, uintattr_t bg, const char *str); +int tb_printf(int x, int y, uintattr_t fg, uintattr_t bg, const char *fmt, ...); +int tb_print_ex(int x, int y, uintattr_t fg, uintattr_t bg, size_t *out_w, + const char *str); +int tb_printf_ex(int x, int y, uintattr_t fg, uintattr_t bg, size_t *out_w, + const char *fmt, ...); -/* Wait for an event forever and fill the 'event' structure with it, when the - * event is available. Returns the type of the event (one of TB_EVENT_* - * constants) or -1 if there was an error. +/* Send raw bytes to terminal. */ +int tb_send(const char *buf, size_t nbuf); +int tb_sendf(const char *fmt, ...); + +/* Set custom functions. fn_type is one of TB_FUNC_* constants, fn is a + * compatible function pointer, or NULL to clear. + * + * TB_FUNC_EXTRACT_PRE: + * If specified, invoke this function BEFORE termbox tries to extract any + * escape sequences from the input buffer. + * + * TB_FUNC_EXTRACT_POST: + * If specified, invoke this function AFTER termbox tries (and fails) to + * extract any escape sequences from the input buffer. */ -SO_IMPORT int tb_poll_event(struct tb_event *event); +int tb_set_func(int fn_type, int (*fn)(struct tb_event *, size_t *)); + +/* Utility functions. */ +int tb_utf8_char_length(char c); +int tb_utf8_char_to_unicode(uint32_t *out, const char *c); +int tb_utf8_unicode_to_char(char *out, uint32_t c); +int tb_last_errno(); +const char *tb_strerror(int err); -/* Utility utf8 functions. */ -#define TB_EOF -1 -SO_IMPORT int tb_utf8_char_length(char c); -SO_IMPORT int tb_utf8_char_to_unicode(uint32_t *out, const char *c); -SO_IMPORT int tb_utf8_unicode_to_char(char *out, uint32_t c); +struct tb_cell *tb_cell_buffer(); #ifdef __cplusplus +} // __ffi_strip +#endif + +#endif /* __TERMBOX_H */ + +#ifdef TB_IMPL + +#ifndef TB_OPT_TRUECOLOR +#define TB_OUTPUT_TRUECOLOR (-1) +#endif + +#define if_err_return(rv, expr) \ + if (((rv) = (expr)) != TB_OK) \ + return (rv) +#define if_err_break(rv, expr) \ + if (((rv) = (expr)) != TB_OK) \ + break +#define if_ok_return(rv, expr) \ + if (((rv) = (expr)) == TB_OK) \ + return (rv) +#define if_ok_or_need_more_return(rv, expr) \ + if (((rv) = (expr)) == TB_OK || (rv) == TB_ERR_NEED_MORE) \ + return (rv) + +#define send_literal(rv, a) \ + if_err_return((rv), bytebuf_nputs(&global.out, (a), sizeof(a) - 1)) + +#define send_num(rv, nbuf, n) \ + if_err_return((rv), \ + bytebuf_nputs(&global.out, (nbuf), convert_num((n), (nbuf)))) + +#define snprintf_or_return(rv, str, sz, fmt, ...) \ + do { \ + (rv) = snprintf((str), (sz), (fmt), __VA_ARGS__); \ + if ((rv) < 0 || (rv) >= (int)(sz)) \ + return TB_ERR; \ + } while (0) + +#define if_not_init_return() \ + if (!global.initialized) \ + return TB_ERR_NOT_INIT + +struct bytebuf_t { + char *buf; + size_t len; + size_t cap; +}; + +struct cellbuf_t { + int width; + int height; + struct tb_cell *cells; +}; + +struct cap_trie_t { + char c; + struct cap_trie_t *children; + size_t nchildren; + int is_leaf; + uint16_t key; + uint8_t mod; +}; + +struct tb_global_t { + int ttyfd; + int rfd; + int wfd; + int ttyfd_open; + int resize_pipefd[2]; + int width; + int height; + int cursor_x; + int cursor_y; + int last_x; + int last_y; + uintattr_t fg; + uintattr_t bg; + uintattr_t last_fg; + uintattr_t last_bg; + int input_mode; + int output_mode; + char *terminfo; + size_t nterminfo; + const char *caps[TB_CAP__COUNT]; + struct cap_trie_t cap_trie; + struct bytebuf_t in; + struct bytebuf_t out; + struct cellbuf_t back; + struct cellbuf_t front; + struct termios orig_tios; + int has_orig_tios; + int last_errno; + int initialized; + int need_resize; + int (*fn_extract_esc_pre)(struct tb_event *, size_t *); + int (*fn_extract_esc_post)(struct tb_event *, size_t *); +}; + +struct tb_global_t global = {0}; + +/* BEGIN codegen c */ +/* Produced by ./codegen.sh on Fri, 18 Feb 2022 13:51:39 +0000 */ + +static const int16_t terminfo_cap_indexes[] = { + 66, // kf1 (TB_CAP_F1) + 68, // kf2 (TB_CAP_F2) + 69, // kf3 (TB_CAP_F3) + 70, // kf4 (TB_CAP_F4) + 71, // kf5 (TB_CAP_F5) + 72, // kf6 (TB_CAP_F6) + 73, // kf7 (TB_CAP_F7) + 74, // kf8 (TB_CAP_F8) + 75, // kf9 (TB_CAP_F9) + 67, // kf10 (TB_CAP_F10) + 216, // kf11 (TB_CAP_F11) + 217, // kf12 (TB_CAP_F12) + 77, // kich1 (TB_CAP_INSERT) + 59, // kdch1 (TB_CAP_DELETE) + 76, // khome (TB_CAP_HOME) + 164, // kend (TB_CAP_END) + 82, // kpp (TB_CAP_PGUP) + 81, // knp (TB_CAP_PGDN) + 87, // kcuu1 (TB_CAP_ARROW_UP) + 61, // kcud1 (TB_CAP_ARROW_DOWN) + 79, // kcub1 (TB_CAP_ARROW_LEFT) + 83, // kcuf1 (TB_CAP_ARROW_RIGHT) + 148, // kcbt (TB_CAP_BACK_TAB) + 28, // smcup (TB_CAP_ENTER_CA) + 40, // rmcup (TB_CAP_EXIT_CA) + 16, // cnorm (TB_CAP_SHOW_CURSOR) + 13, // civis (TB_CAP_HIDE_CURSOR) + 5, // clear (TB_CAP_CLEAR_SCREEN) + 39, // sgr0 (TB_CAP_SGR0) + 36, // smul (TB_CAP_UNDERLINE) + 27, // bold (TB_CAP_BOLD) + 26, // blink (TB_CAP_BLINK) + 311, // sitm (TB_CAP_ITALIC) + 34, // rev (TB_CAP_REVERSE) + 89, // smkx (TB_CAP_ENTER_KEYPAD) + 88, // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// xterm +static const char *xterm_caps[] = { + "\033OP", // kf1 (TB_CAP_F1) + "\033OQ", // kf2 (TB_CAP_F2) + "\033OR", // kf3 (TB_CAP_F3) + "\033OS", // kf4 (TB_CAP_F4) + "\033[15~", // kf5 (TB_CAP_F5) + "\033[17~", // kf6 (TB_CAP_F6) + "\033[18~", // kf7 (TB_CAP_F7) + "\033[19~", // kf8 (TB_CAP_F8) + "\033[20~", // kf9 (TB_CAP_F9) + "\033[21~", // kf10 (TB_CAP_F10) + "\033[23~", // kf11 (TB_CAP_F11) + "\033[24~", // kf12 (TB_CAP_F12) + "\033[2~", // kich1 (TB_CAP_INSERT) + "\033[3~", // kdch1 (TB_CAP_DELETE) + "\033OH", // khome (TB_CAP_HOME) + "\033OF", // kend (TB_CAP_END) + "\033[5~", // kpp (TB_CAP_PGUP) + "\033[6~", // knp (TB_CAP_PGDN) + "\033OA", // kcuu1 (TB_CAP_ARROW_UP) + "\033OB", // kcud1 (TB_CAP_ARROW_DOWN) + "\033OD", // kcub1 (TB_CAP_ARROW_LEFT) + "\033OC", // kcuf1 (TB_CAP_ARROW_RIGHT) + "\033[Z", // kcbt (TB_CAP_BACK_TAB) + "\033[?1049h\033[22;0;0t", // smcup (TB_CAP_ENTER_CA) + "\033[?1049l\033[23;0;0t", // rmcup (TB_CAP_EXIT_CA) + "\033[?12l\033[?25h", // cnorm (TB_CAP_SHOW_CURSOR) + "\033[?25l", // civis (TB_CAP_HIDE_CURSOR) + "\033[H\033[2J", // clear (TB_CAP_CLEAR_SCREEN) + "\033(B\033[m", // sgr0 (TB_CAP_SGR0) + "\033[4m", // smul (TB_CAP_UNDERLINE) + "\033[1m", // bold (TB_CAP_BOLD) + "\033[5m", // blink (TB_CAP_BLINK) + "\033[3m", // sitm (TB_CAP_ITALIC) + "\033[7m", // rev (TB_CAP_REVERSE) + "\033[?1h\033=", // smkx (TB_CAP_ENTER_KEYPAD) + "\033[?1l\033>", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// linux +static const char *linux_caps[] = { + "\033[[A", // kf1 (TB_CAP_F1) + "\033[[B", // kf2 (TB_CAP_F2) + "\033[[C", // kf3 (TB_CAP_F3) + "\033[[D", // kf4 (TB_CAP_F4) + "\033[[E", // kf5 (TB_CAP_F5) + "\033[17~", // kf6 (TB_CAP_F6) + "\033[18~", // kf7 (TB_CAP_F7) + "\033[19~", // kf8 (TB_CAP_F8) + "\033[20~", // kf9 (TB_CAP_F9) + "\033[21~", // kf10 (TB_CAP_F10) + "\033[23~", // kf11 (TB_CAP_F11) + "\033[24~", // kf12 (TB_CAP_F12) + "\033[2~", // kich1 (TB_CAP_INSERT) + "\033[3~", // kdch1 (TB_CAP_DELETE) + "\033[1~", // khome (TB_CAP_HOME) + "\033[4~", // kend (TB_CAP_END) + "\033[5~", // kpp (TB_CAP_PGUP) + "\033[6~", // knp (TB_CAP_PGDN) + "\033[A", // kcuu1 (TB_CAP_ARROW_UP) + "\033[B", // kcud1 (TB_CAP_ARROW_DOWN) + "\033[D", // kcub1 (TB_CAP_ARROW_LEFT) + "\033[C", // kcuf1 (TB_CAP_ARROW_RIGHT) + "\033\011", // kcbt (TB_CAP_BACK_TAB) + "", // smcup (TB_CAP_ENTER_CA) + "", // rmcup (TB_CAP_EXIT_CA) + "\033[?25h\033[?0c", // cnorm (TB_CAP_SHOW_CURSOR) + "\033[?25l\033[?1c", // civis (TB_CAP_HIDE_CURSOR) + "\033[H\033[J", // clear (TB_CAP_CLEAR_SCREEN) + "\033[m\017", // sgr0 (TB_CAP_SGR0) + "\033[4m", // smul (TB_CAP_UNDERLINE) + "\033[1m", // bold (TB_CAP_BOLD) + "\033[5m", // blink (TB_CAP_BLINK) + "", // sitm (TB_CAP_ITALIC) + "\033[7m", // rev (TB_CAP_REVERSE) + "", // smkx (TB_CAP_ENTER_KEYPAD) + "", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// screen +static const char *screen_caps[] = { + "\033OP", // kf1 (TB_CAP_F1) + "\033OQ", // kf2 (TB_CAP_F2) + "\033OR", // kf3 (TB_CAP_F3) + "\033OS", // kf4 (TB_CAP_F4) + "\033[15~", // kf5 (TB_CAP_F5) + "\033[17~", // kf6 (TB_CAP_F6) + "\033[18~", // kf7 (TB_CAP_F7) + "\033[19~", // kf8 (TB_CAP_F8) + "\033[20~", // kf9 (TB_CAP_F9) + "\033[21~", // kf10 (TB_CAP_F10) + "\033[23~", // kf11 (TB_CAP_F11) + "\033[24~", // kf12 (TB_CAP_F12) + "\033[2~", // kich1 (TB_CAP_INSERT) + "\033[3~", // kdch1 (TB_CAP_DELETE) + "\033[1~", // khome (TB_CAP_HOME) + "\033[4~", // kend (TB_CAP_END) + "\033[5~", // kpp (TB_CAP_PGUP) + "\033[6~", // knp (TB_CAP_PGDN) + "\033OA", // kcuu1 (TB_CAP_ARROW_UP) + "\033OB", // kcud1 (TB_CAP_ARROW_DOWN) + "\033OD", // kcub1 (TB_CAP_ARROW_LEFT) + "\033OC", // kcuf1 (TB_CAP_ARROW_RIGHT) + "\033[Z", // kcbt (TB_CAP_BACK_TAB) + "\033[?1049h", // smcup (TB_CAP_ENTER_CA) + "\033[?1049l", // rmcup (TB_CAP_EXIT_CA) + "\033[34h\033[?25h", // cnorm (TB_CAP_SHOW_CURSOR) + "\033[?25l", // civis (TB_CAP_HIDE_CURSOR) + "\033[H\033[J", // clear (TB_CAP_CLEAR_SCREEN) + "\033[m\017", // sgr0 (TB_CAP_SGR0) + "\033[4m", // smul (TB_CAP_UNDERLINE) + "\033[1m", // bold (TB_CAP_BOLD) + "\033[5m", // blink (TB_CAP_BLINK) + "", // sitm (TB_CAP_ITALIC) + "\033[7m", // rev (TB_CAP_REVERSE) + "\033[?1h\033=", // smkx (TB_CAP_ENTER_KEYPAD) + "\033[?1l\033>", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// rxvt-256color +static const char *rxvt_256color_caps[] = { + "\033[11~", // kf1 (TB_CAP_F1) + "\033[12~", // kf2 (TB_CAP_F2) + "\033[13~", // kf3 (TB_CAP_F3) + "\033[14~", // kf4 (TB_CAP_F4) + "\033[15~", // kf5 (TB_CAP_F5) + "\033[17~", // kf6 (TB_CAP_F6) + "\033[18~", // kf7 (TB_CAP_F7) + "\033[19~", // kf8 (TB_CAP_F8) + "\033[20~", // kf9 (TB_CAP_F9) + "\033[21~", // kf10 (TB_CAP_F10) + "\033[23~", // kf11 (TB_CAP_F11) + "\033[24~", // kf12 (TB_CAP_F12) + "\033[2~", // kich1 (TB_CAP_INSERT) + "\033[3~", // kdch1 (TB_CAP_DELETE) + "\033[7~", // khome (TB_CAP_HOME) + "\033[8~", // kend (TB_CAP_END) + "\033[5~", // kpp (TB_CAP_PGUP) + "\033[6~", // knp (TB_CAP_PGDN) + "\033[A", // kcuu1 (TB_CAP_ARROW_UP) + "\033[B", // kcud1 (TB_CAP_ARROW_DOWN) + "\033[D", // kcub1 (TB_CAP_ARROW_LEFT) + "\033[C", // kcuf1 (TB_CAP_ARROW_RIGHT) + "\033[Z", // kcbt (TB_CAP_BACK_TAB) + "\0337\033[?47h", // smcup (TB_CAP_ENTER_CA) + "\033[2J\033[?47l\0338", // rmcup (TB_CAP_EXIT_CA) + "\033[?25h", // cnorm (TB_CAP_SHOW_CURSOR) + "\033[?25l", // civis (TB_CAP_HIDE_CURSOR) + "\033[H\033[2J", // clear (TB_CAP_CLEAR_SCREEN) + "\033[m\017", // sgr0 (TB_CAP_SGR0) + "\033[4m", // smul (TB_CAP_UNDERLINE) + "\033[1m", // bold (TB_CAP_BOLD) + "\033[5m", // blink (TB_CAP_BLINK) + "", // sitm (TB_CAP_ITALIC) + "\033[7m", // rev (TB_CAP_REVERSE) + "\033=", // smkx (TB_CAP_ENTER_KEYPAD) + "\033>", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// rxvt-unicode +static const char *rxvt_unicode_caps[] = { + "", // kf1 (TB_CAP_F1) + "", // kf2 (TB_CAP_F2) + "", // kf3 (TB_CAP_F3) + "", // kf4 (TB_CAP_F4) + "", // kf5 (TB_CAP_F5) + "", // kf6 (TB_CAP_F6) + "", // kf7 (TB_CAP_F7) + "", // kf8 (TB_CAP_F8) + "", // kf9 (TB_CAP_F9) + "", // kf10 (TB_CAP_F10) + "", // kf11 (TB_CAP_F11) + "", // kf12 (TB_CAP_F12) + "", // kich1 (TB_CAP_INSERT) + "", // kdch1 (TB_CAP_DELETE) + "", // khome (TB_CAP_HOME) + "", // kend (TB_CAP_END) + "", // kpp (TB_CAP_PGUP) + "", // knp (TB_CAP_PGDN) + "", // kcuu1 (TB_CAP_ARROW_UP) + "", // kcud1 (TB_CAP_ARROW_DOWN) + "", // kcub1 (TB_CAP_ARROW_LEFT) + "", // kcuf1 (TB_CAP_ARROW_RIGHT) + "", // kcbt (TB_CAP_BACK_TAB) + "", // smcup (TB_CAP_ENTER_CA) + "", // rmcup (TB_CAP_EXIT_CA) + "", // cnorm (TB_CAP_SHOW_CURSOR) + "", // civis (TB_CAP_HIDE_CURSOR) + "", // clear (TB_CAP_CLEAR_SCREEN) + "", // sgr0 (TB_CAP_SGR0) + "", // smul (TB_CAP_UNDERLINE) + "", // bold (TB_CAP_BOLD) + "", // blink (TB_CAP_BLINK) + "", // sitm (TB_CAP_ITALIC) + "", // rev (TB_CAP_REVERSE) + "", // smkx (TB_CAP_ENTER_KEYPAD) + "", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +// Eterm +static const char *eterm_caps[] = { + "\033[11~", // kf1 (TB_CAP_F1) + "\033[12~", // kf2 (TB_CAP_F2) + "\033[13~", // kf3 (TB_CAP_F3) + "\033[14~", // kf4 (TB_CAP_F4) + "\033[15~", // kf5 (TB_CAP_F5) + "\033[17~", // kf6 (TB_CAP_F6) + "\033[18~", // kf7 (TB_CAP_F7) + "\033[19~", // kf8 (TB_CAP_F8) + "\033[20~", // kf9 (TB_CAP_F9) + "\033[21~", // kf10 (TB_CAP_F10) + "\033[23~", // kf11 (TB_CAP_F11) + "\033[24~", // kf12 (TB_CAP_F12) + "\033[2~", // kich1 (TB_CAP_INSERT) + "\033[3~", // kdch1 (TB_CAP_DELETE) + "\033[7~", // khome (TB_CAP_HOME) + "\033[8~", // kend (TB_CAP_END) + "\033[5~", // kpp (TB_CAP_PGUP) + "\033[6~", // knp (TB_CAP_PGDN) + "\033[A", // kcuu1 (TB_CAP_ARROW_UP) + "\033[B", // kcud1 (TB_CAP_ARROW_DOWN) + "\033[D", // kcub1 (TB_CAP_ARROW_LEFT) + "\033[C", // kcuf1 (TB_CAP_ARROW_RIGHT) + "", // kcbt (TB_CAP_BACK_TAB) + "\0337\033[?47h", // smcup (TB_CAP_ENTER_CA) + "\033[2J\033[?47l\0338", // rmcup (TB_CAP_EXIT_CA) + "\033[?25h", // cnorm (TB_CAP_SHOW_CURSOR) + "\033[?25l", // civis (TB_CAP_HIDE_CURSOR) + "\033[H\033[2J", // clear (TB_CAP_CLEAR_SCREEN) + "\033[m\017", // sgr0 (TB_CAP_SGR0) + "\033[4m", // smul (TB_CAP_UNDERLINE) + "\033[1m", // bold (TB_CAP_BOLD) + "\033[5m", // blink (TB_CAP_BLINK) + "", // sitm (TB_CAP_ITALIC) + "\033[7m", // rev (TB_CAP_REVERSE) + "", // smkx (TB_CAP_ENTER_KEYPAD) + "", // rmkx (TB_CAP_EXIT_KEYPAD) +}; + +static struct { + const char *name; + const char **caps; + const char *alias; +} builtin_terms[] = { + {"xterm", xterm_caps, "" }, + {"linux", linux_caps, "" }, + {"screen", screen_caps, "tmux"}, + {"rxvt-256color", rxvt_256color_caps, "" }, + {"rxvt-unicode", rxvt_unicode_caps, "rxvt"}, + {"Eterm", eterm_caps, "" }, + {NULL, NULL, NULL }, +}; + +/* END codegen c */ + +static struct { + const char *cap; + const uint16_t key; + const uint8_t mod; +} builtin_mod_caps[] = { + // xterm arrows + {"\x1b[1;2A", TB_KEY_ARROW_UP, TB_MOD_SHIFT }, + {"\x1b[1;3A", TB_KEY_ARROW_UP, TB_MOD_ALT }, + {"\x1b[1;4A", TB_KEY_ARROW_UP, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5A", TB_KEY_ARROW_UP, TB_MOD_CTRL }, + {"\x1b[1;6A", TB_KEY_ARROW_UP, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7A", TB_KEY_ARROW_UP, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8A", TB_KEY_ARROW_UP, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2B", TB_KEY_ARROW_DOWN, TB_MOD_SHIFT }, + {"\x1b[1;3B", TB_KEY_ARROW_DOWN, TB_MOD_ALT }, + {"\x1b[1;4B", TB_KEY_ARROW_DOWN, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5B", TB_KEY_ARROW_DOWN, TB_MOD_CTRL }, + {"\x1b[1;6B", TB_KEY_ARROW_DOWN, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7B", TB_KEY_ARROW_DOWN, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8B", TB_KEY_ARROW_DOWN, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2C", TB_KEY_ARROW_RIGHT, TB_MOD_SHIFT }, + {"\x1b[1;3C", TB_KEY_ARROW_RIGHT, TB_MOD_ALT }, + {"\x1b[1;4C", TB_KEY_ARROW_RIGHT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5C", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL }, + {"\x1b[1;6C", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7C", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8C", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2D", TB_KEY_ARROW_LEFT, TB_MOD_SHIFT }, + {"\x1b[1;3D", TB_KEY_ARROW_LEFT, TB_MOD_ALT }, + {"\x1b[1;4D", TB_KEY_ARROW_LEFT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5D", TB_KEY_ARROW_LEFT, TB_MOD_CTRL }, + {"\x1b[1;6D", TB_KEY_ARROW_LEFT, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7D", TB_KEY_ARROW_LEFT, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8D", TB_KEY_ARROW_LEFT, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + // xterm keys + {"\x1b[1;2H", TB_KEY_HOME, TB_MOD_SHIFT }, + {"\x1b[1;3H", TB_KEY_HOME, TB_MOD_ALT }, + {"\x1b[1;4H", TB_KEY_HOME, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5H", TB_KEY_HOME, TB_MOD_CTRL }, + {"\x1b[1;6H", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7H", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8H", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2F", TB_KEY_END, TB_MOD_SHIFT }, + {"\x1b[1;3F", TB_KEY_END, TB_MOD_ALT }, + {"\x1b[1;4F", TB_KEY_END, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5F", TB_KEY_END, TB_MOD_CTRL }, + {"\x1b[1;6F", TB_KEY_END, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7F", TB_KEY_END, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8F", TB_KEY_END, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[2;2~", TB_KEY_INSERT, TB_MOD_SHIFT }, + {"\x1b[2;3~", TB_KEY_INSERT, TB_MOD_ALT }, + {"\x1b[2;4~", TB_KEY_INSERT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[2;5~", TB_KEY_INSERT, TB_MOD_CTRL }, + {"\x1b[2;6~", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[2;7~", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[2;8~", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[3;2~", TB_KEY_DELETE, TB_MOD_SHIFT }, + {"\x1b[3;3~", TB_KEY_DELETE, TB_MOD_ALT }, + {"\x1b[3;4~", TB_KEY_DELETE, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[3;5~", TB_KEY_DELETE, TB_MOD_CTRL }, + {"\x1b[3;6~", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[3;7~", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[3;8~", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[5;2~", TB_KEY_PGUP, TB_MOD_SHIFT }, + {"\x1b[5;3~", TB_KEY_PGUP, TB_MOD_ALT }, + {"\x1b[5;4~", TB_KEY_PGUP, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[5;5~", TB_KEY_PGUP, TB_MOD_CTRL }, + {"\x1b[5;6~", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[5;7~", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[5;8~", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[6;2~", TB_KEY_PGDN, TB_MOD_SHIFT }, + {"\x1b[6;3~", TB_KEY_PGDN, TB_MOD_ALT }, + {"\x1b[6;4~", TB_KEY_PGDN, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[6;5~", TB_KEY_PGDN, TB_MOD_CTRL }, + {"\x1b[6;6~", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[6;7~", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[6;8~", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2P", TB_KEY_F1, TB_MOD_SHIFT }, + {"\x1b[1;3P", TB_KEY_F1, TB_MOD_ALT }, + {"\x1b[1;4P", TB_KEY_F1, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5P", TB_KEY_F1, TB_MOD_CTRL }, + {"\x1b[1;6P", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7P", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8P", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2Q", TB_KEY_F2, TB_MOD_SHIFT }, + {"\x1b[1;3Q", TB_KEY_F2, TB_MOD_ALT }, + {"\x1b[1;4Q", TB_KEY_F2, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5Q", TB_KEY_F2, TB_MOD_CTRL }, + {"\x1b[1;6Q", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7Q", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8Q", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2R", TB_KEY_F3, TB_MOD_SHIFT }, + {"\x1b[1;3R", TB_KEY_F3, TB_MOD_ALT }, + {"\x1b[1;4R", TB_KEY_F3, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5R", TB_KEY_F3, TB_MOD_CTRL }, + {"\x1b[1;6R", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7R", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8R", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[1;2S", TB_KEY_F4, TB_MOD_SHIFT }, + {"\x1b[1;3S", TB_KEY_F4, TB_MOD_ALT }, + {"\x1b[1;4S", TB_KEY_F4, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[1;5S", TB_KEY_F4, TB_MOD_CTRL }, + {"\x1b[1;6S", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[1;7S", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[1;8S", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[15;2~", TB_KEY_F5, TB_MOD_SHIFT }, + {"\x1b[15;3~", TB_KEY_F5, TB_MOD_ALT }, + {"\x1b[15;4~", TB_KEY_F5, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[15;5~", TB_KEY_F5, TB_MOD_CTRL }, + {"\x1b[15;6~", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[15;7~", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[15;8~", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[17;2~", TB_KEY_F6, TB_MOD_SHIFT }, + {"\x1b[17;3~", TB_KEY_F6, TB_MOD_ALT }, + {"\x1b[17;4~", TB_KEY_F6, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[17;5~", TB_KEY_F6, TB_MOD_CTRL }, + {"\x1b[17;6~", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[17;7~", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[17;8~", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[18;2~", TB_KEY_F7, TB_MOD_SHIFT }, + {"\x1b[18;3~", TB_KEY_F7, TB_MOD_ALT }, + {"\x1b[18;4~", TB_KEY_F7, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[18;5~", TB_KEY_F7, TB_MOD_CTRL }, + {"\x1b[18;6~", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[18;7~", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[18;8~", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[19;2~", TB_KEY_F8, TB_MOD_SHIFT }, + {"\x1b[19;3~", TB_KEY_F8, TB_MOD_ALT }, + {"\x1b[19;4~", TB_KEY_F8, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[19;5~", TB_KEY_F8, TB_MOD_CTRL }, + {"\x1b[19;6~", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[19;7~", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[19;8~", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[20;2~", TB_KEY_F9, TB_MOD_SHIFT }, + {"\x1b[20;3~", TB_KEY_F9, TB_MOD_ALT }, + {"\x1b[20;4~", TB_KEY_F9, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[20;5~", TB_KEY_F9, TB_MOD_CTRL }, + {"\x1b[20;6~", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[20;7~", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[20;8~", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[21;2~", TB_KEY_F10, TB_MOD_SHIFT }, + {"\x1b[21;3~", TB_KEY_F10, TB_MOD_ALT }, + {"\x1b[21;4~", TB_KEY_F10, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[21;5~", TB_KEY_F10, TB_MOD_CTRL }, + {"\x1b[21;6~", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[21;7~", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[21;8~", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[23;2~", TB_KEY_F11, TB_MOD_SHIFT }, + {"\x1b[23;3~", TB_KEY_F11, TB_MOD_ALT }, + {"\x1b[23;4~", TB_KEY_F11, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[23;5~", TB_KEY_F11, TB_MOD_CTRL }, + {"\x1b[23;6~", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[23;7~", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[23;8~", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b[24;2~", TB_KEY_F12, TB_MOD_SHIFT }, + {"\x1b[24;3~", TB_KEY_F12, TB_MOD_ALT }, + {"\x1b[24;4~", TB_KEY_F12, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[24;5~", TB_KEY_F12, TB_MOD_CTRL }, + {"\x1b[24;6~", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[24;7~", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b[24;8~", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + // rxvt arrows + {"\x1b[a", TB_KEY_ARROW_UP, TB_MOD_SHIFT }, + {"\x1b\x1b[A", TB_KEY_ARROW_UP, TB_MOD_ALT }, + {"\x1b\x1b[a", TB_KEY_ARROW_UP, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1bOa", TB_KEY_ARROW_UP, TB_MOD_CTRL }, + {"\x1b\x1bOa", TB_KEY_ARROW_UP, TB_MOD_CTRL | TB_MOD_ALT }, + + {"\x1b[b", TB_KEY_ARROW_DOWN, TB_MOD_SHIFT }, + {"\x1b\x1b[B", TB_KEY_ARROW_DOWN, TB_MOD_ALT }, + {"\x1b\x1b[b", TB_KEY_ARROW_DOWN, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1bOb", TB_KEY_ARROW_DOWN, TB_MOD_CTRL }, + {"\x1b\x1bOb", TB_KEY_ARROW_DOWN, TB_MOD_CTRL | TB_MOD_ALT }, + + {"\x1b[c", TB_KEY_ARROW_RIGHT, TB_MOD_SHIFT }, + {"\x1b\x1b[C", TB_KEY_ARROW_RIGHT, TB_MOD_ALT }, + {"\x1b\x1b[c", TB_KEY_ARROW_RIGHT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1bOc", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL }, + {"\x1b\x1bOc", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL | TB_MOD_ALT }, + + {"\x1b[d", TB_KEY_ARROW_LEFT, TB_MOD_SHIFT }, + {"\x1b\x1b[D", TB_KEY_ARROW_LEFT, TB_MOD_ALT }, + {"\x1b\x1b[d", TB_KEY_ARROW_LEFT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1bOd", TB_KEY_ARROW_LEFT, TB_MOD_CTRL }, + {"\x1b\x1bOd", TB_KEY_ARROW_LEFT, TB_MOD_CTRL | TB_MOD_ALT }, + + // rxvt keys + {"\x1b[7$", TB_KEY_HOME, TB_MOD_SHIFT }, + {"\x1b\x1b[7~", TB_KEY_HOME, TB_MOD_ALT }, + {"\x1b\x1b[7$", TB_KEY_HOME, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[7^", TB_KEY_HOME, TB_MOD_CTRL }, + {"\x1b[7@", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b\x1b[7^", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[7@", TB_KEY_HOME, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + + {"\x1b\x1b[8~", TB_KEY_END, TB_MOD_ALT }, + {"\x1b\x1b[8$", TB_KEY_END, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[8^", TB_KEY_END, TB_MOD_CTRL }, + {"\x1b\x1b[8^", TB_KEY_END, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[8@", TB_KEY_END, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[8@", TB_KEY_END, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[8$", TB_KEY_END, TB_MOD_SHIFT }, + + {"\x1b\x1b[2~", TB_KEY_INSERT, TB_MOD_ALT }, + {"\x1b\x1b[2$", TB_KEY_INSERT, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[2^", TB_KEY_INSERT, TB_MOD_CTRL }, + {"\x1b\x1b[2^", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[2@", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[2@", TB_KEY_INSERT, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[2$", TB_KEY_INSERT, TB_MOD_SHIFT }, + + {"\x1b\x1b[3~", TB_KEY_DELETE, TB_MOD_ALT }, + {"\x1b\x1b[3$", TB_KEY_DELETE, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[3^", TB_KEY_DELETE, TB_MOD_CTRL }, + {"\x1b\x1b[3^", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[3@", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[3@", TB_KEY_DELETE, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[3$", TB_KEY_DELETE, TB_MOD_SHIFT }, + + {"\x1b\x1b[5~", TB_KEY_PGUP, TB_MOD_ALT }, + {"\x1b\x1b[5$", TB_KEY_PGUP, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[5^", TB_KEY_PGUP, TB_MOD_CTRL }, + {"\x1b\x1b[5^", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[5@", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[5@", TB_KEY_PGUP, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[5$", TB_KEY_PGUP, TB_MOD_SHIFT }, + + {"\x1b\x1b[6~", TB_KEY_PGDN, TB_MOD_ALT }, + {"\x1b\x1b[6$", TB_KEY_PGDN, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[6^", TB_KEY_PGDN, TB_MOD_CTRL }, + {"\x1b\x1b[6^", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[6@", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[6@", TB_KEY_PGDN, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[6$", TB_KEY_PGDN, TB_MOD_SHIFT }, + + {"\x1b\x1b[11~", TB_KEY_F1, TB_MOD_ALT }, + {"\x1b\x1b[23~", TB_KEY_F1, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[11^", TB_KEY_F1, TB_MOD_CTRL }, + {"\x1b\x1b[11^", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[23^", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[23^", TB_KEY_F1, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[23~", TB_KEY_F1, TB_MOD_SHIFT }, + + {"\x1b\x1b[12~", TB_KEY_F2, TB_MOD_ALT }, + {"\x1b\x1b[24~", TB_KEY_F2, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[12^", TB_KEY_F2, TB_MOD_CTRL }, + {"\x1b\x1b[12^", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[24^", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[24^", TB_KEY_F2, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[24~", TB_KEY_F2, TB_MOD_SHIFT }, + + {"\x1b\x1b[13~", TB_KEY_F3, TB_MOD_ALT }, + {"\x1b\x1b[25~", TB_KEY_F3, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[13^", TB_KEY_F3, TB_MOD_CTRL }, + {"\x1b\x1b[13^", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[25^", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[25^", TB_KEY_F3, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[25~", TB_KEY_F3, TB_MOD_SHIFT }, + + {"\x1b\x1b[14~", TB_KEY_F4, TB_MOD_ALT }, + {"\x1b\x1b[26~", TB_KEY_F4, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[14^", TB_KEY_F4, TB_MOD_CTRL }, + {"\x1b\x1b[14^", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[26^", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[26^", TB_KEY_F4, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[26~", TB_KEY_F4, TB_MOD_SHIFT }, + + {"\x1b\x1b[15~", TB_KEY_F5, TB_MOD_ALT }, + {"\x1b\x1b[28~", TB_KEY_F5, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[15^", TB_KEY_F5, TB_MOD_CTRL }, + {"\x1b\x1b[15^", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[28^", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[28^", TB_KEY_F5, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[28~", TB_KEY_F5, TB_MOD_SHIFT }, + + {"\x1b\x1b[17~", TB_KEY_F6, TB_MOD_ALT }, + {"\x1b\x1b[29~", TB_KEY_F6, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[17^", TB_KEY_F6, TB_MOD_CTRL }, + {"\x1b\x1b[17^", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[29^", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[29^", TB_KEY_F6, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[29~", TB_KEY_F6, TB_MOD_SHIFT }, + + {"\x1b\x1b[18~", TB_KEY_F7, TB_MOD_ALT }, + {"\x1b\x1b[31~", TB_KEY_F7, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[18^", TB_KEY_F7, TB_MOD_CTRL }, + {"\x1b\x1b[18^", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[31^", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[31^", TB_KEY_F7, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[31~", TB_KEY_F7, TB_MOD_SHIFT }, + + {"\x1b\x1b[19~", TB_KEY_F8, TB_MOD_ALT }, + {"\x1b\x1b[32~", TB_KEY_F8, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[19^", TB_KEY_F8, TB_MOD_CTRL }, + {"\x1b\x1b[19^", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[32^", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[32^", TB_KEY_F8, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[32~", TB_KEY_F8, TB_MOD_SHIFT }, + + {"\x1b\x1b[20~", TB_KEY_F9, TB_MOD_ALT }, + {"\x1b\x1b[33~", TB_KEY_F9, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[20^", TB_KEY_F9, TB_MOD_CTRL }, + {"\x1b\x1b[20^", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[33^", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[33^", TB_KEY_F9, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[33~", TB_KEY_F9, TB_MOD_SHIFT }, + + {"\x1b\x1b[21~", TB_KEY_F10, TB_MOD_ALT }, + {"\x1b\x1b[34~", TB_KEY_F10, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[21^", TB_KEY_F10, TB_MOD_CTRL }, + {"\x1b\x1b[21^", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[34^", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[34^", TB_KEY_F10, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[34~", TB_KEY_F10, TB_MOD_SHIFT }, + + {"\x1b\x1b[23~", TB_KEY_F11, TB_MOD_ALT }, + {"\x1b\x1b[23$", TB_KEY_F11, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[23^", TB_KEY_F11, TB_MOD_CTRL }, + {"\x1b\x1b[23^", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[23@", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[23@", TB_KEY_F11, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[23$", TB_KEY_F11, TB_MOD_SHIFT }, + + {"\x1b\x1b[24~", TB_KEY_F12, TB_MOD_ALT }, + {"\x1b\x1b[24$", TB_KEY_F12, TB_MOD_ALT | TB_MOD_SHIFT }, + {"\x1b[24^", TB_KEY_F12, TB_MOD_CTRL }, + {"\x1b\x1b[24^", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1b\x1b[24@", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_ALT | TB_MOD_SHIFT}, + {"\x1b[24@", TB_KEY_F12, TB_MOD_CTRL | TB_MOD_SHIFT }, + {"\x1b[24$", TB_KEY_F12, TB_MOD_SHIFT }, + + // linux console/putty arrows + {"\x1b[A", TB_KEY_ARROW_UP, TB_MOD_SHIFT }, + {"\x1b[B", TB_KEY_ARROW_DOWN, TB_MOD_SHIFT }, + {"\x1b[C", TB_KEY_ARROW_RIGHT, TB_MOD_SHIFT }, + {"\x1b[D", TB_KEY_ARROW_LEFT, TB_MOD_SHIFT }, + + // more putty arrows + {"\x1bOA", TB_KEY_ARROW_UP, TB_MOD_CTRL }, + {"\x1b\x1bOA", TB_KEY_ARROW_UP, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1bOB", TB_KEY_ARROW_DOWN, TB_MOD_CTRL }, + {"\x1b\x1bOB", TB_KEY_ARROW_DOWN, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1bOC", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL }, + {"\x1b\x1bOC", TB_KEY_ARROW_RIGHT, TB_MOD_CTRL | TB_MOD_ALT }, + {"\x1bOD", TB_KEY_ARROW_LEFT, TB_MOD_CTRL }, + {"\x1b\x1bOD", TB_KEY_ARROW_LEFT, TB_MOD_CTRL | TB_MOD_ALT }, + + {NULL, 0, 0 }, +}; + +static const unsigned char utf8_length[256] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, + 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, + 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, + 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 1, 1}; + +static const unsigned char utf8_mask[6] = {0x7f, 0x1f, 0x0f, 0x07, 0x03, 0x01}; + +static int tb_reset(); +static int tb_printf_inner(int x, int y, uintattr_t fg, uintattr_t bg, + size_t *out_w, const char *fmt, va_list vl); +static int init_term_attrs(); +static int init_term_caps(); +static int init_cap_trie(); +static int cap_trie_add(const char *cap, uint16_t key, uint8_t mod); +static int cap_trie_find(const char *buf, size_t nbuf, struct cap_trie_t **last, + size_t *depth); +static int cap_trie_deinit(struct cap_trie_t *node); +static int init_resize_handler(); +static int send_init_escape_codes(); +static int send_clear(); +static int update_term_size(); +static int update_term_size_via_esc(); +static int init_cellbuf(); +static int tb_deinit(); +static int load_terminfo(); +static int load_terminfo_from_path(const char *path, const char *term); +static int read_terminfo_path(const char *path); +static int parse_terminfo_caps(); +static int load_builtin_caps(); +static const char *get_terminfo_string(int16_t str_offsets_pos, + int16_t str_table_pos, int16_t str_table_len, int16_t str_index); +static int wait_event(struct tb_event *event, int timeout); +static int extract_event(struct tb_event *event); +static int extract_esc(struct tb_event *event); +static int extract_esc_user(struct tb_event *event, int is_post); +static int extract_esc_cap(struct tb_event *event); +static int extract_esc_mouse(struct tb_event *event); +static int resize_if_needed(); +static void handle_resize(int sig); +static int send_attr(uintattr_t fg, uintattr_t bg); +static int send_sgr(uintattr_t fg, uintattr_t bg); +static int send_cursor_if(int x, int y); +static int send_char(int x, int y, uint32_t ch); +static int send_cluster(int x, int y, uint32_t *ch, size_t nch); +static int convert_num(uint32_t num, char *buf); +static int cell_cmp(struct tb_cell *a, struct tb_cell *b); +static int cell_copy(struct tb_cell *dst, struct tb_cell *src); +static int cell_set(struct tb_cell *cell, uint32_t *ch, size_t nch, + uintattr_t fg, uintattr_t bg); +static int cell_reserve_ech(struct tb_cell *cell, size_t n); +static int cell_free(struct tb_cell *cell); +static int cellbuf_init(struct cellbuf_t *c, int w, int h); +static int cellbuf_free(struct cellbuf_t *c); +static int cellbuf_clear(struct cellbuf_t *c); +static int cellbuf_get(struct cellbuf_t *c, int x, int y, struct tb_cell **out); +static int cellbuf_resize(struct cellbuf_t *c, int w, int h); +static int bytebuf_puts(struct bytebuf_t *b, const char *str); +static int bytebuf_nputs(struct bytebuf_t *b, const char *str, size_t nstr); +static int bytebuf_shift(struct bytebuf_t *b, size_t n); +static int bytebuf_flush(struct bytebuf_t *b, int fd); +static int bytebuf_reserve(struct bytebuf_t *b, size_t sz); +static int bytebuf_free(struct bytebuf_t *b); + +int tb_init() { + return tb_init_file("/dev/tty"); +} + +int tb_init_file(const char *path) { + if (global.initialized) { + return TB_ERR_INIT_ALREADY; + } + int ttyfd = open(path, O_RDWR); + if (ttyfd < 0) { + global.last_errno = errno; + return TB_ERR_INIT_OPEN; + } + global.ttyfd_open = 1; + return tb_init_fd(ttyfd); +} + +int tb_init_fd(int ttyfd) { + return tb_init_rwfd(ttyfd, ttyfd); +} + +int tb_init_rwfd(int rfd, int wfd) { + int rv; + + tb_reset(); + global.ttyfd = rfd == wfd && isatty(rfd) ? rfd : -1; + global.rfd = rfd; + global.wfd = wfd; + + do { + if_err_break(rv, init_term_attrs()); + if_err_break(rv, init_term_caps()); + if_err_break(rv, init_cap_trie()); + if_err_break(rv, init_resize_handler()); + if_err_break(rv, send_init_escape_codes()); + if_err_break(rv, send_clear()); + if_err_break(rv, update_term_size()); + if_err_break(rv, init_cellbuf()); + global.initialized = 1; + } while (0); + + if (rv != TB_OK) { + tb_deinit(); + } + + return rv; +} + +int tb_shutdown() { + if_not_init_return(); + tb_deinit(); + return TB_OK; +} + +int tb_width() { + if_not_init_return(); + return global.width; +} + +int tb_height() { + if_not_init_return(); + return global.height; +} + +int tb_clear() { + int rv; + if_not_init_return(); + if_err_return(rv, resize_if_needed()); + return cellbuf_clear(&global.back); +} + +int tb_set_clear_attrs(uintattr_t fg, uintattr_t bg) { + if_not_init_return(); + global.fg = fg; + global.bg = bg; + return TB_OK; +} + +int tb_present() { + if_not_init_return(); + + int rv; + if_err_return(rv, resize_if_needed()); + + // TODO Assert global.back.(width,height) == global.front.(width,height) + + global.last_x = -1; + global.last_y = -1; + + int x, y, i; + for (y = 0; y < global.front.height; y++) { + for (x = 0; x < global.front.width;) { + struct tb_cell *back, *front; + if_err_return(rv, cellbuf_get(&global.back, x, y, &back)); + if_err_return(rv, cellbuf_get(&global.front, x, y, &front)); + + int w; + { +#ifdef TB_OPT_EGC + if (back->nech > 0) + w = wcswidth((wchar_t *)back->ech, back->nech); + else +#endif + w = wcwidth(back->ch); + } + if (w < 1) { + w = 1; + } + + if (cell_cmp(back, front) != 0) { + cell_copy(front, back); + + send_attr(back->fg, back->bg); + if (w > 1 && x >= global.front.width - (w - 1)) { + for (i = x; i < global.front.width; i++) { + send_char(i, y, ' '); + } + } else { + { +#ifdef TB_OPT_EGC + if (back->nech > 0) + send_cluster(x, y, back->ech, back->nech); + else +#endif + send_char(x, y, back->ch); + } + for (i = 1; i < w; i++) { + struct tb_cell *front_wide; + if_err_return(rv, + cellbuf_get(&global.front, x + i, y, &front_wide)); + cell_set(front_wide, 0, 1, back->fg, back->bg); + } + } + } + x += w; + } + } + + if_err_return(rv, send_cursor_if(global.cursor_x, global.cursor_y)); + if_err_return(rv, bytebuf_flush(&global.out, global.wfd)); + + return TB_OK; +} + +int tb_set_cursor(int cx, int cy) { + if_not_init_return(); + int rv; + if (cx < 0) + cx = 0; + if (cy < 0) + cy = 0; + if (global.cursor_x == -1) { + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_SHOW_CURSOR])); + } + if_err_return(rv, send_cursor_if(cx, cy)); + global.cursor_x = cx; + global.cursor_y = cy; + return TB_OK; +} + +int tb_hide_cursor() { + if_not_init_return(); + int rv; + if (global.cursor_x >= 0) { + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_HIDE_CURSOR])); + } + global.cursor_x = -1; + global.cursor_y = -1; + return TB_OK; +} + +int tb_set_cell(int x, int y, uint32_t ch, uintattr_t fg, uintattr_t bg) { + if_not_init_return(); + return tb_set_cell_ex(x, y, &ch, 1, fg, bg); +} + +int tb_set_cell_ex(int x, int y, uint32_t *ch, size_t nch, uintattr_t fg, + uintattr_t bg) { + if_not_init_return(); + int rv; + struct tb_cell *cell; + if_err_return(rv, cellbuf_get(&global.back, x, y, &cell)); + if_err_return(rv, cell_set(cell, ch, nch, fg, bg)); + return TB_OK; +} + +int tb_extend_cell(int x, int y, uint32_t ch) { + if_not_init_return(); +#ifdef TB_OPT_EGC + int rv; + struct tb_cell *cell; + size_t nech; + if_err_return(rv, cellbuf_get(&global.back, x, y, &cell)); + if (cell->nech > 0) { // append to ech + nech = cell->nech + 1; + if_err_return(rv, cell_reserve_ech(cell, nech)); + cell->ech[nech - 1] = ch; + } else { // make new ech + nech = 2; + if_err_return(rv, cell_reserve_ech(cell, nech)); + cell->ech[0] = cell->ch; + cell->ech[1] = ch; + } + cell->ech[nech] = '\0'; + cell->nech = nech; + return TB_OK; +#else + (void)x; + (void)y; + (void)ch; + return TB_ERR; +#endif +} + +int tb_set_input_mode(int mode) { + if_not_init_return(); + if (mode == TB_INPUT_CURRENT) { + return global.input_mode; + } + + if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == 0) { + mode |= TB_INPUT_ESC; + } + + if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == (TB_INPUT_ESC | TB_INPUT_ALT)) + { + mode &= ~TB_INPUT_ALT; + } + + if (mode & TB_INPUT_MOUSE) { + bytebuf_puts(&global.out, TB_HARDCAP_ENTER_MOUSE); + bytebuf_flush(&global.out, global.wfd); + } else { + bytebuf_puts(&global.out, TB_HARDCAP_EXIT_MOUSE); + bytebuf_flush(&global.out, global.wfd); + } + + global.input_mode = mode; + return TB_OK; +} + +int tb_set_output_mode(int mode) { + if_not_init_return(); + if (mode == TB_OUTPUT_CURRENT) { + return global.output_mode; + } + global.output_mode = mode; + return TB_OK; +} + +int tb_peek_event(struct tb_event *event, int timeout_ms) { + if_not_init_return(); + return wait_event(event, timeout_ms); +} + +int tb_poll_event(struct tb_event *event) { + if_not_init_return(); + return wait_event(event, -1); +} + +int tb_get_fds(int *ttyfd, int *resizefd) { + if_not_init_return(); + + *ttyfd = global.rfd; + *resizefd = global.resize_pipefd[0]; + + return TB_OK; +} + +int tb_print(int x, int y, uintattr_t fg, uintattr_t bg, const char *str) { + return tb_print_ex(x, y, fg, bg, NULL, str); +} + +int tb_print_ex(int x, int y, uintattr_t fg, uintattr_t bg, size_t *out_w, + const char *str) { + int rv; + uint32_t uni; + int w, ix = x; + if (out_w) { + *out_w = 0; + } + while (*str) { + str += tb_utf8_char_to_unicode(&uni, str); + w = wcwidth(uni); + if (w < 0) { + w = 1; + } + if (w == 0 && x > ix) { + if_err_return(rv, tb_extend_cell(x - 1, y, uni)); + } else { + if_err_return(rv, tb_set_cell(x, y, uni, fg, bg)); + } + x += w; + if (out_w) { + *out_w += w; + } + } + return TB_OK; +} + +int tb_printf(int x, int y, uintattr_t fg, uintattr_t bg, const char *fmt, + ...) { + int rv; + va_list vl; + va_start(vl, fmt); + rv = tb_printf_inner(x, y, fg, bg, NULL, fmt, vl); + va_end(vl); + return rv; +} + +int tb_printf_ex(int x, int y, uintattr_t fg, uintattr_t bg, size_t *out_w, + const char *fmt, ...) { + int rv; + va_list vl; + va_start(vl, fmt); + rv = tb_printf_inner(x, y, fg, bg, out_w, fmt, vl); + va_end(vl); + return rv; +} + +int tb_send(const char *buf, size_t nbuf) { + return bytebuf_nputs(&global.out, buf, nbuf); +} + +int tb_sendf(const char *fmt, ...) { + int rv; + char buf[TB_OPT_PRINTF_BUF]; + va_list vl; + va_start(vl, fmt); + rv = vsnprintf(buf, sizeof(buf), fmt, vl); + va_end(vl); + if (rv < 0 || rv >= (int)sizeof(buf)) { + return TB_ERR; + } + return tb_send(buf, (size_t)rv); +} + +int tb_set_func(int fn_type, int (*fn)(struct tb_event *, size_t *)) { + switch (fn_type) { + case TB_FUNC_EXTRACT_PRE: + global.fn_extract_esc_pre = fn; + return TB_OK; + case TB_FUNC_EXTRACT_POST: + global.fn_extract_esc_post = fn; + return TB_OK; + } + return TB_ERR; +} + +struct tb_cell *tb_cell_buffer() { + if (!global.initialized) + return NULL; + return global.back.cells; +} + +int tb_utf8_char_length(char c) { + return utf8_length[(unsigned char)c]; +} + +int tb_utf8_char_to_unicode(uint32_t *out, const char *c) { + if (*c == 0) { + return TB_ERR; + } + + int i; + unsigned char len = tb_utf8_char_length(*c); + unsigned char mask = utf8_mask[len - 1]; + uint32_t result = c[0] & mask; + for (i = 1; i < len; ++i) { + result <<= 6; + result |= c[i] & 0x3f; + } + + *out = result; + return (int)len; +} + +int tb_utf8_unicode_to_char(char *out, uint32_t c) { + int len = 0; + int first; + int i; + + if (c < 0x80) { + first = 0; + len = 1; + } else if (c < 0x800) { + first = 0xc0; + len = 2; + } else if (c < 0x10000) { + first = 0xe0; + len = 3; + } else if (c < 0x200000) { + first = 0xf0; + len = 4; + } else if (c < 0x4000000) { + first = 0xf8; + len = 5; + } else { + first = 0xfc; + len = 6; + } + + for (i = len - 1; i > 0; --i) { + out[i] = (c & 0x3f) | 0x80; + c >>= 6; + } + out[0] = c | first; + + return len; +} + +int tb_last_errno() { + return global.last_errno; +} + +const char *tb_strerror(int err) { + switch (err) { + case TB_OK: + return "Success"; + case TB_ERR_NEED_MORE: + return "Not enough input"; + case TB_ERR_INIT_ALREADY: + return "Termbox initialized already"; + case TB_ERR_MEM: + return "Out of memory"; + case TB_ERR_NO_EVENT: + return "No event"; + case TB_ERR_NO_TERM: + return "No TERM in environment"; + case TB_ERR_NOT_INIT: + return "Termbox not initialized"; + case TB_ERR_OUT_OF_BOUNDS: + return "Out of bounds"; + case TB_ERR_UNSUPPORTED_TERM: + return "Unsupported terminal"; + case TB_ERR_CAP_COLLISION: + return "Termcaps collision"; + case TB_ERR_RESIZE_SSCANF: + return "Terminal width/height not received by sscanf() after " + "resize"; + case TB_ERR: + case TB_ERR_INIT_OPEN: + case TB_ERR_READ: + case TB_ERR_RESIZE_IOCTL: + case TB_ERR_RESIZE_PIPE: + case TB_ERR_RESIZE_SIGACTION: + case TB_ERR_POLL: + case TB_ERR_TCGETATTR: + case TB_ERR_TCSETATTR: + case TB_ERR_RESIZE_WRITE: + case TB_ERR_RESIZE_POLL: + case TB_ERR_RESIZE_READ: + default: + return strerror(global.last_errno); + } +} + +static int tb_reset() { + int ttyfd_open = global.ttyfd_open; + memset(&global, 0, sizeof(global)); + global.ttyfd = -1; + global.rfd = -1; + global.wfd = -1; + global.ttyfd_open = ttyfd_open; + global.resize_pipefd[0] = -1; + global.resize_pipefd[1] = -1; + global.width = -1; + global.height = -1; + global.cursor_x = -1; + global.cursor_y = -1; + global.last_x = -1; + global.last_y = -1; + global.fg = TB_DEFAULT; + global.bg = TB_DEFAULT; + global.last_fg = ~global.fg; + global.last_bg = ~global.bg; + global.input_mode = TB_INPUT_ESC; + global.output_mode = TB_OUTPUT_NORMAL; + return TB_OK; +} + +static int init_term_attrs() { + if (global.ttyfd < 0) { + return TB_OK; + } + + if (tcgetattr(global.ttyfd, &global.orig_tios) != 0) { + global.last_errno = errno; + return TB_ERR_TCGETATTR; + } + + struct termios tios; + memcpy(&tios, &global.orig_tios, sizeof(tios)); + global.has_orig_tios = 1; + + cfmakeraw(&tios); + tios.c_cc[VMIN] = 1; + tios.c_cc[VTIME] = 0; + + if (tcsetattr(global.ttyfd, TCSAFLUSH, &tios) != 0) { + global.last_errno = errno; + return TB_ERR_TCSETATTR; + } + + return TB_OK; +} + +int tb_printf_inner(int x, int y, uintattr_t fg, uintattr_t bg, size_t *out_w, + const char *fmt, va_list vl) { + int rv; + char buf[TB_OPT_PRINTF_BUF]; + rv = vsnprintf(buf, sizeof(buf), fmt, vl); + if (rv < 0 || rv >= (int)sizeof(buf)) { + return TB_ERR; + } + return tb_print_ex(x, y, fg, bg, out_w, buf); +} + +static int init_term_caps() { + if (load_terminfo() == TB_OK) { + return parse_terminfo_caps(); + } + return load_builtin_caps(); +} + +static int init_cap_trie() { + int rv, i; + + // Add caps from terminfo or built-in + for (i = 0; i < TB_CAP__COUNT_KEYS; i++) { + if_err_return(rv, cap_trie_add(global.caps[i], tb_key_i(i), 0)); + } + + // Add built-in mod caps + for (i = 0; builtin_mod_caps[i].cap != NULL; i++) { + rv = cap_trie_add(builtin_mod_caps[i].cap, builtin_mod_caps[i].key, + builtin_mod_caps[i].mod); + // Collisions are OK. This can happen if global.caps collides with + // builtin_mod_caps. It is desirable to give precedence to global.caps + // here. + if (rv != TB_OK && rv != TB_ERR_CAP_COLLISION) { + return rv; + } + } + + return TB_OK; +} + +static int cap_trie_add(const char *cap, uint16_t key, uint8_t mod) { + struct cap_trie_t *next, *node = &global.cap_trie; + size_t i, j; + for (i = 0; cap[i] != '\0'; i++) { + char c = cap[i]; + next = NULL; + + // Check if c is already a child of node + for (j = 0; j < node->nchildren; j++) { + if (node->children[j].c == c) { + next = &node->children[j]; + break; + } + } + if (!next) { + // We need to add a new child to node + node->nchildren += 1; + node->children = + tb_realloc(node->children, sizeof(*node) * node->nchildren); + if (!node->children) { + return TB_ERR_MEM; + } + next = &node->children[node->nchildren - 1]; + memset(next, 0, sizeof(*next)); + next->c = c; + } + + // Continue + node = next; + } + + if (node->is_leaf) { + // Already a leaf here + return TB_ERR_CAP_COLLISION; + } + + node->is_leaf = 1; + node->key = key; + node->mod = mod; + return TB_OK; +} + +static int cap_trie_find(const char *buf, size_t nbuf, struct cap_trie_t **last, + size_t *depth) { + struct cap_trie_t *next, *node = &global.cap_trie; + size_t i, j; + *last = node; + *depth = 0; + for (i = 0; i < nbuf; i++) { + char c = buf[i]; + next = NULL; + + // Find c in node.children + for (j = 0; j < node->nchildren; j++) { + if (node->children[j].c == c) { + next = &node->children[j]; + break; + } + } + if (!next) { + // Not found + return TB_OK; + } + node = next; + *last = node; + *depth += 1; + if (node->is_leaf && node->nchildren < 1) { + break; + } + } + return TB_OK; +} + +static int cap_trie_deinit(struct cap_trie_t *node) { + size_t j; + for (j = 0; j < node->nchildren; j++) { + cap_trie_deinit(&node->children[j]); + } + if (node->children) { + tb_free(node->children); + } + memset(node, 0, sizeof(*node)); + return TB_OK; +} + +static int init_resize_handler() { + if (pipe(global.resize_pipefd) != 0) { + global.last_errno = errno; + return TB_ERR_RESIZE_PIPE; + } + + struct sigaction sa; + memset(&sa, 0, sizeof(sa)); + sa.sa_handler = handle_resize; + if (sigaction(SIGWINCH, &sa, NULL) != 0) { + global.last_errno = errno; + return TB_ERR_RESIZE_SIGACTION; + } + + return TB_OK; +} + +static int send_init_escape_codes() { + int rv; + if_err_return(rv, bytebuf_puts(&global.out, global.caps[TB_CAP_ENTER_CA])); + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_ENTER_KEYPAD])); + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_HIDE_CURSOR])); + return TB_OK; +} + +static int send_clear() { + int rv; + + if_err_return(rv, send_attr(global.fg, global.bg)); + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_CLEAR_SCREEN])); + + if_err_return(rv, send_cursor_if(global.cursor_x, global.cursor_y)); + if_err_return(rv, bytebuf_flush(&global.out, global.wfd)); + + global.last_x = -1; + global.last_y = -1; + + return TB_OK; +} + +static int update_term_size() { + int rv; + + if (global.ttyfd < 0) { + return TB_OK; + } + + struct winsize sz; + memset(&sz, 0, sizeof(sz)); + + if (ioctl(global.ttyfd, TIOCGWINSZ, &sz) == 0) { + global.width = sz.ws_col; + global.height = sz.ws_row; + return TB_OK; + } + + // If TB_RESIZE_FALLBACK deinfed, try >cursor(9999,9999), >u7, <u6 + //#ifdef TB_RESIZE_FALLBACK + if_ok_return(rv, update_term_size_via_esc()); + //#endif + + global.last_errno = errno; + return TB_ERR_RESIZE_IOCTL; +} + +static int update_term_size_via_esc() { +#ifndef TB_RESIZE_FALLBACK_MS +#define TB_RESIZE_FALLBACK_MS 1000 +#endif + + char *move_and_report = "\x1b[9999;9999H\x1b[6n"; + ssize_t write_rv = + write(global.wfd, move_and_report, strlen(move_and_report)); + if (write_rv != (ssize_t)strlen(move_and_report)) { + return TB_ERR_RESIZE_WRITE; + } + +#if !TB_OPT_SELECT + struct pollfd fd = {.fd = global.rfd, .events = POLLIN}; + int poll_rv = poll(&fd, 1, TB_RESIZE_FALLBACK_MS); +#else + fd_set fds; + FD_ZERO(&fds); + FD_SET(global.rfd, &fds); + + struct timeval timeout; + timeout.tv_sec = 0; + timeout.tv_usec = TB_RESIZE_FALLBACK_MS * 1000; + + int poll_rv = select(global.rfd + 1, &fds, NULL, NULL, &timeout); +#endif + + if (poll_rv != 1) { + global.last_errno = errno; + return TB_ERR_RESIZE_POLL; + } + + char buf[TB_OPT_READ_BUF]; + ssize_t read_rv = read(global.rfd, buf, sizeof(buf) - 1); + if (read_rv < 1) { + global.last_errno = errno; + return TB_ERR_RESIZE_READ; + } + buf[read_rv] = '\0'; + + int rw, rh; + if (sscanf(buf, "\x1b[%d;%dR", &rh, &rw) != 2) { + return TB_ERR_RESIZE_SSCANF; + } + + global.width = rw; + global.height = rh; + return TB_OK; +} + +static int init_cellbuf() { + int rv; + if_err_return(rv, cellbuf_init(&global.back, global.width, global.height)); + if_err_return(rv, cellbuf_init(&global.front, global.width, global.height)); + if_err_return(rv, cellbuf_clear(&global.back)); + if_err_return(rv, cellbuf_clear(&global.front)); + return TB_OK; +} + +static int tb_deinit() { + if (global.caps[0] != NULL && global.wfd >= 0) { + bytebuf_puts(&global.out, global.caps[TB_CAP_SHOW_CURSOR]); + bytebuf_puts(&global.out, global.caps[TB_CAP_SGR0]); + bytebuf_puts(&global.out, global.caps[TB_CAP_CLEAR_SCREEN]); + bytebuf_puts(&global.out, global.caps[TB_CAP_EXIT_CA]); + bytebuf_puts(&global.out, global.caps[TB_CAP_EXIT_KEYPAD]); + bytebuf_puts(&global.out, TB_HARDCAP_EXIT_MOUSE); + bytebuf_flush(&global.out, global.wfd); + } + if (global.ttyfd >= 0) { + if (global.has_orig_tios) { + tcsetattr(global.ttyfd, TCSAFLUSH, &global.orig_tios); + } + if (global.ttyfd_open) { + close(global.ttyfd); + global.ttyfd_open = 0; + } + } + + sigaction(SIGWINCH, &(struct sigaction){.sa_handler = SIG_DFL}, NULL); + if (global.resize_pipefd[0] >= 0) + close(global.resize_pipefd[0]); + if (global.resize_pipefd[1] >= 0) + close(global.resize_pipefd[1]); + + cellbuf_free(&global.back); + cellbuf_free(&global.front); + bytebuf_free(&global.in); + bytebuf_free(&global.out); + + if (global.terminfo) + tb_free(global.terminfo); + + cap_trie_deinit(&global.cap_trie); + + tb_reset(); + return TB_OK; +} + +static int load_terminfo() { + int rv; + char tmp[PATH_MAX]; + + // See terminfo(5) "Fetching Compiled Descriptions" for a description of + // this behavior. Some of these paths are compile-time ncurses options, so + // best guesses are used here. + const char *term = getenv("TERM"); + if (!term) { + return TB_ERR; + } + + // If TERMINFO is set, try that directory and stop + const char *terminfo = getenv("TERMINFO"); + if (terminfo) { + return load_terminfo_from_path(terminfo, term); + } + + // Next try ~/.terminfo + const char *home = getenv("HOME"); + if (home) { + snprintf_or_return(rv, tmp, sizeof(tmp), "%s/.terminfo", home); + if_ok_return(rv, load_terminfo_from_path(tmp, term)); + } + + // Next try TERMINFO_DIRS + // + // Note, empty entries are supposed to be interpretted as the "compiled-in + // default", which is of course system-dependent. Previously /etc/terminfo + // was used here. Let's skip empty entries altogether rather than give + // precedence to a guess, and check common paths after this loop. + const char *dirs = getenv("TERMINFO_DIRS"); + if (dirs) { + snprintf_or_return(rv, tmp, sizeof(tmp), "%s", dirs); + char *dir = strtok(tmp, ":"); + while (dir) { + const char *cdir = dir; + if (*cdir != '\0') { + if_ok_return(rv, load_terminfo_from_path(cdir, term)); + } + dir = strtok(NULL, ":"); + } + } + +#ifdef TB_TERMINFO_DIR + if_ok_return(rv, load_terminfo_from_path(TB_TERMINFO_DIR, term)); +#endif + if_ok_return(rv, load_terminfo_from_path("/usr/local/etc/terminfo", term)); + if_ok_return(rv, + load_terminfo_from_path("/usr/local/share/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/usr/local/lib/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/etc/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/usr/share/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/usr/lib/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/usr/share/lib/terminfo", term)); + if_ok_return(rv, load_terminfo_from_path("/lib/terminfo", term)); + + return TB_ERR; +} + +static int load_terminfo_from_path(const char *path, const char *term) { + int rv; + char tmp[PATH_MAX]; + + // Look for term at this terminfo location, e.g., <terminfo>/x/xterm + snprintf_or_return(rv, tmp, sizeof(tmp), "%s/%c/%s", path, term[0], term); + if_ok_return(rv, read_terminfo_path(tmp)); + +#ifdef __APPLE__ + // Try the Darwin equivalent path, e.g., <terminfo>/78/xterm + snprintf_or_return(rv, tmp, sizeof(tmp), "%s/%x/%s", path, term[0], term); + return read_terminfo_path(tmp); +#endif + + return TB_ERR; +} + +static int read_terminfo_path(const char *path) { + FILE *fp = fopen(path, "rb"); + if (!fp) { + return TB_ERR; + } + + struct stat st; + if (fstat(fileno(fp), &st) != 0) { + fclose(fp); + return TB_ERR; + } + + size_t fsize = st.st_size; + char *data = tb_malloc(fsize); + if (!data) { + fclose(fp); + return TB_ERR; + } + + if (fread(data, 1, fsize, fp) != fsize) { + fclose(fp); + tb_free(data); + return TB_ERR; + } + + global.terminfo = data; + global.nterminfo = fsize; + + fclose(fp); + return TB_OK; +} + +static int parse_terminfo_caps() { + // See term(5) "LEGACY STORAGE FORMAT" and "EXTENDED STORAGE FORMAT" for a + // description of this behavior. + + // Ensure there's at least a header's worth of data + if (global.nterminfo < 6) { + return TB_ERR; + } + + int16_t *header = (int16_t *)global.terminfo; + // header[0] the magic number (octal 0432 or 01036) + // header[1] the size, in bytes, of the names section + // header[2] the number of bytes in the boolean section + // header[3] the number of short integers in the numbers section + // header[4] the number of offsets (short integers) in the strings section + // header[5] the size, in bytes, of the string table + + // Legacy ints are 16-bit, extended ints are 32-bit + const int bytes_per_int = header[0] == 01036 ? 4 // 32-bit + : 2; // 16-bit + + // > Between the boolean section and the number section, a null byte will be + // > inserted, if necessary, to ensure that the number section begins on an + // > even byte + const int align_offset = (header[1] + header[2]) % 2 != 0 ? 1 : 0; + + const int pos_str_offsets = + (6 * sizeof(int16_t)) // header (12 bytes) + + header[1] // length of names section + + header[2] // length of boolean section + + align_offset + + (header[3] * bytes_per_int); // length of numbers section + + const int pos_str_table = + pos_str_offsets + + (header[4] * sizeof(int16_t)); // length of string offsets table + + // Load caps + int i; + for (i = 0; i < TB_CAP__COUNT; i++) { + const char *cap = get_terminfo_string(pos_str_offsets, pos_str_table, + header[5], terminfo_cap_indexes[i]); + if (!cap) { + // Something is not right + return TB_ERR; + } + global.caps[i] = cap; + } + + return TB_OK; +} + +static int load_builtin_caps() { + int i, j; + const char *term = getenv("TERM"); + + if (!term) { + return TB_ERR_NO_TERM; + } + + // Check for exact TERM match + for (i = 0; builtin_terms[i].name != NULL; i++) { + if (strcmp(term, builtin_terms[i].name) == 0) { + for (j = 0; j < TB_CAP__COUNT; j++) { + global.caps[j] = builtin_terms[i].caps[j]; + } + return TB_OK; + } + } + + // Check for partial TERM or alias match + for (i = 0; builtin_terms[i].name != NULL; i++) { + if (strstr(term, builtin_terms[i].name) != NULL || + (*(builtin_terms[i].alias) != '\0' && + strstr(term, builtin_terms[i].alias) != NULL)) + { + for (j = 0; j < TB_CAP__COUNT; j++) { + global.caps[j] = builtin_terms[i].caps[j]; + } + return TB_OK; + } + } + + return TB_ERR_UNSUPPORTED_TERM; } + +static const char *get_terminfo_string(int16_t str_offsets_pos, + int16_t str_table_pos, int16_t str_table_len, int16_t str_index) { + const int16_t *str_offset = + (int16_t *)(global.terminfo + (int)str_offsets_pos + + ((int)str_index * (int)sizeof(int16_t))); + if (*str_offset >= str_table_len) { + // Invalid string offset + return NULL; + } + if (((size_t)((int)str_table_pos + (int)*str_offset)) >= global.nterminfo) { + // Truncated/corrupt terminfo? + return NULL; + } + return ( + const char *)(global.terminfo + (int)str_table_pos + (int)*str_offset); +} + +static int wait_event(struct tb_event *event, int timeout) { + int rv; + char buf[TB_OPT_READ_BUF]; + + memset(event, 0, sizeof(*event)); + if_ok_return(rv, extract_event(event)); + +#if !TB_OPT_SELECT + struct pollfd fds[2] = { + {.fd = global.rfd, .events = POLLIN}, + {.fd = global.resize_pipefd[0], .events = POLLIN} + }; +#else + fd_set fds; + struct timeval tv; + tv.tv_sec = timeout / 1000; + tv.tv_usec = (timeout - (tv.tv_sec * 1000)) * 1000; +#endif + + do { +#if !TB_OPT_SELECT + int poll_rv = poll(fds, 2, timeout); +#else + FD_ZERO(&fds); + FD_SET(global.rfd, &fds); + FD_SET(global.resize_pipefd[0], &fds); + + int maxfd = global.resize_pipefd[0] > global.rfd + ? global.resize_pipefd[0] + : global.rfd; + + int poll_rv = + select(maxfd + 1, &fds, NULL, NULL, (timeout < 0) ? NULL : &tv); +#endif + + if (poll_rv < 0) { + // Let EINTR/EAGAIN bubble up + global.last_errno = errno; + return TB_ERR_POLL; + } else if (poll_rv == 0) { + return TB_ERR_NO_EVENT; + } + +#if !TB_OPT_SELECT + int tty_has_events = (fds[0].revents & POLLIN); + int resize_has_events = (fds[1].revents & POLLIN); +#else + int tty_has_events = (FD_ISSET(global.rfd, &fds)); + int resize_has_events = (FD_ISSET(global.resize_pipefd[0], &fds)); #endif + + if (tty_has_events) { + ssize_t read_rv = read(global.rfd, buf, sizeof(buf)); + if (read_rv < 0) { + global.last_errno = errno; + return TB_ERR_READ; + } else if (read_rv > 0) { + bytebuf_nputs(&global.in, buf, read_rv); + } + } + + if (resize_has_events) { + int ignore = 0; + read(global.resize_pipefd[0], &ignore, sizeof(ignore)); + if_err_return(rv, update_term_size()); + event->type = TB_EVENT_RESIZE; + event->w = global.width; + event->h = global.height; + global.need_resize = 1; + return TB_OK; + } + + memset(event, 0, sizeof(*event)); + if_ok_return(rv, extract_event(event)); + } while (timeout == -1); + + return rv; +} + +static int extract_event(struct tb_event *event) { + int rv; + struct bytebuf_t *in = &global.in; + + if (in->len == 0) { + return TB_ERR; + } + + if (in->buf[0] == '\x1b') { + // Escape sequence? + // In TB_INPUT_ESC, skip if the buffer is a single escape char + if (!((global.input_mode & TB_INPUT_ESC) && in->len == 1)) { + if_ok_or_need_more_return(rv, extract_esc(event)); + } + + // Escape key? + if (global.input_mode & TB_INPUT_ESC) { + event->type = TB_EVENT_KEY; + event->ch = 0; + event->key = TB_KEY_ESC; + event->mod = 0; + bytebuf_shift(in, 1); + return TB_OK; + } + + // Recurse for alt key + event->mod |= TB_MOD_ALT; + bytebuf_shift(in, 1); + return extract_event(event); + } + + // ASCII control key? + if ((uint16_t)in->buf[0] < TB_KEY_SPACE || in->buf[0] == TB_KEY_BACKSPACE2) + { + event->type = TB_EVENT_KEY; + event->ch = 0; + event->key = (uint16_t)in->buf[0]; + event->mod |= TB_MOD_CTRL; + bytebuf_shift(in, 1); + return TB_OK; + } + + // UTF-8? + if (in->len >= (size_t)tb_utf8_char_length(in->buf[0])) { + event->type = TB_EVENT_KEY; + tb_utf8_char_to_unicode(&event->ch, in->buf); + event->key = 0; + bytebuf_shift(in, tb_utf8_char_length(in->buf[0])); + return TB_OK; + } + + // Need more input + return TB_ERR; +} + +static int extract_esc(struct tb_event *event) { + int rv; + if_ok_or_need_more_return(rv, extract_esc_user(event, 0)); + if_ok_or_need_more_return(rv, extract_esc_cap(event)); + if_ok_or_need_more_return(rv, extract_esc_mouse(event)); + if_ok_or_need_more_return(rv, extract_esc_user(event, 1)); + return TB_ERR; +} + +static int extract_esc_user(struct tb_event *event, int is_post) { + int rv; + size_t consumed = 0; + struct bytebuf_t *in = &global.in; + int (*fn)(struct tb_event *, size_t *); + + fn = is_post ? global.fn_extract_esc_post : global.fn_extract_esc_pre; + + if (!fn) { + return TB_ERR; + } + + rv = fn(event, &consumed); + if (rv == TB_OK) { + bytebuf_shift(in, consumed); + } + + if_ok_or_need_more_return(rv, rv); + return TB_ERR; +} + +static int extract_esc_cap(struct tb_event *event) { + int rv; + struct bytebuf_t *in = &global.in; + struct cap_trie_t *node; + size_t depth; + + if_err_return(rv, cap_trie_find(in->buf, in->len, &node, &depth)); + if (node->is_leaf) { + // Found a leaf node + event->type = TB_EVENT_KEY; + event->ch = 0; + event->key = node->key; + event->mod = node->mod; + bytebuf_shift(in, depth); + return TB_OK; + } else if (node->nchildren > 0 && in->len <= depth) { + // Found a branch node (not enough input) + return TB_ERR_NEED_MORE; + } + + return TB_ERR; +} + +static int extract_esc_mouse(struct tb_event *event) { + struct bytebuf_t *in = &global.in; + + enum type { TYPE_VT200 = 0, TYPE_1006, TYPE_1015, TYPE_MAX }; + + char *cmp[TYPE_MAX] = {// + // X10 mouse encoding, the simplest one + // \x1b [ M Cb Cx Cy + [TYPE_VT200] = "\x1b[M", + // xterm 1006 extended mode or urxvt 1015 extended mode + // xterm: \x1b [ < Cb ; Cx ; Cy (M or m) + [TYPE_1006] = "\x1b[<", + // urxvt: \x1b [ Cb ; Cx ; Cy M + [TYPE_1015] = "\x1b["}; + + enum type type = 0; + int ret = TB_ERR; + + // Unrolled at compile-time (probably) + for (; type < TYPE_MAX; type++) { + size_t size = strlen(cmp[type]); + + if (in->len >= size && (strncmp(cmp[type], in->buf, size)) == 0) { + break; + } + } + + if (type == TYPE_MAX) { + ret = TB_ERR; // No match + return ret; + } + + size_t buf_shift = 0; + + switch (type) { + case TYPE_VT200: + if (in->len >= 6) { + int b = in->buf[3] - 0x20; + int fail = 0; + + switch (b & 3) { + case 0: + event->key = ((b & 64) != 0) ? TB_KEY_MOUSE_WHEEL_UP + : TB_KEY_MOUSE_LEFT; + break; + case 1: + event->key = ((b & 64) != 0) ? TB_KEY_MOUSE_WHEEL_DOWN + : TB_KEY_MOUSE_MIDDLE; + break; + case 2: + event->key = TB_KEY_MOUSE_RIGHT; + break; + case 3: + event->key = TB_KEY_MOUSE_RELEASE; + break; + default: + ret = TB_ERR; + fail = 1; + break; + } + + if (!fail) { + if ((b & 32) != 0) { + event->mod |= TB_MOD_MOTION; + } + + // the coord is 1,1 for upper left + event->x = ((uint8_t)in->buf[4]) - 0x21; + event->y = ((uint8_t)in->buf[5]) - 0x21; + + ret = TB_OK; + } + + buf_shift = 6; + } + break; + case TYPE_1006: + // fallthrough + case TYPE_1015: { + size_t index_fail = (size_t)-1; + + enum { + FIRST_M = 0, + FIRST_SEMICOLON, + LAST_SEMICOLON, + FIRST_LAST_MAX + }; + + size_t indices[FIRST_LAST_MAX] = {index_fail, index_fail, + index_fail}; + int m_is_capital = 0; + + for (size_t i = 0; i < in->len; i++) { + if (in->buf[i] == ';') { + if (indices[FIRST_SEMICOLON] == index_fail) { + indices[FIRST_SEMICOLON] = i; + } else { + indices[LAST_SEMICOLON] = i; + } + } else if (indices[FIRST_M] == index_fail) { + if (in->buf[i] == 'm' || in->buf[i] == 'M') { + m_is_capital = (in->buf[i] == 'M'); + indices[FIRST_M] = i; + } + } + } + + if (indices[FIRST_M] == index_fail || + indices[FIRST_SEMICOLON] == index_fail || + indices[LAST_SEMICOLON] == index_fail) + { + ret = TB_ERR; + } else { + int start = (type == TYPE_1015 ? 2 : 3); + + int n1 = strtoul(&in->buf[start], NULL, 10); + int n2 = + strtoul(&in->buf[indices[FIRST_SEMICOLON] + 1], NULL, 10); + int n3 = + strtoul(&in->buf[indices[LAST_SEMICOLON] + 1], NULL, 10); + + if (type == TYPE_1015) { + n1 -= 0x20; + } + + int fail = 0; + + switch (n1 & 3) { + case 0: + event->key = ((n1 & 64) != 0) ? TB_KEY_MOUSE_WHEEL_UP + : TB_KEY_MOUSE_LEFT; + break; + case 1: + event->key = ((n1 & 64) != 0) ? TB_KEY_MOUSE_WHEEL_DOWN + : TB_KEY_MOUSE_MIDDLE; + break; + case 2: + event->key = TB_KEY_MOUSE_RIGHT; + break; + case 3: + event->key = TB_KEY_MOUSE_RELEASE; + break; + default: + ret = TB_ERR; + fail = 1; + break; + } + + buf_shift = in->len; + + if (!fail) { + if (!m_is_capital) { + // on xterm mouse release is signaled by lowercase m + event->key = TB_KEY_MOUSE_RELEASE; + } + + if ((n1 & 32) != 0) { + event->mod |= TB_MOD_MOTION; + } + + event->x = ((uint8_t)n2) - 1; + event->y = ((uint8_t)n3) - 1; + + ret = TB_OK; + } + } + } break; + case TYPE_MAX: + ret = TB_ERR; + } + + if (buf_shift > 0) { + bytebuf_shift(in, buf_shift); + } + + if (ret == TB_OK) { + event->type = TB_EVENT_MOUSE; + } + + return ret; +} + +static int resize_if_needed() { + int rv; + if (!global.need_resize) { + return TB_OK; + } + if_err_return(rv, update_term_size()); // TODO is this needed? + if_err_return(rv, + cellbuf_resize(&global.back, global.width, global.height)); + if_err_return(rv, + cellbuf_resize(&global.front, global.width, global.height)); + if_err_return(rv, cellbuf_clear(&global.front)); + if_err_return(rv, send_clear()); + global.need_resize = 0; + return TB_OK; +} + +static void handle_resize(int sig) { + int errno_copy = errno; + write(global.resize_pipefd[1], &sig, sizeof(sig)); + errno = errno_copy; +} + +static int send_attr(uintattr_t fg, uintattr_t bg) { + int rv; + + if (fg == global.last_fg && bg == global.last_bg) { + return TB_OK; + } + + if_err_return(rv, bytebuf_puts(&global.out, global.caps[TB_CAP_SGR0])); + + uintattr_t cfg, cbg; + switch (global.output_mode) { + default: + case TB_OUTPUT_NORMAL: + cfg = fg & 0x0f; + cbg = bg & 0x0f; + break; + + case TB_OUTPUT_256: + cfg = fg & 0xff; + cbg = bg & 0xff; + break; + + case TB_OUTPUT_216: + cfg = fg & 0xff; + if (cfg > 215) + cfg = 7; + cbg = bg & 0xff; + if (cbg > 215) + cbg = 0; + cfg += 0x10; + cbg += 0x10; + break; + + case TB_OUTPUT_GRAYSCALE: + cfg = fg & 0xff; + if (cfg > 23) + cfg = 23; + cbg = bg & 0xff; + if (cbg > 23) + cbg = 0; + cfg += 0xe8; + cbg += 0xe8; + break; + + case TB_OUTPUT_TRUECOLOR: + cfg = fg; + cbg = bg; + break; + } + + if (global.output_mode != TB_OUTPUT_TRUECOLOR) { + if (fg & TB_BOLD) + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_BOLD])); + + if (bg & TB_BOLD) + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_BLINK])); + + if (fg & TB_UNDERLINE) + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_UNDERLINE])); + + if (fg & TB_ITALIC) + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_ITALIC])); + + if ((fg & TB_REVERSE) || (bg & TB_REVERSE)) + if_err_return(rv, + bytebuf_puts(&global.out, global.caps[TB_CAP_REVERSE])); + } + + if_err_return(rv, send_sgr(cfg, cbg)); + + global.last_fg = fg; + global.last_bg = bg; + + return TB_OK; +} + +static int send_sgr(uintattr_t fg, uintattr_t bg) { + int rv; + char nbuf[32]; + + if (global.output_mode != TB_OUTPUT_TRUECOLOR && fg == TB_DEFAULT && + bg == TB_DEFAULT) + { + return TB_OK; + } + + switch (global.output_mode) { + default: + case TB_OUTPUT_NORMAL: + send_literal(rv, "\x1b["); + if (fg != TB_DEFAULT) { + send_literal(rv, "3"); + send_num(rv, nbuf, fg - 1); + if (bg != TB_DEFAULT) { + send_literal(rv, ";"); + } + } + if (bg != TB_DEFAULT) { + send_literal(rv, "4"); + send_num(rv, nbuf, bg - 1); + } + send_literal(rv, "m"); + break; + + case TB_OUTPUT_256: + case TB_OUTPUT_216: + case TB_OUTPUT_GRAYSCALE: + send_literal(rv, "\x1b["); + if (fg != TB_DEFAULT) { + send_literal(rv, "38;5;"); + send_num(rv, nbuf, fg); + if (bg != TB_DEFAULT) { + send_literal(rv, ";"); + } + } + if (bg != TB_DEFAULT) { + send_literal(rv, "48;5;"); + send_num(rv, nbuf, bg); + } + send_literal(rv, "m"); + break; + + case TB_OUTPUT_TRUECOLOR: + send_literal(rv, "\x1b[38;2;"); + send_num(rv, nbuf, (fg >> 16) & 0xff); + send_literal(rv, ";"); + send_num(rv, nbuf, (fg >> 8) & 0xff); + send_literal(rv, ";"); + send_num(rv, nbuf, fg & 0xff); + send_literal(rv, ";48;2;"); + send_num(rv, nbuf, (bg >> 16) & 0xff); + send_literal(rv, ";"); + send_num(rv, nbuf, (bg >> 8) & 0xff); + send_literal(rv, ";"); + send_num(rv, nbuf, bg & 0xff); + send_literal(rv, "m"); + break; + } + return TB_OK; +} + +static int send_cursor_if(int x, int y) { + int rv; + char nbuf[32]; + if (x < 0 || y < 0) { + return TB_OK; + } + send_literal(rv, "\x1b["); + send_num(rv, nbuf, y + 1); + send_literal(rv, ";"); + send_num(rv, nbuf, x + 1); + send_literal(rv, "H"); + return TB_OK; +} + +static int send_char(int x, int y, uint32_t ch) { + return send_cluster(x, y, &ch, 1); +} + +static int send_cluster(int x, int y, uint32_t *ch, size_t nch) { + int rv; + char abuf[8]; + + if (global.last_x != x - 1 || global.last_y != y) { + if_err_return(rv, send_cursor_if(x, y)); + } + global.last_x = x; + global.last_y = y; + + int i; + for (i = 0; i < (int)nch; i++) { + uint32_t ach = *(ch + i); + int aw = tb_utf8_unicode_to_char(abuf, ach); + if (!ach) { + abuf[0] = ' '; + } + if_err_return(rv, bytebuf_nputs(&global.out, abuf, (size_t)aw)); + } + + return TB_OK; +} + +static int convert_num(uint32_t num, char *buf) { + int i, l = 0; + int ch; + do { + buf[l++] = '0' + (num % 10); + num /= 10; + } while (num); + for (i = 0; i < l / 2; i++) { + ch = buf[i]; + buf[i] = buf[l - 1 - i]; + buf[l - 1 - i] = ch; + } + return l; +} + +static int cell_cmp(struct tb_cell *a, struct tb_cell *b) { + if (a->ch != b->ch || a->fg != b->fg || a->bg != b->bg) { + return 1; + } +#ifdef TB_OPT_EGC + if (a->nech != b->nech) { + return 1; + } else if (a->nech > 0) { // a->nech == b->nech + return memcmp(a->ech, b->ech, a->nech); + } +#endif + return 0; +} + +static int cell_copy(struct tb_cell *dst, struct tb_cell *src) { +#ifdef TB_OPT_EGC + if (src->nech > 0) { + return cell_set(dst, src->ech, src->nech, src->fg, src->bg); + } +#endif + return cell_set(dst, &src->ch, 1, src->fg, src->bg); +} + +static int cell_set(struct tb_cell *cell, uint32_t *ch, size_t nch, + uintattr_t fg, uintattr_t bg) { + cell->ch = ch ? *ch : 0; + cell->fg = fg; + cell->bg = bg; +#ifdef TB_OPT_EGC + if (nch <= 1) { + cell->nech = 0; + } else { + cell_reserve_ech(cell, nch + 1); + memcpy(cell->ech, ch, nch); + cell->ech[nch] = '\0'; + cell->nech = nch; + } +#else + (void)nch; + (void)cell_reserve_ech; +#endif + return TB_OK; +} + +static int cell_reserve_ech(struct tb_cell *cell, size_t n) { +#ifdef TB_OPT_EGC + if (cell->cech >= n) { + return TB_OK; + } + if (!(cell->ech = tb_realloc(cell->ech, n * sizeof(cell->ch)))) { + return TB_ERR_MEM; + } + cell->cech = n; + return TB_OK; +#else + (void)cell; + (void)n; + return TB_ERR; +#endif +} + +static int cell_free(struct tb_cell *cell) { +#ifdef TB_OPT_EGC + if (cell->ech) { + tb_free(cell->ech); + } +#endif + memset(cell, 0, sizeof(*cell)); + return TB_OK; +} + +static int cellbuf_init(struct cellbuf_t *c, int w, int h) { + c->cells = tb_malloc(sizeof(struct tb_cell) * w * h); + if (!c->cells) { + return TB_ERR_MEM; + } + memset(c->cells, 0, sizeof(struct tb_cell) * w * h); + c->width = w; + c->height = h; + return TB_OK; +} + +static int cellbuf_free(struct cellbuf_t *c) { + if (c->cells) { + int i; + for (i = 0; i < c->width * c->height; i++) { + cell_free(&c->cells[i]); + } + tb_free(c->cells); + } + memset(c, 0, sizeof(*c)); + return TB_OK; +} + +static int cellbuf_clear(struct cellbuf_t *c) { + int i; + uint32_t space = (uint32_t)' '; + for (i = 0; i < c->width * c->height; i++) { + cell_set(&c->cells[i], &space, 1, global.fg, global.bg); + } + return TB_OK; +} + +static int cellbuf_get(struct cellbuf_t *c, int x, int y, + struct tb_cell **out) { + if (x < 0 || x >= c->width || y < 0 || y >= c->height) { + *out = NULL; + return TB_ERR_OUT_OF_BOUNDS; + } + *out = &c->cells[(y * c->width) + x]; + return TB_OK; +} + +static int cellbuf_resize(struct cellbuf_t *c, int w, int h) { + int rv; + + int ow = c->width; + int oh = c->height; + + if (ow == w && oh == h) { + return TB_OK; + } + + w = w < 1 ? 1 : w; + h = h < 1 ? 1 : h; + + int minw = (w < ow) ? w : ow; + int minh = (h < oh) ? h : oh; + + struct tb_cell *prev = c->cells; + + if_err_return(rv, cellbuf_init(c, w, h)); + if_err_return(rv, cellbuf_clear(c)); + + int x, y; + for (x = 0; x < minw; x++) { + for (y = 0; y < minh; y++) { + struct tb_cell *src, *dst; + src = &prev[(y * ow) + x]; + if_err_return(rv, cellbuf_get(c, x, y, &dst)); + if_err_return(rv, cell_copy(dst, src)); + } + } + + tb_free(prev); + + return TB_OK; +} + +static int bytebuf_puts(struct bytebuf_t *b, const char *str) { + return bytebuf_nputs(b, str, (size_t)strlen(str)); +} + +static int bytebuf_nputs(struct bytebuf_t *b, const char *str, size_t nstr) { + int rv; + if_err_return(rv, bytebuf_reserve(b, b->len + nstr + 1)); + memcpy(b->buf + b->len, str, nstr); + b->len += nstr; + b->buf[b->len] = '\0'; + return TB_OK; +} + +static int bytebuf_shift(struct bytebuf_t *b, size_t n) { + if (n > b->len) { + n = b->len; + } + size_t nmove = b->len - n; + memmove(b->buf, b->buf + n, nmove); + b->len -= n; + return TB_OK; +} + +static int bytebuf_flush(struct bytebuf_t *b, int fd) { + if (b->len <= 0) { + return TB_OK; + } + ssize_t write_rv = write(fd, b->buf, b->len); + if (write_rv < 0 || (size_t)write_rv != b->len) { + // Note, errno will be 0 on partial write + global.last_errno = errno; + return TB_ERR; + } + b->len = 0; + return TB_OK; +} + +static int bytebuf_reserve(struct bytebuf_t *b, size_t sz) { + if (b->cap >= sz) { + return TB_OK; + } + size_t newcap = b->cap > 0 ? b->cap : 1; + while (newcap < sz) { + newcap *= 2; + } + char *newbuf; + if (b->buf) { + newbuf = tb_realloc(b->buf, newcap); + } else { + newbuf = tb_malloc(newcap); + } + if (!newbuf) { + return TB_ERR_MEM; + } + b->buf = newbuf; + b->cap = newcap; + return TB_OK; +} + +static int bytebuf_free(struct bytebuf_t *b) { + if (b->buf) { + tb_free(b->buf); + } + memset(b, 0, sizeof(*b)); + return TB_OK; +} + +#endif /* TB_IMPL */ diff --git a/utf8.c b/utf8.c @@ -1,79 +0,0 @@ -#include "termbox.h" - -static const unsigned char utf8_length[256] = { - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, - 2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, - 3,3,3,3,3,3,3,3,3,3,3,3,3,3,3,3,4,4,4,4,4,4,4,4,5,5,5,5,6,6,1,1 -}; - -static const unsigned char utf8_mask[6] = { - 0x7F, - 0x1F, - 0x0F, - 0x07, - 0x03, - 0x01 -}; - -int tb_utf8_char_length(char c) -{ - return utf8_length[(unsigned char)c]; -} - -int tb_utf8_char_to_unicode(uint32_t *out, const char *c) -{ - if (*c == 0) - return TB_EOF; - - int i; - unsigned char len = tb_utf8_char_length(*c); - unsigned char mask = utf8_mask[len-1]; - uint32_t result = c[0] & mask; - for (i = 1; i < len; ++i) { - result <<= 6; - result |= c[i] & 0x3f; - } - - *out = result; - return (int)len; -} - -int tb_utf8_unicode_to_char(char *out, uint32_t c) -{ - int len = 0; - int first; - int i; - - if (c < 0x80) { - first = 0; - len = 1; - } else if (c < 0x800) { - first = 0xc0; - len = 2; - } else if (c < 0x10000) { - first = 0xe0; - len = 3; - } else if (c < 0x200000) { - first = 0xf0; - len = 4; - } else if (c < 0x4000000) { - first = 0xf8; - len = 5; - } else { - first = 0xfc; - len = 6; - } - - for (i = len - 1; i > 0; --i) { - out[i] = (c & 0x3f) | 0x80; - c >>= 6; - } - out[0] = c | first; - - return len; -} diff --git a/util.c b/util.c @@ -18,13 +18,13 @@ u8_string_to_unicode(uint32_t* us, const char* s, const size_t max) int u8_len = 0; size_t added = 0; - while (s && *s && u8_len != TB_EOF) + while (s && *s && u8_len != TB_ERR) { u8_len = tb_utf8_char_to_unicode(&uch, s); if (added + 1 > max) break; *pus++ = uch; - if (u8_len != TB_EOF) + if (u8_len != TB_ERR) { s += u8_len; added++;