/*

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

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

 *           (C) 2001 Dirk Mueller (mueller@kde.org)

 *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)

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

 *

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

 *

 */

#ifndef Document_h

#define Document_h

#include "Attr.h"

#include "Color.h"

#include "DeprecatedPtrList.h"

#include "DeprecatedValueList.h"

#include "DocumentMarker.h"

#include "HTMLCollection.h"

#include "HTMLFormElement.h"

#include "KURL.h"

#include "SecurityOrigin.h"

#include "StringHash.h"

#include "Timer.h"

#include <wtf/HashCountedSet.h>

#include <wtf/ListHashSet.h>

namespace WebCore {

    class AXObjectCache;

    class Attr;

    class Attribute;

    class CDATASection;

    class CSSStyleDeclaration;

    class CSSStyleSelector;

    class CSSStyleSheet;

    class Comment;

    class DOMImplementation;

    class DOMWindow;

    class DocLoader;

    class DocumentFragment;

    class DocumentType;

    class EditingText;

    class Element;

    class EntityReference;

    class Event;

    class EventListener;

    class Frame;

    class FrameView;

    class HTMLDocument;

    class HTMLElement;

    class HTMLFormControlElementWithState;

    class HTMLFormElement;

    class HTMLHeadElement;

    class HTMLImageLoader;

    class HTMLInputElement;

    class HTMLMapElement;

    class IntPoint;

    class JSEditor;

    class MouseEventWithHitTestResults;

    class NameNodeList;

    class NodeFilter;

    class NodeIterator;

    class NodeList;

    class Page;

    class PlatformMouseEvent;

    class ProcessingInstruction;

    class Range;

    class RegisteredEventListener;

    class RenderArena;

    class Settings;

    class StyleSheet;

    class StyleSheetList;

    class Text;

    class TextResourceDecoder;

    class Tokenizer;

    class TreeWalker;

#if ENABLE(XBL)

    class XBLBindingManager;

#endif

#if ENABLE(XPATH)

    class XPathEvaluator;

    class XPathExpression;

    class XPathNSResolver;

    class XPathResult;

#endif

    struct DashboardRegionValue;

    struct HitTestRequest;

#if ENABLE(SVG)

    class SVGDocumentExtensions;

#endif

    typedef int ExceptionCode;

class FormElementKey {

public:

    FormElementKey(AtomicStringImpl* = 0, AtomicStringImpl* = 0);

    ~FormElementKey();

    FormElementKey(const FormElementKey&);

    FormElementKey& operator=(const FormElementKey&);

    AtomicStringImpl* name() const { return m_name; }

    AtomicStringImpl* type() const { return m_type; }

private:

    void ref() const;

    void deref() const;

    AtomicStringImpl* m_name;

    AtomicStringImpl* m_type;

};

inline bool operator==(const FormElementKey& a, const FormElementKey& b)

    { return a.name() == b.name() && a.type() == b.type(); }

struct FormElementKeyHash {

    static unsigned hash(const FormElementKey&);

    static bool equal(const FormElementKey& a, const FormElementKey& b) { return a == b; }

};

struct FormElementKeyHashTraits : WTF::GenericHashTraits<FormElementKey> {

    static FormElementKey deletedValue();

};

class Document : public ContainerNode {

public:

    Document(DOMImplementation*, Frame*, bool isXHTML = false);

    virtual ~Document();

    virtual void removedLastRef();

    // Nodes belonging to this document hold "self-only" references -

    // these are enough to keep the document from being destroyed, but

    // not enough to keep it from removing its children. This allows a

    // node that outlives its document to still have a valid document

    // pointer without introducing reference cycles

    void selfOnlyRef()

    {

        ASSERT(!m_deletionHasBegun);

        ++m_selfOnlyRefCount;

    }

    void selfOnlyDeref()

    {

        ASSERT(!m_deletionHasBegun);

        --m_selfOnlyRefCount;

        if (!m_selfOnlyRefCount && !refCount()) {

#ifndef NDEBUG

            m_deletionHasBegun = true;

#endif

            delete this;

        }

    }

