diff options
Diffstat (limited to 'util/sbase/paste.c')
-rw-r--r-- | util/sbase/paste.c | 144 |
1 files changed, 144 insertions, 0 deletions
diff --git a/util/sbase/paste.c b/util/sbase/paste.c new file mode 100644 index 00000000..4fa9fc5a --- /dev/null +++ b/util/sbase/paste.c @@ -0,0 +1,144 @@ +/* See LICENSE file for copyright and license details. */ +#include <stdlib.h> +#include <string.h> + +#include "utf.h" +#include "util.h" + +struct fdescr { + FILE *fp; + const char *name; +}; + +static void +sequential(struct fdescr *dsc, int fdescrlen, Rune *delim, size_t delimlen) +{ + Rune c, last; + size_t i, d; + + for (i = 0; i < fdescrlen; i++) { + d = 0; + last = 0; + + while (efgetrune(&c, dsc[i].fp, dsc[i].name)) { + if (last == '\n') { + if (delim[d] != '\0') + efputrune(&delim[d], stdout, "<stdout>"); + d = (d + 1) % delimlen; + } + + if (c != '\n') + efputrune(&c, stdout, "<stdout>"); + last = c; + } + + if (last == '\n') + efputrune(&last, stdout, "<stdout>"); + } +} + +static void +parallel(struct fdescr *dsc, int fdescrlen, Rune *delim, size_t delimlen) +{ + Rune c, d; + size_t i, m; + ssize_t last; + +nextline: + last = -1; + + for (i = 0; i < fdescrlen; i++) { + d = delim[i % delimlen]; + c = 0; + + while (efgetrune(&c, dsc[i].fp, dsc[i].name)) { + for (m = last + 1; m < i; m++) { + if (delim[m % delimlen] != '\0') + efputrune(&delim[m % delimlen], stdout, "<stdout>"); + } + last = i; + if (c == '\n') { + if (i != fdescrlen - 1) + c = d; + efputrune(&c, stdout, "<stdout>"); + break; + } + efputrune(&c, stdout, "<stdout>"); + } + + if (c == 0 && last != -1) { + if (i == fdescrlen - 1) + putchar('\n'); + else if (d != '\0') + efputrune(&d, stdout, "<stdout>"); + last++; + } + } + if (last != -1) + goto nextline; +} + +static void +usage(void) +{ + eprintf("usage: %s [-s] [-d list] file ...\n", argv0); +} + +int +main(int argc, char *argv[]) +{ + struct fdescr *dsc; + Rune *delim_rune = NULL; + size_t delim_runelen, i, delim_bytelen = 1; + int seq = 0, ret = 0; + char *delim = "\t"; + + ARGBEGIN { + case 's': + seq = 1; + break; + case 'd': + delim = EARGF(usage()); + delim_bytelen = unescape(delim); + break; + default: + usage(); + } ARGEND + + if (!argc) + usage(); + + /* populate delimiters */ + delim_rune = ereallocarray(NULL, + utfmemlen(delim, delim_bytelen) + 1, sizeof(*delim_rune)); + if (!(delim_runelen = utfntorunestr(delim, delim_bytelen, delim_rune))) { + usage(); + } + + /* populate file list */ + dsc = ereallocarray(NULL, argc, sizeof(*dsc)); + + for (i = 0; i < argc; i++) { + if (!strcmp(argv[i], "-")) { + argv[i] = "<stdin>"; + dsc[i].fp = stdin; + } else if (!(dsc[i].fp = fopen(argv[i], "r"))) { + eprintf("fopen %s:", argv[i]); + } + dsc[i].name = argv[i]; + } + + if (seq) { + sequential(dsc, argc, delim_rune, delim_runelen); + } else { + parallel(dsc, argc, delim_rune, delim_runelen); + } + + for (i = 0; i < argc; i++) + if (dsc[i].fp != stdin && fshut(dsc[i].fp, argv[i])) + ret |= fshut(dsc[i].fp, argv[i]); + + ret |= fshut(stdin, "<stdin>") | fshut(stdout, "<stdout>"); + + return ret; +} |