diff options
Diffstat (limited to 'toys')
| -rw-r--r-- | toys/posix/grep.c | 101 |
1 files changed, 72 insertions, 29 deletions
diff --git a/toys/posix/grep.c b/toys/posix/grep.c index 105ab3bc..2cf078b1 100644 --- a/toys/posix/grep.c +++ b/toys/posix/grep.c @@ -4,11 +4,13 @@ * * See http://pubs.opengroup.org/onlinepubs/9699919799/utilities/grep.html * - * TODO: --color - * * Posix doesn't even specify -r, documenting deviations from it is silly. +* echo hello | grep -w '' +* echo '' | grep -w '' +* echo hello | grep -f </dev/null +* -USE_GREP(NEWTOY(grep, "S(exclude)*M(include)*ZzEFHIabhinorsvwclqe*f*C#B#A#m#x[!wx][!EFw]", TOYFLAG_BIN)) +USE_GREP(NEWTOY(grep, "(color):;S(exclude)*M(include)*ZzEFHIabhinorsvwclqe*f*C#B#A#m#x[!wx][!EFw]", TOYFLAG_BIN)) USE_EGREP(OLDTOY(egrep, grep, TOYFLAG_BIN)) USE_FGREP(OLDTOY(fgrep, grep, TOYFLAG_BIN)) @@ -66,7 +68,9 @@ config FGREP GLOBALS( long m, A, B, C; struct arg_list *f, *e, *M, *S; + char *color; + char *purple, *cyan, *red, *green, *grey; struct double_list *reg; char indelim, outdelim; int found, tried; @@ -74,22 +78,30 @@ GLOBALS( struct reg { struct reg *next, *prev; - regex_t r; int rc; + regex_t r; regmatch_t m; }; +static void numdash(long num, char dash) +{ + printf("%s%ld%s%c", TT.green, num, TT.cyan, dash); +} + // Emit line with various potential prefixes and delimiter static void outline(char *line, char dash, char *name, long lcount, long bcount, int trim) { - if (name && FLAG(H)) printf("%s%c", name, dash); - if (!line || (lcount && FLAG(n))) - printf("%ld%c", lcount, line ? dash : TT.outdelim); - if (bcount && FLAG(b)) printf("%ld%c", bcount-1, dash); + if (name && FLAG(H)) printf("%s%s%s%c", TT.purple, name, TT.cyan, dash); + if (FLAG(c)) { + printf("%s%ld", TT.grey, lcount); + xputc(TT.outdelim); + } else if (lcount && FLAG(n)) numdash(lcount, dash); + if (bcount && FLAG(b)) numdash(bcount-1, dash); if (line) { + if (FLAG(color)) xputsn(FLAG(o) ? TT.red : TT.grey); // support embedded NUL bytes in output - fwrite(line, 1, trim, stdout); + xputsl(line, trim); xputc(TT.outdelim); } } @@ -134,7 +146,7 @@ static void do_grep(int fd, char *name) struct reg *shoe; size_t ulen; long len; - int matched = 0, baseline = 0; + int matched = 0, rc = 1; // get next line, check and trim delimiter lcount++; @@ -152,7 +164,6 @@ static void do_grep(int fd, char *name) // Loop to handle multiple matches in same line do { regmatch_t *mm = (void *)toybuf; - int rc, skip = 0; // Handle "fixed" (literal) matches if (FLAG(F)) { @@ -175,12 +186,14 @@ static void do_grep(int fd, char *name) if (s) { rc = 0; mm->rm_so = (s-line); - skip = mm->rm_eo = (s-line)+strlen(seek->arg); + mm->rm_eo = (s-line)+strlen(seek->arg); } else rc = 1; // Handle regex matches } else { - mm->rm_so = INT_MAX; + int baseline = mm->rm_eo; + + mm->rm_so = mm->rm_eo = INT_MAX; rc = 1; for (shoe = (void *)TT.reg; shoe; shoe = shoe->next) { @@ -195,14 +208,12 @@ static void do_grep(int fd, char *name) // If we got a match, is it a _better_ match? if (!shoe->rc && (shoe->m.rm_so < mm->rm_so || - (shoe->m.rm_so == mm->rm_so && shoe->m.rm_eo >= skip))) + (shoe->m.rm_so == mm->rm_so && shoe->m.rm_eo >= mm->rm_eo))) { mm = &shoe->m; - skip = mm->rm_eo; rc = 0; } } - baseline = skip; } if (!rc && FLAG(x)) @@ -221,21 +232,22 @@ static void do_grep(int fd, char *name) } if (c) { start += mm->rm_so+1; - continue; } } if (FLAG(v)) { if (FLAG(o)) { - if (rc) skip = mm->rm_eo = strlen(start); - else if (!mm->rm_so) { - start += skip; + if (rc) { + mm->rm_so = 0; + mm->rm_eo = ulen-(start-line); + } else if (!mm->rm_so) { + start += mm->rm_eo; continue; } else mm->rm_eo = mm->rm_so; } else { if (!rc) break; - mm->rm_eo = strlen(start); + mm->rm_eo = ulen-(start-line); } mm->rm_so = 0; } else if (rc) break; @@ -260,12 +272,16 @@ static void do_grep(int fd, char *name) if (FLAG(o)) if (mm->rm_eo == mm->rm_so) break; +// TODO checking this twice if (!FLAG(c)) { long bcount = 1 + offset + (start-line) + (FLAG(o) ? mm->rm_so : 0); if (bin) printf("Binary file %s matches\n", name); - else if (!FLAG(o)) { + else if (FLAG(o)) + outline(start+mm->rm_so, ':', name, lcount, bcount, + mm->rm_eo-mm->rm_so); + else { while (dlb) { struct double_list *dl = dlist_pop(&dlb); unsigned *uu = (void *)(dl->data+((strlen(dl->data)+1)|3)+1); @@ -276,19 +292,34 @@ static void do_grep(int fd, char *name) before--; } - outline(line, ':', name, lcount, bcount, ulen); + if (matched==1) + outline(FLAG(color) ? 0 : line, ':', name, lcount, bcount, ulen); + if (FLAG(color)) { + xputsn(TT.grey); + if (mm->rm_so) xputsl(line, mm->rm_so); + xputsn(TT.red); + xputsl(line+mm->rm_so, mm->rm_eo-mm->rm_so); + } + if (TT.A) after = TT.A+1; - } else outline(start+mm->rm_so, ':', name, lcount, bcount, - mm->rm_eo-mm->rm_so); + } } - start += skip; - if (!FLAG(o)) break; + start += mm->rm_eo; + if (mm->rm_so == mm->rm_eo) break; + if (!FLAG(o) && FLAG(color)) break; } while (*start); offset += len; - if (matched) mcount++; - else { + if (matched) { + // Finish off pending line color fragment. + if (FLAG(color) && !FLAG(o)) { + xputsn(TT.grey); + if (ulen > start-line) xputsl(start, ulen-(start-line)); + xputc(TT.outdelim); + } + mcount++; + } else { int discard = (after || TT.B); if (after && --after) { @@ -348,6 +379,7 @@ static void parse_regex(void) // Split lines at \n, add individual lines to new list. do { +// TODO: NUL terminated input shouldn't split -e at \n ss = strchr(s, '\n'); if (ss) *(ss++) = 0; new = xmalloc(sizeof(struct arg_list)); @@ -420,6 +452,17 @@ void grep_main(void) { char **ss = toys.optargs; + if (FLAG(color) && (!TT.color || !strcmp(TT.color, "auto")) && !isatty(1)) + toys.optflags &= ~FLAG_color; + + if (FLAG(color)) { + TT.purple = "\033[35m"; + TT.cyan = "\033[36m"; + TT.red = "\033[1;31m"; + TT.green = "\033[32m"; + TT.grey = "\033[0m"; + } else TT.purple = TT.cyan = TT.red = TT.green = TT.grey = ""; + // Grep exits with 2 for errors toys.exitval = 2; |