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

#ifndef ASH_SYSTEM_TRAY_IMAGED_TRAY_ICON_H_
#define ASH_SYSTEM_TRAY_IMAGED_TRAY_ICON_H_

#include <memory>
#include <string>
#include <variant>

#include "ash/ash_export.h"
#include "ash/system/tray/tray_background_view.h"
#include "ui/base/metadata/metadata_header_macros.h"

namespace ui {
class ImageModel;
class LocatedEvent;
}  // namespace ui

namespace views {
class ImageView;
}  // namespace views

namespace ash {

class Shelf;
class TrayBubbleView;

class ASH_EXPORT ImagedTrayIcon : public TrayBackgroundView {
  METADATA_HEADER(ImagedTrayIcon, TrayBackgroundView)

 public:
  using IconVisibilityCallback =
      base::RepeatingCallback<bool(session_manager::SessionState state)>;
  using StringVariant = std::variant<std::u16string, /*message_id=*/int>;

  ImagedTrayIcon(Shelf* shelf,
                 const ui::ImageModel& image_model,
                 const StringVariant& tooltip,
                 const StringVariant& accessibility_name,
                 const TrayBackgroundViewCatalogName catalog_name);

  ImagedTrayIcon(const ImagedTrayIcon&) = delete;
  ImagedTrayIcon& operator=(const ImagedTrayIcon&) = delete;

  ~ImagedTrayIcon() override;

  // Sets a callback that is run when the session state changes. The callback
  // should return true if the icon should be visible in the given state.
  void set_icon_visibility_callback(IconVisibilityCallback callback);

  void SetTooltip(const StringVariant& tooltip);
  void SetAccessibilityName(const StringVariant& name);

  // TrayBackgroundView:
  void ClickedOutsideBubble(const ui::LocatedEvent& event) override;
  void UpdateTrayItemColor(bool is_active) override;
  void HideBubbleWithView(const TrayBubbleView* bubble_view) override;
  void HideBubble(const TrayBubbleView* bubble_view) override;
  void HandleLocaleChange() override;

  views::ImageView* image_view() { return image_view_; }

 private:
  class IconVisibilityHandler;

  raw_ptr<views::ImageView> image_view_;
  StringVariant tooltip_;
  StringVariant accessibility_name_;

  std::unique_ptr<IconVisibilityHandler> session_visibility_handler_;
};

}  // namespace ash

#endif  // ASH_SYSTEM_TRAY_IMAGED_TRAY_ICON_H_