brave-core/browser/ui/toolbar/app_menu_icons.cc

// Copyright (c) 2023 The Brave Authors. All rights reserved.
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.

#include "brave/browser/ui/toolbar/app_menu_icons.h"

#include "base/containers/fixed_flat_map.h"
#include "base/containers/map_util.h"
#include "base/no_destructor.h"
#include "brave/app/brave_command_ids.h"
#include "brave/components/ai_chat/core/common/buildflags/buildflags.h"
#include "brave/components/vector_icons/vector_icons.h"
#include "chrome/app/chrome_command_ids.h"
#include "ui/base/models/image_model.h"
#include "ui/color/color_id.h"
#include "ui/gfx/vector_icon_types.h"
#include "ui/menus/simple_menu_model.h"

namespace {

constexpr int kLeoIconSize = 16;

constexpr auto kCommandIcons =
    base::MakeFixedFlatMap<int, const gfx::VectorIcon*>({
        // The `App` menu
        {IDC_UPGRADE_DIALOG, &kLeoRocketIcon},
        {IDC_NEW_TAB, &kLeoBrowserAddIcon},
        {IDC_NEW_WINDOW, &kLeoWindowTabNewIcon},
        {IDC_NEW_INCOGNITO_WINDOW, &kLeoWindowTabPrivateIcon},
        {IDC_NEW_TOR_CONNECTION_FOR_SITE, &kLeoWindowTabTorIcon},
        {IDC_NEW_OFFTHERECORD_WINDOW_TOR, &kLeoWindowTabTorIcon},
        {IDC_SHOW_BRAVE_REWARDS, &kLeoProductBatOutlineIcon},
        {IDC_SHOW_BRAVE_WALLET, &kLeoProductBraveWalletIcon},
#if BUILDFLAG(ENABLE_AI_CHAT)
        {IDC_TOGGLE_AI_CHAT, &kLeoProductBraveLeoIcon},
#endif
        {IDC_BRAVE_VPN_MENU, &kLeoProductVpnIcon},
        {IDC_PASSWORDS_AND_AUTOFILL_MENU, &kLeoKeyIcon},
        {IDC_RECENT_TABS_MENU, &kLeoHistoryIcon},
        {IDC_BOOKMARKS_MENU, &kLeoProductBookmarksIcon},
        {IDC_SHOW_DOWNLOADS, &kLeoDownloadIcon},
        {IDC_MANAGE_EXTENSIONS, &kLeoBrowserExtensionsIcon},
        {IDC_EXTENSIONS_SUBMENU_MANAGE_EXTENSIONS, &kLeoBrowserExtensionsIcon},
        {IDC_EXTENSIONS_SUBMENU_VISIT_CHROME_WEB_STORE, &kLeoShoppingBagIcon},
        {IDC_ZOOM_MENU, &kLeoSearchZoomInIcon},
        {IDC_PRINT, &kLeoPrintIcon},
        {IDC_FIND_AND_EDIT_MENU, &kLeoFileSearchIcon},
        {IDC_SAVE_AND_SHARE_MENU, &kLeoFileDownloadIcon},
        {IDC_MORE_TOOLS_MENU, &kLeoWindowScrewdriverIcon},
        {IDC_OPTIONS, &kLeoSettingsIcon},
        {IDC_HELP_MENU, &kLeoHelpOutlineIcon},
        {IDC_EXIT, &kLeoCloseIcon},

        // The `VPN` submenu
        {IDC_TOGGLE_BRAVE_VPN, &kLeoProductVpnIcon},
        {IDC_TOGGLE_BRAVE_VPN_TOOLBAR_BUTTON, &kLeoEyeOnIcon},
        {IDC_SHOW_BRAVE_VPN_PANEL, &kLeoProductVpnIcon},
        {IDC_SEND_BRAVE_VPN_FEEDBACK, &kLeoMessageBubbleHeartIcon},
        {IDC_MANAGE_BRAVE_VPN_PLAN, &kLeoLaunchIcon},
        {IDC_ABOUT_BRAVE_VPN, &kLeoInfoOutlineIcon},

        // The `Bookmarks` submenu
        {IDC_BOOKMARK_THIS_TAB, &kLeoBrowserBookmarkAddIcon},
        {IDC_BOOKMARK_ALL_TABS, &kLeoBrowserBookmarkPluralIcon},
        {IDC_BRAVE_BOOKMARK_BAR_SUBMENU, &kLeoProductBookmarksIcon},
        {IDC_SHOW_BOOKMARK_MANAGER, &kLeoWindowBookmarkIcon},
        {IDC_SHOW_BOOKMARK_SIDE_PANEL, &kLeoProductBookmarksIcon},
        {IDC_IMPORT_SETTINGS, &kLeoImportArrowIcon},
        {IDC_READING_LIST_MENU, &kLeoReadingListIcon},

        // The `History submenu
        {IDC_SHOW_HISTORY, &kLeoHistoryIcon},
        {IDC_CLEAR_BROWSING_DATA, &kLeoTrashIcon},
        {IDC_RECENT_TABS_NO_DEVICE_TABS, &kLeoSmartphoneLaptopIcon},

        // The `Help` submenu
        {IDC_ABOUT, &kLeoBraveIconMonochromeIcon},
        {IDC_HELP_PAGE_VIA_MENU, &kLeoHelpOutlineIcon},
        {IDC_SHOW_BRAVE_WEBCOMPAT_REPORTER, &kLeoWarningTriangleOutlineIcon},

        // The `Passwords and autofill` submenu
        {IDC_SHOW_PASSWORD_MANAGER, &kLeoKeyIcon},
        {IDC_SHOW_EMAIL_ALIASES, &kLeoEmailShieldIcon},
        {IDC_SHOW_PAYMENT_METHODS, &kLeoCreditCardIcon},
        {IDC_SHOW_ADDRESSES, &kLeoLocationOnIcon},

        // The `Find and edit` submenu
        {IDC_FIND, &kLeoSearchIcon},
        // Use when cut icon is ready.
        // {IDC_CUT, &kLeoCutIcon},
        {IDC_COPY, &kLeoCopyIcon},
        {IDC_PASTE, &kLeoClipboardIcon},

        // The `Save and share` submenu
        {IDC_SAVE_PAGE, &kLeoFileDownloadIcon},
        {IDC_INSTALL_PWA, &kLeoPwaInstallIcon},
        {IDC_OPEN_IN_PWA_WINDOW, &kLeoLaunchIcon},
        {IDC_CREATE_SHORTCUT, &kLeoShareWindowsIcon},
        {IDC_COPY_URL, &kLeoLinkNormalIcon},
        {IDC_SEND_TAB_TO_SELF, &kLeoSmartphoneLaptopIcon},
        {IDC_QRCODE_GENERATOR, &kLeoQrCodeIcon},
        {IDC_SHARING_HUB_SCREENSHOT, &kLeoScreenshotIcon},

        // The `Reading list` submenu
        {IDC_READING_LIST_MENU_ADD_TAB, &kLeoReadingListAddIcon},
        {IDC_READING_LIST_MENU_SHOW_UI, &kLeoReadingListIcon},

        // The `More tools` submenu
        {IDC_ADD_NEW_PROFILE, &kLeoUserAddIcon},
        {IDC_OPEN_GUEST_PROFILE, &kLeoUserCircleIcon},
        {IDC_NAME_WINDOW, &kLeoEditBoxIcon},
        {IDC_COMMANDER, &kLeoArrowSmallRightIcon},
        {IDC_SHOW_APPS_PAGE, &kLeoGrid04Icon},
        {IDC_PERFORMANCE, &kLeoNetworkSpeedFastIcon},
        {IDC_DEV_TOOLS, &kLeoCodeIcon},
        {IDC_TASK_MANAGER_APP_MENU, &kLeoWindowBinaryCodeIcon},
        {IDC_SHOW_BRAVE_SYNC, &kLeoProductSyncIcon},
        {IDC_ROUTE_MEDIA, &kLeoChromeCastIcon},
        {IDC_SIDEBAR_SHOW_OPTION_MENU, &kLeoBrowserSidebarRightIcon},
    });

}  // namespace

void ApplyLeoIcons(ui::SimpleMenuModel* menu) {
  // Menu can be null in some tests.
  if (!menu) {
    return;
  }

  for (size_t i = 0; i < menu->GetItemCount(); ++i) {
    if (auto* icon =
            base::FindPtrOrNull(kCommandIcons, menu->GetCommandIdAt(i))) {
      menu->SetIcon(i, ui::ImageModel::FromVectorIcon(
                           *icon, ui::ColorIds::kColorMenuIcon, kLeoIconSize));
    }
  }
}