chromium/ios/chrome/browser/dialogs/ui_bundled/java_script_dialog_blocking_state.h

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_CHROME_BROWSER_DIALOGS_UI_BUNDLED_JAVA_SCRIPT_DIALOG_BLOCKING_STATE_H_
#define IOS_CHROME_BROWSER_DIALOGS_UI_BUNDLED_JAVA_SCRIPT_DIALOG_BLOCKING_STATE_H_

#import <Foundation/Foundation.h>

#import "base/memory/raw_ptr.h"
#include "ios/web/public/web_state_observer.h"
#import "ios/web/public/web_state_user_data.h"

namespace web {
class NavigationItem;
}

// A helper object that tracks whether JavaScript dialogs should be blocked for
// a given WebState.  State is reset every time a navigation is committed.
class JavaScriptDialogBlockingState
    : public web::WebStateUserData<JavaScriptDialogBlockingState>,
      public web::WebStateObserver {
 public:
  JavaScriptDialogBlockingState(const JavaScriptDialogBlockingState&) = delete;
  JavaScriptDialogBlockingState& operator=(
      const JavaScriptDialogBlockingState&) = delete;

  ~JavaScriptDialogBlockingState() override;

  // Whether to show the blocking option for the next JavaScript dialog from its
  // WebState.
  bool show_blocking_option() { return dialog_count_ > 0; }
  // Whether the blocking option has been selected for its WebState's most
  // recent navigation.
  bool blocked() { return blocked_item_ != nullptr; }

  // Notifies the blocking state that a JavaScript dialog has been shown for its
  // WebState.
  void JavaScriptDialogWasShown() { ++dialog_count_; }
  // Notifies the blocking state that the JavaScript dialog blocking option has
  // been selected for its WebState's most recent navigation.
  void JavaScriptDialogBlockingOptionSelected();

 private:
  friend class web::WebStateUserData<JavaScriptDialogBlockingState>;

  // Private constructor.
  explicit JavaScriptDialogBlockingState(web::WebState* web_state);

  // WebStateObserver:
  void DidStartNavigation(web::WebState* web_state,
                          web::NavigationContext* navigation_context) override;
  void WebStateDestroyed(web::WebState* web_state) override;

  // The WebState this instance is observing. Will be null after
  // WebStateDestroyed has been called.
  raw_ptr<web::WebState> web_state_ = nullptr;

  // Whether to show the blocking option.
  size_t dialog_count_ = 0;
  // The NavigationItem for which JavaScript dialogs were blocked.
  raw_ptr<web::NavigationItem> blocked_item_ = nullptr;

  WEB_STATE_USER_DATA_KEY_DECL();
};

#endif  // IOS_CHROME_BROWSER_DIALOGS_UI_BUNDLED_JAVA_SCRIPT_DIALOG_BLOCKING_STATE_H_