/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "WebBrowserPersistSerializeParent.h"

#include "nsReadableUtils.h"
#include "nsThreadUtils.h"

namespace mozilla {

WebBrowserPersistSerializeParent::WebBrowserPersistSerializeParent(
    nsIWebBrowserPersistDocument* aDocument, nsIOutputStream* aStream,
    nsIWebBrowserPersistWriteCompletion* aFinish)
    : mDocument(aDocument),
      mStream(aStream),
      mFinish(aFinish),
      mOutputError(NS_OK) {
  MOZ_ASSERT(aDocument);
  MOZ_ASSERT(aStream);
  MOZ_ASSERT(aFinish);
}

WebBrowserPersistSerializeParent::~WebBrowserPersistSerializeParent() = default;

mozilla::ipc::IPCResult WebBrowserPersistSerializeParent::RecvWriteData(
    nsTArray<uint8_t>&& aData) {
  if (NS_FAILED(mOutputError)) {
    return IPC_OK();
  }

  uint32_t written = 0;
  static_assert(sizeof(char) == sizeof(uint8_t),
                "char must be (at least?) 8 bits");
  const char* data = reinterpret_cast<const char*>(aData.Elements());
  // nsIOutputStream::Write is allowed to return short writes.
  while (written < aData.Length()) {
    uint32_t writeReturn;
    nsresult rv =
        mStream->Write(data + written, aData.Length() - written, &writeReturn);
    if (NS_FAILED(rv)) {
      mOutputError = rv;
      return IPC_OK();
    }
    written += writeReturn;
  }
  return IPC_OK();
}

mozilla::ipc::IPCResult WebBrowserPersistSerializeParent::Recv__delete__(
    const nsACString& aContentType, const nsresult& aStatus) {
  if (NS_SUCCEEDED(mOutputError)) {
    mOutputError = aStatus;
  }
  mFinish->OnFinish(mDocument, mStream, aContentType, mOutputError);
  mFinish = nullptr;
  return IPC_OK();
}

void WebBrowserPersistSerializeParent::ActorDestroy(ActorDestroyReason aWhy) {
  if (mFinish) {
    MOZ_ASSERT(aWhy != Deletion);
    // See comment in WebBrowserPersistDocumentParent::ActorDestroy
    // (or bug 1202887) for why this is deferred.
    nsCOMPtr<nsIRunnable> errorLater =
        NewRunnableMethod<nsCOMPtr<nsIWebBrowserPersistDocument>,
                          nsCOMPtr<nsIOutputStream>, nsCString, nsresult>(
            "nsIWebBrowserPersistWriteCompletion::OnFinish", mFinish,
            &nsIWebBrowserPersistWriteCompletion::OnFinish, mDocument, mStream,
            ""_ns, NS_ERROR_FAILURE);
    NS_DispatchToCurrentThread(errorLater);
    mFinish = nullptr;
  }
}

}  // namespace mozilla