    // DOM methods & attributes for Document

    virtual DocumentType* doctype() const; // May return 0 for HTML documents

    DocumentType* realDocType() const { return m_docType.get(); }

    DOMImplementation* implementation() const;

    virtual void childrenChanged();

    Element* documentElement() const;

    virtual PassRefPtr<Element> createElement(const String& tagName, ExceptionCode&);

    PassRefPtr<DocumentFragment> createDocumentFragment ();

    PassRefPtr<Text> createTextNode(const String& data);

    PassRefPtr<Comment> createComment(const String& data);

    PassRefPtr<CDATASection> createCDATASection(const String& data, ExceptionCode&);

    PassRefPtr<ProcessingInstruction> createProcessingInstruction(const String& target, const String& data, ExceptionCode&);

    PassRefPtr<Attr> createAttribute(const String& name, ExceptionCode& ec) { return createAttributeNS(String(), name, ec); }

    PassRefPtr<Attr> createAttributeNS(const String& namespaceURI, const String& qualifiedName, ExceptionCode&);

    PassRefPtr<EntityReference> createEntityReference(const String& name, ExceptionCode&);

    PassRefPtr<Node> importNode(Node* importedNode, bool deep, ExceptionCode&);

    virtual PassRefPtr<Element> createElementNS(const String& namespaceURI, const String& qualifiedName, ExceptionCode&);

    PassRefPtr<Element> createElement(const QualifiedName&, bool createdByParser, ExceptionCode& ec);

    Element* getElementById(const AtomicString&) const;

    Element* elementFromPoint(int x, int y) const;

    String readyState() const;

    String inputEncoding() const;

    String defaultCharset() const;

    String charset() const { return inputEncoding(); }

    String characterSet() const { return inputEncoding(); }

    void setCharset(const String&);

    String xmlEncoding() const { return m_xmlEncoding; }

    String xmlVersion() const { return m_xmlVersion; }

    bool xmlStandalone() const { return m_xmlStandalone; }

    void setXMLEncoding(const String& encoding) { m_xmlEncoding = encoding; } // read-only property, only to be set from XMLTokenizer

    void setXMLVersion(const String&, ExceptionCode&);

    void setXMLStandalone(bool, ExceptionCode&);

    String documentURI() const;

    void setDocumentURI(const String&);

    virtual String baseURI() const;

    PassRefPtr<Node> adoptNode(PassRefPtr<Node> source, ExceptionCode&);

    PassRefPtr<NameNodeList> getElementsByName(const String& elementName);

    // Actually part of JSHTMLDocument, but used for giving XML documents a window title as well

    String title() const { return m_title; }

    void setTitle(const String&, Element* titleElement = 0);

    void removeTitle(Element* titleElement);

    PassRefPtr<HTMLCollection> images();

    PassRefPtr<HTMLCollection> embeds();

    PassRefPtr<HTMLCollection> plugins(); // an alias for embeds() required for the JS DOM bindings.

    PassRefPtr<HTMLCollection> applets();

    PassRefPtr<HTMLCollection> links();

    PassRefPtr<HTMLCollection> forms();

    PassRefPtr<HTMLCollection> anchors();

    PassRefPtr<HTMLCollection> all();

    PassRefPtr<HTMLCollection> objects();

    PassRefPtr<HTMLCollection> scripts();

    PassRefPtr<HTMLCollection> windowNamedItems(const String& name);

    PassRefPtr<HTMLCollection> documentNamedItems(const String& name);

    HTMLCollection::CollectionInfo* collectionInfo(HTMLCollection::Type type)

    {

        if ((int)type < HTMLCollection::UnnamedCollectionTypes) 

            return m_collectionInfo + type; 

        return 0;

    }

    HTMLCollection::CollectionInfo* nameCollectionInfo(HTMLCollection::Type type, const String& name);

    // DOM methods overridden from  parent classes

    virtual String nodeName() const;

    virtual NodeType nodeType() const;

