/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */

#ifndef mozilla_dom_quota_UsageRequest_h
#define mozilla_dom_quota_UsageRequest_h

#include "nsCOMPtr.h"
#include "nsCycleCollectionParticipant.h"
#include "nsIQuotaRequests.h"

class nsIPrincipal;
class nsIQuotaCallback;
class nsIQuotaUsageCallback;
struct PRThread;

namespace mozilla {
namespace dom {
namespace quota {

class QuotaUsageRequestChild;

class RequestBase
  : public nsIQuotaRequestBase
{
protected:
#ifdef DEBUG
  PRThread* mOwningThread;
#endif

  nsCOMPtr<nsIPrincipal> mPrincipal;

  nsresult mResultCode;
  bool mHaveResultOrErrorCode;

public:
  void
  AssertIsOnOwningThread() const
#ifdef DEBUG
  ;
#else
  { }
#endif

  void
  SetError(nsresult aRv);

  NS_DECL_CYCLE_COLLECTING_ISUPPORTS
  NS_DECL_NSIQUOTAREQUESTBASE
  NS_DECL_CYCLE_COLLECTION_CLASS(RequestBase)

protected:
  RequestBase();

  RequestBase(nsIPrincipal* aPrincipal);

  virtual ~RequestBase()
  {
    AssertIsOnOwningThread();
  }

  virtual void
  FireCallback() = 0;
};

class UsageRequest final
  : public RequestBase
  , public nsIQuotaUsageRequest
{
  nsCOMPtr<nsIQuotaUsageCallback> mCallback;

  nsCOMPtr<nsIVariant> mResult;

  QuotaUsageRequestChild* mBackgroundActor;

  bool mCanceled;

public:
  explicit UsageRequest(nsIQuotaUsageCallback* aCallback);

  UsageRequest(nsIPrincipal* aPrincipal,
               nsIQuotaUsageCallback* aCallback);

  void
  SetBackgroundActor(QuotaUsageRequestChild* aBackgroundActor);

  void
  ClearBackgroundActor()
  {
    AssertIsOnOwningThread();

    mBackgroundActor = nullptr;
  }

  void
  SetResult(nsIVariant* aResult);

  NS_DECL_ISUPPORTS_INHERITED
  NS_FORWARD_NSIQUOTAREQUESTBASE(RequestBase::)
  NS_DECL_NSIQUOTAUSAGEREQUEST
  NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(UsageRequest, RequestBase)

private:
  ~UsageRequest();

  virtual void
  FireCallback() override;
};

class Request final
  : public RequestBase
  , public nsIQuotaRequest
{
  nsCOMPtr<nsIQuotaCallback> mCallback;

public:
  Request();

  explicit Request(nsIPrincipal* aPrincipal);

  void
  SetResult();

  NS_DECL_ISUPPORTS_INHERITED
  NS_FORWARD_NSIQUOTAREQUESTBASE(RequestBase::)
  NS_DECL_NSIQUOTAREQUEST
  NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(Request, RequestBase)

private:
  ~Request();

  virtual void
  FireCallback() override;
};

} // namespace quota
} // namespace dom
} // namespace mozilla

#endif // mozilla_dom_quota_UsageRequest_h
