1 /* 2 * Copyright (C) 2010 Google, 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 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 INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #ifndef HTMLViewSourceParser_h 27 #define HTMLViewSourceParser_h 28 29 #include "DecodedDataDocumentParser.h" 30 #include "HTMLInputStream.h" 31 #include "HTMLSourceTracker.h" 32 #include "HTMLToken.h" 33 #include "HTMLTokenizer.h" 34 #include "HTMLViewSourceDocument.h" 35 #include <wtf/PassOwnPtr.h> 36 37 namespace WebCore { 38 39 class HTMLTokenizer; 40 class HTMLScriptRunner; 41 class HTMLTreeBuilder; 42 class HTMLPreloadScanner; 43 class ScriptController; 44 class ScriptSourceCode; 45 46 class HTMLViewSourceParser : public DecodedDataDocumentParser { 47 public: create(HTMLViewSourceDocument * document)48 static PassRefPtr<HTMLViewSourceParser> create(HTMLViewSourceDocument* document) 49 { 50 return adoptRef(new HTMLViewSourceParser(document)); 51 } 52 virtual ~HTMLViewSourceParser(); 53 54 protected: 55 explicit HTMLViewSourceParser(HTMLViewSourceDocument*); 56 tokenizer()57 HTMLTokenizer* tokenizer() const { return m_tokenizer.get(); } 58 59 private: 60 // DocumentParser 61 virtual void insert(const SegmentedString&); 62 virtual void append(const SegmentedString&); 63 virtual void finish(); 64 virtual bool finishWasCalled(); 65 document()66 HTMLViewSourceDocument* document() const { return static_cast<HTMLViewSourceDocument*>(DecodedDataDocumentParser::document()); } 67 68 void pumpTokenizer(); 69 String sourceForToken(); 70 void updateTokenizerState(); 71 72 HTMLInputStream m_input; 73 HTMLToken m_token; 74 HTMLSourceTracker m_sourceTracker; 75 OwnPtr<HTMLTokenizer> m_tokenizer; 76 }; 77 78 } 79 80 #endif 81