| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | #include <stddef.h> | ||
| 2 | #include <string.h> | ||
| 3 | #include <unistd.h> | ||
| 4 | #include "vars.h" | ||
| 5 | #include "editor.h" | ||
| 6 | #include "selection.h" | ||
| 7 | #include "util/bsearch.h" | ||
| 8 | #include "util/numtostr.h" | ||
| 9 | #include "util/path.h" | ||
| 10 | #include "util/str-util.h" | ||
| 11 | #include "util/xmalloc.h" | ||
| 12 | #include "view.h" | ||
| 13 | |||
| 14 | // TODO: Make expand() functions return a StringVariant type and | ||
| 15 | // only do dynamic allocations when necessary | ||
| 16 | typedef struct { | ||
| 17 | char name[12]; | ||
| 18 | char *(*expand)(const EditorState *e); | ||
| 19 | } BuiltinVar; | ||
| 20 | |||
| 21 | 2 | static char *expand_dte_home(const EditorState *e) | |
| 22 | { | ||
| 23 | 2 | return xstrdup(e->user_config_dir); | |
| 24 | } | ||
| 25 | |||
| 26 | 1 | static char *expand_file(const EditorState *e) | |
| 27 | { | ||
| 28 |
1/4✗ Branch 2 → 3 not taken.
✓ Branch 2 → 5 taken 1 time.
✗ Branch 3 → 4 not taken.
✗ Branch 3 → 5 not taken.
|
1 | if (!e->buffer || !e->buffer->abs_filename) { |
| 29 | return NULL; | ||
| 30 | } | ||
| 31 | ✗ | return xstrdup(e->buffer->abs_filename); | |
| 32 | } | ||
| 33 | |||
| 34 | 1 | static char *expand_file_dir(const EditorState *e) | |
| 35 | { | ||
| 36 |
1/4✗ Branch 2 → 3 not taken.
✓ Branch 2 → 5 taken 1 time.
✗ Branch 3 → 4 not taken.
✗ Branch 3 → 5 not taken.
|
1 | if (!e->buffer || !e->buffer->abs_filename) { |
| 37 | return NULL; | ||
| 38 | } | ||
| 39 | ✗ | return path_dirname(e->buffer->abs_filename); | |
| 40 | } | ||
| 41 | |||
| 42 | 2 | static char *expand_rfile(const EditorState *e) | |
| 43 | { | ||
| 44 |
1/4✗ Branch 2 → 3 not taken.
✓ Branch 2 → 8 taken 2 times.
✗ Branch 3 → 4 not taken.
✗ Branch 3 → 8 not taken.
|
2 | if (!e->buffer || !e->buffer->abs_filename) { |
| 45 | return NULL; | ||
| 46 | } | ||
| 47 | ✗ | char buf[8192]; | |
| 48 | ✗ | const char *cwd = getcwd(buf, sizeof buf); | |
| 49 | ✗ | const char *abs = e->buffer->abs_filename; | |
| 50 | ✗ | return likely(cwd) ? path_relative(abs, cwd) : xstrdup(abs); | |
| 51 | } | ||
| 52 | |||
| 53 | 1 | static char *expand_rfiledir(const EditorState *e) | |
| 54 | { | ||
| 55 | 1 | char *rfile = expand_rfile(e); | |
| 56 |
1/2✗ Branch 3 → 4 not taken.
✓ Branch 3 → 7 taken 1 time.
|
1 | if (!rfile) { |
| 57 | return NULL; | ||
| 58 | } | ||
| 59 | |||
| 60 | ✗ | StringView dir = path_slice_dirname(rfile); | |
| 61 | ✗ | if (rfile != (char*)dir.data) { | |
| 62 | // rfile contains no directory part (i.e. no slashes), so there's | ||
| 63 | // nothing to "slice" and we simply return "." instead | ||
| 64 | ✗ | free(rfile); | |
| 65 | ✗ | return xstrdup("."); | |
| 66 | } | ||
| 67 | |||
| 68 | // Overwrite the last slash in rfile with a null terminator, so as | ||
| 69 | // to remove the last path component (the filename) | ||
| 70 | ✗ | rfile[dir.length] = '\0'; | |
| 71 | ✗ | return rfile; | |
| 72 | } | ||
| 73 | |||
| 74 | 1 | static char *expand_filetype(const EditorState *e) | |
| 75 | { | ||
| 76 |
1/2✗ Branch 2 → 3 not taken.
✓ Branch 2 → 4 taken 1 time.
|
1 | return e->buffer ? xstrdup(e->buffer->options.filetype) : NULL; |
| 77 | } | ||
| 78 | |||
| 79 | 1 | static char *expand_colno(const EditorState *e) | |
| 80 | { | ||
| 81 |
1/2✗ Branch 2 → 3 not taken.
✓ Branch 2 → 5 taken 1 time.
|
1 | return e->view ? xstrdup(umax_to_str(e->view->cx_display + 1)) : NULL; |
| 82 | } | ||
| 83 | |||
| 84 | 1 | static char *expand_lineno(const EditorState *e) | |
| 85 | { | ||
| 86 |
1/2✗ Branch 2 → 3 not taken.
✓ Branch 2 → 5 taken 1 time.
|
1 | return e->view ? xstrdup(umax_to_str(e->view->cy + 1)) : NULL; |
| 87 | } | ||
| 88 | |||
| 89 | 4 | static char *msgpos(const EditorState *e, size_t idx) | |
| 90 | { | ||
| 91 | 4 | return xstrdup(umax_to_str(e->messages[idx].pos + 1)); | |
| 92 | } | ||
| 93 | |||
| 94 | 2 | static char *expand_msgpos_a(const EditorState *e) {return msgpos(e, 0);} | |
| 95 | 1 | static char *expand_msgpos_b(const EditorState *e) {return msgpos(e, 1);} | |
| 96 | 1 | static char *expand_msgpos_c(const EditorState *e) {return msgpos(e, 2);} | |
| 97 | |||
| 98 | 4 | static char *expand_word(const EditorState *e) | |
| 99 | { | ||
| 100 | 4 | const View *view = e->view; | |
| 101 |
2/2✓ Branch 2 → 3 taken 3 times.
✓ Branch 2 → 14 taken 1 time.
|
4 | if (unlikely(!view)) { |
| 102 | return NULL; | ||
| 103 | } | ||
| 104 | |||
| 105 |
1/2✗ Branch 3 → 4 not taken.
✓ Branch 3 → 11 taken 3 times.
|
3 | if (view->selection) { |
| 106 | ✗ | SelectionInfo info = init_selection(view); | |
| 107 | ✗ | size_t len = info.eo - info.so; | |
| 108 | ✗ | if (unlikely(len == 0)) { | |
| 109 | return NULL; | ||
| 110 | } | ||
| 111 | ✗ | char *selection = block_iter_get_bytes(&info.si, len); | |
| 112 | ✗ | BUG_ON(!selection); | |
| 113 | ✗ | selection[len] = '\0'; // See comment in block_iter_get_bytes() | |
| 114 | ✗ | return selection; | |
| 115 | } | ||
| 116 | |||
| 117 | 3 | StringView word = view_get_word_under_cursor(view); | |
| 118 |
1/2✓ Branch 12 → 13 taken 3 times.
✗ Branch 12 → 14 not taken.
|
3 | return word.length ? xstrcut(word.data, word.length) : NULL; |
| 119 | } | ||
| 120 | |||
| 121 | static const BuiltinVar normal_vars[] = { | ||
| 122 | {"COLNO", expand_colno}, | ||
| 123 | {"DTE_HOME", expand_dte_home}, | ||
| 124 | {"FILE", expand_file}, | ||
| 125 | {"FILEDIR", expand_file_dir}, | ||
| 126 | {"FILETYPE", expand_filetype}, | ||
| 127 | {"LINENO", expand_lineno}, | ||
| 128 | {"MSGPOS", expand_msgpos_a}, | ||
| 129 | {"MSGPOS_A", expand_msgpos_a}, | ||
| 130 | {"MSGPOS_B", expand_msgpos_b}, | ||
| 131 | {"MSGPOS_C", expand_msgpos_c}, | ||
| 132 | {"RFILE", expand_rfile}, | ||
| 133 | {"RFILEDIR", expand_rfiledir}, | ||
| 134 | {"WORD", expand_word}, | ||
| 135 | }; | ||
| 136 | |||
| 137 | 24 | UNITTEST { | |
| 138 | 24 | CHECK_BSEARCH_ARRAY(normal_vars, name); | |
| 139 | 24 | } | |
| 140 | |||
| 141 | // Handles variables like e.g. ${builtin:color/reset} and ${script:file.sh} | ||
| 142 | ✗ | static char *expand_prefixed_var(const char *name, size_t name_len, StringView prefix) | |
| 143 | { | ||
| 144 | ✗ | char buf[64]; | |
| 145 | ✗ | bool name_fits_buf = (name_len < sizeof(buf)); | |
| 146 | |||
| 147 | ✗ | if (strview_equal_cstring(prefix, "builtin")) { | |
| 148 | // Skip past "builtin:" | ||
| 149 | ✗ | name += prefix.length + 1; | |
| 150 | ✗ | } else if (strview_equal_cstring(prefix, "script") && name_fits_buf) { | |
| 151 | // Copy `name` into `buf` and replace ':' with '/' | ||
| 152 | ✗ | memcpy(buf, name, name_len + 1); | |
| 153 | ✗ | buf[prefix.length] = '/'; | |
| 154 | ✗ | name = buf; | |
| 155 | } else { | ||
| 156 | return NULL; | ||
| 157 | } | ||
| 158 | |||
| 159 | ✗ | const BuiltinConfig *cfg = get_builtin_config(name); | |
| 160 | ✗ | return cfg ? xmemdup(cfg->text.data, cfg->text.length + 1) : NULL; | |
| 161 | } | ||
| 162 | |||
| 163 | 25 | char *expand_normal_var(const EditorState *e, const char *name) | |
| 164 | { | ||
| 165 | 25 | size_t name_len = strlen(name); | |
| 166 |
1/2✓ Branch 2 → 3 taken 25 times.
✗ Branch 2 → 12 not taken.
|
25 | if (unlikely(name_len == 0)) { |
| 167 | return NULL; | ||
| 168 | } | ||
| 169 | |||
| 170 | 25 | size_t pos = 0; | |
| 171 | 25 | StringView prefix = get_delim(name, &pos, name_len, ':'); | |
| 172 |
1/2✗ Branch 4 → 5 not taken.
✓ Branch 4 → 6 taken 25 times.
|
25 | if (prefix.length < name_len) { |
| 173 | ✗ | return expand_prefixed_var(name, name_len, prefix); | |
| 174 | } | ||
| 175 | |||
| 176 | 25 | const BuiltinVar *var = BSEARCH(name, normal_vars, vstrcmp); | |
| 177 |
2/2✓ Branch 7 → 8 taken 17 times.
✓ Branch 7 → 9 taken 8 times.
|
25 | if (var) { |
| 178 | 17 | return var->expand(e); | |
| 179 | } | ||
| 180 | |||
| 181 | 8 | const char *str = xgetenv(name); | |
| 182 |
2/2✓ Branch 10 → 11 taken 7 times.
✓ Branch 10 → 12 taken 1 time.
|
8 | return str ? xstrdup(str) : NULL; |
| 183 | } | ||
| 184 | |||
| 185 | 3 | void collect_normal_vars ( | |
| 186 | PointerArray *a, | ||
| 187 | StringView prefix, // Prefix to match against | ||
| 188 | const char *suffix // Suffix to append to collected strings | ||
| 189 | ) { | ||
| 190 | 3 | size_t suffix_len = strlen(suffix) + 1; | |
| 191 |
2/2✓ Branch 8 → 3 taken 39 times.
✓ Branch 8 → 9 taken 3 times.
|
42 | for (size_t i = 0; i < ARRAYLEN(normal_vars); i++) { |
| 192 | 39 | StringView var = strview(normal_vars[i].name); | |
| 193 |
1/2✗ Branch 4 → 5 not taken.
✓ Branch 4 → 7 taken 39 times.
|
39 | if (strview_has_sv_prefix(var, prefix)) { |
| 194 | ✗ | ptr_array_append(a, xmemjoin(var.data, var.length, suffix, suffix_len)); | |
| 195 | } | ||
| 196 | } | ||
| 197 | 3 | } | |
| 198 |