Refactoring: Simplify DocumentWriter by reorganizing its lifetime.

Source/core/loader/DocumentWriter.h

Index: Source/core/loader/DocumentWriter.h
diff --git a/Source/core/loader/DocumentWriter.h b/Source/core/loader/DocumentWriter.h
index 56c8aabd74ebe92c5c721b87e4a9ea6a80a6aa30..35667be2805a7ff158f453a211ba7c12c775f1a0 100644
--- a/Source/core/loader/DocumentWriter.h
+++ b/Source/core/loader/DocumentWriter.h
@@ -30,6 +30,7 @@
 #define DocumentWriter_h
 
 #include "core/loader/TextResourceDecoderBuilder.h"
+#include "wtf/RefCounted.h"
 #include "wtf/text/WTFString.h"
 
 namespace WebCore {
@@ -41,49 +42,43 @@ class KURL;
 class SecurityOrigin;
 class TextResourceDecoder;
 
-class DocumentWriter {
+class DocumentWriter : public RefCounted<DocumentWriter> {
     WTF_MAKE_NONCOPYABLE(DocumentWriter);
 public:
-    explicit DocumentWriter(Frame*);
+    static PassRefPtr<DocumentWriter> create(Document*, const String& mimeType = "", const String& encoding = "", bool encodingUserChoosen = false);
+
+    ~DocumentWriter();
+
+    void end();
 
     // This is only called by ScriptController::executeScriptIfJavaScriptURL
     // and always contains the result of evaluating a javascript: url.
     void replaceDocument(const String&, Document* ownerDocument);
 
-    void begin();
-    void begin(const KURL&, bool dispatchWindowObjectAvailable = true, Document* ownerDocument = 0);
     void addData(const char* bytes, size_t length);
-    void end();
     
-    void setFrame(Frame* frame) { m_frame = frame; }
-
     const String& mimeType() const { return m_decoderBuilder.mimeType(); }
-    void setMIMEType(const String& type) { m_decoderBuilder.setMIMEType(type); }
-    void setEncoding(const String& encoding, bool userChosen) { m_decoderBuilder.setEncoding(encoding, userChosen); }
+    const String& encoding() const { return m_decoderBuilder.encoding(); }
+    bool encodingWasChosenByUser() const { return m_decoderBuilder.encodingWasChosenByUser(); }
 
     // Exposed for DocumentParser::appendBytes.
     void reportDataReceived();
+    // Exposed for DocumentLoader::replaceDocument.
+    void appendReplacingData(const String&);
 
     void setDocumentWasLoadedAsPartOfNavigation();
 
 private:
-    PassRefPtr<Document> createDocument(const KURL&);
-    void clear();
+    DocumentWriter(Document*, const String& mimeType, const String& encoding, bool encodingUserChoosen);
 
-    Frame* m_frame;
+    PassRefPtr<Document> createDocument(const KURL&);
 
+    Document* m_document;
     bool m_hasReceivedSomeData;
     TextResourceDecoderBuilder m_decoderBuilder;
 
     RefPtr<TextResourceDecoder> m_decoder;
     RefPtr<DocumentParser> m_parser;
-
-    enum WriterState {
-        NotStartedWritingState,
-        StartedWritingState,
-        FinishedWritingState,
-    };
-    WriterState m_state;
 };
 
 } // namespace WebCore