/* 
 * Description: display .desktop files content found on system
 *     Application-Name command
 * Version:  0.1
 * Licence:  MIT
 * Author:  prx prx@si3t.ch
 * TODO : locale
 */

#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <dirent.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define LEN(X)               (sizeof X / sizeof X[0])
#define PLEDGEORDIE(prom) if (pledge(prom, NULL) == -1) { err(1, "pledge"); } 

int          efclose(FILE *stream);
FILE *       efopen(const char *path, const char *mode);
int          endswith(const char *str, const char *end);
size_t       esnprintf(char *str, size_t size, const char *fmt, ...);
char *       estrdup(const char *s);
int          filter_desktop(const struct dirent *d);
char *       get_desktop_item(const char *f, const char *item);
void         print_name(const char *f);
void         print_exec(const char *f);
void         print_category(const char *f);
void         trim(char **str);
int          startswith(const char *str, const char *start);

static const char *categories[][2] = {
	{ "Graphics", "Graphics" },
	{ "Development", "Development" },
	{ "Settings", "Settings" },
	{ "System", "System" },
	{ "Utility", "Utility" },
	{ "Game", "Game" },
	{ "Network", "Network" },
	{ "Office", "Office" },
	{ "Audio", "Multimedia" },
	{ "Video", "Multimedia" },
};
static const char *misc = "Misc";

int
efclose(FILE *stream)
{
	int ret = 0;


	ret = fclose(stream);
	if (ferror(stream)) {
		err(1,"closefile");
	}
	return ret;
}

FILE *
efopen(const char *path, const char *mode)
{
	FILE *f = NULL;


	if ((f = fopen(path, mode)) == NULL)
		err(1, "%s: %s", "Fail to open", path);

	return f;
}

size_t
esnprintf(char *str, size_t size, const char *fmt, ...)
{
    va_list ap;
    size_t ret = 0;


    va_start(ap, fmt);
    ret = vsnprintf(str, size, fmt, ap);
    if (ret < 0) {
        err(1,"vsnprintf:");
    } else if (ret >= size) {
        err(1,"vsnprintf: Output truncated");
	}

    va_end(ap);

    return ret;
}

char *
estrdup(const char *s)
{
	char *p = strdup(s);


	if (p == NULL)
		err(1, "strdup");

	return p;
}

int 
startswith(const char *str, const char *start)
{
	size_t str_len = strlen(str);
	size_t start_len = strlen(start);


	int ret = 0;
	if ((str_len >= start_len) && 
		(strncmp(str, start, start_len) == 0)) {
		ret = 1;
	}

	return ret;
}

int
endswith(const char *str, const char *end)
{
  size_t str_len = strlen(str);
  size_t end_len = strlen(end);

  int ret = 0;
  if ((str_len >= end_len) &&
    (strcmp(str + (str_len -end_len), end) == 0)) {
    ret = 1;
  }

  return ret;
}

char *
get_desktop_item(const char *f, const char *item)
{
	char *match = NULL;
	char *tmp = NULL;
	char *line = NULL;
	size_t linesize = 0;
	ssize_t linelen;
	FILE *fp;


	fp = efopen(f, "r");

	while ((linelen = getline(&line, &linesize, fp)) != -1) {
		if (startswith(line, item)) {
			tmp = strsep(&line, "=");
			trim(&line);
			match = estrdup(line);
			break;
		}
	}

	free(tmp);
	if (ferror(fp)) {
		err(1, "getline");
	}

	efclose(fp);
	return match;
}

void
print_name(const char *f)
{
	char *res              = NULL;
	char l[3]              = {'\0'};
	char namel[10]         = {'\0'};
	const char *lang       = getenv("LANG");
	
	
	strncpy(l, lang, 2);
	esnprintf(namel, sizeof(namel), "Name[%s]=", l);
	res = get_desktop_item(f, namel);
	if (res == NULL) {
		res = get_desktop_item(f, "Name=");
	}
	printf("%s", res);
	free(res);
}

