Live the dream

sky/engine/core/html/imports/HTMLImportLoader.cpp

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 14 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 #include "core/html/imports/HTMLImportLoader.h"
 
 #include "core/dom/Document.h"
+#include "core/dom/DocumentParser.h"
 #include "core/dom/StyleEngine.h"
 #include "core/dom/custom/CustomElementSyncMicrotaskQueue.h"
 #include "core/html/HTMLDocument.h"
 #include "core/html/imports/HTMLImportChild.h"
 #include "core/html/imports/HTMLImportsController.h"
-#include "core/loader/DocumentWriter.h"
-
 
 namespace blink {
 
 HTMLImportLoader::HTMLImportLoader(HTMLImportsController* controller)
     : m_controller(controller)
     , m_state(StateLoading)
     , m_microtaskQueue(CustomElementSyncMicrotaskQueue::create())
 {
 }
 
@@ skipping 12 matching lines @@
 
 void HTMLImportLoader::clear()
 {
     m_controller = nullptr;
     if (m_document) {
         m_document->setImportsController(0);
         m_document->cancelParsing();
         m_document.clear();
     }
     m_fetcher.clear();
-    m_drainer.clear();
 }
 #endif
 
 void HTMLImportLoader::startLoading(const KURL& url)
 {
     m_fetcher = adoptPtr(new MojoFetcher(this, url));
 }
 
 void HTMLImportLoader::OnReceivedResponse(mojo::URLResponsePtr response)
 {
     if (response->error || response->status_code >= 400) {
         setState(StateError);
         return;
     }
-    mojo::ScopedDataPipeConsumerHandle body = response->body.Pass();
     setState(startWritingAndParsing(response.Pass()));
-    m_drainer = adoptPtr(new DataPipeDrainer(this, body.Pass()));
-}
-
-void HTMLImportLoader::OnDataAvailable(const void* data, size_t length)
-{
-    RefPtrWillBeRawPtr<DocumentWriter> protectingWriter(m_writer.get());
-    m_writer->addData(static_cast<const char*>(data), length);
-}
-
-void HTMLImportLoader::OnDataComplete()
-{
-    setState(finishWriting());
 }
 
 HTMLImportLoader::State HTMLImportLoader::startWritingAndParsing(mojo::URLResponsePtr response)
 {
     ASSERT(!m_imports.isEmpty());
     KURL url(ParsedURLString, String::fromUTF8(response->url));
     DocumentInit init = DocumentInit(url, 0, m_controller->master()->contextDocument(), m_controller)
         .withRegistrationContext(m_controller->master()->registrationContext());
     m_document = HTMLDocument::create(init);
-    m_writer = DocumentWriter::create(m_document.get());
-
+    m_document->startParsing();
+    m_document->parser()->parse(response->body.Pass());
     return StateLoading;
 }
 
 HTMLImportLoader::State HTMLImportLoader::finishWriting()
 {
     return StateWritten;
 }
 
 HTMLImportLoader::State HTMLImportLoader::finishParsing()
 {
     return StateParsed;
 }
 
 HTMLImportLoader::State HTMLImportLoader::finishLoading()
 {
     return StateLoaded;
 }
 
 void HTMLImportLoader::setState(State state)
 {
     if (m_state == state)
         return;
 
     m_state = state;
 
-    if (m_state == StateParsed || m_state == StateError || m_state == StateWritten) {
-        if (RefPtrWillBeRawPtr<DocumentWriter> writer = m_writer.release())
-            writer->end();
-    }
+    if (m_state == StateParsed || m_state == StateError || m_state == StateWritten)
+        m_document->cancelParsing();
 
     // Since DocumentWriter::end() can let setState() reenter, we shouldn't refer to m_state here.
     if (state == StateLoaded || state == StateError)
         didFinishLoading();
 }
 
 void HTMLImportLoader::didFinishParsing()
 {
     setState(finishParsing());
     setState(finishLoading());
@@ skipping 49 matching lines @@
     return m_microtaskQueue;
 }
 
 void HTMLImportLoader::trace(Visitor* visitor)
 {
     visitor->trace(m_controller);
 #if ENABLE(OILPAN)
     visitor->trace(m_imports);
 #endif
     visitor->trace(m_document);
-    visitor->trace(m_writer);
     visitor->trace(m_microtaskQueue);
 }
 
 } // namespace blink