mirror of git://git.suckless.org/sbase
84 lines
1.3 KiB
C
84 lines
1.3 KiB
C
/* 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 *);
|
|
|
|
static char mode = 't';
|
|
static const char *sep = "\t";
|
|
static long incr = 1;
|
|
static regex_t preg;
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
eprintf("usage: %s [-b style] [-i increment] [-s sep] [FILE...]\n",
|
|
argv0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
FILE *fp;
|
|
char *r;
|
|
|
|
ARGBEGIN {
|
|
case 'b':
|
|
r = EARGF(usage());
|
|
mode = r[0];
|
|
if(r[0] == 'p') {
|
|
regcomp(&preg, &r[1], REG_NOSUB);
|
|
} else if(!strchr("ant", mode)) {
|
|
usage();
|
|
}
|
|
break;
|
|
case 'i':
|
|
incr = estrtol(EARGF(usage()), 0);
|
|
break;
|
|
case 's':
|
|
sep = EARGF(usage());
|
|
break;
|
|
default:
|
|
usage();
|
|
} ARGEND;
|
|
|
|
if(argc == 0) {
|
|
nl(stdin);
|
|
} else for(; argc > 0; argc--, argv++) {
|
|
if(!(fp = fopen(argv[0], "r"))) {
|
|
weprintf("fopen %s:", argv[0]);
|
|
continue;
|
|
}
|
|
nl(fp);
|
|
fclose(fp);
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
void
|
|
nl(FILE *fp)
|
|
{
|
|
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);
|
|
}
|
|
|