// Copyright 2025 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/tabs/split_tab_swap_menu_model.h"

#include "chrome/app/vector_icons/vector_icons.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/grit/generated_resources.h"
#include "components/split_tabs/split_tab_visual_data.h"
#include "components/tabs/public/split_tab_data.h"
#include "components/tabs/public/tab_interface.h"
#include "ui/base/interaction/element_identifier.h"
#include "ui/base/l10n/l10n_util.h"
#include "ui/base/ui_base_features.h"

DEFINE_CLASS_ELEMENT_IDENTIFIER_VALUE(SplitTabSwapMenuModel,
                                      kSwapStartTabMenuItem);
DEFINE_CLASS_ELEMENT_IDENTIFIER_VALUE(SplitTabSwapMenuModel,
                                      kSwapEndTabMenuItem);

SplitTabSwapMenuModel::SplitTabSwapMenuModel(TabStripModel* tab_strip_model,
                                             int tab_index)
    : ui::SimpleMenuModel(this),
      tab_strip_model_(tab_strip_model),
      tab_index_(tab_index) {
  AddItem(static_cast<int>(CommandId::kSwapStartTab), std::u16string());
  AddItem(static_cast<int>(CommandId::kSwapEndTab), std::u16string());

  SetElementIdentifierAt(
      GetIndexOfCommandId(static_cast<int>(CommandId::kSwapStartTab)).value(),
      kSwapStartTabMenuItem);
  SetElementIdentifierAt(
      GetIndexOfCommandId(static_cast<int>(CommandId::kSwapEndTab)).value(),
      kSwapEndTabMenuItem);
}

SplitTabSwapMenuModel::~SplitTabSwapMenuModel() = default;

bool SplitTabSwapMenuModel::IsItemForCommandIdDynamic(int command_id) const {
  const CommandId id = static_cast<CommandId>(command_id);
  return id == CommandId::kSwapStartTab || id == CommandId::kSwapEndTab;
}

std::u16string SplitTabSwapMenuModel::GetLabelForCommandId(
    int command_id) const {
  const CommandId id = static_cast<CommandId>(command_id);
  const auto layout = GetSplitLayout();
  const bool is_side_by_side =
      !layout.has_value() || *layout == split_tabs::SplitTabLayout::kSideBySide;

  if (id == CommandId::kSwapStartTab) {
    return l10n_util::GetStringUTF16(is_side_by_side
                                         ? IDS_SPLIT_TAB_SWAP_LEFT_VIEW
                                         : IDS_SPLIT_TAB_SWAP_TOP_VIEW);
  } else if (id == CommandId::kSwapEndTab) {
    return l10n_util::GetStringUTF16(is_side_by_side
                                         ? IDS_SPLIT_TAB_SWAP_RIGHT_VIEW
                                         : IDS_SPLIT_TAB_SWAP_BOTTOM_VIEW);
  } else {
    NOTREACHED() << "There are no other commands that are dynamic so this case "
                    "should not be reached.";
  }
}

ui::ImageModel SplitTabSwapMenuModel::GetIconForCommandId(
    int command_id) const {
  const CommandId id = static_cast<CommandId>(command_id);
  const gfx::VectorIcon* icon = nullptr;
  const auto layout = GetSplitLayout();
  const bool is_side_by_side =
      !layout.has_value() || *layout == split_tabs::SplitTabLayout::kSideBySide;

  if (id == CommandId::kSwapStartTab) {
    icon = is_side_by_side
               ? &(features::IsRoundedIconsEnabled() ? kSplitSceneLeftIcon
                                                     : kSplitSceneLeftOldIcon)
               : &(features::IsRoundedIconsEnabled() ? kSplitSceneUpIcon
                                                     : kSplitSceneUpOldIcon);
  } else if (id == CommandId::kSwapEndTab) {
    icon = is_side_by_side
               ? &(features::IsRoundedIconsEnabled() ? kSplitSceneRightIcon
                                                     : kSplitSceneRightOldIcon)
               : &(features::IsRoundedIconsEnabled() ? kSplitSceneDownIcon
                                                     : kSplitSceneDownOldIcon);
  }
  CHECK(icon);
  return ui::ImageModel::FromVectorIcon(*icon, ui::kColorMenuIcon,
                                        ui::SimpleMenuModel::kDefaultIconSize);
}

void SplitTabSwapMenuModel::ExecuteCommand(int command_id, int event_flags) {
  const CommandId id = static_cast<CommandId>(command_id);
  const auto split_id = GetSplitTabId();
  // It's possible that the active tab is no longer in a split layout after the
  // menu was created. This primarily happens if a new tab was opened while the
  // menu was live and stole focus.
  if (!split_id) {
    return;
  }
  split_tabs::SplitTabData* const split_tab_data =
      tab_strip_model_->GetSplitData(*split_id);
  std::vector<tabs::TabInterface*> tabs_in_split = split_tab_data->ListTabs();
  CHECK_EQ(tabs_in_split.size(), 2U);

  if (id == CommandId::kSwapStartTab) {
    tab_strip_model_->UpdateTabInSplit(tabs_in_split[0], tab_index_,
                                       TabStripModel::SplitUpdateType::kSwap);
  } else if (id == CommandId::kSwapEndTab) {
    tab_strip_model_->UpdateTabInSplit(tabs_in_split[1], tab_index_,
                                       TabStripModel::SplitUpdateType::kSwap);
  }
}

std::optional<split_tabs::SplitTabId> SplitTabSwapMenuModel::GetSplitTabId()
    const {
  tabs::TabInterface* const tab = tab_strip_model_->GetActiveTab();
  return tab->GetSplit();
}

std::optional<split_tabs::SplitTabLayout>
SplitTabSwapMenuModel::GetSplitLayout() const {
  const auto split_id = GetSplitTabId();
  if (!split_id.has_value()) {
    return std::nullopt;
  }
  split_tabs::SplitTabVisualData* const visual_data =
      tab_strip_model_->GetSplitData(*split_id)->visual_data();
  return visual_data->split_layout();
}