/* Copyright (C) 2020  C. McEnroe <june@causal.agency>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * Additional permission under GNU GPL version 3 section 7:
 *
 * If you modify this Program, or any covered work, by linking or
 * combining it with OpenSSL (or a modified version of that library),
 * containing parts covered by the terms of the OpenSSL License and the
 * original SSLeay license, the licensors of this Program grant you
 * additional permission to convey the resulting work. Corresponding
 * Source for a non-source form of such a combination shall include the
 * source code for the parts of OpenSSL used as well as that of the
 * covered work.
 */

#include <assert.h>
#include <err.h>
#include <regex.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <time.h>

#include "archive.h"

static char *htmlMailto(struct Address addr) {
	struct Variable vars[] = {
		{ "mailbox", addr.mailbox },
		{ "host", addr.host },
		{0},
	};
	return templateString("mailto:[mailbox]@[host]", vars, escapeURL);
}

static int htmlAddress(FILE *file, struct Address addr, bool last) {
	char *mailto = htmlMailto(addr);
	const char *template;
	if (addr.host) {
		template = Q(
			<li><a href="[mailto]">[name]</a>[,]</li>
		);
	} else if (addr.mailbox) {
		template = Q(
			<li>[mailbox]:
				<ul>
		);
	} else {
		template = Q(
				</ul>;
			</li>
		);
	}
	struct Variable vars[] = {
		{ "mailto", mailto },
		{ "name", addressName(addr) },
		{ "mailbox", addr.mailbox },
		{ ",", (last ? "" : ", ") },
		{0},
	};
	int error = templateRender(file, template, vars, escapeXML);
	free(mailto);
	return error;
}

static int
htmlAddressList(FILE *file, const char *name, struct AddressList list) {
	if (!list.len) return 0;
	const char *template = Q(
		<div class="[name]">
			[name]:
			<ul class="address">
	);
	struct Variable vars[] = {
		{ "name", name },
		{0},
	};
	int error = templateRender(file, template, vars, escapeXML);
	if (error) return error;
	for (size_t i = 0; i < list.len; ++i) {
		error = htmlAddress(file, list.addrs[i], i == list.len - 1);
		if (error) return error;
	}
	return templateRender(file, Q(</ul></div>), NULL, NULL);
}

static char *htmlReply(const struct Envelope *envelope) {
	const char *template = {
		"mailto:[mailbox]@[host]"
		"?cc=[cc]"
		"&subject=[re][subject]"
		"&In-Reply-To=[<][messageID][>]"
	};
	struct Variable vars[] = {
		{ "mailbox", envelope->replyTo.mailbox },
		{ "host", envelope->replyTo.host },
		{ "cc", baseMailto },
		{ "re", (strncmp(envelope->subject, "Re: ", 4) ? "Re: " : "") },
		{ "subject", envelope->subject },
		{ "messageID", envelope->messageID },
		{ "<", "<" },
		{ ">", ">" },
		{0},
	};
	return templateString(template, vars, escapeURL);
}

static char *htmlFragment(const char *messageID) {
	struct Variable vars[] = {
		{ "messageID", messageID },
		{0},
	};
	return templateString("#[messageID]", vars, escapeURL);
}

static char *htmlMbox(const char *messageID) {
	struct Variable vars[] = {
		{ "messageID", messageID },
		{ "type", "mbox" },
		{0},
	};
	return templateString("../" PATH_MESSAGE, vars, escapeURL);
}

