/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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 "IPCStreamDestination.h"
#include "mozilla/dom/nsIContentParent.h"
#include "mozilla/ipc/PBackgroundParent.h"
#include "mozilla/ipc/PChildToParentStreamParent.h"
#include "mozilla/ipc/PParentToChildStreamParent.h"
#include "mozilla/Unused.h"

namespace mozilla {
namespace ipc {

// Child to Parent implementation
// ----------------------------------------------------------------------------

namespace {

class IPCStreamSourceParent final : public PParentToChildStreamParent
                                  , public IPCStreamSource
{
public:
  static IPCStreamSourceParent*
  Create(nsIAsyncInputStream* aInputStream)
  {
    MOZ_ASSERT(aInputStream);

    IPCStreamSourceParent* source = new IPCStreamSourceParent(aInputStream);
    if (!source->Initialize()) {
      delete source;
      return nullptr;
    }

    return source;
  }

  // PParentToChildStreamParent methods

  void
  ActorDestroy(ActorDestroyReason aReason) override
  {
    ActorDestroyed();
  }

  IPCResult
  RecvStartReading() override
  {
    Start();
    return IPC_OK();
  }

  IPCResult
  RecvRequestClose(const nsresult& aRv) override
  {
    OnEnd(aRv);
    return IPC_OK();
  }

  void
  Close(nsresult aRv) override
  {
    MOZ_ASSERT(IPCStreamSource::mState == IPCStreamSource::eClosed);
    Unused << SendClose(aRv);
  }

  void
  SendData(const wr::ByteBuffer& aBuffer) override
  {
    Unused << SendBuffer(aBuffer);
  }

private:
  explicit IPCStreamSourceParent(nsIAsyncInputStream* aInputStream)
    :IPCStreamSource(aInputStream)
  {}
};

} // anonymous namespace

/* static */ PParentToChildStreamParent*
IPCStreamSource::Create(nsIAsyncInputStream* aInputStream,
                        dom::nsIContentParent* aManager)
{
  MOZ_ASSERT(aInputStream);
  MOZ_ASSERT(aManager);

  // PContent can only be used on the main thread
  MOZ_ASSERT(NS_IsMainThread());

  IPCStreamSourceParent* source = IPCStreamSourceParent::Create(aInputStream);
  if (!source) {
    return nullptr;
  }

  if (!aManager->SendPParentToChildStreamConstructor(source)) {
    // no delete here, the manager will delete the actor for us.
    return nullptr;
  }

  source->ActorConstructed();
  return source;
}

/* static */ PParentToChildStreamParent*
IPCStreamSource::Create(nsIAsyncInputStream* aInputStream,
                        PBackgroundParent* aManager)
{
  MOZ_ASSERT(aInputStream);
  MOZ_ASSERT(aManager);

  IPCStreamSourceParent* source = IPCStreamSourceParent::Create(aInputStream);
  if (!source) {
    return nullptr;
  }

  if (!aManager->SendPParentToChildStreamConstructor(source)) {
    // no delete here, the manager will delete the actor for us.
    return nullptr;
  }

  source->ActorConstructed();
  return source;
}

/* static */ IPCStreamSource*
IPCStreamSource::Cast(PParentToChildStreamParent* aActor)
{
  MOZ_ASSERT(aActor);
  return static_cast<IPCStreamSourceParent*>(aActor);
}

// Child to Parent implementation
// ----------------------------------------------------------------------------

namespace {

class IPCStreamDestinationParent final : public PChildToParentStreamParent
                                       , public IPCStreamDestination
{
public:
  nsresult Initialize()
  {
    return IPCStreamDestination::Initialize();
  }

  ~IPCStreamDestinationParent()
  {}

private:
  // PChildToParentStreamParent methods

  void
  ActorDestroy(ActorDestroyReason aReason) override
  {
    ActorDestroyed();
  }

  IPCResult
  RecvBuffer(const wr::ByteBuffer& aBuffer) override
  {
    BufferReceived(aBuffer);
    return IPC_OK();
  }

  IPCResult
  RecvClose(const nsresult& aRv) override
  {
    CloseReceived(aRv);
    return IPC_OK();
  }

  // IPCStreamDestination methods

  void
  StartReading() override
  {
    MOZ_ASSERT(HasDelayedStart());
    Unused << SendStartReading();
  }

  void
  RequestClose(nsresult aRv) override
  {
    Unused << SendRequestClose(aRv);
  }

  void
  TerminateDestination() override
  {
    Unused << Send__delete__(this);
  }
};

} // anonymous namespace

PChildToParentStreamParent*
AllocPChildToParentStreamParent()
{
  IPCStreamDestinationParent* actor = new IPCStreamDestinationParent();

  if (NS_WARN_IF(NS_FAILED(actor->Initialize()))) {
    delete actor;
    actor = nullptr;
  }

  return actor;
}

void
DeallocPChildToParentStreamParent(PChildToParentStreamParent* aActor)
{
  delete aActor;
}

/* static */ IPCStreamDestination*
IPCStreamDestination::Cast(PChildToParentStreamParent* aActor)
{
  MOZ_ASSERT(aActor);
  return static_cast<IPCStreamDestinationParent*>(aActor);
}

} // namespace ipc
} // namespace mozilla