diff options
Diffstat (limited to 'src/nvim/ex_getln.c')
-rw-r--r-- | src/nvim/ex_getln.c | 27 |
1 files changed, 13 insertions, 14 deletions
diff --git a/src/nvim/ex_getln.c b/src/nvim/ex_getln.c index c0c8a5605b..5018c9268b 100644 --- a/src/nvim/ex_getln.c +++ b/src/nvim/ex_getln.c @@ -29,6 +29,7 @@ #include "nvim/edit.h" #include "nvim/eval.h" #include "nvim/eval/typval.h" +#include "nvim/eval/vars.h" #include "nvim/ex_cmds.h" #include "nvim/ex_cmds_defs.h" #include "nvim/ex_docmd.h" @@ -4561,39 +4562,37 @@ bool is_in_cmdwin(void) char *script_get(exarg_T *const eap, size_t *const lenp) FUNC_ATTR_NONNULL_ALL FUNC_ATTR_WARN_UNUSED_RESULT FUNC_ATTR_MALLOC { - const char *const cmd = eap->arg; + char *cmd = eap->arg; if (cmd[0] != '<' || cmd[1] != '<' || eap->getline == NULL) { *lenp = strlen(eap->arg); return eap->skip ? NULL : xmemdupz(eap->arg, *lenp); } + cmd += 2; garray_T ga = { .ga_data = NULL, .ga_len = 0 }; + + list_T *const l = heredoc_get(eap, cmd, true); + if (l == NULL) { + return NULL; + } + if (!eap->skip) { ga_init(&ga, 1, 0x400); } - const char *const end_pattern = (cmd[2] != NUL ? skipwhite(cmd + 2) : "."); - while (true) { - char *const theline = eap->getline(eap->cstack->cs_looplevel > 0 ? -1 : NUL, eap->cookie, 0, - true); - - if (theline == NULL || strcmp(end_pattern, theline) == 0) { - xfree(theline); - break; - } - + TV_LIST_ITER_CONST(l, li, { if (!eap->skip) { - ga_concat(&ga, theline); + ga_concat(&ga, tv_get_string(TV_LIST_ITEM_TV(li))); ga_append(&ga, '\n'); } - xfree(theline); - } + }); *lenp = (size_t)ga.ga_len; // Set length without trailing NUL. if (!eap->skip) { ga_append(&ga, NUL); } + tv_list_free(l); return (char *)ga.ga_data; } |