#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <time.h>

#include "util.h"
#include "xml.h"

#define ISINCONTENT(ctx)  ((ctx).iscontent && !((ctx).iscontenttag))
#define ISCONTENTTAG(ctx) (!((ctx).iscontent) && (ctx).iscontenttag)
/* string and size */
#define STRP(s)           s,sizeof(s)-1
/* length of string */
#define STRSIZ(s)         (sizeof(s)-1)

enum FeedType { FeedTypeNone = 0, FeedTypeRSS = 1, FeedTypeAtom = 2 };
static const char *feedtypes[] = { "", "rss", "atom" };

enum ContentType { ContentTypeNone = 0, ContentTypePlain = 1, ContentTypeHTML = 2 };
static const char *contenttypes[] = { "", "plain", "html" };

static const int FieldSeparator = '\t'; /* output field seperator character */
static const char *baseurl = "";

enum TagId {
	TagUnknown = 0,
	/* RSS */
	RSSTagDcdate, RSSTagPubdate, RSSTagTitle,
	RSSTagLink, RSSTagDescription, RSSTagContentencoded,
	RSSTagGuid, RSSTagAuthor, RSSTagDccreator,
	/* Atom */
	AtomTagPublished, AtomTagUpdated, AtomTagTitle,
	AtomTagSummary, AtomTagContent,
	AtomTagId, AtomTagLink, AtomTagAuthor
};

/* String data / memory pool */
typedef struct string {
	char   *data;   /* data */
	size_t  len;    /* string length */
	size_t  bufsiz; /* allocated size */
} String;

/* Feed item */
typedef struct feeditem {
	String timestamp;
	String title;
	String link;
	String content;
	int    contenttype; /* ContentTypePlain or ContentTypeHTML */
	String id;
	String author;
	int    feedtype;    /* FeedTypeRSS or FeedTypeAtom */
} FeedItem;

typedef struct feedtag {
	char       *name;
	size_t      namelen;
	enum TagId  id;
} FeedTag;

typedef struct feedcontext {
	String   *field;        /* pointer to current FeedItem field String */
	FeedItem  item;         /* data for current feed item */
	char      tag[256];     /* current tag _inside_ a feeditem */
	int       tagid;        /* unique number for parsed tag (faster comparison) */
	size_t    taglen;
	int       iscontent;    /* in content data */
	int       iscontenttag; /* in content tag */
	int       attrcount;
} FeedContext;

static enum TagId gettag(enum FeedType, const char *, size_t);
static int    gettimetz(const char *, char *, size_t, int *);
static int    isattr(const char *, size_t, const char *, size_t);
static int    istag(const char *, size_t, const char *, size_t);
static int    parsetime(const char *, char *, size_t, time_t *);
static void   printfields(void);
static void   string_append(String *, const char *, size_t);
static void   string_buffer_init(String *, size_t);
static void   string_buffer_realloc(String *, size_t);
static void   string_clear(String *);
static void   string_print_encoded(String *);
static void   string_print_trimmed(String *);
static void   xml_handler_attr(XMLParser *, const char *, size_t,
                               const char *, size_t, const char *, size_t);
static void   xml_handler_attr_start(XMLParser *, const char *, size_t,
                                     const char *, size_t);
static void   xml_handler_attr_end(struct xmlparser *, const char *, size_t,
                                   const char *, size_t);
static void   xml_handler_cdata(XMLParser *, const char *, size_t);
static void   xml_handler_data(XMLParser *, const char *, size_t);
static void   xml_handler_data_entity(XMLParser *, const char *, size_t);
static void   xml_handler_end_element(XMLParser *, const char *, size_t, int);
static void   xml_handler_start_element(XMLParser *, const char *, size_t);
static void   xml_handler_start_element_parsed(XMLParser *, const char *,
                                               size_t, int);

static FeedContext ctx;
static XMLParser parser; /* XML parser state */

