/*
 * Portions Copyright (c) 1999 GMRS Software GmbH
 * Carl-von-Linde-Str. 38, D-85716 Unterschleissheim, http://www.gmrs.de
 * All rights reserved.
 *
 * Author: Arno Unkrig <arno@unkrig.de>
 *
 * 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 2 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 in the file COPYING for more details.
 */

#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "html.h"
#include "HTMLControl.h"
#include "HTMLDriver.h"
#include "HTMLParser.hh"
#include "sgml.h"
#include "cmp_nocase.h"
#include "istr.h"

#ifndef nelems
#define nelems(array) (sizeof(array) / sizeof((array)[0]))
#endif

enum {
	NOT_A_TAG,
	START_TAG, BLOCK_START_TAG,
	END_TAG, BLOCK_END_TAG,
	NON_CONTAINER_TAG
};

void
HTMLControl::htmlparser_yyerror(const char *p)
{
	/*
	 * Swallow parse error messages if not in "syntax check" mode.
	 */
	if (mode != HTMLDriver::SYNTAX_CHECK &&
			!strncmp(p, "syntax error", sizeof("syntax error") - 1))
		return;

	std::cerr
		<< "File \""
		<< file_name
		<< "\", line "
		<< current_line
		<< ", column "
		<< current_column
		<< ": "
		<< p
		<< std::endl;
}

struct htmlparsertoken *HTMLControl::get_nth_token(int id)
{
	int i;
	htmlparsertoken *tokw;
	htmlparsertoken *tail = NULL;

	for (i = 0, tokw = next_tokens; tokw != NULL; tokw = tokw->next, i++) {
		if (i == id)
			return tokw;
		tail = tokw;
	}

	if (i == id) {
		if (tail == NULL)
			next_tokens = tokw = new htmlparsertoken;
		else
			tokw = tail->next = new htmlparsertoken;
		tokw->next_token = yylex2(&tokw->next_token_value,
								  &tokw->next_token_tag_type);
		tokw->next = NULL;

		/* do some space normalisation on PCDATA here to ease whitespace
		 * logics lateron */
		if (tokw->next_token == HTMLParser_token::PCDATA && !literal_mode) {
			istr &s(*tokw->next_token_value.strinG);
			/* empty string, just skip, get the next one */
			if (s.empty()) {
				del_nth_token(id);
				return get_nth_token(id);
			}

			/* we got something, collate sequences of whitespace
			 * within the string */
			for (string::size_type x = 0; x < s.length(); x++) {
				if (isspace(s[x])) {
					string::size_type y;
					for (y = x + 1; y < s.length() && isspace(s[y]); y++)
						;
					s.replace(x, y - x, " ");
				}
			}

			/* if we have trailing or leading spaces, transform them
			 * into their own token, so collapsing them is easier */
			if (s.length() > 1) {
				if (isspace(s[s.length() - 1])) {
					s.erase(s.length() - 1, 1);  /* drop space */
					/* append token */
					htmlparsertoken *space = new htmlparsertoken;
					space->next_token = HTMLParser_token::PCDATA;
					space->next_token_tag_type = NOT_A_TAG;
					space->next_token_value.strinG = new istr(" ");
					space->next = NULL;
					tokw->next = space;
				}
				if (isspace(s[0])) {
					s.erase(0, 1);  /* drop space */
					/* prepend token */
					htmlparsertoken *space = new htmlparsertoken;
					space->next = tokw;
					if (next_tokens == tokw) {
						next_tokens = space;
					} else {
						tail->next = space;
					}
					space->next_token = HTMLParser_token::PCDATA;
					space->next_token_tag_type = NOT_A_TAG;
					space->next_token_value.strinG = new istr(" ");
					tokw = space;
				}
			}
		}

		return tokw;
	}

	return NULL;
}

void HTMLControl::del_nth_token(int id)
{
	int i;
	htmlparsertoken *tokw;
	htmlparsertoken *tail = NULL;

	for (i = 0, tokw = next_tokens; tokw != NULL; tokw = tokw->next, i++) {
		if (i == id)
			break;
		tail = tokw;
	}

	if (i == id) {
		if (tail == NULL) {
			next_tokens =  next_tokens->next;
		} else {
			tail->next = tokw->next;
		}
		if (tokw->next_token == HTMLParser_token::PCDATA)
			delete tokw->next_token_value.strinG;
		delete tokw;
	}
}

