/* 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_QuotaManagerService_h
#define mozilla_dom_quota_QuotaManagerService_h

#include "ErrorList.h"
#include "mozilla/AlreadyAddRefed.h"
#include "mozilla/HalBatteryInformation.h"
#include "mozilla/dom/ipc/IdType.h"
#include "nsIObserver.h"
#include "nsIQuotaManagerService.h"
#include "nsIQuotaManagerServiceInternal.h"
#include "nsISupports.h"

#define QUOTAMANAGER_SERVICE_CONTRACTID \
  "@mozilla.org/dom/quota-manager-service;1"

class nsIPrincipal;
class nsIQuotaRequest;
class nsIQuotaUsageCallback;
class nsIQuotaUsageRequest;

namespace mozilla {
namespace ipc {

class PBackgroundChild;

}  // namespace ipc

namespace hal {
class BatteryInformation;
}

namespace dom::quota {

class QuotaChild;
class QuotaManager;

class QuotaManagerService final : public nsIQuotaManagerService,
                                  public nsIQuotaManagerServiceInternal,
                                  public nsIObserver,
                                  public hal::BatteryObserver {
  using PBackgroundChild = mozilla::ipc::PBackgroundChild;

  class BackgroundCreateCallback;
  class PendingRequestInfo;
  class UsageRequestInfo;
  class RequestInfo;
  class IdleMaintenanceInfo;

  QuotaChild* mBackgroundActor;

  bool mBackgroundActorFailed;
  bool mIdleObserverRegistered;

 public:
  // Returns a non-owning reference.
  static QuotaManagerService* GetOrCreate();

  // Returns a non-owning reference.
  static QuotaManagerService* Get();

  // No one should call this but the factory.
  static already_AddRefed<QuotaManagerService> FactoryCreate();

  void ClearBackgroundActor();

  // Called when a process is being shot down. Aborts any running operations
  // for the given process.
  void AbortOperationsForProcess(ContentParentId aContentParentId);

 private:
  QuotaManagerService();
  ~QuotaManagerService();

  nsresult Init();

  void Destroy();

  nsresult EnsureBackgroundActor();

  nsresult InitiateRequest(PendingRequestInfo& aInfo);

  void PerformIdleMaintenance();

  void RemoveIdleObserver();

  NS_DECL_ISUPPORTS
  NS_DECL_NSIQUOTAMANAGERSERVICE
  NS_DECL_NSIQUOTAMANAGERSERVICEINTERNAL
  NS_DECL_NSIOBSERVER

  // BatteryObserver override
  void Notify(const hal::BatteryInformation& aBatteryInfo) override;
};

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

#endif /* mozilla_dom_quota_QuotaManagerService_h */