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