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

#include "mozilla/dom/BrowsingContext.h"
#include "mozilla/dom/BrowsingContextGroup.h"
#include "mozilla/dom/Document.h"
#include "nsCOMPtr.h"
#include "nsPIDOMWindow.h"
#include "nsTArray.h"

namespace mozilla::dom {

/**
 * Suppresses event handling and suspends for all in-process documents in a
 * BrowsingContext subtree.
 */
class MOZ_RAII AutoSuppressEventHandling : public AutoWalkBrowsingContextGroup {
 public:
  AutoSuppressEventHandling() = default;

  explicit AutoSuppressEventHandling(BrowsingContext* aContext) {
    if (aContext) {
      SuppressBrowsingContext(aContext);
    }
  }

  ~AutoSuppressEventHandling();

 protected:
  virtual void SuppressDocument(Document* aDocument) override;
  void UnsuppressDocument(Document* aDocument) override;
};

/**
 * Suppresses event handling and suspends the active inner window for all
 * in-process documents in a BrowsingContextGroup. This should be used while
 * spinning the event loop for a synchronous operation (like `window.open()`)
 * which affects operations in any other window in the same BrowsingContext
 * group.
 */
class MOZ_RAII AutoSuppressEventHandlingAndSuspend
    : private AutoSuppressEventHandling {
 public:
  explicit AutoSuppressEventHandlingAndSuspend(BrowsingContextGroup* aGroup) {
    if (aGroup) {
      SuppressBrowsingContextGroup(aGroup);
    }
  }

  ~AutoSuppressEventHandlingAndSuspend();

 protected:
  void SuppressDocument(Document* aDocument) override;

 private:
  AutoTArray<nsCOMPtr<nsPIDOMWindowInner>, 16> mWindows;
};
}  // namespace mozilla::dom

#endif