    // Other methods (not part of DOM)

    virtual bool isDocumentNode() const { return true; }

    virtual bool isHTMLDocument() const { return false; }

    virtual bool isImageDocument() const { return false; }

#if ENABLE(SVG)

    virtual bool isSVGDocument() const { return false; }

#endif

    virtual bool isPluginDocument() const { return false; }

    CSSStyleSelector* styleSelector() const { return m_styleSelector; }

    Element* getElementByAccessKey(const String& key) const;

    /**

     * Updates the pending sheet count and then calls updateStyleSelector.

     */

    void removePendingSheet();

    /**

     * This method returns true if all top-level stylesheets have loaded (including

     * any @imports that they may be loading).

     */

    bool haveStylesheetsLoaded() const

    {

        return m_pendingStylesheets <= 0 || m_ignorePendingStylesheets

#if USE(LOW_BANDWIDTH_DISPLAY)

            || m_inLowBandwidthDisplay

#endif

            ;

    }

    /**

     * Increments the number of pending sheets.  The <link> elements

     * invoke this to add themselves to the loading list.

     */

    void addPendingSheet() { m_pendingStylesheets++; }

    bool gotoAnchorNeededAfterStylesheetsLoad() { return m_gotoAnchorNeededAfterStylesheetsLoad; }

    void setGotoAnchorNeededAfterStylesheetsLoad(bool b) { m_gotoAnchorNeededAfterStylesheetsLoad = b; }

    /**

     * Called when one or more stylesheets in the document may have been added, removed or changed.

     *

     * Creates a new style selector and assign it to this document. This is done by iterating through all nodes in

     * document (or those before <BODY> in a HTML document), searching for stylesheets. Stylesheets can be contained in

     * <LINK>, <STYLE> or <BODY> elements, as well as processing instructions (XML documents only). A list is

     * constructed from these which is used to create the a new style selector which collates all of the stylesheets

     * found and is used to calculate the derived styles for all rendering objects.

     */

    void updateStyleSelector();

    void recalcStyleSelector();

    bool usesDescendantRules() const { return m_usesDescendantRules; }

    void setUsesDescendantRules(bool b) { m_usesDescendantRules = b; }

    bool usesSiblingRules() const { return m_usesSiblingRules; }

    void setUsesSiblingRules(bool b) { m_usesSiblingRules = b; }

    bool usesFirstLineRules() const { return m_usesFirstLineRules; }

    void setUsesFirstLineRules(bool b) { m_usesFirstLineRules = b; }

    bool usesFirstLetterRules() const { return m_usesFirstLetterRules; }

    void setUsesFirstLetterRules(bool b) { m_usesFirstLetterRules = b; }

    // Machinery for saving and restoring state when you leave and then go back to a page.

    void registerFormElementWithState(HTMLFormControlElementWithState* e) { m_formElementsWithState.add(e); }

    void unregisterFormElementWithState(HTMLFormControlElementWithState* e) { m_formElementsWithState.remove(e); }

    Vector<String> formElementsState() const;

    void setStateForNewFormElements(const Vector<String>&);

    bool hasStateForNewFormElements() const;

    bool takeStateForFormElement(AtomicStringImpl* name, AtomicStringImpl* type, String& state);

    FrameView* view() const; // can be NULL

    Frame* frame() const { return m_frame; } // can be NULL

    Page* page() const; // can be NULL

    Settings* settings() const; // can be NULL

    PassRefPtr<Range> createRange();

    PassRefPtr<NodeIterator> createNodeIterator(Node* root, unsigned whatToShow,

        NodeFilter*, bool expandEntityReferences, ExceptionCode&);

    PassRefPtr<TreeWalker> createTreeWalker(Node* root, unsigned whatToShow, 

        NodeFilter*, bool expandEntityReferences, ExceptionCode&);

    // Special support for editing

    PassRefPtr<CSSStyleDeclaration> createCSSStyleDeclaration();

    PassRefPtr<EditingText> createEditingTextNode(const String&);

