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

#include <type_traits>
#include <utility>

#include "ErrorList.h"

namespace mozilla::dom::quota {

template <typename Initialization, typename StringGenerator>
class FirstInitializationAttempts {
  Initialization mFirstInitializationAttempts = Initialization::None;

 public:
  class FirstInitializationAttemptImpl {
    using FirstInitializationAttemptsType =
        FirstInitializationAttempts<Initialization, StringGenerator>;

    FirstInitializationAttemptsType& mOwner;
    const Initialization mInitialization;

   public:
    FirstInitializationAttemptImpl(FirstInitializationAttemptsType& aOwner,
                                   const Initialization aInitialization)
        : mOwner(aOwner), mInitialization(aInitialization) {}

    bool Recorded() const {
      return mOwner.FirstInitializationAttemptRecorded(mInitialization);
    }

    void Record(const nsresult aRv) const {
      mOwner.RecordFirstInitializationAttempt(mInitialization, aRv);
    }
  };

  template <typename Func>
  auto WithFirstInitializationAttempt(const Initialization aInitialization,
                                      Func&& aFunc)
      -> std::invoke_result_t<Func, FirstInitializationAttemptImpl&&> {
    return std::forward<Func>(aFunc)(
        FirstInitializationAttemptImpl(*this, aInitialization));
  }

  bool FirstInitializationAttemptRecorded(
      const Initialization aInitialization) const {
    return static_cast<bool>(mFirstInitializationAttempts & aInitialization);
  }

  void RecordFirstInitializationAttempt(const Initialization aInitialization,
                                        nsresult aRv);

  void ResetFirstInitializationAttempts() {
    mFirstInitializationAttempts = Initialization::None;
  }
};

template <typename Initialization, typename StringGenerator>
using FirstInitializationAttempt = typename FirstInitializationAttempts<
    Initialization, StringGenerator>::FirstInitializationAttemptImpl;

}  // namespace mozilla::dom::quota

#endif  // DOM_QUOTA_FIRSTINITIALIZATIONATTEMPTS_H_