void
print_category(const char *f)
{
	char *res              = NULL;
	

	res = get_desktop_item(f, "Categories=");
	if (res != NULL) {
		for (size_t k = 0; k < LEN(categories); k++) {
			if (strstr(res, categories[k][0]) != NULL) {
				printf("%s", categories[k][1]);
				break;
			}
		}
	} else {
		printf("%s", misc);
	}
	free(res);
}

void
print_exec(const char *f)
{
	char *res              = NULL;
	char *tmp              = NULL;
	

	/* remove %f and stuff */
	res = get_desktop_item(f, "Exec=");
	tmp = strsep(&res, "%");

	printf("%s", tmp);
	free(tmp);
}

int
filter_desktop(const struct dirent *d)
{
	int ret = 0;
	const char *homedir     = getenv("HOME");
	char exclude[BUFSIZ] = {'\0'};
	char *line = NULL;
	FILE *fp;
	size_t linesize = 0;
	ssize_t linelen;

	/* exclude list is in ~/.config/lsdesktop.exclude */
	esnprintf(exclude, sizeof(exclude), "%s/.config/lsdesktop.exclude", homedir);


	if (endswith(d->d_name, ".desktop")) {
		ret = 1;

		if (access(exclude, F_OK ) != -1 ) {
			fp = efopen(exclude, "r");
			while ((linelen = getline(&line, &linesize, fp)) != -1) {
				trim(&line);
				if (strcmp(d->d_name, line) == 0) {
					ret = 0;
					break;
				}
			}

			free(line);
			if (ferror(fp)) {
				err(1, "getline");
			}

			efclose(fp);
		}
	} 

	return ret;
}

/* return a trimmed copy of str */
/* trim(&str); */
void
trim(char **str)
{
    size_t begin = 0;
    size_t end = strlen(*str);


    while (isspace((*str)[begin])) {
        begin++;
    }

    while (isspace((*str)[end-1])) {
        end--;
    }
    for (size_t i = begin, j=0; i < end; i++, j++) {
        (*str)[j] = (*str)[i];
    }
    (*str)[end - begin] = '\0';
}

int 
main (int argc, char *argv[])
{
#ifdef __OpenBSD__
	PLEDGEORDIE("stdio rpath cpath wpath");
#endif

	int n                   = 0;
	char *sep               = "|";
	char *item              = NULL;
	char homedirapp[BUFSIZ] = {'\0'};
	char desktop[BUFSIZ]    = {'\0'};
	const char *homedir     = getenv("HOME");
	struct dirent **namelist;


	if (argc == 2) {
		if (strcmp(argv[1], "-h") == 0) {
			printf("usage: %s [-h|-s]\n", argv[0]);
			printf("    -h: show this message\n");
			printf("    -s: set separator string between name and command\n\n");
			printf("every filename listed in ~/.config/lsdesktop.exclude"
				" will be ignored (one per line)\n");
			exit(0);
		}
	} else if (argc == 3) {
		sep = argv[2];
	}

	esnprintf(homedirapp, sizeof(homedirapp), "%s/.local/share/applications", homedir);
	/* .desktop files directories */
	const char *dfd[] = { 
		"/usr/local/share/applications",
		"/usr/share/applications", 
		homedirapp
	};

	for (size_t i = 0; i < LEN(dfd); i++) {
		if ((n = scandir(dfd[i], &namelist, filter_desktop, NULL)) < 0) {
			warn("scandir:%s", dfd[i]);
		} else {
			for(int j = 0; j < n; j++) {
				esnprintf(desktop, sizeof(desktop), "%s/%s",
					dfd[i], namelist[j]->d_name);

				print_category(desktop);
				printf("%s", sep);
				print_name(desktop);
				printf("%s", sep);
				print_exec(desktop);
				printf("\n");
				free(namelist[j]);
			}
			free(namelist);
		}
	}

	return 0;
}