// Copyright 2018 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_TOOLBAR_LEGACY_UI_BUNDLED_BUTTONS_TOOLBAR_TAB_GRID_BUTTON_H_
#define IOS_CHROME_BROWSER_TOOLBAR_LEGACY_UI_BUNDLED_BUTTONS_TOOLBAR_TAB_GRID_BUTTON_H_

#import "ios/chrome/browser/toolbar/legacy/ui_bundled/buttons/legacy_toolbar_button.h"

enum class ToolbarTabGroupState;

using ToolbarTabGridButtonImageLoader =
    UIImage* (^)(ToolbarTabGroupState tabGroupState);

// LegacyToolbarButton for displaying the number of tab.
@interface ToolbarTabGridButton : LegacyToolbarButton

// Sets the number of tabs displayed by this button to `tabCount`. If `tabCount`
// is more than 99, it shows a smiley instead. But the value stored in tabCount
// can be bigger than 100.
@property(nonatomic, assign) int tabCount;

// The Tab Group state of the toolbar.
// If kNormal, the button features the simple square with the tab count label
// inside.
// If kTabGroup, the button features a filled square on a square with the tab
// count label inside the filled square.
@property(nonatomic, assign) ToolbarTabGroupState tabGroupState;

// Initialized the button with an image loader that can return a different image
// based on the tab group state.
- (instancetype)initWithTabGroupStateImageLoader:
    (ToolbarTabGridButtonImageLoader)tabGroupStateImageLoader
    NS_DESIGNATED_INITIALIZER;
- (instancetype)initWithImageLoader:(ToolbarButtonImageLoader)imageLoader
    NS_UNAVAILABLE;
- (instancetype)initWithImageLoader:(ToolbarButtonImageLoader)imageLoader
          IPHHighlightedImageLoader:
              (ToolbarButtonImageLoader)IPHHighlightedImageLoader
    NS_UNAVAILABLE;

@end

#endif  // IOS_CHROME_BROWSER_TOOLBAR_LEGACY_UI_BUNDLED_BUTTONS_TOOLBAR_TAB_GRID_BUTTON_H_