/* 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 "QuotaObject.h"

#include "CanonicalQuotaObject.h"
#include "RemoteQuotaObject.h"
#include "mozilla/dom/quota/IPCQuotaObject.h"
#include "mozilla/dom/quota/PRemoteQuotaObject.h"
#include "mozilla/dom/quota/RemoteQuotaObjectChild.h"
#include "mozilla/dom/quota/RemoteQuotaObjectParent.h"
#include "mozilla/dom/quota/RemoteQuotaObjectParentTracker.h"
#include "mozilla/ipc/BackgroundParent.h"
#include "nsIInterfaceRequestor.h"
#include "nsIInterfaceRequestorUtils.h"

namespace mozilla::dom::quota {

CanonicalQuotaObject* QuotaObject::AsCanonicalQuotaObject() {
  return mIsRemote ? nullptr : static_cast<CanonicalQuotaObject*>(this);
}

RemoteQuotaObject* QuotaObject::AsRemoteQuotaObject() {
  return mIsRemote ? static_cast<RemoteQuotaObject*>(this) : nullptr;
}

IPCQuotaObject QuotaObject::Serialize(nsIInterfaceRequestor* aCallbacks) {
  MOZ_RELEASE_ASSERT(XRE_IsParentProcess());
  MOZ_RELEASE_ASSERT(!NS_IsMainThread());
  MOZ_RELEASE_ASSERT(!mozilla::ipc::IsOnBackgroundThread());
  MOZ_RELEASE_ASSERT(!GetCurrentThreadWorkerPrivate());
  MOZ_RELEASE_ASSERT(!mIsRemote);

  mozilla::ipc::Endpoint<PRemoteQuotaObjectParent> parentEndpoint;
  mozilla::ipc::Endpoint<PRemoteQuotaObjectChild> childEndpoint;
  MOZ_ALWAYS_SUCCEEDS(
      PRemoteQuotaObject::CreateEndpoints(&parentEndpoint, &childEndpoint));

  nsCOMPtr<RemoteQuotaObjectParentTracker> tracker =
      do_GetInterface(aCallbacks);

  auto actor =
      MakeRefPtr<RemoteQuotaObjectParent>(AsCanonicalQuotaObject(), tracker);

  if (tracker) {
    tracker->RegisterRemoteQuotaObjectParent(WrapNotNull(actor));
  }

  parentEndpoint.Bind(actor);

  IPCQuotaObject ipcQuotaObject;
  ipcQuotaObject.childEndpoint() = std::move(childEndpoint);

  return ipcQuotaObject;
}

// static
RefPtr<QuotaObject> QuotaObject::Deserialize(IPCQuotaObject& aQuotaObject) {
  MOZ_RELEASE_ASSERT(!NS_IsMainThread());
  MOZ_RELEASE_ASSERT(!mozilla::ipc::IsOnBackgroundThread());
  MOZ_RELEASE_ASSERT(!GetCurrentThreadWorkerPrivate());

  auto actor = MakeRefPtr<RemoteQuotaObjectChild>();

  aQuotaObject.childEndpoint().Bind(actor);

  return MakeRefPtr<RemoteQuotaObject>(actor);
}

}  // namespace mozilla::dom::quota