/* radare - LGPL - Copyright 2009-2010 pancake<nopcode.org> nibble<develsec.org> */

#include <r_cons.h>

R_API void r_cons_grep(const char *str) {
	RCons *cons;
	char buf[1024];
	char *ptr, *optr, *ptr2, *ptr3;
	cons = r_cons_singleton ();
	cons->grep.str = NULL;
	cons->grep.nstrings = 0;
	cons->grep.token = cons->grep.line = -1;
	cons->grep.counter = cons->grep.neg = 0;

	if (str == NULL || !*str)
		return;

	if (*str == '!') { // neg
		cons->grep.neg = 1;
		str = str + 1;
	}
	if (*str == '?') { // counter
		cons->grep.counter = 1;
		str = str + 1;
	}

	strncpy (buf, str, sizeof (buf));
	ptr = buf;
	ptr3 = strchr (ptr, '['); // column number
	if (ptr3) {
		ptr3[0]='\0';
		cons->grep.token = atoi (ptr3+1);
		if (cons->grep.token<0)
			cons->grep.token = -1;
	}
	ptr2 = strchr (ptr, ':'); // line number
	if (ptr2) {
		ptr2[0]='\0';
		cons->grep.line = atoi (ptr2+1);
		if (cons->grep.line<0)
			cons->grep.line = -1;
	}
	if (*ptr) {
		cons->grep.str = (char *)strdup (ptr);
		do {
			optr = ptr;
			ptr = strchr (ptr, ','); // grep keywords
			if (ptr) {
				ptr[0] = '\0';
				ptr = ptr+1;
			}
			// TODO: check if keyword > 64
			strncpy (cons->grep.strings[cons->grep.nstrings], optr, 63);
			cons->grep.nstrings++;
		} while (ptr);
	}
}

R_API int r_cons_grepbuf(char *buf, int len) {
	RCons *cons = r_cons_singleton ();
	char tline[1024], *tbuf, *p, *out, *in = buf;
	int ret, buffer_len = 0, l = 0;

	out = tbuf = calloc (1, len);
	cons->lines = 0;
	while (in-buf<len) {
		p = strchr (in, '\n');
		if (!p) {
			free (tbuf);
			return 0;
		}
		l = p-in;
		if (l>0 && l<sizeof (tline)-1) {
			memset (tline, 0, sizeof (tline));
			memcpy (tline, in, l);
			ret = r_cons_grep_line (tline, l);
			if (ret > 0) {
				if (cons->grep.line == -1 ||
					(cons->grep.line != -1 && cons->grep.line == cons->lines)) {
					memcpy (out, tline, ret);
					out += ret;
					buffer_len += ret;
				}
				cons->lines++;
			} else if (ret < 0) {
				free (tbuf);
				return 0;
			} 
			in += l+1;
		} else in++;
	}
	memcpy (buf, tbuf, len);
	cons->buffer_len = buffer_len;
	free (tbuf);
	return cons->lines;
}

R_API int r_cons_grep_line(char *buf, int len) {
	RCons *cons = r_cons_singleton ();
	const char delims[6][2] = { "|", "/", "\\", ",", ";", "\t" };
	int hit = cons->grep.neg;
	int i, j;

	if (cons->grep.nstrings>0) {
		for (i=0; i<cons->grep.nstrings; i++)
			if (strstr (buf, cons->grep.strings[i])) {
				hit = !cons->grep.neg;
				break;
			}
	} else hit = 1;

	if (hit) {
		if (cons->grep.token != -1 && (cons->grep.line == -1 ||
			(cons->grep.line != -1 && cons->grep.line == cons->lines))) {
			char ptr[1024], *tok = NULL;
			strncpy (ptr, buf, 1023);
			for (i=0; i<len; i++) for (j=0; j<6; j++)
				if (ptr[i] == delims[j][0])
					ptr[i] = ' ';
			for (tok = buf, i=0; i<=cons->grep.token; i++) {
				if (i==0) tok = (char *)strtok (ptr, " ");
				else tok = (char *)strtok (NULL, " ");
				if (tok == NULL)
					return -1;
			}
			len = strlen (tok);
			memcpy (buf, tok, len);
		}
		memcpy (buf+len, "\n", 1);
		len += 1;
	} else len = 0;

	return len;
}

