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

#include "GroupInfo.h"
#include "GroupInfoPair.h"
#include "OriginInfo.h"
#include "mozilla/dom/quota/QuotaManager.h"

namespace mozilla::dom::quota {

template <typename F>
auto QuotaManager::WithOriginInfo(const OriginMetadata& aOriginMetadata,
                                  F aFunction)
    -> std::invoke_result_t<F, const RefPtr<OriginInfo>&> {
  MutexAutoLock lock(mQuotaMutex);

  RefPtr<OriginInfo> originInfo =
      LockedGetOriginInfo(aOriginMetadata.mPersistenceType, aOriginMetadata);
  MOZ_ASSERT(originInfo);

  return aFunction(originInfo);
}

template <typename P>
void QuotaManager::CollectPendingOriginsForListing(P aPredicate) {
  MutexAutoLock lock(mQuotaMutex);

  for (const auto& entry : mGroupInfoPairs) {
    const auto& pair = entry.GetData();

    MOZ_ASSERT(!entry.GetKey().IsEmpty());
    MOZ_ASSERT(pair);

    RefPtr<GroupInfo> groupInfo =
        pair->LockedGetGroupInfo(PERSISTENCE_TYPE_DEFAULT);
    if (groupInfo) {
      for (const auto& originInfo : groupInfo->mOriginInfos) {
        if (!originInfo->LockedDirectoryExists()) {
          aPredicate(originInfo);
        }
      }
    }
  }
}

template <typename F>
void QuotaManager::MaybeRecordQuotaManagerShutdownStepWith(F&& aFunc) {
  // Callable on any thread.

  if (IsShuttingDown()) {
    RecordShutdownStep(Nothing{}, std::forward<F>(aFunc)());
  }
}

}  // namespace mozilla::dom::quota

#endif  // DOM_QUOTA_QUOTAMANAGERIMPL_H_