    virtual void recalcStyle( StyleChange = NoChange );

    static DeprecatedPtrList<Document>* changedDocuments;

    virtual void updateRendering();

    void updateLayout();

    void updateLayoutIgnorePendingStylesheets();

    static void updateDocumentsRendering();

    DocLoader* docLoader() { return m_docLoader; }

    virtual void attach();

    virtual void detach();

    RenderArena* renderArena() { return m_renderArena; }

    AXObjectCache* axObjectCache() const;

    // to get visually ordered hebrew and arabic pages right

    void setVisuallyOrdered();

    void updateSelection();

    void open();

    void implicitOpen();

    void close();

    void implicitClose();

    void cancelParsing();

    void write(const String& text);

    void write(const DeprecatedString &text);

    void writeln(const String& text);

    void finishParsing();

    void clear();

    bool wellFormed() const { return m_wellFormed; }

    DeprecatedString URL() const { return m_url.isEmpty() ? "about:blank" : m_url; }

    void setURL(const DeprecatedString& url);

    DeprecatedString baseURL() const { return m_baseURL.isEmpty() ? URL() : m_baseURL; }

    void setBaseURL(const DeprecatedString& baseURL);

    String baseTarget() const { return m_baseTarget; }

    void setBaseTarget(const String& baseTarget) { m_baseTarget = baseTarget; }

    DeprecatedString completeURL(const DeprecatedString &);

    String completeURL(const String&);

    // from cachedObjectClient

    virtual void setCSSStyleSheet(const String& url, const String& charset, const String& sheetStr);

    void setUserStyleSheet(const String& sheet);

    const String& userStyleSheet() const { return m_usersheet; }

    void setPrintStyleSheet(const String& sheet) { m_printSheet = sheet; }

    const String& printStyleSheet() const { return m_printSheet; }

    CSSStyleSheet* elementSheet();

    virtual Tokenizer* createTokenizer();

    Tokenizer* tokenizer() { return m_tokenizer; }

    bool printing() const { return m_printing; }

    void setPrinting(bool p) { m_printing = p; }

    enum ParseMode { Compat, AlmostStrict, Strict };

    virtual void determineParseMode(const String&);

    void setParseMode(ParseMode m) { pMode = m; }

    ParseMode parseMode() const { return pMode; }

    bool inCompatMode() const { return pMode == Compat; }

    bool inAlmostStrictMode() const { return pMode == AlmostStrict; }

    bool inStrictMode() const { return pMode == Strict; }

    enum HTMLMode { Html3, Html4, XHtml };

    void setHTMLMode(HTMLMode m) { hMode = m; }

    HTMLMode htmlMode() const { return hMode; }

    void setParsing(bool);

    bool parsing() const { return m_bParsing; }

    int minimumLayoutDelay();

    bool shouldScheduleLayout();

    int elapsedTime() const;

    void setTextColor(const Color& color) { m_textColor = color; }

    Color textColor() const { return m_textColor; }

    const Color& linkColor() const { return m_linkColor; }

    const Color& visitedLinkColor() const { return m_visitedLinkColor; }

    const Color& activeLinkColor() const { return m_activeLinkColor; }

    void setLinkColor(const Color& c) { m_linkColor = c; }

    void setVisitedLinkColor(const Color& c) { m_visitedLinkColor = c; }

    void setActiveLinkColor(const Color& c) { m_activeLinkColor = c; }

    void resetLinkColor();

    void resetVisitedLinkColor();

    void resetActiveLinkColor();

    MouseEventWithHitTestResults prepareMouseEvent(const HitTestRequest&, const IntPoint&, const PlatformMouseEvent&);

    virtual bool childTypeAllowed(NodeType);

    virtual PassRefPtr<Node> cloneNode(bool deep);

    virtual bool canReplaceChild(Node* newChild, Node* oldChild);

    StyleSheetList* styleSheets();

    /* Newly proposed CSS3 mechanism for selecting alternate

       stylesheets using the DOM. May be subject to change as

       spec matures. - dwh

    */