/*
 * Effectively, this method simply invokes "yylex2()", but it does some
 * postprocessing on PCDATA tokens that would be difficult to do in "yylex2()".
 */
int HTMLControl::htmlparser_yylex(
		html2text::HTMLParser::semantic_type *value_return)
{
	for (;;) { // Notice the "return" at the end of the body!
		int token;
		int tag_type;
		int next_token;
		int next_token_tag_type;
		html2text::HTMLParser::semantic_type next_token_value;
		struct htmlparsertoken *firsttok;

		firsttok      = get_nth_token(0);
		token         = firsttok->next_token;
		tag_type      = firsttok->next_token_tag_type;
		*value_return = firsttok->next_token_value;

		/* pop token */
		next_tokens = next_tokens->next;
		delete firsttok;

		/* Switch on/off "literal mode" on "<PRE>" and "</PRE>" */
		if (token == HTMLParser_token::PRE) {
			literal_mode = true;

			firsttok            = get_nth_token(0);
			next_token          = firsttok->next_token;
			next_token_value    = firsttok->next_token_value;

			if (next_token == HTMLParser_token::PCDATA) {
				/* Swallow '\n' immediately following "<PRE>" */
				istr &s(*next_token_value.strinG);
				if (!s.empty() && s[0] == '\n')
					s.erase(0, 1);
			}
		}

		if (token == HTMLParser_token::END_PRE)
			literal_mode = false;

		if (tag_type == BLOCK_START_TAG ||
			tag_type == BLOCK_END_TAG ||
			token == HTMLParser_token::BR ||
			token == HTMLParser_token::HR)
			document_start = true;  /* swallow whitespace */

		if (token == HTMLParser_token::PCDATA) {
			if (literal_mode) {
				firsttok            = get_nth_token(0);
				next_token          = firsttok->next_token;
				next_token_value    = firsttok->next_token_value;

				/* Erase " '\n' { ' ' } " immediately before "</PRE>".  */
				if (next_token == HTMLParser_token::END_PRE) {
					istr &s(*value_return->strinG);
					string::size_type x = s.length();
					while (x > 0 && s[x - 1] == ' ')
						--x;
					if (x > 0 && s[x - 1] == '\n')
						s.erase(x - 1, string::npos);
				}
			} else {
				/* In order to post-process the PCDATA token, we need to
			 	 * look ahead until we find another PCDATA token.  We're
			 	 * trying to eliminate empty ones here, so as to avoid too
			 	 * little and too much whitespace.  During reading,
			 	 * strings are whitespace normalised, and trailing or
			 	 * leading spaces are converted into separate elements
			 	 * here. */

				istr &s(*value_return->strinG);
				bool keepspace = document_start ? false : true;
				if (isspace(s[0])) {  /* this must be a space on its own */
					for (int i = 0; ;) {
						firsttok            = get_nth_token(i);
						next_token          = firsttok->next_token;
						next_token_tag_type = firsttok->next_token_tag_type;
						next_token_value    = firsttok->next_token_value;

						if (next_token == EOF ||
							next_token == HTMLParser_token::PRE ||
							next_token_tag_type == BLOCK_START_TAG ||
							next_token_tag_type == BLOCK_END_TAG)
						{
							keepspace = false;
							break;
						}

						if (next_token == HTMLParser_token::PCDATA) {
							istr &ns(*next_token_value.strinG);
							if (isspace(ns[0])) {
								/* duplicate, drop */
								del_nth_token(i);
								continue;
							} else {
								break;  /* we're done here */
							}
						}
						i++;
					}

					if (!keepspace) {
						/* drop, we're excessive */
						delete value_return->strinG;
						continue;
					}
				} else {
					document_start = false;
				}
			}
		}

		return token;
	}
}

/*
 * Keep this array sorted alphabetically!
 */
