ios/chrome/browser/infobars/infobar_badge_tab_helper.h

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

#ifndef IOS_CHROME_BROWSER_INFOBARS_INFOBAR_BADGE_TAB_HELPER_H_
#define IOS_CHROME_BROWSER_INFOBARS_INFOBAR_BADGE_TAB_HELPER_H_

#include "base/scoped_observer.h"
#import "ios/web/public/web_state/web_state_user_data.h"

#include "components/infobars/core/infobar_manager.h"

namespace web {
class WebState;
}

@protocol InfobarBadgeTabHelperDelegate;

// TabHelper that observes InfoBarManager. It updates an InfobarBadge delegate
// for relevant Infobar changes.
class InfobarBadgeTabHelper
    : public infobars::InfoBarManager::Observer,
      public web::WebStateUserData<InfobarBadgeTabHelper> {
 public:
  // Creates the tab helper for |web_state| if it does not exist.
  static void CreateForWebState(web::WebState* web_state);
  // Sets the InfobarBadgeTabHelperDelegate to |delegate|.
  void SetDelegate(id<InfobarBadgeTabHelperDelegate> delegate);
  // Updates Infobar badge for the case where an InfobarBanner was dismissed.
  void UpdateBadgeForInfobarBannerDismissed();
  // Updates Infobar badge for the case where an InfobarModal was presented.
  void UpdateBadgeForInfobarModalPresented();
  // Updates Infobar badge for the case where an InfobarModal was dismissed.
  void UpdateBadgeForInfobarModalDismissed();
  // Updates Infobar badge for the case where an Infobar was accepted.
  void UpdateBadgeForInfobarAccepted();

  // Returns wheter an Infobar badge is being displayed for the TabHelper
  // Webstate.
  bool IsInfobarBadgeDisplaying();
  ~InfobarBadgeTabHelper() override;

 private:
  friend class web::WebStateUserData<InfobarBadgeTabHelper>;
  // Constructor.
  explicit InfobarBadgeTabHelper(web::WebState* web_state);
  // InfoBarManagerObserver implementation.
  void OnInfoBarAdded(infobars::InfoBar* infobar) override;
  void OnInfoBarRemoved(infobars::InfoBar* infobar, bool animate) override;
  void OnManagerShuttingDown(infobars::InfoBarManager* manager) override;
  // Updates the badge delegate for |infobar|.
  void UpdateBadgeForInfobar(infobars::InfoBar* infobar, bool display);

  // Manages this object as an observer of infobars.
  ScopedObserver<infobars::InfoBarManager, infobars::InfoBarManager::Observer>
      infobar_observer_;
  // Delegate which displays the Infobar badge.
  __weak id<InfobarBadgeTabHelperDelegate> delegate_ = nil;
  // Returns wheter an Infobar is being displayed.
  bool is_infobar_displaying_;

  WEB_STATE_USER_DATA_KEY_DECL();
  DISALLOW_COPY_AND_ASSIGN(InfobarBadgeTabHelper);
};

#endif  // IOS_CHROME_BROWSER_INFOBARS_INFOBAR_BADGE_TAB_HELPER_H_