/* unique number for parsed tag (faster comparison) */
static enum TagId
gettag(enum FeedType feedtype, const char *name, size_t namelen)
{
	/* RSS, alphabetical order */
	static FeedTag rsstag[] = {
		{ STRP("author"),          RSSTagAuthor         },
		{ STRP("content:encoded"), RSSTagContentencoded },
		{ STRP("dc:creator"),      RSSTagDccreator      },
		{ STRP("dc:date"),         RSSTagDcdate         },
		{ STRP("description"),     RSSTagDescription    },
		{ STRP("guid"),            RSSTagGuid           },
		{ STRP("link"),            RSSTagLink           },
		{ STRP("pubdate"),         RSSTagPubdate        },
		{ STRP("title"),           RSSTagTitle          },
		{ NULL, 0, -1 }
	};
	/* Atom, alphabetical order */
	static FeedTag atomtag[] = {
		{ STRP("author"),    AtomTagAuthor    },
		{ STRP("content"),   AtomTagContent   },
		{ STRP("id"),        AtomTagId        },
		{ STRP("link"),      AtomTagLink      },
		{ STRP("published"), AtomTagPublished },
		{ STRP("summary"),   AtomTagSummary   },
		{ STRP("title"),     AtomTagTitle     },
		{ STRP("updated"),   AtomTagUpdated   },
		{ NULL, 0, -1 }
	};
	int i, n;

	/* optimization: these are always non-matching */
	if (namelen < 2 || namelen > 15)
		return TagUnknown;

	switch (feedtype) {
	case FeedTypeRSS:
		for (i = 0; rsstag[i].name; i++) {
			if (!(n = strncasecmp(rsstag[i].name, name, rsstag[i].namelen)))
				return rsstag[i].id; /* found */
			/* optimization: it's sorted so nothing after it matches. */
			if (n > 0)
				return TagUnknown;
		}
		break;
	case FeedTypeAtom:
		for (i = 0; atomtag[i].name; i++) {
			if (!(n = strncasecmp(atomtag[i].name, name, atomtag[i].namelen)))
				return atomtag[i].id; /* found */
			/* optimization: it's sorted so nothing after it matches. */
			if (n > 0)
				return TagUnknown;
		}
		break;
	default:
		return TagUnknown;
	}
	return TagUnknown; /* NOTREACHED */
}

/* clear string only; don't free, prevents unnecessary reallocation */
static void
string_clear(String *s)
{
	if (s->data)
		s->data[0] = '\0';
	s->len = 0;
}

static void
string_buffer_init(String *s, size_t len)
{
	if (!(s->data = malloc(len)))
		err(1, "malloc");
	s->bufsiz = len;
	string_clear(s);
}

static void
string_buffer_realloc(String *s, size_t newlen)
{
	char *p;
	size_t alloclen;

	for (alloclen = 16; alloclen <= newlen; alloclen *= 2)
		;
	if (!(p = realloc(s->data, alloclen)))
		err(1, "realloc");
	s->bufsiz = alloclen;
	s->data = p;
}

static void
string_append(String *s, const char *data, size_t len)
{
	if (!len || *data == '\0')
		return;
	/* check if allocation is necesary, don't shrink buffer
	   should be more than bufsiz ofcourse */
	if (s->len + len >= s->bufsiz)
		string_buffer_realloc(s, s->len + len + 1);
	memcpy(s->data + s->len, data, len);
	s->len += len;
	s->data[s->len] = '\0';
}

/* get timezone from string, return as formatted string and time offset,
 * for the offset it assumes UTC.
 * NOTE: only parses timezones in RFC-822, other timezones are ambiguous
 * anyway. If needed you can add ones yourself, like "cest", "cet" etc. */