static const struct TextToIntP {
	char name[11];
	char block_tag;
	const int start_tag_code;
	const int end_tag_code;
}
tag_names[] = {
#define pack1(tag) \
	{ #tag, 0, HTMLParser_token::tag, 0 }
#define pack2(tag) \
	{ #tag, 0, HTMLParser_token::tag, HTMLParser_token::END_ ## tag }
#define pack3(tag) \
	{ #tag, 1, HTMLParser_token::tag, HTMLParser_token::END_ ## tag }
	pack2(A),
	pack3(ADDRESS),
	pack2(APPLET),
	pack1(AREA),
	pack2(B),
	pack1(BASE),
	pack1(BASEFONT),
	pack2(BIG),
	pack3(BLOCKQUOTE),
	pack3(BODY),
	pack1(BR),
	pack3(CAPTION),
	pack3(CENTER),
	pack3(CITE),
	pack2(CODE),
	pack3(DD),
	pack2(DFN),
	pack3(DIR),
	pack3(DIV),
	pack3(DL),
	pack3(DT),
	pack2(EM),
	pack2(FONT),
	pack3(FORM),
	pack3(H1),
	pack3(H2),
	pack3(H3),
	pack3(H4),
	pack3(H5),
	pack3(H6),
	pack3(HEAD),
	pack1(HR),
	pack3(HTML),
	pack2(I),
	pack1(IMG),
	pack1(INPUT),
	pack1(ISINDEX),
	pack2(KBD),
	pack3(LI),
	pack1(LINK),
	pack2(MAP),
	pack3(MENU),
	pack1(META),
	pack2(NOBR),
	pack3(OL),
	pack3(OPTION),
	pack3(P),
	pack1(PARAM),
	pack3(PRE),
	pack2(SAMP),
	pack3(SCRIPT),
	pack2(SELECT),
	pack2(SMALL),
	pack2(SPAN),
	pack2(STRIKE),
	pack2(STRONG),
	pack3(STYLE),
	pack2(SUB),
	pack2(SUP),
	pack3(TABLE),
	pack3(TD),
	pack2(TEXTAREA),
	pack3(TH),
	pack3(TITLE),
	pack3(TR),
	pack2(TT),
	pack2(U),
	pack3(UL),
	pack2(VAR),
#undef pack3
#undef pack2
#undef pack1
};

int
HTMLControl::yylex2(html2text::HTMLParser::semantic_type *value_return,
		int *tag_type_return)
{
	int c;

	*tag_type_return = NOT_A_TAG;
	for (;;) { // Notice the "return" at the end of this loop.
		/*
		 * Get the first character of the token.
		 */
		c = get_char();
		if (c == EOF)
			return EOF;

		if (c == '<') {
			/*
			 * Examine the first character of the tag.
			 */
			c = get_char();
			if (c == '!') {
				c = get_char();
				if (c == '-') {
					c = get_char();
					if (c != '-')
						return HTMLParser_token::SCAN_ERROR;

					/*
					 * This is a comment... skip it!
					 *
					 *   <!-- Single-line comment -->
					 *
					 *   <!-- Multi-
					 *        line
					 *        comment //-->
					 *
					 * EXTENSION: Allow "-->" as the terminator of a
					 *            multi-line comment.
					 */
					int state = 0;
					do {
						c = get_char();
						if (c == EOF)
							return HTMLParser_token::SCAN_ERROR;
						switch (state) {
						case 0: if (c == '-')
								state = 1;
						break;
						case 1: state = c == '-' ? 2 : 0;
							break;
						case 2: state = c == '>' ? 3 : c == '-' ? 2 : 0;
							break;
						}
					} while (state != 3);
					continue; // Start over
				}

				/* scan <![if .... ]> <![endif]> kind of crap */
				if (c == '[') {
					do {
						c = get_char();
						if (c == EOF)
							return HTMLParser_token::SCAN_ERROR;
					} while (c != ']');
					do {
						c = get_char();
					} while (c != EOF && isspace(c));
					if (c != '>')
						return HTMLParser_token::SCAN_ERROR;
					continue;  /* ignore this thing (start over) */
				}

				/*
				 * Scan "<!DOCTYPE ...>" tag.
				 */
				if (!isalpha(c))
					return HTMLParser_token::SCAN_ERROR;
				string tag_name(1, '!');
				tag_name += c;
				for (;;) {
					c = get_char();
					if (!isalnum(c) && c != '-')
						break;
					tag_name += c;
				}
				if (cmp_nocase(tag_name, "!DOCTYPE") != 0)
					return HTMLParser_token::SCAN_ERROR;
				while (c != '>') {
					c = get_char();
					if (c == EOF)
						return HTMLParser_token::SCAN_ERROR;
					// Let newline not close the DOCTYPE tag - Arno
				}
				return HTMLParser_token::DOCTYPE;
			}

			if (c == '/' || isalpha(c) || c == '_' || c == '?') {
				string tag_name;
				bool is_end_tag = false;

				if (c == '/') {
					is_end_tag = true;
					c = get_char();
				}
				if (c == '?') {
					c = get_char();
				}
				if (!isalpha(c) && c != '_')
					return HTMLParser_token::SCAN_ERROR;
				tag_name += c;
				for (;;) {
					c = get_char();
					/* ID and NAME tokens must begin with a letter
					 * ([A-Za-z]) -- isalpha tested before -- and may be
					 * followed by any number of letters, digits
					 * ([0-9]), hyphens ("-"), underscores ("_"), colons
					 * (":"), and periods (".") */
					if (!isalnum(c) &&
							c != '-' && c != '_' && c != ':' && c != '.')
						break;
					tag_name += c;
				}

				while (isspace(c))
					c = get_char();

				/*
				 * Scan tag attributes (only for opening tags). Create
				 * the "tag_attributes" only on demand; this saves a lot
				 * of overhead.
				 */
				auto_ptr<list<TagAttribute>> tag_attributes;
				if (!is_end_tag) {
					while (c != EOF && c != '>') {
						bool         attrseparate = false;
						TagAttribute attribute;

						/* attribute := name [ '=' quote string quote ]
						 * name := first [ rest ... ]
						 * first := alpha | '_'
						 * rest := alpha | num | '_' | '-' | ':' | '.'
						 * quote := '\'' | '"'
						 * string := * !quote
						 *
						 * issue #61, be loose here, allow all kinds of
						 * garbage to happen, optimistically try to find
						 * attributes, until the closing '>' */
						if (!isalpha(c) && c != '_')
							goto eat_garbage;

						/* Scan attribute name, see the ID and NAME rule
						 * mentioned above */
						attribute.first = c;
						for (;;) {
							c = get_char();
							if (!isalnum(c) &&
									c != '-' && c != '_' &&
									c != ':' && c != '.')
								break;
							attribute.first += c;
						}

						/* skip whitespace */
						while (isspace(c)) {
							c = get_char();
							attrseparate = true;
						}

						/*
						 * Scan (optional) attribute value.
						 */
						if (c == '=') {
							c = get_char();
							while (isspace(c))
								c = get_char();
							if (c == '"' || c == '\'') {
								int closing_quote = c; // Same as opening quote!
								for (;;) {
									c = get_char();
									if (c == EOF)
										return HTMLParser_token::SCAN_ERROR;
									// Accept multiple-line elements - Arno
									if (c == closing_quote)
										break;

									/*
									 * Do *not* interpret "&auml;" and
									 * consorts here! This would ruin
									 * tag attributes like
									 * "HREF=hhh?a=1&b=2".
									 */
									attribute.second += c;
								}
								c = get_char(); // Get next after closing quote
							} else {
								/* unquoted value */
								while (isalnum(c)) {
									attribute.second += c;
									c = get_char();
								}
								if (c == EOF)
									return HTMLParser_token::SCAN_ERROR;
								if (!isspace(c))
									goto eat_garbage;
							}

							/* skip whitespace */
							while (isspace(c))
								c = get_char();
						} else {
							if (!attrseparate)
								goto eat_garbage;
							/* short form, expand */
							attribute.second = attribute.first;
						}

						/*
						 * Store the attribute.
						 */
						if (!tag_attributes.get()) {
							tag_attributes.reset(new list<TagAttribute>);
						}
						tag_attributes->push_back(attribute);

						continue;
eat_garbage:
						while (c != ' ' &&
							   c != '>' &&
							   c != EOF)
							c = get_char();

						/* skip whitespace */
						while (isspace(c))
							c = get_char();
					}
					if (c == EOF)
						return HTMLParser_token::SCAN_ERROR;
				}

				// accept XHTML tags like <hr /> - Alexander Solovey

				if (c != '>') {
					if (c == '/') {
						c = get_char();
						if (c != '>') {
							return HTMLParser_token::SCAN_ERROR;
						}
					} else {
						return HTMLParser_token::SCAN_ERROR;
					}
				}

				if (debug_scanner) {
					std::cerr << "Scanned tag \"<" <<
						(is_end_tag ? "/" : "") << tag_name;
					if (!is_end_tag && tag_attributes.get()) {
						const list<TagAttribute>           &ta(*tag_attributes);
						list<TagAttribute>::const_iterator j;
						for (j = ta.begin(); j != ta.end(); ++j) {
							std::cerr << " " << (*j).first <<
								"=\"" << (*j).second.c_str() << "\"";
						}
					}
					std::cerr << ">\"" << std::endl;
				}

				/*
				 * Look up the tag in the table of recognized tags.
				 */
				static int(*const f)(const char *, const char *) = cmp_nocase;
				const TextToIntP *tag = (const TextToIntP *) bsearch(
						tag_name.c_str(),
						tag_names, nelems(tag_names), sizeof(TextToIntP),
						(int (*)(const void *, const void *))f);
				if (tag == NULL) { /* EXTENSION: Swallow unknown tags. */
					if (debug_scanner) {
						std::cerr << "Tag unknown (" << tag_name.c_str()
							<< ") -- swallowed." << std::endl;
					}
					continue;
				}

				/*
				 * Return the BISON token code for the tag.
				 */
				if (is_end_tag) {
					if (!tag->end_tag_code) {
						if (debug_scanner) {
							std::cerr << "Non-container end tag scanned." <<
								std::endl;
						}
						continue;
					}
					*tag_type_return = tag->block_tag ? BLOCK_END_TAG : END_TAG;
					return tag->end_tag_code;
				} else {
					*tag_type_return = (
						!tag->end_tag_code ? NON_CONTAINER_TAG :
						tag->block_tag     ? BLOCK_START_TAG   : START_TAG
						);
					value_return->tag_attributes = tag_attributes.release();
					return tag->start_tag_code;
				}
			}

			/*
			 * EXTENSION: This tag did not match "<!", and not "</", and not
			 * "<[A-Za-z-]", so take it as literal text.
			 */
			unget_char(c);
			c = '<';
		}

		if (c == '\n' || (unsigned char) c >= (unsigned char) ' ') {
			// Same as in line 402
			istr *s = value_return->strinG = new istr;

			while (c != EOF) {
				/*
				 * Accept literal '<' in some cases.
				 */
				if (c == '<') {
					int c2;
					unget_char(c2 = get_char());
					if (c2 == '!' || c2 == '/' || c2 == '?' || isalpha(c2)) {
						unget_char(c);
						break;
					}
				}

				*s += c;
				c = get_char();
			}

			/* need to do this here, because space calculations want to
			 * know the final form of the data */
			replace_sgml_entities(s); // Replace "&auml;" and consorts.

			/*
			 * Swallow empty PCDATAs.
			 */
			if (s->empty()) {
				delete s;
				continue;
			}

			if (debug_scanner)
				std::cerr << "Scanned PCDATA \"" << s->c_str() << "\""
					<< std::endl;

			return HTMLParser_token::PCDATA;
		}

		return HTMLParser_token::SCAN_ERROR;
	}
}

bool
HTMLControl::read_cdata(const char *terminal, string *value_return)
{
	string &s(*value_return);
	int c;
	int state = 0;

	for (;;) {
		c = get_char();
		if (c == EOF)
			return false;
		if (toupper(c) == terminal[state]) {
			state++;
			if (terminal[state] == '\0') {
				// s.erase(s.length() - state);
				// caused core dump on empty STYLE and
				// SCRIPT elements - Johannes Geiger
				s.erase(s.length() - state + 1);
				return true;
			}
		} else {
			state = 0;
		}
		s += c;
	}
}

int
HTMLControl::get_char()
{
	/* our input is always converted to UTF-8, so load bytes as required */
	unsigned int c = 0;

	if (number_of_ungotten_chars > 0) {
		c = ungotten_chars[--number_of_ungotten_chars];
	} else {
		c = is.get();
		while (c == '\r')
			c = is.get();

		if (c == (unsigned int)EOF) {
			;
		} else if (c == '\n') {
			current_line++;
			current_column = 0;
		} else {
			current_column++;
			if ((c >> 7) & 1) {
				unsigned char nextpoint = 1;

				/* we assume iconv produced valid UTF-8 here */
				while ((c >> (7 - nextpoint)) & 1)
					c |= ((is.get() & 0xFF) << (8 * nextpoint++));
			}
		}
	}

#if 0
	if ((c >> 7) & 1) {
		unsigned int d = c;
		unsigned char point = 1;
		fprintf(stderr, "utf-8 start %02x\n", c & 0xFF);
		while ((c >> (7 - point++)) & 1) {
			d >>= 8;
			fprintf(stderr, "utf-8 join  %02x\n", d & 0xFF);
		};
	}
#endif

	return c;
}

/* ------------------------------------------------------------------------- */

void
HTMLControl::unget_char(int c)
{
	if (number_of_ungotten_chars == nelems(ungotten_chars)) {
		//error("Too many chars ungotten");
		return;
	}
	ungotten_chars[number_of_ungotten_chars++] = c;
}