add nl

nl.c

@@ -0,0 +1,69 @@
+/* See LICENSE file for copyright and license details. */
+#include <regex.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include "text.h"
+#include "util.h"
+
+static void nl(FILE *, const char *);
+
+static char mode = 't';
+static const char *sep = "\t";
+static long incr = 1;
+static regex_t preg;
+
+int
+main(int argc, char *argv[])
+{
+	char c, *end;
+	FILE *fp;
+
+	while((c = getopt(argc, argv, "b:i:s:")) != -1)
+		switch(c) {
+		case 'b':
+			mode = optarg[0];
+			if(optarg[0] == 'p')
+				regcomp(&preg, &optarg[1], REG_NOSUB);
+			else if(!strchr("ant", optarg[0]) || optarg[1] != '\0')
+				eprintf("usage: %s [-b mode] [file...]\n", argv[0]);
+			break;
+		case 'i':
+			incr = strtol(optarg, &end, 0);
+			if(*end != '\0')
+				eprintf("%s: not a number\n", optarg);
+			break;
+		case 's':
+			sep = optarg;
+			break;
+		default:
+			exit(EXIT_FAILURE);
+		}
+	if(optind == argc)
+		nl(stdin, "<stdin>");
+	else for(; optind < argc; optind++) {
+		if(!(fp = fopen(argv[optind], "r")))
+			eprintf("fopen %s:", argv[optind]);
+		nl(fp, argv[optind]);
+		fclose(fp);
+	}
+	return EXIT_SUCCESS;
+}
+
+void
+nl(FILE *fp, const char *str)
+{
+	char *buf = NULL;
+	long n = 0;
+	size_t size = 0;
+
+	while(afgets(&buf, &size, fp))
+		if((mode == 'a')
+		|| (mode == 'p' && !regexec(&preg, buf, 0, NULL, 0))
+		|| (mode == 't' && buf[0] != '\n'))
+			printf("%6ld%s%s", n += incr, sep, buf);
+		else
+			printf("       %s", buf);
+	free(buf);
+}