static int
gettimetz(const char *s, char *buf, size_t bufsiz, int *tzoffset)
{
	static struct tzone {
		char *name;
		int offhour;
		int offmin;
	} tzones[] = {
		{ "CDT", -5, 0 },
		{ "CST", -6, 0 },
		{ "EDT", -4, 0 },
		{ "EST", -5, 0 },
		{ "GMT",  0, 0 },
		{ "MDT", -6, 0 },
		{ "MST", -7, 0 },
		{ "PDT", -7, 0 },
		{ "PST", -8, 0 },
		{ "UT",   0, 0 },
		{ "UTC",  0, 0 },
		{ "A",   -1, 0 },
		{ "M",  -12, 0 },
		{ "N",    1, 0 },
		{ "Y",   12, 0 },
		{ "Z",    0, 0 }
	};
	char tzbuf[5] = "", *tz = "", c = '+';
	int tzhour = 0, tzmin = 0, r;
	size_t i;

	if (!*s || *s == 'Z' || *s == 'z')
		goto time_ok;
	/* skip whitespace */
	s = &s[strspn(s, " 	")];

	/* look until some common timezone delimiters are found */
	for (i = 0; s[i] && isalpha((int)s[i]); i++)
		;
	/* copy tz name */
	if (i >= sizeof(tzbuf))
		return -1; /* timezone too long */
	memcpy(tzbuf, s, i);
	tzbuf[i] = '\0';

	if ((sscanf(s, "%c%02d:%02d", &c, &tzhour, &tzmin)) == 3)
		;
	else if (sscanf(s, "%c%02d%02d", &c, &tzhour, &tzmin) == 3)
		;
	else if (sscanf(s, "%c%d", &c, &tzhour) == 2)
		tzmin = 0;
	else
		tzhour = tzmin = 0;
	if (!tzhour && !tzmin)
		c = '+';

	/* compare tz and adjust offset relative to UTC */
	for (i = 0; i < LEN(tzones); i++) {
		if (!strcmp(tzbuf, tzones[i].name)) {
			tz = "UTC";
			tzhour = tzones[i].offhour;
			tzmin = tzones[i].offmin;
			c = tzones[i].offhour < 0 ? '-' : '+';
			break;
		}
	}
	tzhour = abs(tzhour);
	tzmin = abs(tzmin);

time_ok:
	/* timezone set but non-match */
	if (tzbuf[0] && !tz[0]) {
		r = snprintf(buf, bufsiz, "%s", tzbuf);
		tzhour = tzmin = 0;
		c = '+';
	} else {
		r = snprintf(buf, bufsiz, "%s%c%02d:%02d",
		             tz[0] ? tz : "UTC", c, tzhour, tzmin);
	}
	if (r < 0 || (size_t)r >= bufsiz)
		return -1; /* truncation or error */
	if (tzoffset)
		*tzoffset = ((tzhour * 3600) + (tzmin * 60)) *
		            (c == '-' ? -1 : 1);
	return 0;
}

static int
parsetime(const char *s, char *buf, size_t bufsiz, time_t *tp)
{
	time_t t;
	char tz[64] = "";
	struct tm tm;
	const char *formats[] = {
		"%a, %d %b %Y %H:%M:%S",
		"%Y-%m-%d %H:%M:%S",
		"%Y-%m-%dT%H:%M:%S",
		NULL
	};
	char *p;
	size_t i;
	int tzoffset, r;

	memset(&tm, 0, sizeof(tm));
	for (i = 0; formats[i]; i++) {
		if (!(p = strptime(s, formats[i], &tm)))
			continue;
		tm.tm_isdst = -1; /* don't use DST */
		if ((t = timegm(&tm)) == -1) /* error */
			return -1;
		if (gettimetz(p, tz, sizeof(tz), &tzoffset) == -1)
			return -1;
		t -= tzoffset;
		if (buf) {
			r = snprintf(buf, bufsiz,
			         "%04d-%02d-%02d %02d:%02d:%02d %s",
			         tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
			         tm.tm_hour, tm.tm_min, tm.tm_sec, tz);
			if (r == -1 || (size_t)r >= bufsiz)
				return -1; /* truncation */
		}
		if (tp)
			*tp = t;
		return 0;
	}
	return -1;
}

/* Print text, encode TABs, newlines and '\', remove other whitespace.
 * Remove leading and trailing whitespace. */
static void
string_print_encoded(String *s)
{
	const char *p, *e;

	/* skip leading whitespace */
	for (p = s->data; *p && isspace((int)*p); p++)
		;
	/* seek offset of trailing whitespace */
	for (e = p + strlen(p); e > p && isspace((int)*(e - 1)); e--)
		;

	for (; *p && p != e; p++) {
		if (isspace((int)*p) && *p != ' ') {
			switch (*p) {
			case '\n': fputs("\\n",  stdout); break;
			case '\\': fputs("\\\\", stdout); break;
			case '\t': fputs("\\t",  stdout); break;
			default: break; /* ignore other whitespace chars */
			}
		} else if (!iscntrl((int)*p)) { /* ignore control chars */
			putchar(*p);
		}
	}
}

