1/* See LICENSE file for copyright and license details. */
  2
  3#include <stdlib.h>
  4#include <string.h>
  5
  6#include "utf.h"
  7#include "util.h"
  8
  9struct fdescr {
 10  FILE       *fp;
 11  const char *name;
 12};
 13
 14static void
 15sequential(struct fdescr *dsc, int fdescrlen, Rune *delim, size_t delimlen)
 16{
 17  Rune   c, last;
 18  int    i;
 19  size_t d;
 20
 21  for (i = 0; i < fdescrlen; i++) {
 22    d    = 0;
 23    last = 0;
 24
 25    while (efgetrune(&c, dsc[i].fp, dsc[i].name)) {
 26      if (last == '\n') {
 27        if (delim[d] != '\0')
 28          efputrune(&delim[d], stdout, "<stdout>");
 29        d = (d + 1) % delimlen;
 30      }
 31
 32      if (c != '\n')
 33        efputrune(&c, stdout, "<stdout>");
 34      last = c;
 35    }
 36
 37    if (last == '\n')
 38      efputrune(&last, stdout, "<stdout>");
 39  }
 40}
 41
 42static void
 43parallel(struct fdescr *dsc, int fdescrlen, Rune *delim, size_t delimlen)
 44{
 45  Rune c, d;
 46  int  i, m, last;
 47
 48nextline:
 49  last = -1;
 50
 51  for (i = 0; i < fdescrlen; i++) {
 52    d = delim[(size_t)i % delimlen];
 53    c = 0;
 54
 55    while (efgetrune(&c, dsc[i].fp, dsc[i].name)) {
 56      for (m = last + 1; m < i; m++) {
 57        if (delim[(size_t)m % delimlen] != '\0')
 58          efputrune(&delim[(size_t)m % delimlen], stdout, "<stdout>");
 59      }
 60      last = i;
 61      if (c == '\n') {
 62        if (i != fdescrlen - 1)
 63          c = d;
 64        efputrune(&c, stdout, "<stdout>");
 65        break;
 66      }
 67      efputrune(&c, stdout, "<stdout>");
 68    }
 69
 70    if (c == 0 && last != -1) {
 71      if (i == fdescrlen - 1)
 72        putchar('\n');
 73      else if (d != '\0')
 74        efputrune(&d, stdout, "<stdout>");
 75      last++;
 76    }
 77  }
 78  if (last != -1)
 79    goto nextline;
 80}
 81
 82static void
 83usage(void)
 84{
 85  eprintf("usage: %s [-s] [-d list] file ...\n", argv0);
 86}
 87
 88// ?man paste: merge lines of files
 89// ?man arguments: file ...
 90// ?man merge corresponding lines of files side by side
 91int
 92main(int argc, char *argv[])
 93{
 94  struct fdescr *dsc;
 95  Rune          *delim_rune = NULL;
 96  size_t         delim_runelen, delim_bytelen = 1;
 97  int            seq = 0, ret = 0, i;
 98  char          *delim = "\t";
 99
100  ARGBEGIN
101  {
102    // ?man -s: silent mode or print summary
103    case 's':
104      seq = 1;
105      break;
106    // ?man -d:str: specify directory
107    case 'd':
108      delim         = EARGF(usage());
109      delim_bytelen = unescape(delim);
110      break;
111    default:
112      usage();
113  }
114  ARGEND
115
116  if (!argc)
117    usage();
118
119  /* populate delimiters */
120  delim_rune = ereallocarray(NULL, utfmemlen(delim, delim_bytelen) + 1, sizeof(*delim_rune));
121  if (!(delim_runelen = utfntorunestr(delim, delim_bytelen, delim_rune))) {
122    usage();
123  }
124
125  /* populate file list */
126  dsc = ereallocarray(NULL, argc, sizeof(*dsc));
127
128  for (i = 0; i < argc; i++) {
129    if (!strcmp(argv[i], "-")) {
130      argv[i]   = "<stdin>";
131      dsc[i].fp = stdin;
132    } else if (!(dsc[i].fp = fopen(argv[i], "r"))) {
133      eprintf("fopen %s:", argv[i]);
134    }
135    dsc[i].name = argv[i];
136  }
137
138  if (seq) {
139    sequential(dsc, argc, delim_rune, delim_runelen);
140  } else {
141    parallel(dsc, argc, delim_rune, delim_runelen);
142  }
143
144  for (i = 0; i < argc; i++)
145    if (dsc[i].fp != stdin && fshut(dsc[i].fp, argv[i]))
146      ret |= fshut(dsc[i].fp, argv[i]);
147
148  ret |= fshut(stdin, "<stdin>") | fshut(stdout, "<stdout>");
149
150  return ret;
151}