    String preferredStylesheetSet() const;

    String selectedStylesheetSet() const;

    void setSelectedStylesheetSet(const String&);

    bool setFocusedNode(PassRefPtr<Node>);

    Node* focusedNode() const { return m_focusedNode.get(); }

    void setHoverNode(PassRefPtr<Node>);

    Node* hoverNode() const { return m_hoverNode.get(); }

    void setActiveNode(PassRefPtr<Node>);

    Node* activeNode() const { return m_activeNode.get(); }

    void focusedNodeRemoved(Node*);

    void hoveredNodeDetached(Node*);

    void activeChainNodeDetached(Node*);

    // Updates for :target (CSS3 selector).

    void setCSSTarget(Node*);

    Node* getCSSTarget() const;

    void setDocumentChanged(bool);

    void attachNodeIterator(NodeIterator*);

    void detachNodeIterator(NodeIterator*);

    void notifyBeforeNodeRemoval(Node*);

    DOMWindow* defaultView() const;

    PassRefPtr<Event> createEvent(const String& eventType, ExceptionCode&);

    // keep track of what types of event listeners are registered, so we don't

    // dispatch events unnecessarily

    enum ListenerType {

        DOMSUBTREEMODIFIED_LISTENER          = 0x01,

        DOMNODEINSERTED_LISTENER             = 0x02,

        DOMNODEREMOVED_LISTENER              = 0x04,

        DOMNODEREMOVEDFROMDOCUMENT_LISTENER  = 0x08,

        DOMNODEINSERTEDINTODOCUMENT_LISTENER = 0x10,

        DOMATTRMODIFIED_LISTENER             = 0x20,

        DOMCHARACTERDATAMODIFIED_LISTENER    = 0x40,

        OVERFLOWCHANGED_LISTENER             = 0x80

    };

    bool hasListenerType(ListenerType listenerType) const { return (m_listenerTypes & listenerType); }

    void addListenerType(ListenerType listenerType) { m_listenerTypes = m_listenerTypes | listenerType; }

    CSSStyleDeclaration* getOverrideStyle(Element*, const String& pseudoElt);

    virtual void defaultEventHandler(Event*);

    void handleWindowEvent(Event*, bool useCapture);

    void setHTMLWindowEventListener(const AtomicString &eventType, PassRefPtr<EventListener>);

    EventListener* getHTMLWindowEventListener(const AtomicString &eventType);

    void removeHTMLWindowEventListener(const AtomicString &eventType);

    void setHTMLWindowEventListener(const AtomicString& eventType, Attribute*);

    void addWindowEventListener(const AtomicString& eventType, PassRefPtr<EventListener>, bool useCapture);

    void removeWindowEventListener(const AtomicString& eventType, EventListener*, bool useCapture);

    bool hasWindowEventListener(const AtomicString& eventType);

    PassRefPtr<EventListener> createHTMLEventListener(const String& functionName, const String& code, Node*);

    /**

     * Searches through the document, starting from fromNode, for the next selectable element that comes after fromNode.

     * The order followed is as specified in section 17.11.1 of the HTML4 spec, which is elements with tab indexes

     * first (from lowest to highest), and then elements without tab indexes (in document order).

     *

     * @param fromNode The node from which to start searching. The node after this will be focused. May be null.

     *

     * @return The focus node that comes after fromNode

     *

     * See http://www.w3.org/TR/html4/interact/forms.html#h-17.11.1

     */

    Node* nextFocusableNode(Node* start, KeyboardEvent*);

    /**

     * Searches through the document, starting from fromNode, for the previous selectable element (that comes _before_)

     * fromNode. The order followed is as specified in section 17.11.1 of the HTML4 spec, which is elements with tab

     * indexes first (from lowest to highest), and then elements without tab indexes (in document order).

     *

     * @param fromNode The node from which to start searching. The node before this will be focused. May be null.

     *

     * @return The focus node that comes before fromNode

     *

     * See http://www.w3.org/TR/html4/interact/forms.html#h-17.11.1