1 files changed, 98 insertions, 0 deletions
diff --git a/util/sbase/cols.c b/util/sbase/cols.c
new file mode 100644
index 00000000..428cd79d
--- /dev/null
+++ b/util/sbase/cols.c
@@ -0,0 +1,98 @@
+/* See LICENSE file for copyright and license details. */
+#include <sys/ioctl.h>
+
+#include <limits.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "text.h"
+#include "util.h"
+
+static void
+usage(void)
+{
+	eprintf("usage: %s [-c num] [file ...]\n", argv0);
+}
+
+int
+main(int argc, char *argv[])
+{
+	FILE *fp;
+	struct winsize w;
+	struct linebuf b = EMPTY_LINEBUF;
+	size_t chars = 65, maxlen = 0, i, j, k, len, cols, rows;
+	int cflag = 0, ret = 0;
+	char *p;
+
+	ARGBEGIN {
+	case 'c':
+		cflag = 1;
+		chars = estrtonum(EARGF(usage()), 1, MIN(LLONG_MAX, SIZE_MAX));
+		break;
+	default:
+		usage();
+	} ARGEND
+
+	if (!cflag) {
+		if ((p = getenv("COLUMNS")))
+			chars = estrtonum(p, 1, MIN(LLONG_MAX, SIZE_MAX));
+		else if (!ioctl(STDOUT_FILENO, TIOCGWINSZ, &w) && w.ws_col > 0)
+			chars = w.ws_col;
+	}
+
+	if (!argc) {
+		getlines(stdin, &b);
+	} else {
+		for (; *argv; argc--, argv++) {
+			if (!strcmp(*argv, "-")) {
+				*argv = "<stdin>";
+				fp = stdin;
+			} else if (!(fp = fopen(*argv, "r"))) {
+				weprintf("fopen %s:", *argv);
+				ret = 1;
+				continue;
+			}
+			getlines(fp, &b);
+			if (fp != stdin && fshut(fp, *argv))
+				ret = 1;
+		}
+	}
+
+	for (i = 0; i < b.nlines; i++) {
+		for (j = 0, len = 0; j < b.lines[i].len; j++) {
+			if (UTF8_POINT(b.lines[i].data[j]))
+				len++;
+		}
+		if (len && b.lines[i].data[b.lines[i].len - 1] == '\n') {
+			b.lines[i].data[--(b.lines[i].len)] = '\0';
+			len--;
+		}
+		if (len > maxlen)
+			maxlen = len;
+	}
+
+	for (cols = 1; (cols + 1) * maxlen + cols <= chars; cols++);
+	rows = b.nlines / cols + (b.nlines % cols > 0);
+
+	for (i = 0; i < rows; i++) {
+		for (j = 0; j < cols && i + j * rows < b.nlines; j++) {
+			for (k = 0, len = 0; k < b.lines[i + j * rows].len; k++) {
+				if (UTF8_POINT(b.lines[i + j * rows].data[k]))
+					len++;
+			}
+			fwrite(b.lines[i + j * rows].data, 1,
+			       b.lines[i + j * rows].len, stdout);
+			if (j < cols - 1)
+				for (k = len; k < maxlen + 1; k++)
+					putchar(' ');
+		}
+		putchar('\n');
+	}
+
+	ret |= fshut(stdin, "<stdin>") | fshut(stdout, "<stdout>");
+
+	return ret;
+}