1/*
2 * Copyright (C) 2006, 2008, 2009 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 *    notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 *    notice, this list of conditions and the following disclaimer in the
11 *    documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
20 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
22 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25#ifndef HTMLViewSourceDocument_h
26#define HTMLViewSourceDocument_h
27
28#include "HTMLDocument.h"
29
30namespace WebCore {
31
32class HTMLTableCellElement;
33class HTMLTableSectionElement;
34class HTMLToken;
35
36class HTMLViewSourceDocument FINAL : public HTMLDocument {
37public:
38    static PassRefPtr<HTMLViewSourceDocument> create(Frame* frame, const KURL& url, const String& mimeType)
39    {
40        return adoptRef(new HTMLViewSourceDocument(frame, url, mimeType));
41    }
42
43    void addSource(const String&, HTMLToken&);
44
45private:
46    HTMLViewSourceDocument(Frame*, const KURL&, const String& mimeType);
47
48    // Returns HTMLViewSourceParser or TextDocumentParser based on m_type.
49    virtual PassRefPtr<DocumentParser> createParser();
50
51    void processDoctypeToken(const String& source, HTMLToken&);
52    void processTagToken(const String& source, HTMLToken&);
53    void processCommentToken(const String& source, HTMLToken&);
54    void processCharacterToken(const String& source, HTMLToken&);
55
56    void createContainingTable();
57    PassRefPtr<Element> addSpanWithClassName(const AtomicString&);
58    void addLine(const AtomicString& className);
59    void finishLine();
60    void addText(const String& text, const AtomicString& className);
61    int addRange(const String& source, int start, int end, const String& className, bool isLink = false, bool isAnchor = false, const String& link = String());
62    PassRefPtr<Element> addLink(const AtomicString& url, bool isAnchor);
63    PassRefPtr<Element> addBase(const AtomicString& href);
64
65    String m_type;
66    RefPtr<Element> m_current;
67    RefPtr<HTMLTableSectionElement> m_tbody;
68    RefPtr<HTMLTableCellElement> m_td;
69};
70
71}
72
73#endif // HTMLViewSourceDocument_h
74