/*
 * This file is part of Hubbub.
 * Licensed under the MIT License,
 *                http://www.opensource.org/licenses/mit-license.php
 * Copyright 2008 Andrew Sidwell <takkaria@netsurf-browser.org>
 */

#include <assert.h>
#include <string.h>

#include "treebuilder/modes.h"
#include "treebuilder/internal.h"
#include "treebuilder/treebuilder.h"
#include "utils/utils.h"


/**
 * Clear the stack back to a table body context.
 *
 * \param treebuilder	The treebuilder instance
 */
static void table_clear_stack(hubbub_treebuilder *treebuilder)
{
	element_type cur_node = current_node(treebuilder);

	while (cur_node != TBODY && cur_node != TFOOT &&
			cur_node != THEAD && cur_node != HTML) {
		hubbub_ns ns;
		element_type type;
		void *node;

		element_stack_pop(treebuilder, &ns, &type, &node);

		treebuilder->tree_handler->unref_node(
				treebuilder->tree_handler->ctx,
				node);

		cur_node = current_node(treebuilder);
	}

	return;
}


/**
 * Handle the case common to some start tag and the table end tag cases.
 *
 * \param treebuilder	The treebuilder instance
 * \return Whether to reprocess the current token
 */
static hubbub_error table_sub_start_or_table_end(hubbub_treebuilder *treebuilder)
{
	if (element_in_scope(treebuilder, TBODY, true) ||
			element_in_scope(treebuilder, THEAD, true) ||
			element_in_scope(treebuilder, TFOOT, true)) {
		hubbub_ns ns;
		element_type otype;
		void *node;

		table_clear_stack(treebuilder);

		/* "Act as if an end tag with the same name as the current
		 * node had been seen" -- this behaviour should be identical
		 * to handling for (tbody/tfoot/thead) end tags in this mode */
		element_stack_pop(treebuilder, &ns, &otype, &node);

		treebuilder->tree_handler->unref_node(
				treebuilder->tree_handler->ctx,
				node);

		treebuilder->context.mode = IN_TABLE;

		return HUBBUB_REPROCESS;
	} else {
		/** \todo parse error */
	}

	return HUBBUB_OK;
}


/**
 * Handle tokens in "in table body" insertion mode
 *
 * \param treebuilder  The treebuilder instance
 * \param token        The token to process
 * \return True to reprocess the token, false otherwise
 */
hubbub_error handle_in_table_body(hubbub_treebuilder *treebuilder,
		const hubbub_token *token)
{
	hubbub_error err = HUBBUB_OK;

	switch (token->type) {
	case HUBBUB_TOKEN_START_TAG:
	{
		element_type type = element_type_from_name(treebuilder,
				&token->data.tag.name);

		if (type == TR) {
			table_clear_stack(treebuilder);

			err = insert_element(treebuilder, &token->data.tag, 
					true);
			if (err == HUBBUB_OK)
				treebuilder->context.mode = IN_ROW;
		} else if (type == TH || type == TD) {
			hubbub_tag tag;

			/** \todo parse error */

			/* Manufacture tr tag */
			tag.ns = HUBBUB_NS_HTML;
			tag.name.ptr = (const uint8_t *) "tr";
			tag.name.len = SLEN("tr");

			tag.n_attributes = 0;
			tag.attributes = NULL;

			table_clear_stack(treebuilder);

			err = insert_element(treebuilder, &tag, true);
			if (err == HUBBUB_OK) {
				treebuilder->context.mode = IN_ROW;

				err = HUBBUB_REPROCESS;
			}
		} else if (type == CAPTION || type == COL ||
				type == COLGROUP || type == TBODY ||
				type == TFOOT || type == THEAD) {
			err = table_sub_start_or_table_end(treebuilder);
		} else {
			err = handle_in_table(treebuilder, token);
		}
	}
		break;
	case HUBBUB_TOKEN_END_TAG:
	{
		element_type type = element_type_from_name(treebuilder,
				&token->data.tag.name);

		if (type == TBODY || type == TFOOT || type == THEAD) {
			if (!element_in_scope(treebuilder, type, true)) {
				/** \todo parse error */
				/* Ignore the token */
			} else {
				hubbub_ns ns;
				element_type otype;
				void *node;

				table_clear_stack(treebuilder);

				element_stack_pop(treebuilder, &ns,
						&otype, &node);

				treebuilder->tree_handler->unref_node(
						treebuilder->tree_handler->ctx,
						node);

				treebuilder->context.mode = IN_TABLE;
			}
		} else if (type == TABLE) {
			err = table_sub_start_or_table_end(treebuilder);
		} else if (type == BODY || type == CAPTION || type == COL ||
				type == COLGROUP || type == HTML ||
				type == TD || type == TH || type == TR) {
			/** \todo parse error */
			/* Ignore the token */
		} else {
			err = handle_in_table(treebuilder, token);
		}
	}
		break;
	case HUBBUB_TOKEN_CHARACTER:
	case HUBBUB_TOKEN_COMMENT:
	case HUBBUB_TOKEN_DOCTYPE:
	case HUBBUB_TOKEN_EOF:
		err = handle_in_table(treebuilder, token);
		break;
	}

	return err;
}