/*
 * Copyright (C) 2010. Adam Barth. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1.  Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 * 2.  Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
 *     its contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, 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 "core/loader/DocumentWriter.h"

#include "core/dom/Document.h"
#include "core/dom/ScriptableDocumentParser.h"
#include "core/frame/FrameView.h"
#include "core/frame/LocalDOMWindow.h"
#include "core/frame/LocalFrame.h"
#include "core/frame/Settings.h"
#include "core/html/parser/TextResourceDecoder.h"
#include "core/loader/FrameLoader.h"
#include "core/loader/FrameLoaderStateMachine.h"
#include "platform/weborigin/KURL.h"
#include "platform/weborigin/SecurityOrigin.h"
#include <memory>

namespace blink {

DocumentWriter* DocumentWriter::create(
    Document* document,
    ParserSynchronizationPolicy parsingPolicy,
    const AtomicString& mimeType,
    const AtomicString& encoding) {
  return new DocumentWriter(document, parsingPolicy, mimeType, encoding);
}

DocumentWriter::DocumentWriter(Document* document,
                               ParserSynchronizationPolicy parserSyncPolicy,
                               const AtomicString& mimeType,
                               const AtomicString& encoding)
    : m_document(document),
      m_decoderBuilder(mimeType, encoding),
      // We grab a reference to the parser so that we'll always send data to the
      // original parser, even if the document acquires a new parser (e.g., via
      // document.open).
      m_parser(m_document->implicitOpen(parserSyncPolicy)) {
  if (m_document->frame()) {
    if (FrameView* view = m_document->frame()->view())
      view->setContentsSize(IntSize());
  }
}

DocumentWriter::~DocumentWriter() {}

DEFINE_TRACE(DocumentWriter) {
  visitor->trace(m_document);
  visitor->trace(m_parser);
}

void DocumentWriter::appendReplacingData(const String& source) {
  m_document->setCompatibilityMode(Document::NoQuirksMode);

  // FIXME: This should call DocumentParser::appendBytes instead of append
  // to support RawDataDocumentParsers.
  if (DocumentParser* parser = m_document->parser())
    parser->append(source);
}

void DocumentWriter::addData(const char* bytes, size_t length) {
  DCHECK(m_parser);
  if (m_parser->needsDecoder() && 0 < length) {
    std::unique_ptr<TextResourceDecoder> decoder =
        m_decoderBuilder.buildFor(m_document);
    m_parser->setDecoder(std::move(decoder));
  }
  // appendBytes() can result replacing DocumentLoader::m_writer.
  m_parser->appendBytes(bytes, length);
}

void DocumentWriter::end() {
  DCHECK(m_document);

  if (!m_parser)
    return;

  if (m_parser->needsDecoder()) {
    std::unique_ptr<TextResourceDecoder> decoder =
        m_decoderBuilder.buildFor(m_document);
    m_parser->setDecoder(std::move(decoder));
  }

  m_parser->finish();
  m_parser = nullptr;
  m_document = nullptr;
}

void DocumentWriter::setDocumentWasLoadedAsPartOfNavigation() {
  DCHECK(m_parser);
  DCHECK(!m_parser->isStopped());
  m_parser->setDocumentWasLoadedAsPartOfNavigation();
}

}  // namespace blink