/* 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/>.
 */

#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <sysexits.h>
#include <time.h>

#include "archive.h"

static char *atomID(const struct Envelope *envelope) {
	struct Variable vars[] = {
		{ "messageID", envelope->messageID },
		{0},
	};
	return templateURL("mid:[messageID]", vars);
}

static int atomAuthor(FILE *file, struct Address addr) {
	const char *template = TEMPLATE(
		<author>
			<name>[name]</name>
			<email>[mailbox]@[host]</email>
		</author>
	);
	struct Variable vars[] = {
		{ "name", addressName(addr) },
		{ "mailbox", addr.mailbox },
		{ "host", addr.host },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

static char *atomEntryURL(const struct Envelope *envelope) {
	struct Variable vars[] = {
		{ "base", baseURL },
		{ "messageID", envelope->messageID },
		{ "type", "mbox" },
		{0},
	};
	return templateURL("[base]/" PATH_MESSAGE, vars);
}

int atomEntryOpen(FILE *file, const struct Envelope *envelope) {
	const char *template = TEMPLATE(
		<entry>
		<id>[id]</id>
		<title>[title]</title>
		<updated>[updated]</updated>
		<link rel="alternate" type="application/mbox" href="[url]"/>
	);
	char *id = atomID(envelope);
	char updated[sizeof("0000-00-00T00:00:00Z")];
	strftime(updated, sizeof(updated), "%FT%TZ", gmtime(&envelope->time));
	char *url = atomEntryURL(envelope);
	struct Variable vars[] = {
		{ "id", id },
		{ "title", envelope->subject },
		{ "updated", updated },
		{ "url", url },
		{0},
	};
	int error = 0
		|| templateRender(file, template, vars, escapeXML)
		|| atomAuthor(file, envelope->from);
	free(id);
	free(url);
	return error;
}

int atomContent(FILE *file, const char *content) {
	const char *template = TEMPLATE(
		<content type="xhtml">
			<div xmlns="http://www.w3.org/1999/xhtml">
				<pre>[content]</pre>
			</div>
		</content>
	);
	struct Variable vars[] = {
		{ "content", content },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

int atomEntryClose(FILE *file) {
	return templateRender(file, TEMPLATE(</entry>), NULL, NULL);
}

static char *atomThreadURL(const struct Envelope *envelope, const char *type) {
	struct Variable vars[] = {
		{ "base", baseURL },
		{ "messageID", envelope->messageID },
		{ "type", type },
		{0},
	};
	return templateURL("[base]/" PATH_THREAD, vars);
}

int atomThreadOpen(FILE *file, const struct Envelope *envelope) {
	const char *template = TEMPLATE(
		<[q]xml version="1.0" encoding="utf-8"[q]>
		<feed xmlns="http://www.w3.org/2005/Atom">
		<id>[id]</id>
		<title>[title]</title>
		<updated>[updated]</updated>
		<link rel="self" href="[atom]"/>
		<link rel="alternate" type="text/html" href="[html]"/>
		<link rel="alternate" type="application/mbox" href="[mbox]"/>
	);
	char *id = atomID(envelope);
	time_t now = time(NULL);
	char updated[sizeof("0000-00-00T00:00:00Z")];
	strftime(updated, sizeof(updated), "%FT%TZ", gmtime(&now));
	char *atom = atomThreadURL(envelope, "atom");
	char *html = atomThreadURL(envelope, "html");
	char *mbox = atomThreadURL(envelope, "mbox");
	struct Variable vars[] = {
		{ "q", "?" },
		{ "id", id },
		{ "title", envelope->subject },
		{ "updated", updated },
		{ "atom", atom },
		{ "html", html },
		{ "mbox", mbox },
		{0},
	};
	int error = 0
		|| templateRender(file, template, vars, escapeXML)
		|| atomAuthor(file, envelope->from);
	free(id);
	free(atom);
	free(html);
	free(mbox);
	return error;
}

int atomThreadClose(FILE *file) {
	return templateRender(file, TEMPLATE(</feed>), NULL, NULL);
}

int atomIndexOpen(FILE *file) {
	const char *template = TEMPLATE(
		<[q]xml version="1.0" encoding="utf-8"[q]>
		<feed xmlns="http://www.w3.org/2005/Atom">
		<id>mailto:[addr]</id>
		<title>[title]</title>
		<updated>[updated]</updated>
		<link rel="self" href="[base]/index.atom"/>
		<link rel="alternate" type="text/html" href="[base]/index.html"/>
	);
	time_t now = time(NULL);
	char updated[sizeof("0000-00-00T00:00:00Z")];
	strftime(updated, sizeof(updated), "%FT%TZ", gmtime(&now));
	struct Variable vars[] = {
		{ "q", "?" },
		{ "addr", baseAddress },
		{ "base", (baseURL ? baseURL : "") },
		{ "title", baseTitle },
		{ "updated", updated },
		{0},
	};
	return templateRender(file, template, vars, escapeXML);
}

int atomIndexClose(FILE *file) {
	return templateRender(file, TEMPLATE(</feed>), NULL, NULL);
}