/* Print text, replace TABs, carriage return and other whitespace with ' '.
 * Other control chars are removed. Remove leading and trailing whitespace. */
static void
string_print_trimmed(String *s)
{
	const char *p, *e;

	/* skip leading whitespace */
	for (p = s->data; *p && isspace((int)*p); p++)
		;
	/* seek offset of trailing whitespace */
	for (e = p + strlen(p); e > p && isspace((int)*(e - 1)); e--)
		;

	for (; *p && p != e; p++) {
		if (isspace((int)*p))
			putchar(' ');
		else if (!iscntrl((int)*p))
			/* ignore other control chars */
			putchar((int)*p);
	}
}

static void
printfields(void)
{
	char link[4096], timebuf[64];
	time_t t;
	int r;

	/* parse time, timestamp and formatted timestamp field is empty
	 * if the parsed time is invalid */
	r = parsetime((&ctx.item.timestamp)->data, timebuf,
	              sizeof(timebuf), &t);
	if (r != -1)
		printf("%ld", (long)t);
	putchar(FieldSeparator);
	if (r != -1)
		fputs(timebuf, stdout);
	putchar(FieldSeparator);
	string_print_trimmed(&ctx.item.title);
	putchar(FieldSeparator);
	/* always print absolute urls */
	if (absuri(ctx.item.link.data, baseurl, link, sizeof(link)) != -1)
		fputs(link, stdout);
	putchar(FieldSeparator);
	string_print_encoded(&ctx.item.content);
	putchar(FieldSeparator);
	fputs(contenttypes[ctx.item.contenttype], stdout);
	putchar(FieldSeparator);
	string_print_trimmed(&ctx.item.id);
	putchar(FieldSeparator);
	string_print_trimmed(&ctx.item.author);
	putchar(FieldSeparator);
	fputs(feedtypes[ctx.item.feedtype], stdout);
	putchar('\n');
}

static int
istag(const char *name, size_t len, const char *name2, size_t len2)
{
	return (len == len2 && !strcasecmp(name, name2));
}

static int
isattr(const char *name, size_t len, const char *name2, size_t len2)
{
	return (len == len2 && !strcasecmp(name, name2));
}

/* NOTE: this handler can be called multiple times if the data in this
 * block is bigger than the buffer */
static void
xml_handler_data(XMLParser *p, const char *s, size_t len)
{
	if (ctx.field) {
		/* add only data from <name> inside <author> tag
		 * or any other non-<author> tag  */
		if (ctx.tagid != AtomTagAuthor || !strcmp(p->tag, "name"))
			string_append(ctx.field, s, len);
	}
}

static void
xml_handler_cdata(XMLParser *p, const char *s, size_t len)
{
	(void)p;

	if (ctx.field)
		string_append(ctx.field, s, len);
}

static void
xml_handler_attr_start(XMLParser *p, const char *tag, size_t taglen,
	const char *name, size_t namelen)
{
	(void)tag;
	(void)taglen;

	if (!ISINCONTENT(ctx))
		return;

	/* handles transforming inline XML to data */
	if (!ctx.attrcount)
		xml_handler_data(p, " ", 1);
	ctx.attrcount++;
	xml_handler_data(p, name, namelen);
	xml_handler_data(p, "=\"", 2);
}

static void
xml_handler_attr_end(struct xmlparser *p, const char *tag, size_t taglen,
	const char *name, size_t namelen)
{
	(void)tag;
	(void)taglen;
	(void)name;
	(void)namelen;

	if (!ISINCONTENT(ctx))
		return;

	/* handles transforming inline XML to data */
	xml_handler_data(p, "\"", 1);
	ctx.attrcount = 0;
}

static void
xml_handler_start_element_parsed(XMLParser *p, const char *tag, size_t taglen,
	int isshort)
{
	(void)tag;
	(void)taglen;

	if (!ISINCONTENT(ctx))
		return;

	if (isshort)
		xml_handler_data(p, "/>", 2);
	else
		xml_handler_data(p, ">", 1);
}

