/*

 * Copyright (C) 1999 Lars Knoll (knoll@kde.org)

 *           (C) 1999 Antti Koivisto (koivisto@kde.org)

 * Copyright (C) 2004, 2005, 2006, 2007, 2008 Apple Inc. All rights reserved.

 * Copyright (C) 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Library General Public

 * License as published by the Free Software Foundation; either

 * version 2 of the License, or (at your option) any later version.

 *

 * This library 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

 * Library General Public License for more details.

 *

 * You should have received a copy of the GNU Library General Public License

 * along with this library; see the file COPYING.LIB.  If not, write to

 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

 * Boston, MA 02110-1301, USA.

 *

 */

#include "config.h"

#include "HTMLElement.h"

#include "Attribute.h"

#include "CSSPropertyNames.h"

#include "CSSValueKeywords.h"

#include "DocumentFragment.h"

#include "Event.h"

#include "EventListener.h"

#include "EventNames.h"

#include "ExceptionCode.h"

#include "Frame.h"

#include "HTMLBRElement.h"

#include "HTMLCollection.h"

#include "HTMLDocument.h"

#include "HTMLElementFactory.h"

#include "HTMLFormElement.h"

#include "HTMLNames.h"

#include "RenderWordBreak.h"

#include "ScriptEventListener.h"

#include "Settings.h"

#include "Text.h"

#include "TextIterator.h"

#include "markup.h"

#include <wtf/StdLibExtras.h>

namespace WebCore {

using namespace HTMLNames;

using std::min;

using std::max;

PassRefPtr<HTMLElement> HTMLElement::create(const QualifiedName& tagName, Document* document)

{

    return adoptRef(new HTMLElement(tagName, document));

}

String HTMLElement::nodeName() const

{

    // FIXME: Would be nice to have an atomicstring lookup based off uppercase chars that does not have to copy

    // the string on a hit in the hash.

    // FIXME: We should have a way to detect XHTML elements and replace the hasPrefix() check with it.

    if (document()->isHTMLDocument() && !tagQName().hasPrefix())

        return tagQName().localNameUpper();

    return Element::nodeName();

}

HTMLTagStatus HTMLElement::endTagRequirement() const

{

    if (hasLocalName(wbrTag))

        return TagStatusForbidden;

    if (hasLocalName(dtTag) || hasLocalName(ddTag) || hasLocalName(rpTag) || hasLocalName(rtTag))

        return TagStatusOptional;

    // Same values as <span>.  This way custom tag name elements will behave like inline spans.

    return TagStatusRequired;

}

struct Empty1IntHashTraits : HashTraits<int> {

    static const bool emptyValueIsZero = false;

