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

#include "ServiceWorkerInfo.h"

namespace mozilla {
namespace dom {

ServiceWorkerImpl::~ServiceWorkerImpl() { MOZ_DIAGNOSTIC_ASSERT(!mOuter); }

void ServiceWorkerImpl::AddServiceWorker(ServiceWorker* aWorker) {
  MOZ_DIAGNOSTIC_ASSERT(!mOuter);
  MOZ_DIAGNOSTIC_ASSERT(aWorker);
  mOuter = aWorker;
}

void ServiceWorkerImpl::RemoveServiceWorker(ServiceWorker* aWorker) {
  MOZ_DIAGNOSTIC_ASSERT(mOuter);
  MOZ_DIAGNOSTIC_ASSERT(mOuter == aWorker);
  mOuter = nullptr;
}

void ServiceWorkerImpl::GetRegistration(
    ServiceWorkerRegistrationCallback&& aSuccessCB,
    ServiceWorkerFailureCallback&& aFailureCB) {
  // While we could immediate call success with our registration descriptor
  // we instead queue a runnable to do this.  This ensures that
  // GetRegistration() is always async to match how the IPC implementation will
  // work.  It also ensure that if any events are triggered from providing the
  // registration that they are fired from a runnable on the correct global's
  // event target.

  if (!mOuter) {
    aFailureCB(CopyableErrorResult(NS_ERROR_DOM_INVALID_STATE_ERR));
    return;
  }

  nsIGlobalObject* global = mOuter->GetParentObject();
  if (!global) {
    aFailureCB(CopyableErrorResult(NS_ERROR_DOM_INVALID_STATE_ERR));
    return;
  }

  nsCOMPtr<nsIRunnable> r = NS_NewRunnableFunction(
      "ServiceWorkerImpl::GetRegistration",
      [reg = mReg, successCB = std::move(aSuccessCB)]() mutable {
        successCB(reg->Descriptor());
      });

  nsresult rv = global->EventTargetFor(TaskCategory::Other)
                    ->Dispatch(r.forget(), NS_DISPATCH_NORMAL);
  if (NS_FAILED(rv)) {
    aFailureCB(CopyableErrorResult(rv));
    return;
  }
}

void ServiceWorkerImpl::PostMessage(RefPtr<ServiceWorkerCloneData>&& aData,
                                    const ClientInfo& aClientInfo,
                                    const ClientState& aClientState) {
  mInfo->PostMessage(std::move(aData), aClientInfo, aClientState);
}

ServiceWorkerImpl::ServiceWorkerImpl(ServiceWorkerInfo* aInfo,
                                     ServiceWorkerRegistrationInfo* aReg)
    : mInfo(aInfo), mReg(aReg), mOuter(nullptr) {
  MOZ_DIAGNOSTIC_ASSERT(mInfo);
  MOZ_DIAGNOSTIC_ASSERT(mReg);
}

}  // namespace dom
}  // namespace mozilla