gecko-dev/dom/serviceworkers/ServiceWorkerImpl.cpp

/* -*- 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