Mercurial > hg > index.fcgi > rcg > rcg-1

view rcg.c @ 13:2fb87616ab65

find changesets by author, revision, files, or words in the commit message
rcg.c: fix bug when dealing with ".*"
author paulo@thepaulopc
date Sat, 06 Nov 2010 02:57:06 -0700
parents 9ec52ba4b994
children 669266246c6b
line source
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <regex.h>
4
5 #include <string.h>
6
7 #include <unistd.h>
8
9 #define _GNU_SOURCE
10 #include <getopt.h>
11
12 const size_t BUFSIZE = 5000;
13
14 const char *_CLR_START = "\x1B[";
15 const char *_CLR_BOLD = "1;";
16 char *CLR_START;
17 const char *CLR_END = "m";
18 const char *CLR_CLEAR = "\x1B[0m";
19
20 typedef struct _colors {
21 char *name;
22 char *fg;
23 char *bg;
24 } Colors;
25
26 const Colors COLORS[] = {
27 { "black", "30", "40" },
28 { "red", "31", "41" },
29 { "green", "32", "42" },
30 { "brown", "33", "43" },
31 { "blue", "34", "44" },
32 { "magenta", "35", "45" },
33 { "cyan", "36", "46" },
34 { "white", "37", "47" },
35 { "default", "39", "49" },
36 { NULL },
37 };
38
39 char *getFGColor(char *cname)
40 {
41 Colors *c;
42 for (c = COLORS; c->name; c++)
43 {
44 if (strncmp(cname, c->name, strlen(c->name)) == 0)
45 return c->fg;
46 }
47 return NULL;
48 }
49
50 char *getBGColor(char *cname)
51 {
52 Colors *c;
53 for (c = COLORS; c->name; c++)
54 {
55 if (strncmp(cname, c->name, strlen(c->name)) == 0)
56 return c->bg;
57 }
58 return NULL;
59 }
60
61 int colorLine = 0;
62 int embolden = 0;
63 char *g_fg = "";
64 char *g_bg = "";
65
66 typedef enum _exit_code {
67 EXIT_OK,
68 EXIT_REALLOC_ERROR,
69 EXIT_ARGS_ERROR,
70 } Exit_code;
71
72 int re_error(int errcode, const regex_t *re)
73 {
74 char *err_string = calloc(BUFSIZE, sizeof(char));
75 regerror(errcode, re, err_string, BUFSIZE*sizeof(char));
76 fprintf(stderr, "%s \n", err_string);
77 free(err_string);
78 return errcode;
79 }
80
81 Exit_code realloc_error()
82 {
83 fprintf(stderr, "realloc() failure \n");
84 return EXIT_REALLOC_ERROR;
85 }
86
87 Exit_code args_error()
88 {
89 fprintf(stderr, "Invalid or nonexistent argument. \n");
90 return EXIT_ARGS_ERROR;
91 }
92
93 // returns regular expression argument
94 char *parseArgs(int argc, char **argv)
95 {
96 static struct option long_options[] =
97 {
98 { "line", 0, 0, 'l' },
99 { "bold", 0, 0, 'B' },
100 { 0, 0, 0, 0 },
101 };
102
103 int c;
104 int l;
105 while ((c = getopt_long(argc, argv, "lBf:b:", long_options, &l)) >= 0)
106 {
107 if (c == 'l')
108 colorLine = 1;
109 else if (c == 'B')
110 embolden = 1;
111 else if (c == 'f')
112 g_fg = strdup(optarg);
113 else if (c == 'b')
114 g_bg = strdup(optarg);
115 }
116
117 if (optind >= argc)
118 exit(args_error());
119
120 return argv[optind];
121 }
122
123 int main(int argc, char **argv)
124 {
125 char *re_expression = parseArgs(argc, argv);
126
127 //fprintf(stderr, "re_expression = %s \n", re_expression); //d/ 20100405 PBA
128
129 char *buf = calloc(BUFSIZE, sizeof(char));
130
131 int re_err;
132 regex_t *re = calloc(1, sizeof(regex_t));
133 re_err = regcomp(re, re_expression, REG_EXTENDED | REG_NEWLINE);
134 if (re_err != 0)
135 exit(re_error(re_err, re));
136
137 regmatch_t *rem = calloc(1, sizeof(regmatch_t));
138
139 if (embolden)
140 {
141 CLR_START = calloc(strlen(_CLR_START) + strlen(_CLR_BOLD) + 1, sizeof(char));
142 strcpy(CLR_START, _CLR_START);
143 strcat(CLR_START, _CLR_BOLD);
144 }
145 else
146 CLR_START = strdup(_CLR_START);
147
148 if (strlen(g_fg) == 0 && strlen(g_bg) == 0)
149 g_fg = "red";
150 else if (strlen(g_fg) == 0 && strlen(g_bg) > 0)
151 g_fg = "default";
152
153 char *fgcolor = getFGColor(g_fg);
154 if (!fgcolor)
155 exit(args_error());
156
157 char *clr = strdup(fgcolor);
158 if (strlen(g_bg) > 0)
159 {
160 char *bgcolor = getBGColor(g_bg);
161 if (bgcolor)
162 {
163 size_t l = strlen(clr) + 1 + strlen(bgcolor);
164 clr = realloc(clr, (l + 1)*sizeof(char));
165 if (!clr)
166 exit(realloc_error());
167 strcat(clr, ";");
168 strcat(clr, bgcolor);
169 }
170 else
171 exit(args_error());
172 }
173
174 size_t CLR_START_len = strlen(CLR_START);
175 size_t clr_len = strlen(clr);
176 size_t CLR_END_len = strlen(CLR_END);
177 size_t CLR_CLEAR_len = strlen(CLR_CLEAR);
178
179 char *out = NULL;
180
181 while (fgets(buf, BUFSIZE, stdin))
182 {
183 int so = -1;
184 int eo = -1;
185
186 size_t out_len = strlen(buf) + 1;
187 unsigned int out_pos = 0;
188 unsigned int buf_pos = 0;
189
190 for (re_err = 0; re_err != REG_NOMATCH;)
191 {
192 re_err = regexec(re, &buf[buf_pos], 1, rem, 0);
193
194 //fprintf(stderr, "%s", &buf[buf_pos]); //d// 20100405 PBA
195
196 if (out_len >= (BUFSIZE - buf_pos))
197 out_len = (BUFSIZE - buf_pos) - 1;
198
199 out = realloc(out, out_len*sizeof(char));
200 if (!out)
201 exit(realloc_error());
202
203 so = rem[0].rm_so;
204 eo = rem[0].rm_eo;
205
206 int match_len = eo - so;
207
208 if (colorLine && match_len > 0)
209 {
210 so = 0;
211 eo = out_len - 2;
212 match_len = eo - so;
213 }
214
215 //fprintf(stderr, "%d %d \n", so, eo); //d// 20100327 PBA
216
217 if (re_err != REG_NOMATCH && so >= 0 && eo >= 0 && match_len > 0)
218 {
219 out_len += CLR_START_len + clr_len + CLR_END_len + CLR_CLEAR_len;
220 out = realloc(out, out_len*sizeof(char));
221 if (!out)
222 exit(realloc_error());
223 strncpy(&out[out_pos], &buf[buf_pos], so);
224 out_pos += so;
225 buf_pos += so;
226 strncpy(&out[out_pos], CLR_START, CLR_START_len);
227 out_pos += CLR_START_len;
228 strncpy(&out[out_pos], clr, clr_len);
229 out_pos += clr_len;
230 strncpy(&out[out_pos], CLR_END, CLR_END_len);
231 out_pos += CLR_END_len;
232 strncpy(&out[out_pos], &buf[buf_pos], match_len);
233 out_pos += match_len;
234 strncpy(&out[out_pos], CLR_CLEAR, CLR_CLEAR_len);
235 out_pos += CLR_CLEAR_len;
236
237 buf_pos += match_len;
238 }
239 else
240 {
241 strncpy(&out[out_pos], &buf[buf_pos], out_len - out_pos);
242 out[out_len - 1] = '\0';
243 re_err = REG_NOMATCH;
244 }
245 }
246
247 fputs(out, stdout);
248 fflush(stdout);
249 }
250
251 return EXIT_OK;
252 }