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 DoctypeToken;
33class HTMLTableCellElement;
34class HTMLTableSectionElement;
35
36struct Token;
37
38class HTMLViewSourceDocument : public HTMLDocument {
39public:
40    static PassRefPtr<HTMLViewSourceDocument> create(Frame* frame, const String& mimeType)
41    {
42        return adoptRef(new HTMLViewSourceDocument(frame, mimeType));
43    }
44
45    void addViewSourceToken(Token*); // Used by the HTML tokenizer.
46    void addViewSourceText(const String&); // Used by the plaintext tokenizer.
47    void addViewSourceDoctypeToken(DoctypeToken*);
48
49private:
50    HTMLViewSourceDocument(Frame*, const String& mimeType);
51
52    // Returns HTMLTokenizer or TextTokenizer based on m_type.
53    virtual Tokenizer* createTokenizer();
54
55    void createContainingTable();
56    PassRefPtr<Element> addSpanWithClassName(const String&);
57    void addLine(const String& className);
58    void addText(const String& text, const String& className);
59    PassRefPtr<Element> addLink(const String& url, bool isAnchor);
60
61    String m_type;
62    RefPtr<Element> m_current;
63    RefPtr<HTMLTableSectionElement> m_tbody;
64    RefPtr<HTMLTableCellElement> m_td;
65};
66
67}
68
69#endif // HTMLViewSourceDocument_h
70