/* Copyright (C) 2021  C. McEnroe <june@causal.agency>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <ctype.h>
#include <err.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <unistd.h>

static char *deregex(const char *patt) {
	char *buf = malloc(strlen(patt) + 1);
	if (!buf) err(EX_OSERR, "malloc");
	char *ptr = buf;
	if (*patt == '^') patt++;
	for (; *patt; ++patt) {
		if (patt[0] == '$' && !patt[1]) {
			*ptr++ = '\n';
			break;
		}
		if (patt[0] == '\\' && patt[1]) patt++;
		*ptr++ = *patt;
	}
	*ptr = '\0';
	return buf;
}

static size_t escape(bool esc, const char *ptr, size_t len) {
	if (!esc) {
		fwrite(ptr, len, 1, stdout);
		return len;
	}
	for (size_t i = 0; i < len; ++i) {
		switch (ptr[i]) {
			break; case '&': printf("&amp;");
			break; case '<': printf("&lt;");
			break; case '"': printf("&quot;");
			break; default:  putchar(ptr[i]);
		}
	}
	return len;
}

static void id(const char *tag) {
	for (const char *ch = tag; *ch; ++ch) {
		if (isalnum(*ch) || strchr("-._", *ch)) {
			putchar(*ch);
		} else {
			putchar('_');
		}
	}
}

static char *hstrstr(const char *haystack, const char *needle) {
	while (haystack) {
		char *elem = strchr(haystack, '<');
		char *match = strstr(haystack, needle);
		if (!match) return NULL;
		if (!elem || match < elem) return match;
		haystack = strchr(elem, '>');
	}
	return NULL;
}

int main(int argc, char *argv[]) {
	bool pre = false;
	bool pipe = false;
	bool main = false;
	bool index = false;
	const char *tagsPath = "tags";
	for (int opt; 0 < (opt = getopt(argc, argv, "f:impx"));) {
		switch (opt) {
			break; case 'f': tagsPath = optarg;
			break; case 'i': pipe = true;
			break; case 'm': main = true;
			break; case 'p': pre = true;
			break; case 'x': index = true;
			break; default:  return EX_USAGE;
		}
	}
	if (optind == argc) errx(EX_USAGE, "name required");
	const char *name = argv[optind];

	FILE *file = fopen(name, "r");
	if (!file) err(EX_NOINPUT, "%s", name);

	FILE *tagsFile = fopen(tagsPath, "r");
	if (!tagsFile) err(EX_NOINPUT, "%s", tagsPath);

#ifdef __OpenBSD__
	int error = pledge("stdio", NULL);
	if (error) err(EX_OSERR, "pledge");
#endif

	size_t len = 0;
	size_t cap = 256;
	struct Tag {
		char *tag;
		int num;
		char *str;
		size_t len;
	} *tags = malloc(cap * sizeof(*tags));
	if (!tags) err(EX_OSERR, "malloc");

	char *buf = NULL;
	size_t bufCap = 0;
	while (0 < getline(&buf, &bufCap, tagsFile)) {
		char *line = buf;
		char *tag = strsep(&line, "\t");
		char *file = strsep(&line, "\t");
		char *def = strsep(&line, "\n");
		if (!tag || !file || !def) errx(EX_DATAERR, "malformed tags file");

		if (strcmp(file, name)) continue;
		if (len == cap) {
			tags = realloc(tags, (cap *= 2) * sizeof(*tags));
			if (!tags) err(EX_OSERR, "realloc");
		}
		tags[len].tag = strdup(tag);
		if (!tags[len].tag) err(EX_OSERR, "strdup");

		tags[len].num = 0;
		if (def[0] == '/' || def[0] == '?') {
			def++;
			def[strlen(def)-1] = '\0';
			if (def[0] != '^') {
				warnx("unanchored regex for tag %s: %s", tag, def);
			}
			tags[len].str = deregex(def);
			tags[len].len = strlen(tags[len].str);
		} else {
			tags[len].num = strtol(def, &def, 10);
			if (*def) {
				warnx("invalid line number for tag %s: %s", tag, def);
				continue;
			}
		}
		len++;
	}
	fclose(tagsFile);

	int num = 0;
	printf(pre ? "<pre>" : index ? "<ul class=\"index\">\n" : "");
	while (0 < getline(&buf, &bufCap, file) && ++num) {
		char *tag = NULL;
		for (size_t i = 0; i < len; ++i) {
			if (tags[i].num) {
				if (num != tags[i].num) continue;
			} else {
				if (strncmp(tags[i].str, buf, tags[i].len)) continue;
			}
			tag = tags[i].tag;
			tags[i] = tags[--len];
			break;
		}
		if (index) {
			if (!tag) continue;
			printf("<li><a class=\"tag\" href=\"#");
			id(tag);
			printf("\">");
			escape(true, tag, strlen(tag));
			printf("</a></li>\n");
			continue;
		}
		if (pipe) {
			ssize_t len = getline(&buf, &bufCap, stdin);
			if (len < 0) {
				errx(EX_DATAERR, "missing line %d on standard input", num);
			}
		}
		if (!tag) {
			escape(!pipe, buf, strlen(buf));
			continue;
		}

		size_t mlen = strlen(tag);
		char *match = (pipe ? hstrstr : strstr)(buf, tag);
		while (match > buf && isalnum(match[-1])) {
			match = (pipe ? hstrstr : strstr)(&match[mlen], tag);
		}
		if (!match && tag[0] == 'M') {
			mlen = 4;
			match = (pipe ? hstrstr : strstr)(buf, "main");
			if (main) tag = "main";
		}
		if (!match) {
			mlen = strlen(buf) - 1;
			match = buf;
		}
		escape(!pipe, buf, match - buf);
		printf("<a class=\"tag\" id=\"");
		id(tag);
		printf("\" href=\"#");
		id(tag);
		printf("\">");
		match += escape(!pipe, match, mlen);
		printf("</a>");
		escape(!pipe, match, strlen(match));
	}
	printf(pre ? "</pre>" : index ? "</ul>\n" : "");
}