/* 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 "ExtensionPort.h"
#include "ExtensionBrowser.h"
#include "ExtensionEventManager.h"

#include "mozilla/dom/BindingUtils.h"  // SequenceRooter
#include "mozilla/dom/ExtensionPortBinding.h"
#include "mozilla/dom/ScriptSettings.h"  // AutoEntryScript
#include "nsIGlobalObject.h"

namespace mozilla {
namespace extensions {

NS_IMPL_CYCLE_COLLECTING_ADDREF(ExtensionPort);
NS_IMPL_CYCLE_COLLECTING_RELEASE(ExtensionPort)

NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE_CLASS(ExtensionPort)

NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(ExtensionPort)
  // Clean the entry for this instance from the ports lookup map
  // stored in the related ExtensionBrowser instance.
  tmp->ForgetReleasedPort();
  NS_IMPL_CYCLE_COLLECTION_UNLINK(mExtensionBrowser, mOnDisconnectEventMgr,
                                  mOnMessageEventMgr)
  NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER
  NS_IMPL_CYCLE_COLLECTION_UNLINK_WEAK_PTR
NS_IMPL_CYCLE_COLLECTION_UNLINK_END

NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(ExtensionPort)
  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mExtensionBrowser, mOnDisconnectEventMgr,
                                    mOnMessageEventMgr)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END

NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(ExtensionPort)
  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
  NS_INTERFACE_MAP_ENTRY(nsISupports)
NS_INTERFACE_MAP_END

NS_IMPL_WEBEXT_EVENTMGR(ExtensionPort, u"onMessage"_ns, OnMessage)
NS_IMPL_WEBEXT_EVENTMGR(ExtensionPort, u"onDisconnect"_ns, OnDisconnect)

// static
already_AddRefed<ExtensionPort> ExtensionPort::Create(
    nsIGlobalObject* aGlobal, ExtensionBrowser* aExtensionBrowser,
    UniquePtr<dom::ExtensionPortDescriptor>&& aPortDescriptor) {
  RefPtr<ExtensionPort> port =
      new ExtensionPort(aGlobal, aExtensionBrowser, std::move(aPortDescriptor));
  return port.forget();
}

// static
UniquePtr<dom::ExtensionPortDescriptor> ExtensionPort::ToPortDescriptor(
    JS::Handle<JS::Value> aDescriptorValue, ErrorResult& aRv) {
  if (!aDescriptorValue.isObject()) {
    aRv.Throw(NS_ERROR_UNEXPECTED);
    return nullptr;
  }

  dom::AutoEntryScript aes(&aDescriptorValue.toObject(), __func__);
  JSContext* acx = aes.cx();
  auto portDescriptor = MakeUnique<dom::ExtensionPortDescriptor>();
  if (!portDescriptor->Init(acx, aDescriptorValue, __func__)) {
    aRv.Throw(NS_ERROR_UNEXPECTED);
    return nullptr;
  }

  return portDescriptor;
}

ExtensionPort::ExtensionPort(
    nsIGlobalObject* aGlobal, ExtensionBrowser* aExtensionBrowser,
    UniquePtr<dom::ExtensionPortDescriptor>&& aPortDescriptor)
    : mGlobal(aGlobal),
      mExtensionBrowser(aExtensionBrowser),
      mPortDescriptor(std::move(aPortDescriptor)) {
  MOZ_DIAGNOSTIC_ASSERT(mGlobal);
  MOZ_DIAGNOSTIC_ASSERT(mExtensionBrowser);
}

void ExtensionPort::ForgetReleasedPort() {
  if (mExtensionBrowser) {
    mExtensionBrowser->ForgetReleasedPort(mPortDescriptor->mPortId);
    mExtensionBrowser = nullptr;
  }
  mPortDescriptor = nullptr;
  mGlobal = nullptr;
}

nsString ExtensionPort::GetAPIObjectId() const {
  return mPortDescriptor->mPortId;
}

JSObject* ExtensionPort::WrapObject(JSContext* aCx,
                                    JS::Handle<JSObject*> aGivenProto) {
  return dom::ExtensionPort_Binding::Wrap(aCx, this, aGivenProto);
}

nsIGlobalObject* ExtensionPort::GetParentObject() const { return mGlobal; }

void ExtensionPort::GetName(nsAString& aString) {
  aString.Assign(mPortDescriptor->mName);
}

}  // namespace extensions
}  // namespace mozilla
