/*
 * euses is the first working incarnation of something
 * I called `guses' (for Gentoo uses, see
 * <http://www.xs4all.nl/~rooversj/gentoo/guses>) which
 * is a bash script I wrote that searches crudely for USE
 * flags in use.desc and use.local.desc and returns the
 * matching lines with some fancy colouring.
 * I originally intended that to be the ultimate, but I
 * opted to learn a bit of C instead and have a much
 * faster working program. Not that speed matters much in
 * this case...
 *
 * It should compile on any basic Linux system, but since it
 * is intended to search in two Gentoo-specific files (and
 * because right now the paths are hard-wired and no error-
 * checking is in place) compiling it (with whatever gcc
 * options you prefer) will probably simply work.
 *
 * Thanks to KillerFox and #gentoo-hppa for their programming clues. :)
 * 
 * This program was written by JeR in 2005, released on May 9, 2005
 * and is distributed under the GNU General Public License
 * version 2 or later.
 *
 * Mail your suggestions, improvements and complaints to
 * jer at xs4all dot nl
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>

#define MAX_LINE_LENGTH 1024

/*Color definitions: */
#define BG "\033[01;32m"	/* Bright Green */
#define FC "\033[00m"		/* Normal	*/
#define BW "\033[01;29m"	/* Bright White */
#define NC "\033[00;36m"	/* Normal Cyan	*/
#define BB "\033[01;34m"	/* Bright Blue	*/

void print_usage(char *program_name, int interactive)
{
	program_name = strrchr(program_name, '/');
	if (program_name == NULL) program_name = "euses";
	if (interactive == 1) {
		(void)printf("Usage: %s%s <USE flag to search>%s\n", BG, program_name,FC);
	} else {
		(void)printf("Usage: %s <USE flag to search>\n", program_name);
	}
	(void)printf("%s searches use.desc and use.local.desc for your USE flag.\n", program_name);
	exit(EXIT_SUCCESS);
}

void toerrishuman(int whichone) {
	switch (whichone) {
		case 0: puts("/etc/make.conf not found\n"); break;
		case 1: puts("use.desc not found\n"); break;
		case 2: puts("use.local.desc not found\n"); break;
		default: puts("Do you Gentoo?\n"); break;
	}
	exit(EXIT_FAILURE);
}

void use_desc_format(char *use_line)
{
	int i;
	size_t use_len;
	use_len = strlen(use_line);
	(void)printf("%s", BG);
	for (i = 0; i < (int)use_len; ++i) {
		if (use_line[i] == ' ') {
			(void)printf("%s", FC);
		}
		(void)printf("%c", use_line[i]);
	}
	return;
}

void use_local_desc_format(char *use_line)
{
	int i;
	size_t use_len;
	int past_colon = 0;
	use_len = strlen(use_line);
	(void)printf("%s", NC);
	for (i = 0; i < (int)use_len; ++i) {
		if (&use_line[i] >= strstr(use_line, ":")) {
			switch (use_line[i]) {
			case ':':{
					if (past_colon == 0)
						(void)printf("%s%c%s", FC, use_line[i], BG);
				}
				past_colon = 1;
				break;
			case ' ':
				(void)printf("%s%c", FC, use_line[i]);
				break;
			default:
				(void)printf("%c", use_line[i]);
				break;
			}
		} else
			printf("%c", use_line[i]);

	}
	return;
}

char *getportagepath(char *portage_path)
{
	char etc_line[MAX_LINE_LENGTH * 2];
	FILE *ptr = NULL;
	const char *etc_make_conf_path = "/etc/make.conf";
	const char *portdir = "PORTDIR=\"";
	int i = 9;
	int j = 0;

	ptr = fopen(etc_make_conf_path, "r");
	if (ptr == NULL) { toerrishuman(0) ; exit(EXIT_FAILURE); }

	while (feof(ptr) == 0) {
		while (fgets(etc_line, MAX_LINE_LENGTH, ptr) != NULL)
		{
			if (strstr(etc_line, portdir) != 0)
			{
				while (etc_line[i] != '\0')
				{
					if (etc_line[i] != '"')
						portage_path[j++] = etc_line[i++];
					else { ++i; ++j; }
				}
				portage_path[j] = '\0';
			}
		}
		continue;
	}
	(void)fclose(ptr);
	ptr = NULL;
	if (strlen(portage_path) == 0) portage_path = "/usr/portage";
	return portage_path;
}

int main(int argc, char **argv)
{
	FILE *ptr = NULL;
	char *match_char;
	int use_desc_matches = 0;
	int use_local_desc_matches = 0;
	char use_desc_path[MAX_LINE_LENGTH * 2];
	char use_local_desc_path[MAX_LINE_LENGTH * 2];
	const char *use_desc_file = "/profiles/use.desc";
	const char *use_local_desc_file = "/profiles/use.local.desc";
	int interactive = 0;
	char use_line[MAX_LINE_LENGTH];
	char path[MAX_LINE_LENGTH];
	char *portage_path = path;

	memset(use_desc_path, 0, sizeof(use_desc_path));
	memset(use_local_desc_path, 0, sizeof(use_local_desc_path));

	/* Check in /etc/make.conf where Portage is: */
	portage_path = getportagepath(portage_path);

	strcat(use_desc_path, portage_path);
	strcat(use_desc_path, use_desc_file);
	strcat(use_local_desc_path, portage_path);
	strcat(use_local_desc_path, use_local_desc_file);

	if (isatty(STDOUT_FILENO) == 1) interactive = 1;

	/* First look in use.desc: */

	if (argc < 2) print_usage(argv[0], interactive);
	ptr = fopen(use_desc_path, "r");
	if (ptr == 0) { toerrishuman(1) ; exit(EXIT_FAILURE); }
	while (feof(ptr) == 0) {
		while(fgets(use_line, MAX_LINE_LENGTH, ptr) != NULL)
		{
			if (use_line[0] == '#') continue;
			match_char = strstr(use_line, argv[1]);
			if (match_char != 0) {
				if (interactive == 1) {
					if (use_desc_matches == 0) {
						printf(" %s*%s Found '%s%s%s' in %suse.desc%s:\n", BG, FC, BW, argv[1], FC, BB, FC);
						use_desc_matches = 1;
					}
					(void)use_desc_format(use_line);	/* Format a single line */
				} else {
					printf("      use.desc: %s", use_line);	/* Don't format */
				}
			}
		}
	}
	(void)fclose(ptr);
	ptr = NULL;

	/* Then look in use.local.desc: */

	ptr = fopen(use_local_desc_path, "r");
	if (ptr == NULL) { toerrishuman(2) ; exit(EXIT_FAILURE); }
	while (feof(ptr) == 0) {
		while(fgets(use_line, MAX_LINE_LENGTH, ptr) != NULL)
		{
			if (use_line[0] == '#') continue;
			if (strstr(use_line, argv[1]) != 0) {
				if (interactive == 1) {
					if (use_local_desc_matches == 0) {
						printf(" %s*%s Found %s%s%s in %suse.local.desc%s:\n", BG, FC, BW, argv[1], FC, BB, FC);
						use_local_desc_matches = 1;
					}
					(void)use_local_desc_format(use_line);
				} else {
					printf("use.local.desc: %s", use_line);
				}
			}
		}
	}
	(void)fclose(ptr);
	ptr = NULL;
	exit(EXIT_SUCCESS);
}