static void
xml_handler_attr(XMLParser *p, const char *tag, size_t taglen,
	const char *name, size_t namelen, const char *value,
	size_t valuelen)
{
	(void)tag;
	(void)taglen;

	/* handles transforming inline XML to data */
	if (ISINCONTENT(ctx)) {
		xml_handler_data(p, value, valuelen);
		return;
	}

	if (ctx.item.feedtype == FeedTypeAtom) {
		if (ISCONTENTTAG(ctx)) {
			if (isattr(name, namelen, STRP("type")) &&
			   (isattr(value, valuelen, STRP("xhtml")) ||
			    isattr(value, valuelen, STRP("text/xhtml")) ||
			    isattr(value, valuelen, STRP("html")) ||
			    isattr(value, valuelen, STRP("text/html"))))
			{
				ctx.item.contenttype = ContentTypeHTML;
				ctx.iscontent = 1;
/*				p->xmldataentity = NULL;*/ /* TODO: don't convert entities? test this */
				p->xmlattrstart = xml_handler_attr_start;
				p->xmlattrend = xml_handler_attr_end;
				p->xmltagstartparsed = xml_handler_start_element_parsed;
			}
		} else if (ctx.tagid == AtomTagLink &&
		          isattr(name, namelen, STRP("href")))
		{
			/* link href attribute */
			string_append(&ctx.item.link, value, valuelen);
		}
	}
}

static void
xml_handler_start_element(XMLParser *p, const char *name, size_t namelen)
{
	/* starts with div, handle as XML, don't convert entities
	 * (set handler to NULL) */
	if (ISCONTENTTAG(ctx) && ctx.item.feedtype == FeedTypeAtom &&
	   namelen == STRSIZ("div") && !strncmp(name, STRP("div"))) {
		p->xmldataentity = NULL;
	}
	if (ctx.iscontent) {
		ctx.attrcount = 0;
		ctx.iscontenttag = 0;
		xml_handler_data(p, "<", 1);
		xml_handler_data(p, name, namelen);
		return;
	}

	/* start of RSS or Atom item / entry */
	if (ctx.item.feedtype == FeedTypeNone) {
		if (istag(name, namelen, STRP("entry"))) {
			/* Atom */
			ctx.item.feedtype = FeedTypeAtom;
			/* default content type for Atom */
			ctx.item.contenttype = ContentTypePlain;
			ctx.field = NULL; /* XXX: optimization */
		} else if (istag(name, namelen, STRP("item"))) {
			/* RSS */
			ctx.item.feedtype = FeedTypeRSS;
			/* default content type for RSS */
			ctx.item.contenttype = ContentTypeHTML;
			ctx.field = NULL; /* XXX: optimization */
		}
		return;
	}

	/* tag already set: return */
	if (ctx.tag[0] != '\0')
		return;

	/* in item */
	strlcpy(ctx.tag, name, sizeof(ctx.tag)); /* NOTE: truncation ignored */
	ctx.taglen = namelen;
	ctx.tagid = gettag(ctx.item.feedtype, ctx.tag, ctx.taglen);

	switch (ctx.tagid) {
	case RSSTagPubdate:
	case RSSTagDcdate:
		ctx.field = &ctx.item.timestamp;
		break;
	case AtomTagPublished:
	case AtomTagUpdated:
		/* prefer published over updated if set */
		if (ctx.tagid != AtomTagUpdated || !ctx.item.timestamp.len)
			ctx.field = &ctx.item.timestamp;
		break;
	case RSSTagTitle:
	case AtomTagTitle:
		ctx.field = &ctx.item.title;
		break;
	case RSSTagLink:
	case AtomTagLink:
		ctx.field = &ctx.item.link;
		break;
	case RSSTagDescription:
	case RSSTagContentencoded:
		/* prefer content:encoded over description if set */
		if (ctx.tagid != RSSTagDescription || !ctx.item.content.len) {
			ctx.iscontenttag = 1;
			ctx.field = &ctx.item.content;
		}
		break;
	case AtomTagSummary:
	case AtomTagContent:
		/* prefer content over summary if set */
		if (ctx.tagid != AtomTagSummary || !ctx.item.content.len) {
			ctx.iscontenttag = 1;
			ctx.field = &ctx.item.content;
		}
		break;
	case RSSTagGuid:
	case AtomTagId:
		ctx.field = &ctx.item.id;
		break;
	case RSSTagAuthor:
	case RSSTagDccreator:
	case AtomTagAuthor:
		ctx.field = &ctx.item.author;
		break;
	default:
		ctx.field = NULL;
	}
	/* clear field */
	if (ctx.field)
		string_clear(ctx.field);
}