// XXX: rename char *r_cons_filter_html(const char *ptr)
R_API int r_cons_html_print(const char *ptr) {
	const char *str = ptr;
	int color = 0;
	int esc = 0;
	int len = 0;
	int inv = 0;

	for (;ptr[0]; ptr = ptr + 1) {
		if (ptr[0] == '\n') {
			printf ("<br />");
			fflush (stdout);
		}
		if (ptr[0] == 0x1b) {
			esc = 1;
			write (1, str, ptr-str);
			str = ptr + 1;
			continue;
		}
		if (esc == 1) {
			// \x1b[2J
			if (ptr[0] != '[') {
				eprintf ("Oops invalid escape char\n");
				esc = 0;
				str = ptr + 1;
				continue;
			}
			esc = 2;
			continue;
		} else 
		if (esc == 2) {
			if (ptr[0]=='2'&&ptr[1]=='J') {
				ptr = ptr +1;
				printf ("<hr />\n"); fflush(stdout);
				esc = 0;
				str = ptr;
				continue;
			} else
			if (ptr[0]=='0'&&ptr[1]==';'&&ptr[2]=='0') {
				ptr = ptr + 4;
				r_cons_gotoxy (0,0);
				esc = 0;
				str = ptr;
				continue;
			} else
			if (ptr[0]=='0'&&ptr[1]=='m') {
				ptr = ptr + 1;
				str = ptr + 1;
				inv = 0;
				esc = 0;
				continue;
				// reset color
			} else
			if (ptr[0]=='7'&&ptr[1]=='m') {
				inv = 128;
				ptr = ptr + 1;
				str = ptr + 1;
				esc = 0;
				continue;
				// reset color
			} else
			if (ptr[0]=='3' && ptr[2]=='m') {
				color = 1;
				switch(ptr[1]) {
				case '0': // BLACK
					printf ("<font color=black>"); fflush(stdout);
					break;
				case '1': // RED
					printf ("<font color=red>"); fflush(stdout);
					break;
				case '2': // GREEN
					printf ("<font color=green>"); fflush(stdout);
					break;
				case '3': // YELLOW
					printf ("<font color=yellow>"); fflush(stdout);
					break;
				case '4': // BLUE
					printf ("<font color=blue>"); fflush(stdout);
					break;
				case '5': // MAGENTA
					printf ("<font color=magenta>"); fflush(stdout);
					break;
				case '6': // TURQOISE
					printf ("<font color=#0ae>"); fflush(stdout);
					break;
				case '7': // WHITE
					printf ("<font color=white>"); fflush(stdout);
					break;
				case '8': // GRAY
					printf ("<font color=#777>"); fflush(stdout);
					break;
				case '9': // ???
					break;
				}
				ptr = ptr + 1;
				str = ptr + 2;
				esc = 0;
				continue;
			} else
			if (ptr[0]=='4' && ptr[2]=='m') {
				/* background color */
				switch (ptr[1]) {
				case '0': // BLACK
					printf ("<font style='background-color:#000'>"); fflush(stdout);
					break;
				case '1': // RED
					printf ("<font style='background-color:#f00'>"); fflush(stdout);
					break;
				case '2': // GREEN
					printf ("<font style='background-color:#0f0'>"); fflush(stdout);
					break;
				case '3': // YELLOW
					printf ("<font style='background-color:#ff0'>"); fflush(stdout);
					break;
				case '4': // BLUE
					printf ("<font style='background-color:#00f'>"); fflush(stdout);
					break;
				case '5': // MAGENTA
					printf ("<font style='background-color:#f0f'>"); fflush(stdout);
					break;
				case '6': // TURQOISE
					printf ("<font style='background-color:#aaf'>"); fflush(stdout);
					break;
				case '7': // WHITE
					printf ("<font style='background-color:#fff'>"); fflush(stdout);
					break;
				case '8': // GRAY
					printf ("<font style='background-color:#777'>"); fflush(stdout);
					break;
				case '9': // ???
					break;
				}
			}
		} 
		len++;
	}
	write (1, str, ptr-str);
	return len;
}