chrome/browser/badging/badge_service_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_impl.h"

#include <utility>

#include "base/logging.h"
#include "chrome/browser/badging/badge_manager.h"
#include "chrome/browser/badging/badge_manager_factory.h"
#include "chrome/browser/extensions/extension_util.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser_finder.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/extensions/hosted_app_browser_controller.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "extensions/common/extension.h"

// static
void BadgeServiceImpl::Create(blink::mojom::BadgeServiceRequest request,
                              content::RenderFrameHost* render_frame_host) {
  DCHECK(render_frame_host);

  // Lifetime managed through FrameServiceBase.
  new BadgeServiceImpl(render_frame_host, std::move(request));
}

void BadgeServiceImpl::SetInteger(uint64_t content) {
  SetBadge(base::Optional<uint64_t>(content));
}

void BadgeServiceImpl::SetFlag() {
  SetBadge(base::nullopt);
}

void BadgeServiceImpl::SetBadge(base::Optional<uint64_t> content) {
  if (!IsInApp())
    return;

  const extensions::Extension* extension = ExtensionFromLastUrl();

  if (!extension)
    return;

  badge_manager_->UpdateBadge(extension->id(), content);
}

void BadgeServiceImpl::ClearBadge() {
  if (!IsInApp())
    return;

  const extensions::Extension* extension = ExtensionFromLastUrl();

  if (!extension)
    return;

  badge_manager_->ClearBadge(extension->id());
}

BadgeServiceImpl::BadgeServiceImpl(content::RenderFrameHost* render_frame_host,
                                   blink::mojom::BadgeServiceRequest request)
    : content::FrameServiceBase<blink::mojom::BadgeService>(render_frame_host,
                                                            std::move(request)),
      render_frame_host_(render_frame_host) {
  web_contents_ = content::WebContents::FromRenderFrameHost(render_frame_host_);
  browser_context_ = web_contents_->GetBrowserContext();
  badge_manager_ = badging::BadgeManagerFactory::GetInstance()->GetForProfile(
      Profile::FromBrowserContext(browser_context_));
}

BadgeServiceImpl::~BadgeServiceImpl() = default;

const extensions::Extension* BadgeServiceImpl::ExtensionFromLastUrl() {
  DCHECK(browser_context_);

  return extensions::util::GetInstalledPwaForUrl(
      browser_context_, render_frame_host_->GetLastCommittedURL());
}

bool BadgeServiceImpl::IsInApp() {
  extensions::HostedAppBrowserController* hosted_app_controller =
      chrome::FindBrowserWithWebContents(web_contents_)
          ->hosted_app_controller();
  return hosted_app_controller &&
         extensions::IsSameScope(hosted_app_controller->GetAppLaunchURL(),
                                 web_contents_->GetLastCommittedURL(),
                                 web_contents_->GetBrowserContext());
}