    static int emptyValue() { return 1; }

};

typedef HashMap<AtomicStringImpl*, int, DefaultHash<AtomicStringImpl*>::Hash, HashTraits<AtomicStringImpl*>, Empty1IntHashTraits> TagPriorityMap;

static const TagPriorityMap* createTagPriorityMap()

{

    TagPriorityMap* map = new TagPriorityMap;

    map->add(wbrTag.localName().impl(), 0);

    map->add(addressTag.localName().impl(), 3);

    map->add(ddTag.localName().impl(), 3);

    map->add(dtTag.localName().impl(), 3);

    map->add(noscriptTag.localName().impl(), 3);

    map->add(rpTag.localName().impl(), 3);

    map->add(rtTag.localName().impl(), 3);

    // 5 is same as <div>'s priority.

    map->add(articleTag.localName().impl(), 5);

    map->add(asideTag.localName().impl(), 5);

    map->add(centerTag.localName().impl(), 5);

    map->add(footerTag.localName().impl(), 5);

    map->add(headerTag.localName().impl(), 5);

    map->add(hgroupTag.localName().impl(), 5);

    map->add(nobrTag.localName().impl(), 5);

    map->add(rubyTag.localName().impl(), 5);

    map->add(navTag.localName().impl(), 5);

    map->add(sectionTag.localName().impl(), 5);

    map->add(noembedTag.localName().impl(), 10);

    map->add(noframesTag.localName().impl(), 10);

    // TagPriorityMap returns 1 for unregistered tags. It's same as <span>.

    // This way custom tag name elements will behave like inline spans.

    return map;

}

int HTMLElement::tagPriority() const

{

    static const TagPriorityMap* tagPriorityMap = createTagPriorityMap();

    return tagPriorityMap->get(localName().impl());

}

bool HTMLElement::mapToEntry(const QualifiedName& attrName, MappedAttributeEntry& result) const

{

    if (attrName == alignAttr ||

        attrName == contenteditableAttr) {

        result = eUniversal;

        return false;

    }

    if (attrName == dirAttr) {

        result = hasLocalName(bdoTag) ? eBDO : eUniversal;

        return false;

    }

    return StyledElement::mapToEntry(attrName, result);

}

void HTMLElement::parseMappedAttribute(Attribute* attr)

{

    if (isIdAttributeName(attr->name()) || attr->name() == classAttr || attr->name() == styleAttr)

        return StyledElement::parseMappedAttribute(attr);

    String indexstring;

    if (attr->name() == alignAttr) {

        if (equalIgnoringCase(attr->value(), "middle"))

            addCSSProperty(attr, CSSPropertyTextAlign, "center");

        else

            addCSSProperty(attr, CSSPropertyTextAlign, attr->value());

    } else if (attr->name() == contenteditableAttr) {

        setContentEditable(attr);

    } else if (attr->name() == tabindexAttr) {

        indexstring = getAttribute(tabindexAttr);

        if (indexstring.length()) {

            bool parsedOK;

            int tabindex = indexstring.toIntStrict(&parsedOK);

            if (parsedOK)

                // Clamp tabindex to the range of 'short' to match Firefox's behavior.

                setTabIndexExplicitly(max(static_cast<int>(std::numeric_limits<short>::min()), min(tabindex, static_cast<int>(std::numeric_limits<short>::max()))));

        }

    } else if (attr->name() == langAttr) {

        // FIXME: Implement

    } else if (attr->name() == dirAttr) {

        addCSSProperty(attr, CSSPropertyDirection, attr->value());

        addCSSProperty(attr, CSSPropertyUnicodeBidi, hasLocalName(bdoTag) ? CSSValueBidiOverride : CSSValueEmbed);

    } else if (attr->name() == draggableAttr) {

        const AtomicString& value = attr->value();

        if (equalIgnoringCase(value, "true")) {

            addCSSProperty(attr, CSSPropertyWebkitUserDrag, CSSValueElement);

            addCSSProperty(attr, CSSPropertyWebkitUserSelect, CSSValueNone);

        } else if (equalIgnoringCase(value, "false"))

            addCSSProperty(attr, CSSPropertyWebkitUserDrag, CSSValueNone);

    }

// standard events

    else if (attr->name() == onclickAttr) {

        setAttributeEventListener(eventNames().clickEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == oncontextmenuAttr) {

        setAttributeEventListener(eventNames().contextmenuEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondblclickAttr) {

        setAttributeEventListener(eventNames().dblclickEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmousedownAttr) {

        setAttributeEventListener(eventNames().mousedownEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmousemoveAttr) {

        setAttributeEventListener(eventNames().mousemoveEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmouseoutAttr) {

        setAttributeEventListener(eventNames().mouseoutEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmouseoverAttr) {

        setAttributeEventListener(eventNames().mouseoverEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmouseupAttr) {

        setAttributeEventListener(eventNames().mouseupEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onmousewheelAttr) {

        setAttributeEventListener(eventNames().mousewheelEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onfocusAttr) {

        setAttributeEventListener(eventNames().focusEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onfocusinAttr) {

        setAttributeEventListener(eventNames().focusinEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onfocusoutAttr) {

        setAttributeEventListener(eventNames().focusoutEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onblurAttr) {

        setAttributeEventListener(eventNames().blurEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onkeydownAttr) {

        setAttributeEventListener(eventNames().keydownEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onkeypressAttr) {

        setAttributeEventListener(eventNames().keypressEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onkeyupAttr) {

        setAttributeEventListener(eventNames().keyupEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onscrollAttr) {

        setAttributeEventListener(eventNames().scrollEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onbeforecutAttr) {

        setAttributeEventListener(eventNames().beforecutEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == oncutAttr) {

        setAttributeEventListener(eventNames().cutEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onbeforecopyAttr) {

        setAttributeEventListener(eventNames().beforecopyEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == oncopyAttr) {

        setAttributeEventListener(eventNames().copyEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onbeforepasteAttr) {

        setAttributeEventListener(eventNames().beforepasteEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onpasteAttr) {

        setAttributeEventListener(eventNames().pasteEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragenterAttr) {

        setAttributeEventListener(eventNames().dragenterEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragoverAttr) {

        setAttributeEventListener(eventNames().dragoverEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragleaveAttr) {

        setAttributeEventListener(eventNames().dragleaveEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondropAttr) {

        setAttributeEventListener(eventNames().dropEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragstartAttr) {

        setAttributeEventListener(eventNames().dragstartEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragAttr) {

        setAttributeEventListener(eventNames().dragEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ondragendAttr) {

        setAttributeEventListener(eventNames().dragendEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onselectstartAttr) {

        setAttributeEventListener(eventNames().selectstartEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onsubmitAttr) {

        setAttributeEventListener(eventNames().submitEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onerrorAttr) {

        setAttributeEventListener(eventNames().errorEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onwebkitanimationstartAttr) {

        setAttributeEventListener(eventNames().webkitAnimationStartEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onwebkitanimationiterationAttr) {

        setAttributeEventListener(eventNames().webkitAnimationIterationEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onwebkitanimationendAttr) {

        setAttributeEventListener(eventNames().webkitAnimationEndEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == onwebkittransitionendAttr) {

        setAttributeEventListener(eventNames().webkitTransitionEndEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == oninputAttr) {

        setAttributeEventListener(eventNames().inputEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == oninvalidAttr) {

        setAttributeEventListener(eventNames().invalidEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontouchstartAttr) {

        setAttributeEventListener(eventNames().touchstartEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontouchmoveAttr) {

        setAttributeEventListener(eventNames().touchmoveEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontouchendAttr) {

        setAttributeEventListener(eventNames().touchendEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontouchcancelAttr) {

        setAttributeEventListener(eventNames().touchcancelEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontransformactionstartAttr) {

        setAttributeEventListener(eventNames().transformactionstartEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontransformactionupdateAttr) {

        setAttributeEventListener(eventNames().transformactionupdateEvent, createAttributeEventListener(this, attr));

    } else if (attr->name() == ontransformactionendAttr) {

        setAttributeEventListener(eventNames().transformactionendEvent, createAttributeEventListener(this, attr));

    }

}

String HTMLElement::innerHTML() const

{

    return createMarkup(this, ChildrenOnly);

}

String HTMLElement::outerHTML() const

{

    return createMarkup(this);

}

PassRefPtr<DocumentFragment> HTMLElement::createContextualFragment(const String& markup, FragmentScriptingPermission scriptingPermission)

{

    // The following is in accordance with the definition as used by IE.

    if (endTagRequirement() == TagStatusForbidden)

        return 0;

    if (hasLocalName(colTag) || hasLocalName(colgroupTag) || hasLocalName(framesetTag) ||

        hasLocalName(headTag) || hasLocalName(styleTag) || hasLocalName(titleTag))

        return 0;

    return Element::createContextualFragment(markup, scriptingPermission);

}

static inline bool hasOneChild(ContainerNode* node)

{

    Node* firstChild = node->firstChild();

    return firstChild && !firstChild->nextSibling();

}

static inline bool hasOneTextChild(ContainerNode* node)

{

    return hasOneChild(node) && node->firstChild()->isTextNode();

}

static void replaceChildrenWithFragment(HTMLElement* element, PassRefPtr<DocumentFragment> fragment, ExceptionCode& ec)

{

    if (!fragment->firstChild()) {

        element->removeChildren();

        return;

    }

    if (hasOneTextChild(element) && hasOneTextChild(fragment.get())) {

        static_cast<Text*>(element->firstChild())->setData(static_cast<Text*>(fragment->firstChild())->data(), ec);

        return;

    }

    if (hasOneChild(element)) {

        element->replaceChild(fragment, element->firstChild(), ec);

        return;

    }

    element->removeChildren();

    element->appendChild(fragment, ec);

}

static void replaceChildrenWithText(HTMLElement* element, const String& text, ExceptionCode& ec)

{

    if (hasOneTextChild(element)) {

        static_cast<Text*>(element->firstChild())->setData(text, ec);

        return;

    }

    RefPtr<Text> textNode = Text::create(element->document(), text);

    if (hasOneChild(element)) {

        element->replaceChild(textNode.release(), element->firstChild(), ec);

        return;

    }

    element->removeChildren();

    element->appendChild(textNode.release(), ec);

}

void HTMLElement::setInnerHTML(const String& html, ExceptionCode& ec)

{

    if (hasLocalName(scriptTag) || hasLocalName(styleTag)) {

        // Script and CSS source shouldn't be parsed as HTML.

        removeChildren();

        appendChild(document()->createTextNode(html), ec);

        return;

    }

    RefPtr<DocumentFragment> fragment = createContextualFragment(html);

    if (!fragment) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    replaceChildrenWithFragment(this, fragment.release(), ec);

}

void HTMLElement::setOuterHTML(const String& html, ExceptionCode& ec)

{

    Node* p = parent();

    if (!p || !p->isHTMLElement()) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    HTMLElement* parent = static_cast<HTMLElement*>(p);

    RefPtr<DocumentFragment> fragment = parent->createContextualFragment(html);

    if (!fragment) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    // FIXME: Why doesn't this have code to merge neighboring text nodes the way setOuterText does?

    parent->replaceChild(fragment.release(), this, ec);

}

void HTMLElement::setInnerText(const String& text, ExceptionCode& ec)

{

    // Follow the IE specs about when this is allowed.

    if (endTagRequirement() == TagStatusForbidden) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    if (hasLocalName(colTag) || hasLocalName(colgroupTag) || hasLocalName(framesetTag) ||

        hasLocalName(headTag) || hasLocalName(htmlTag) || hasLocalName(tableTag) || 

        hasLocalName(tbodyTag) || hasLocalName(tfootTag) || hasLocalName(theadTag) ||

        hasLocalName(trTag)) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    // FIXME: This doesn't take whitespace collapsing into account at all.

    if (!text.contains('\n') && !text.contains('\r')) {

        if (text.isEmpty()) {

            removeChildren();

            return;

        }

        replaceChildrenWithText(this, text, ec);

        return;

    }

    // FIXME: Do we need to be able to detect preserveNewline style even when there's no renderer?

    // FIXME: Can the renderer be out of date here? Do we need to call updateStyleIfNeeded?

    // For example, for the contents of textarea elements that are display:none?

    RenderObject* r = renderer();

    if (r && r->style()->preserveNewline()) {

        if (!text.contains('\r')) {

            replaceChildrenWithText(this, text, ec);

            return;

        }

        String textWithConsistentLineBreaks = text;

        textWithConsistentLineBreaks.replace("\r\n", "\n");

        textWithConsistentLineBreaks.replace('\r', '\n');

        replaceChildrenWithText(this, textWithConsistentLineBreaks, ec);

        return;

    }

    // Add text nodes and <br> elements.

    ec = 0;

    RefPtr<DocumentFragment> fragment = DocumentFragment::create(document());

    int lineStart = 0;

    UChar prev = 0;

    int length = text.length();

    for (int i = 0; i < length; ++i) {

        UChar c = text[i];

        if (c == '\n' || c == '\r') {

            if (i > lineStart) {

                fragment->appendChild(Text::create(document(), text.substring(lineStart, i - lineStart)), ec);

                if (ec)

                    return;

            }

            if (!(c == '\n' && i != 0 && prev == '\r')) {

                fragment->appendChild(HTMLBRElement::create(document()), ec);

                if (ec)

                    return;

            }

            lineStart = i + 1;

        }

        prev = c;

    }

    if (length > lineStart)

        fragment->appendChild(Text::create(document(), text.substring(lineStart, length - lineStart)), ec);

    replaceChildrenWithFragment(this, fragment.release(), ec);

}

void HTMLElement::setOuterText(const String &text, ExceptionCode& ec)

{

    // Follow the IE specs about when this is allowed.

    if (endTagRequirement() == TagStatusForbidden) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    if (hasLocalName(colTag) || hasLocalName(colgroupTag) || hasLocalName(framesetTag) ||

        hasLocalName(headTag) || hasLocalName(htmlTag) || hasLocalName(tableTag) || 

        hasLocalName(tbodyTag) || hasLocalName(tfootTag) || hasLocalName(theadTag) ||

        hasLocalName(trTag)) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    Node* parent = parentNode();

    if (!parent) {

        ec = NO_MODIFICATION_ALLOWED_ERR;

        return;

    }

    // FIXME: This creates a new text node even when the text is empty.

    // FIXME: This creates a single text node even when the text has CR and LF

    // characters in it. Instead it should create <br> elements.

    RefPtr<Text> t = Text::create(document(), text);

    ec = 0;

    parent->replaceChild(t, this, ec);

    if (ec)

        return;

    // Is previous node a text node? If so, merge into it.

    Node* prev = t->previousSibling();

    if (prev && prev->isTextNode()) {

        Text* textPrev = static_cast<Text*>(prev);

        textPrev->appendData(t->data(), ec);

        if (ec)

            return;

        t->remove(ec);

        if (ec)

            return;

        t = textPrev;

    }

    // Is next node a text node? If so, merge it in.

    Node* next = t->nextSibling();

    if (next && next->isTextNode()) {

        Text* textNext = static_cast<Text*>(next);

        t->appendData(textNext->data(), ec);

        if (ec)

            return;

        textNext->remove(ec);

        if (ec)

            return;

    }

}

Node* HTMLElement::insertAdjacent(const String& where, Node* newChild, ExceptionCode& ec)

{

    // In Internet Explorer if the element has no parent and where is "beforeBegin" or "afterEnd",

    // a document fragment is created and the elements appended in the correct order. This document

    // fragment isn't returned anywhere.

    //

    // This is impossible for us to implement as the DOM tree does not allow for such structures,

    // Opera also appears to disallow such usage.

    if (equalIgnoringCase(where, "beforeBegin")) {

        if (Node* p = parent())

            return p->insertBefore(newChild, this, ec) ? newChild : 0;

        return 0;

    }

    if (equalIgnoringCase(where, "afterBegin"))

        return insertBefore(newChild, firstChild(), ec) ? newChild : 0;