// 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 "chromeos/ash/components/editor_menu/public/cpp/icon.h"

#include <string_view>

#include "build/branding_buildflags.h"
#include "chromeos/ash/components/editor_menu/public/cpp/preset_text_query.h"
#include "chromeos/ui/vector_icons/vector_icons.h"
#include "components/vector_icons/vector_icons.h"
#include "ui/base/ui_base_features.h"
#include "ui/gfx/vector_icon_types.h"

#if BUILDFLAG(GOOGLE_CHROME_BRANDING)
#include "chromeos/ash/resources/internal/icons/vector_icons.h"
#endif  // BUILDFLAG(GOOGLE_CHROME_BRANDING)

namespace chromeos::editor_menu {

const gfx::VectorIcon& GetIconForPresetQueryCategory(
    PresetQueryCategory category) {
  switch (category) {
    case PresetQueryCategory::kUnknown:
      return ::features::IsRoundedIconsEnabled()
                 ? vector_icons::kKeyboardIcon
                 : vector_icons::kKeyboardOldIcon;
    case PresetQueryCategory::kProofread:
      return kEditorMenuProofreadIcon;
    case PresetQueryCategory::kShorten:
      return kEditorMenuShortenIcon;
    case PresetQueryCategory::kElaborate:
      return kEditorMenuElaborateIcon;
    case PresetQueryCategory::kRephrase:
      return kEditorMenuRephraseIcon;
    case PresetQueryCategory::kFormalize:
      return kEditorMenuFormalizeIcon;
    case PresetQueryCategory::kEmojify:
      return kEditorMenuEmojifyIcon;
    case PresetQueryCategory::kLobster:
#if BUILDFLAG(GOOGLE_CHROME_BRANDING)
      return kLobsterIcon;
#else
      return ::features::IsRoundedIconsEnabled()
                 ? vector_icons::kKeyboardIcon
                 : vector_icons::kKeyboardOldIcon;
#endif
  }
}

}  // namespace chromeos::editor_menu