1 files changed, 69 insertions, 0 deletions
diff --git a/dom/quota/FirstInitializationAttempts.h b/dom/quota/FirstInitializationAttempts.h
new file mode 100644
index 0000000000..e997985371
--- /dev/null
+++ b/dom/quota/FirstInitializationAttempts.h
@@ -0,0 +1,69 @@
+/* -*- 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/. */
+
+#ifndef DOM_QUOTA_FIRSTINITIALIZATIONATTEMPTS_H_
+#define DOM_QUOTA_FIRSTINITIALIZATIONATTEMPTS_H_
+
+#include <cstdint>
+#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_