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

#include "chrome/browser/ui/views/tabs/tab_strip_nudge_button.h"

#include "chrome/app/vector_icons/vector_icons.h"
#include "chrome/browser/ui/color/chrome_color_id.h"
#include "chrome/browser/ui/views/tabs/tab_strip_control_button.h"
#include "chrome/grit/generated_resources.h"
#include "components/vector_icons/vector_icons.h"
#include "ui/base/l10n/l10n_util.h"
#include "ui/base/metadata/metadata_impl_macros.h"
#include "ui/base/ui_base_features.h"
#include "ui/compositor/layer.h"
#include "ui/gfx/vector_icon_types.h"
#include "ui/views/accessibility/view_accessibility.h"
#include "ui/views/animation/ink_drop.h"
#include "ui/views/controls/highlight_path_generator.h"
#include "ui/views/layout/box_layout.h"
#include "ui/views/view_class_properties.h"

namespace {
constexpr int kTabStripNudgeCornerRadius = 10;
constexpr int kTabStripNudgeFlatCornerRadius = 4;
constexpr int kTabStripNudgeIconMargin = 6;
constexpr int kTabStripNudgeLabelLeftMargin = 4;
constexpr int kTabStripNudgeLabelRightMargin = 12;
constexpr int kTabStripNudgeCloseButtonMargin = 8;
constexpr int kTabStripNudgeCloseButtonSize = 16;

constexpr gfx::Insets GetLabelInsets(bool show_close_button) {
  return show_close_button
             ? gfx::Insets().set_left(kTabStripNudgeLabelLeftMargin)
             :
             // Set a right margin on the label when the close button is hidden.
             gfx::Insets().set_left_right(kTabStripNudgeLabelLeftMargin,
                                          kTabStripNudgeLabelRightMargin);
}

}  // namespace

TabStripNudgeButton::TabStripNudgeButton(
    BrowserWindowInterface* browser_window_interface,
    PressedCallback pressed_callback,
    PressedCallback close_pressed_callback,
    const std::u16string& label_text,
    const ui::ElementIdentifier& element_identifier,
    Edge flat_edge,
    const gfx::VectorIcon& icon,
    const bool show_close_button)
    : TabStripControlButton(browser_window_interface,
                            std::move(pressed_callback),
                            icon,
                            label_text,
                            Edge::kNone,
                            flat_edge) {
  auto* const layout_manager =
      SetLayoutManager(std::make_unique<views::BoxLayout>());
  layout_manager->set_main_axis_alignment(
      views::BoxLayout::MainAxisAlignment::kEnd);

  SetProperty(views::kElementIdentifierKey, element_identifier);

  if (!icon.is_empty()) {
    const gfx::Insets icon_margin =
        gfx::Insets().set_left(kTabStripNudgeIconMargin);
    image_container_view()->SetProperty(views::kMarginsKey, icon_margin);
  }

  SetLabelStyle(views::style::STYLE_BODY_3_EMPHASIS);
  label()->SetElideBehavior(gfx::ElideBehavior::NO_ELIDE);
  label()->SetProperty(views::kMarginsKey, GetLabelInsets(show_close_button));

  SetForegroundFrameActiveColorId(kColorTabSearchButtonCRForegroundFrameActive);
  SetForegroundFrameInactiveColorId(
      kColorTabSearchButtonCRForegroundFrameInactive);
  SetBackgroundFrameActiveColorId(kColorNewTabButtonCRBackgroundFrameActive);
  SetBackgroundFrameInactiveColorId(
      kColorNewTabButtonCRBackgroundFrameInactive);

  set_paint_transparent_for_custom_image_theme(false);

  if (show_close_button) {
    SetCloseButton(std::move(close_pressed_callback));
  }

  UpdateColors();
}

TabStripNudgeButton::~TabStripNudgeButton() = default;

void TabStripNudgeButton::SetOpacity(float factor) {
  if (label()->layer()) {
    label()->layer()->SetOpacity(factor);
  }
  close_button_->layer()->SetOpacity(factor);
}
void TabStripNudgeButton::SetWidthFactor(float factor) {
  width_factor_ = factor;
  PreferredSizeChanged();
}

