chrome/browser/ui/views/badge_service_delegate_impl.cc

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

#include "chrome/browser/badging/badge_service_delegate.h"

#include "build/build_config.h"
#include "chrome/browser/ui/browser_finder.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "content/public/browser/web_contents.h"

#if defined(OS_WIN)
#include "chrome/browser/ui/views/frame/taskbar_decorator_win.cc"
#elif defined(OS_MACOSX)
#include "chrome/browser/apps/app_shim/app_shim_host_mac.h"
#include "chrome/browser/apps/app_shim/extension_app_shim_handler_mac.h"
#include "chrome/common/mac/app_shim.mojom.h"
#endif

namespace {

#if defined(OS_MACOSX)
void SetAppShimBadgeLabel(content::WebContents* contents,
                          const std::string& badge_label) {
  Browser* browser = chrome::FindBrowserWithWebContents(contents);
  auto* shim_handler = apps::ExtensionAppShimHandler::Get();
  if (!shim_handler)
    return;

  AppShimHost* shim_host = shim_handler->GetHostForBrowser(browser);
  if (!shim_host)
    return;

  chrome::mojom::AppShim* shim = shim_host->GetAppShim();
  if (!shim)
    return;

  shim->SetBadgeLabel(badge_label);
}
#endif

}  // namespace

void BadgeServiceDelegate::SetBadge(content::WebContents* contents) {
#if defined(OS_WIN)
  Browser* browser = chrome::FindBrowserWithWebContents(contents);
  auto* window = browser->window()->GetNativeWindow();
  chrome::DrawNumericTaskbarDecoration(window);
#elif defined(OS_MACOSX)
  SetAppShimBadgeLabel(contents, "•");
#endif
}

void BadgeServiceDelegate::ClearBadge(content::WebContents* contents) {
#if defined(OS_WIN)
  Browser* browser = chrome::FindBrowserWithWebContents(contents);
  BrowserView* browser_view = BrowserView::GetBrowserViewForBrowser(browser);

  // Restore the decoration to whatever it is naturally (either nothing or a
  // profile picture badge).
  browser_view->frame()->GetFrameView()->UpdateTaskbarDecoration();
#elif defined(OS_MACOSX)
  SetAppShimBadgeLabel(contents, "");
#endif
}