static void
xml_handler_data_entity(XMLParser *p, const char *data, size_t datalen)
{
	char buffer[16];
	int len;

	/* try to translate entity, else just pass as data to
	 * xml_data_handler */
	len = xml_entitytostr(data, buffer, sizeof(buffer));
	/* this should never happen (buffer too small) */
	if (len < 0)
		return;

	if (len > 0)
		xml_handler_data(p, buffer, (size_t)len);
	else
		xml_handler_data(p, data, datalen);
}

static void
xml_handler_end_element(XMLParser *p, const char *name, size_t namelen, int isshort)
{
	int tagid;

	if (ctx.iscontent) {
		ctx.attrcount = 0;
		tagid = gettag(ctx.item.feedtype, name, namelen);
		/* close content */
		if (ctx.tagid == tagid) {
			ctx.iscontent = 0;
			ctx.iscontenttag = 0;
			ctx.tag[0] = '\0';
			ctx.taglen = 0;
			ctx.tagid = TagUnknown;

			p->xmldataentity = xml_handler_data_entity;
			p->xmlattrstart = NULL;
			p->xmlattrend = NULL;
			p->xmltagstartparsed = NULL;

			return;
		}
		if (!isshort) {
			xml_handler_data(p, "</", 2);
			xml_handler_data(p, name, namelen);
			xml_handler_data(p, ">", 1);
		}
		return;
	}
	if (ctx.item.feedtype == FeedTypeNone)
		return;
	/* end of RSS or Atom entry / item */
	if ((ctx.item.feedtype == FeedTypeAtom &&
	   istag(name, namelen, STRP("entry"))) || /* Atom */
	   (ctx.item.feedtype == FeedTypeRSS &&
	   istag(name, namelen, STRP("item")))) /* RSS */
	{
		printfields();

		/* clear strings */
		string_clear(&ctx.item.timestamp);
		string_clear(&ctx.item.title);
		string_clear(&ctx.item.link);
		string_clear(&ctx.item.content);
		string_clear(&ctx.item.id);
		string_clear(&ctx.item.author);
		ctx.item.feedtype = FeedTypeNone;
		ctx.item.contenttype = ContentTypePlain;
		ctx.tag[0] = '\0'; /* unset tag */
		ctx.taglen = 0;
		ctx.tagid = TagUnknown;

		/* not sure if needed */
		ctx.iscontenttag = 0;
		ctx.iscontent = 0;
	} else if (ctx.taglen == namelen && !strcmp(ctx.tag, name)) {
		/* clear */
		/* XXX: optimize ? */
		ctx.field = NULL;
		ctx.tag[0] = '\0'; /* unset tag */
		ctx.taglen = 0;
		ctx.tagid = TagUnknown;

		/* not sure if needed */
		ctx.iscontenttag = 0;
		ctx.iscontent = 0;
	}
}

int
main(int argc, char *argv[])
{
	if (argc > 1)
		baseurl = argv[1];

	/* init strings and initial memory pool size */
	string_buffer_init(&ctx.item.timestamp, 64);
	string_buffer_init(&ctx.item.title, 256);
	string_buffer_init(&ctx.item.link, 1024);
	string_buffer_init(&ctx.item.content, 4096);
	string_buffer_init(&ctx.item.id, 1024);
	string_buffer_init(&ctx.item.author, 256);
	ctx.item.contenttype = ContentTypePlain;
	ctx.item.feedtype = FeedTypeNone;

	memset(&parser, 0, sizeof(parser));
	parser.xmltagstart = xml_handler_start_element;
	parser.xmltagend = xml_handler_end_element;
	parser.xmldata = xml_handler_data;
	parser.xmldataentity = xml_handler_data_entity;
	parser.xmlattr = xml_handler_attr;
	parser.xmlcdata = xml_handler_cdata;
	xmlparser_parse_fd(&parser, 0);

	return 0;
}