gfx::Size TabStripNudgeButton::CalculatePreferredSize(
    const views::SizeBounds& available_size) const {
  const int full_width =
      GetLayoutManager()->GetPreferredSize(this, available_size).width();
  const int width = full_width * width_factor_;
  const int height = TabStripControlButton::CalculatePreferredSize(
                         views::SizeBounds(width, available_size.height()))
                         .height();
  return gfx::Size(width, height);
}

void TabStripNudgeButton::SetIsShowingNudge(bool is_showing) {
  is_showing_nudge_ = is_showing;
  if (is_showing) {
    SetFocusBehavior(FocusBehavior::ALWAYS);
    SetCloseButtonFocusBehavior(FocusBehavior::ALWAYS);
  } else {
    SetFocusBehavior(FocusBehavior::NEVER);
    SetCloseButtonFocusBehavior(FocusBehavior::NEVER);
  }
}

bool TabStripNudgeButton::GetIsShowingNudge() const {
  return is_showing_nudge_;
}

gfx::SlideAnimation* TabStripNudgeButton::GetExpansionAnimationForTesting() {
  return nullptr;
}

int TabStripNudgeButton::GetCornerRadius() const {
  return kTabStripNudgeCornerRadius;
}

int TabStripNudgeButton::GetFlatCornerRadius() const {
  return kTabStripNudgeFlatCornerRadius;
}

void TabStripNudgeButton::SetCloseButtonFocusBehavior(
    views::View::FocusBehavior focus_behavior) {
  close_button_->SetFocusBehavior(focus_behavior);
}

void TabStripNudgeButton::SetCloseButton(PressedCallback pressed_callback) {
  auto close_button =
      std::make_unique<views::LabelButton>(std::move(pressed_callback));
  close_button->SetTooltipText(
      l10n_util::GetStringUTF16(IDS_TOOLTIP_GLIC_CLOSE));

  const ui::ImageModel icon_image_model = ui::ImageModel::FromVectorIcon(
      features::IsRoundedIconsEnabled()
          ? vector_icons::kCloseIcon
          : vector_icons::kCloseChromeRefreshOldIcon,
      kColorTabSearchButtonCRForegroundFrameActive,
      kTabStripNudgeCloseButtonSize);

  close_button->SetImageModel(views::Button::STATE_NORMAL, icon_image_model);
  close_button->SetImageModel(views::Button::STATE_HOVERED, icon_image_model);
  close_button->SetImageModel(views::Button::STATE_PRESSED, icon_image_model);

  close_button->SetPaintToLayer();
  close_button->layer()->SetFillsBoundsOpaquely(false);

  views::InkDrop::Get(close_button.get())
      ->SetMode(views::InkDropHost::InkDropMode::ON);
  views::InkDrop::Get(close_button.get())->SetHighlightOpacity(0.16f);
  views::InkDrop::Get(close_button.get())->SetVisibleOpacity(0.14f);
  views::InkDrop::Get(close_button.get())
      ->SetBaseColor(kColorTabSearchButtonCRForegroundFrameActive);

  auto ink_drop_highlight_path =
      std::make_unique<views::CircleHighlightPathGenerator>(gfx::Insets());
  views::HighlightPathGenerator::Install(close_button.get(),
                                         std::move(ink_drop_highlight_path));

  close_button->SetPreferredSize(
      gfx::Size(kTabStripNudgeCloseButtonSize, kTabStripNudgeCloseButtonSize));
  close_button->SetBorder(nullptr);

  const gfx::Insets margin = gfx::Insets().set_left_right(
      kTabStripNudgeCloseButtonMargin, kTabStripNudgeCloseButtonMargin);
  close_button->SetProperty(views::kMarginsKey, margin);

  close_button_ = AddChildView(std::move(close_button));
  SetIsShowingNudge(false);
}

BEGIN_METADATA(TabStripNudgeButton)

END_METADATA