static int
htmlNavItem(FILE *file, const char *name, const char *base, const char *url) {
	const char *template = Q(<li><a href="[base][url]">[name]</a> </li>);
	struct Variable vars[] = {
		{ "name", name },
		{ "base", base },
		{ "url", url },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

static int htmlParent(FILE *file, const struct Envelope *envelope) {
	if (!envelope->inReplyTo) return 0;
	char *fragment = htmlFragment(envelope->inReplyTo);
	int error = htmlNavItem(file, "parent", "", fragment);
	free(fragment);
	return error;
}

int htmlMessageNav(FILE *file, const struct Envelope *envelope) {
	char *mbox = htmlMbox(envelope->messageID);
	char *reply = htmlReply(envelope);
	int error = 0
		|| templateRender(file, Q(<nav><ul>), NULL, NULL)
		|| htmlParent(file, envelope)
		|| htmlNavItem(file, "download", "", mbox)
		|| htmlNavItem(file, "reply", "", reply)
		|| templateRender(file, Q(</ul></nav>), NULL, NULL);
	free(mbox);
	free(reply);
	return error;
}

int htmlMessageOpen(FILE *file, const struct Envelope *envelope) {
	char *fragment = htmlFragment(envelope->messageID);
	char *mailto = htmlMailto(envelope->from);
	const char *template = Q(
		<article class="message" id="[messageID]">
		<header>
			<h2 class="Subject"><a href="[fragment]">[subject]</a></h2>
			<div class="From">
				From: <a href="[mailto]">[from]</a>
				<time datetime="[utc]">[date]</time>
			</div>
	);
	struct Variable vars[] = {
		{ "messageID", envelope->messageID },
		{ "fragment", fragment },
		{ "subject", envelope->subject },
		{ "mailto", mailto },
		{ "from", addressName(envelope->from) },
		{ "utc", iso8601(envelope->time).s },
		{ "date", envelope->date },
		{0},
	};
	int error = 0
		|| templateRender(file, template, vars, escapeXML)
		|| htmlAddressList(file, "To", envelope->to)
		|| htmlAddressList(file, "Cc", envelope->cc)
		|| htmlMessageNav(file, envelope)
		|| templateRender(file, Q(</header>), NULL, NULL);
	free(fragment);
	free(mailto);
	return error;
}

static int htmlInlineAttrs(FILE *file, const struct BodyPart *part) {
	const char *template = " " Q([attr]="[value]");
	if (part->contentID) {
		struct Variable vars[] = {
			{ "attr", "id" },
			{ "value", part->contentID },
			{0},
		};
		int error = templateRender(file, template, vars, escapeXML);
		if (error) return error;
	}
	if (part->description) {
		struct Variable vars[] = {
			{ "attr", "title" },
			{ "value", part->description },
			{0},
		};
		int error = templateRender(file, template, vars, escapeXML);
		if (error) return error;
	}
	const char *language = NULL;
	if (part->language.type == String) language = part->language.string;
	if (part->language.type == List && part->language.list.len == 1) {
		language = dataCheck(part->language.list.ptr[0], String).string;
	}
	if (language) {
		struct Variable vars[] = {
			{ "attr", "lang" },
			{ "value", language },
			{0},
		};
		int error = templateRender(file, template, vars, escapeXML);
		if (error) return error;
	}
	return 0;
}

static void swap(char *a, char *b) {
	char ch = *a;
	*a = *b;
	*b = ch;
}

static int htmlMarkupURLs(FILE *file, char *buf) {
	static const char *Pattern = "(^|[[:space:]<])(https?:[^[:space:]>]+)(.|$)";
	static regex_t regex;
	if (!regex.re_nsub) {
		int error = regcomp(&regex, Pattern, REG_EXTENDED);
		assert(!error);
	}

	int error;
	char *ptr;
	regmatch_t match[4];
	for (ptr = buf; !regexec(&regex, ptr, 4, match, 0); ptr += match[2].rm_eo) {
		char nul = '\0';

		swap(&ptr[match[2].rm_so], &nul);
		error = escapeXML(file, ptr);
		if (error) return error;
		swap(&ptr[match[2].rm_so], &nul);

		const char *template = Q(<a href="[url]">[url]</a>);
		swap(&ptr[match[3].rm_so], &nul);
		struct Variable vars[] = {
			{ "url", &ptr[match[2].rm_so] },
			{0},
		};
		error = templateRender(file, template, vars, escapeXML);
		if (error) return error;
		swap(&ptr[match[3].rm_so], &nul);
	}
	return escapeXML(file, ptr);
}

static int htmlMarkupQuote(FILE *file, char *buf) {
	int error;
	size_t level = 0;
	for (char *ch = buf; *ch == '>' || *ch == ' '; level += (*ch++ == '>'));
	for (size_t i = 0; i < level; ++i) {
		error = templateRender(file, Q(<q>), NULL, NULL);
		if (error) return error;
	}
	error = htmlMarkupURLs(file, buf);
	if (error) return error;
	for (size_t i = 0; i < level; ++i) {
		error = templateRender(file, Q(</q>), NULL, NULL);
		if (error) return error;
	}
	return 0;
}

static int htmlMarkup(FILE *file, const char *content) {
	int error = 0;
	size_t cap = 0;
	char *buf = NULL;
	bool patch = false;
	for (const char *nl; (nl = strchr(content, '\n')); content = &nl[1]) {
		size_t len = nl - content;
		if (cap < len + 1) {
			cap = len + 1;
			buf = realloc(buf, cap);
			if (!buf) err(EX_OSERR, "realloc");
		}
		memcpy(buf, content, len);
		buf[len] = '\0';
		struct Variable vars[] = {
			{ "line", buf },
			{0},
		};

		if (!strcmp(buf, "---")) {
			patch = true;
		} else if (patch && !strcmp(buf, "-- ")) {
			patch = false;
		}

		static const char *Pattern = "^(diff|index|---|[+]{3}|@@) ";
		static regex_t regex;
		if (!regex.re_nsub) {
			error = regcomp(&regex, Pattern, REG_EXTENDED);
			assert(!error);
		}
		if (patch && !regexec(&regex, buf, 0, NULL, 0)) {
			error = templateRender(
				file, Q(<b>[line]</b>), vars, escapeXML
			);
		} else if (patch && buf[0] == '-' && strcmp(buf, "---")) {
			error = templateRender(
				file, Q(<del>[line]</del>), vars, escapeXML
			);
		} else if (patch && buf[0] == '+') {
			error = templateRender(
				file, Q(<ins>[line]</ins>), vars, escapeXML
			);
		} else if (patch) {
			error = escapeXML(file, buf);
		} else if (buf[0] == '>') {
			error = htmlMarkupQuote(file, buf);
		} else {
			error = htmlMarkupURLs(file, buf);
		}
		if (error) break;

		error = templateRender(file, "\n", NULL, NULL);
		if (error) break;
	}
	free(buf);
	return error;
}

int htmlInline(FILE *file, const struct BodyPart *part, const char *content) {
	return 0
		|| templateRender(file, Q(<pre), NULL, NULL)
		|| htmlInlineAttrs(file, part)
		|| templateRender(file, Q(>), NULL, NULL)
		|| htmlMarkup(file, content)
		|| templateRender(file, Q(</pre>), NULL, NULL);
}

int htmlAttachmentOpen(FILE *file) {
	return templateRender(file, Q(<ul class="attachment">), NULL, NULL);
}

int htmlAttachment(
	FILE *file, const struct BodyPart *part, const struct Variable *path
) {
	char *url = templateString("../" PATH_ATTACHMENT, path, escapeURL);
	const char *name = paramGet(part->disposition.params, "filename");
	if (!name) name = paramGet(part->params, "name");
	const char *template = Q(
		<li><a href="[url]">[name][type][/][subtype]</a> </li>
	);
	struct Variable vars[] = {
		{ "url", url },
		{ "name", (name ? name : "") },
		{ "type", (name ? "" : part->type) },
		{ "/", (name ? "" : "/") },
		{ "subtype", (name ? "" : part->subtype) },
		{0},
	};
	int error = templateRender(file, template, vars, escapeXML);
	free(url);
	return error;
}

int htmlAttachmentClose(FILE *file) {
	return templateRender(file, Q(</ul>), NULL, NULL);
}

int htmlMessageClose(FILE *file) {
	return templateRender(file, Q(</article>), NULL, NULL);
}

static char *htmlThreadURL(const struct Envelope *envelope, const char *type) {
	struct Variable vars[] = {
		{ "messageID", envelope->messageID },
		{ "type", type },
		{0},
	};
	return templateString("../" PATH_THREAD, vars, escapeURL);
}

int htmlThreadHead(FILE *file, const struct Envelope *envelope) {
	char *atom = htmlThreadURL(envelope, "atom");
	char *mbox = htmlThreadURL(envelope, "mbox");
	const char *template = Q(
		<!DOCTYPE html>
		<meta charset="utf-8">
		<meta name="generator" content="[generator]">
		<title>[subject]</title>
		<link rel="alternate" type="application/atom+xml" href="[atom]">
		<link rel="alternate" type="application/mbox" href="[mbox]">
	);
	struct Variable vars[] = {
		{ "generator", GENERATOR_URL },
		{ "subject", envelope->subject },
		{ "atom", atom },
		{ "mbox", mbox },
		{0},
	};
	int error = templateRender(file, template, vars, escapeXML);
	free(atom);
	free(mbox);
	return error;
}

int htmlThreadOpen(FILE *file, const struct Envelope *envelope) {
	char *atom = htmlThreadURL(envelope, "atom");
	char *mbox = htmlThreadURL(envelope, "mbox");
	const char *template = Q(
		<header class="thread">
			<h1>[subject]</h1>
			<nav>
				<ul>
					<li><a href="../index.html">index</a> </li>
					<li><a href="[atom]">follow</a> </li>
					<li><a href="[mbox]">download</a> </li>
				</ul>
			</nav>
		</header>
		<main class="thread">
	);
	struct Variable vars[] = {
		{ "subject", envelope->subject },
		{ "atom", atom },
		{ "mbox", mbox },
		{0},
	};
	int error = templateRender(file, template, vars, escapeXML);
	free(atom);
	free(mbox);
	return error;
}

static uint32_t threadCount(struct List thread) {
	uint32_t count = 0;
	for (size_t i = 0; i < thread.len; ++i) {
		if (thread.ptr[i].type == List) {
			count += threadCount(thread.ptr[i].list);
		} else {
			count++;
		}
	}
	return count;
}

static int htmlReplies(FILE *file, uint32_t replies) {
	const char *template = Q(
		<data class="replies" value="[replies]">[replies] repl[ies]</data>
	);
	struct Variable vars[] = {
		{ "replies", u32(replies).s },
		{ "ies", (replies != 1 ? "ies" : "y") },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

int htmlSubthreadOpen(FILE *file, struct List thread) {
	const char *template = Q(
		<details class="subthread" open>
			<summary>
	);
	return 0
		|| templateRender(file, template, NULL, NULL)
		|| htmlReplies(file, threadCount(thread))
		|| templateRender(file, Q(</summary>), NULL, NULL);
}

int htmlSubthreadClose(FILE *file) {
	return templateRender(file, Q(</details>), NULL, NULL);
}

static int htmlFooter(FILE *file) {
	const char *template = Q(
		<footer>
			<a href="[generator]">generated</a>
			<time datetime="[time]">[time]</time>
		</footer>
	);
	struct Variable vars[] = {
		{ "generator", GENERATOR_URL },
		{ "time", iso8601(time(NULL)).s },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

int htmlThreadClose(FILE *file) {
	return 0
		|| templateRender(file, Q(</main>), NULL, NULL)
		|| htmlFooter(file);
}

int htmlIndexHead(FILE *file) {
	const char *template = Q(
		<!DOCTYPE html>
		<meta charset="utf-8">
		<meta name="generator" content="[generator]">
		<title>[title]</title>
		<link rel="alternate" type="application/atom+xml" href="index.atom">
	);
	struct Variable vars[] = {
		{ "generator", GENERATOR_URL },
		{ "title", baseTitle },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

static int htmlIndexNav(FILE *file) {
	int error = 0
		|| templateRender(file, Q(<nav><ul>), NULL, NULL)
		|| htmlNavItem(file, "follow", "", "index.atom");
	if (error) return error;
	if (baseSubscribe) {
		error = htmlNavItem(file, "subscribe", "", baseSubscribe);
		if (error) return error;
	}
	return 0
		|| htmlNavItem(file, "write", "mailto:", baseMailto)
		|| templateRender(file, Q(</ul></nav>), NULL, NULL);
}

int htmlIndexOpen(FILE *file) {
	const char *head = Q(
		<header class="index">
			<h1>[title]</h1>
	);
	const char *tail = Q(
		</header>
		<main class="index">
			<ol>
	);
	struct Variable vars[] = {
		{ "title", baseTitle },
		{0},
	};
	return 0
		|| templateRender(file, head, vars, escapeXML)
		|| htmlIndexNav(file)
		|| templateRender(file, tail, vars, escapeXML);
}

static char *htmlIndexURL(const struct Envelope *envelope) {
	struct Variable vars[] = {
		{ "messageID", envelope->messageID },
		{ "type", "html" },
		{0},
	};
	return templateString(PATH_THREAD, vars, escapeURL);
}

int htmlIndexThread(
	FILE *file, const struct Envelope *envelope, struct List thread
) {
	char *url = htmlIndexURL(envelope);
	const char *template = Q(
		<li>
			<h2 class="Subject"><a href="[url]">[subject]</a></h2>
			<div class="From">
				From: [from]
				<time datetime="[utc]">[date]</time>
			</div>
	);
	struct Variable vars[] = {
		{ "url", url },
		{ "subject", envelope->subject },
		{ "from", addressName(envelope->from) },
		{ "utc", iso8601(envelope->time).s },
		{ "date", envelope->date },
		{0},
	};
	int error = 0
		|| templateRender(file, template, vars, escapeXML)
		|| htmlReplies(file, threadCount(thread) - 1)
		|| templateRender(file, Q(</li>), NULL, NULL);
	free(url);
	return error;
}

int htmlIndexClose(FILE *file) {
	return 0
		|| templateRender(file, Q(</ol></main>), NULL, NULL)
		|| htmlFooter(file);
}