sbase/nl.c

124 lines
2.4 KiB
C
Raw Normal View History

2011-05-29 16:30:44 -04:00
/* See LICENSE file for copyright and license details. */
#include <limits.h>
2011-05-29 16:30:44 -04:00
#include <stdio.h>
#include <stdint.h>
2011-05-29 16:30:44 -04:00
#include <stdlib.h>
#include <string.h>
2011-05-29 16:30:44 -04:00
#include "text.h"
#include "util.h"
2015-03-10 07:49:56 -04:00
/* formats here specify line number and separator (not line content) */
#define FORMAT_LN "%-*ld%s"
#define FORMAT_RN "%*ld%s"
#define FORMAT_RZ "%0*ld%s"
2015-02-20 09:12:03 -05:00
static char mode = 't';
2015-03-10 07:49:56 -04:00
static int blines = 1;
2015-02-20 09:12:03 -05:00
static const char *format = FORMAT_RN;
2011-05-29 16:30:44 -04:00
static const char *sep = "\t";
2015-02-20 07:15:43 -05:00
static int width = 6;
2015-02-20 07:05:54 -05:00
static size_t startnum = 1;
static size_t incr = 1;
static regex_t preg;
2015-03-07 07:33:39 -05:00
static void
nl(const char *name, FILE *fp)
{
char *buf = NULL;
2015-03-10 07:49:56 -04:00
int donumber, bl = 1;
2015-02-20 07:05:54 -05:00
size_t size = 0;
while (getline(&buf, &size, fp) != -1) {
2015-03-10 07:49:56 -04:00
donumber = 0;
if ((mode == 't' && buf[0] != '\n')
|| (mode == 'p' && !regexec(&preg, buf, 0, NULL, 0))) {
donumber = 1;
} else if (mode == 'a') {
if (buf[0] == '\n' && bl < blines) {
++bl;
} else {
donumber = 1;
bl = 1;
}
}
if (donumber) {
printf(format, width, startnum, sep);
2015-02-20 07:05:54 -05:00
startnum += incr;
} else {
2015-03-10 07:49:56 -04:00
printf("%*s", width, "");
2015-02-20 07:05:54 -05:00
}
2015-03-10 07:49:56 -04:00
printf("%s", buf);
}
free(buf);
if (ferror(fp))
eprintf("%s: read error:", name);
}
2011-05-29 16:30:44 -04:00
2013-06-14 14:20:47 -04:00
static void
usage(void)
{
2015-03-10 07:49:56 -04:00
eprintf("usage: %s [-b type] [-i incr] [-l num] [-n format] [-s sep] [-v startnum] [-w width] [file]\n", argv0);
2013-06-14 14:20:47 -04:00
}
2011-05-29 16:30:44 -04:00
int
main(int argc, char *argv[])
{
FILE *fp;
2013-06-14 14:20:47 -04:00
char *r;
2011-05-29 16:30:44 -04:00
2013-06-14 14:20:47 -04:00
ARGBEGIN {
case 'b':
r = EARGF(usage());
mode = r[0];
if (r[0] == 'p')
2014-11-16 09:17:24 -05:00
eregcomp(&preg, &r[1], REG_NOSUB);
else if (!strchr("ant", mode))
2013-06-14 14:20:47 -04:00
usage();
break;
case 'i':
incr = estrtonum(EARGF(usage()), 0, MIN(LLONG_MAX, SIZE_MAX));
2013-06-14 14:20:47 -04:00
break;
2015-03-10 07:49:56 -04:00
case 'l':
blines = estrtonum(EARGF(usage()), 0, UINT_MAX);
break;
2015-02-20 09:12:03 -05:00
case 'n':
format = EARGF(usage());
if (!strcmp(format, "ln"))
format = FORMAT_LN;
else if (!strcmp(format, "rn"))
format = FORMAT_RN;
else if (!strcmp(format, "rz"))
format = FORMAT_RZ;
else
eprintf("%s: bad format\n", format);
break;
2013-06-14 14:20:47 -04:00
case 's':
sep = EARGF(usage());
break;
2015-02-20 07:05:54 -05:00
case 'v':
startnum = estrtonum(EARGF(usage()), 0, MIN(LLONG_MAX, SIZE_MAX));
break;
2015-02-20 07:15:43 -05:00
case 'w':
width = estrtonum(EARGF(usage()), 1, INT_MAX);
break;
2013-06-14 14:20:47 -04:00
default:
usage();
} ARGEND;
2014-12-04 07:04:41 -05:00
if (argc > 1)
usage();
if (argc == 0) {
nl("<stdin>", stdin);
2014-12-04 07:04:41 -05:00
} else {
if (!(fp = fopen(argv[0], "r")))
eprintf("fopen %s:", argv[0]);
nl(argv[0], fp);
2011-05-29 16:30:44 -04:00
fclose(fp);
}
2014-12-04 07:04:41 -05:00
return 0;
2011-05-29 16:30:44 -04:00
}