diff options
Diffstat (limited to 'vis-registers.c')
| -rw-r--r-- | vis-registers.c | 249 |
1 files changed, 249 insertions, 0 deletions
diff --git a/vis-registers.c b/vis-registers.c new file mode 100644 index 0000000..f0ebb15 --- /dev/null +++ b/vis-registers.c @@ -0,0 +1,249 @@ +#include <stdlib.h> +#include <string.h> + +#include "vis-core.h" + +static Buffer *register_buffer(Register *reg, size_t slot) { + Buffer *buf = array_get(®->values, slot); + if (buf) + return buf; + if (array_resize(®->values, slot) && (buf = array_get(®->values, slot))) + return buf; + Buffer new; + buffer_init(&new); + if (!array_add(®->values, &new)) + return NULL; + size_t capacity = array_capacity(®->values); + for (size_t i = array_length(®->values); i < capacity; i++) { + if (!array_add(®->values, &new)) + return NULL; + } + return array_get(®->values, slot); +} + +static ssize_t read_buffer(void *context, char *data, size_t len) { + buffer_append(context, data, len); + return len; +} + +bool register_init(Register *reg) { + Buffer buf; + buffer_init(&buf); + array_init_sized(®->values, sizeof(Buffer)); + return array_add(®->values, &buf); +} + +void register_release(Register *reg) { + if (!reg) + return; + size_t n = array_capacity(®->values); + for (size_t i = 0; i < n; i++) + buffer_release(array_get(®->values, i)); + array_release(®->values); +} + +const char *register_slot_get(Vis *vis, Register *reg, size_t slot, size_t *len) { + if (len) + *len = 0; + switch (reg->type) { + case REGISTER_NORMAL: + { + Buffer *buf = array_get(®->values, slot); + if (!buf) + return NULL; + buffer_terminate(buf); + if (len) + *len = buffer_length0(buf); + return buffer_content0(buf); + } + case REGISTER_CLIPBOARD: + { + Buffer buferr; + buffer_init(&buferr); + Buffer *buf = array_get(®->values, slot); + if (!buf) + return NULL; + buffer_clear(buf); + + int status = vis_pipe(vis, vis->win->file, + &(Filerange){ .start = 0, .end = 0 }, + (const char*[]){ VIS_CLIPBOARD, "--paste", NULL }, + buf, read_buffer, &buferr, read_buffer); + + if (status != 0) + vis_info_show(vis, "Command failed %s", buffer_content0(&buferr)); + buffer_release(&buferr); + if (len) + *len = buffer_length0(buf); + return buffer_content0(buf); + } + case REGISTER_BLACKHOLE: + default: + return NULL; + } +} + +const char *register_get(Vis *vis, Register *reg, size_t *len) { + return register_slot_get(vis, reg, 0, len); +} + +bool register_slot_put(Vis *vis, Register *reg, size_t slot, const char *data, size_t len) { + if (reg->type != REGISTER_NORMAL) + return false; + Buffer *buf = register_buffer(reg, slot); + return buf && buffer_put(buf, data, len); +} + +bool register_put(Vis *vis, Register *reg, const char *data, size_t len) { + return register_slot_put(vis, reg, 0, data, len) && + register_resize(reg, 1); +} + +bool register_put0(Vis *vis, Register *reg, const char *data) { + return register_put(vis, reg, data, strlen(data)+1); +} + +static bool register_slot_append_range(Register *reg, size_t slot, Text *txt, Filerange *range) { + switch (reg->type) { + case REGISTER_NORMAL: + { + Buffer *buf = register_buffer(reg, slot); + if (!buf) + return false; + size_t len = text_range_size(range); + if (len == SIZE_MAX || !buffer_grow(buf, len+1)) + return false; + if (buf->len > 0 && buf->data[buf->len-1] == '\0') + buf->len--; + buf->len += text_bytes_get(txt, range->start, len, buf->data + buf->len); + return buffer_append(buf, "\0", 1); + } + default: + return false; + } +} + +bool register_slot_put_range(Vis *vis, Register *reg, size_t slot, Text *txt, Filerange *range) { + if (reg->append) + return register_slot_append_range(reg, slot, txt, range); + + switch (reg->type) { + case REGISTER_NORMAL: + { + Buffer *buf = register_buffer(reg, slot); + if (!buf) + return false; + size_t len = text_range_size(range); + if (len == SIZE_MAX || !buffer_reserve(buf, len+1)) + return false; + buf->len = text_bytes_get(txt, range->start, len, buf->data); + return buffer_append(buf, "\0", 1); + } + case REGISTER_CLIPBOARD: + { + Buffer buferr; + buffer_init(&buferr); + + int status = vis_pipe(vis, vis->win->file, range, + (const char*[]){ VIS_CLIPBOARD, "--copy", NULL }, + NULL, NULL, &buferr, read_buffer); + + if (status != 0) + vis_info_show(vis, "Command failed %s", buffer_content0(&buferr)); + buffer_release(&buferr); + return status == 0; + } + case REGISTER_BLACKHOLE: + return true; + default: + return false; + } +} + +bool register_put_range(Vis *vis, Register *reg, Text *txt, Filerange *range) { + return register_slot_put_range(vis, reg, 0, txt, range) && + register_resize(reg, 1); +} + +size_t register_count(Register *reg) { + return array_length(®->values); +} + +bool register_resize(Register *reg, size_t count) { + return array_truncate(®->values, count); +} + +enum VisRegister vis_register_from(Vis *vis, char reg) { + switch (reg) { + case '+': return VIS_REG_CLIPBOARD; + case '@': return VIS_MACRO_LAST_RECORDED; + } + + if ('a' <= reg && reg <= 'z') + return VIS_REG_a + reg - 'a'; + if ('A' <= reg && reg <= 'Z') + return VIS_REG_A + reg - 'A'; + for (size_t i = 0; i < LENGTH(vis_registers); i++) { + if (vis_registers[i].name == reg) + return i; + } + return VIS_REG_INVALID; +} + +void vis_register(Vis *vis, enum VisRegister reg) { + if (VIS_REG_A <= reg && reg <= VIS_REG_Z) { + vis->action.reg = &vis->registers[VIS_REG_a + reg - VIS_REG_A]; + vis->action.reg->append = true; + } else if (reg < LENGTH(vis->registers)) { + vis->action.reg = &vis->registers[reg]; + vis->action.reg->append = false; + } +} + +static Register *register_from(Vis *vis, enum VisRegister id) { + if (VIS_REG_A <= id && id <= VIS_REG_Z) + id = VIS_REG_a + id - VIS_REG_A; + if (id < LENGTH(vis->registers)) + return &vis->registers[id]; + return NULL; +} + +bool vis_register_put(Vis *vis, enum VisRegister id, const char *data, size_t len) { + Register *reg = register_from(vis, id); + if (!reg) + return false; + return register_put(vis, reg, data, len); +} + +const char *vis_register_get(Vis *vis, enum VisRegister id, size_t *len) { + return vis_register_slot_get(vis, id, 0, len); +} + +const char *vis_register_slot_get(Vis *vis, enum VisRegister id, size_t slot, size_t *len) { + Register *reg = register_from(vis, id); + if (reg) + return register_slot_get(vis, reg, slot, len); + *len = 0; + return NULL; +} + +const RegisterDef vis_registers[] = { + [VIS_REG_DEFAULT] = { '"', VIS_HELP("Unnamed register") }, + [VIS_REG_ZERO] = { '0', VIS_HELP("Yank register") }, + [VIS_REG_1] = { '1', VIS_HELP("1st sub-expression match") }, + [VIS_REG_2] = { '2', VIS_HELP("2nd sub-expression match") }, + [VIS_REG_3] = { '3', VIS_HELP("3rd sub-expression match") }, + [VIS_REG_4] = { '4', VIS_HELP("4th sub-expression match") }, + [VIS_REG_5] = { '5', VIS_HELP("5th sub-expression match") }, + [VIS_REG_6] = { '6', VIS_HELP("6th sub-expression match") }, + [VIS_REG_7] = { '7', VIS_HELP("7th sub-expression match") }, + [VIS_REG_8] = { '8', VIS_HELP("8th sub-expression match") }, + [VIS_REG_9] = { '9', VIS_HELP("9th sub-expression match") }, + [VIS_REG_AMPERSAND] = { '&', VIS_HELP("Last regex match") }, + [VIS_REG_BLACKHOLE] = { '_', VIS_HELP("/dev/null register") }, + [VIS_REG_CLIPBOARD] = { '*', VIS_HELP("System clipboard register, see vis-clipboard(1)") }, + [VIS_REG_DOT] = { '.', VIS_HELP("Last inserted text") }, + [VIS_REG_SEARCH] = { '/', VIS_HELP("Last search pattern") }, + [VIS_REG_COMMAND] = { ':', VIS_HELP("Last :-command") }, + [VIS_REG_SHELL] = { '!', VIS_HELP("Last shell command given to either <, >, |, or !") }, +}; |