1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright 2023 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/controlled_frame/controlled_frame_menu_icon_loader.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/types/expected.h"
#include "chrome/browser/web_applications/isolated_web_apps/isolated_web_app_url_info.h"
#include "chrome/browser/web_applications/web_app_icon_manager.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/browser/web_applications/web_app_provider.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/render_frame_host.h"
#include "extensions/common/extension.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/favicon_size.h"
#include "ui/gfx/geometry/size.h"
#include "ui/gfx/image/image.h"
#include "ui/gfx/image/image_skia.h"
#include "ui/gfx/image/image_skia_operations.h"
namespace controlled_frame {
ControlledFrameMenuIconLoader::ControlledFrameMenuIconLoader() = default;
ControlledFrameMenuIconLoader::~ControlledFrameMenuIconLoader() = default;
void ControlledFrameMenuIconLoader::LoadIcon(
content::BrowserContext* context,
const extensions::Extension* extension,
const extensions::MenuItem::ExtensionKey& extension_key) {
const auto* render_frame_host = content::RenderFrameHost::FromID(
extension_key.webview_embedder_process_id,
extension_key.webview_embedder_frame_id);
CHECK(render_frame_host);
const base::expected<web_app::IsolatedWebAppUrlInfo, std::string>
iwa_url_info = web_app::IsolatedWebAppUrlInfo::Create(
render_frame_host->GetLastCommittedURL());
CHECK(iwa_url_info.has_value());
auto* web_app_provider = web_app::WebAppProvider::GetForWebApps(
Profile::FromBrowserContext(context));
CHECK(web_app_provider);
auto result = pending_icons_.insert(extension_key);
CHECK(result.second);
web_app_provider->icon_manager().ReadSmallestIcon(
iwa_url_info->app_id(), {web_app::IconPurpose::ANY}, gfx::kFaviconSize,
base::BindOnce(&ControlledFrameMenuIconLoader::OnIconLoaded,
weak_ptr_factory_.GetWeakPtr(), extension_key));
}
gfx::Image ControlledFrameMenuIconLoader::GetIcon(
const extensions::MenuItem::ExtensionKey& extension_key) {
auto it = icons_.find(extension_key);
CHECK(it != icons_.end());
gfx::Image result = it->second;
return result;
}
void ControlledFrameMenuIconLoader::RemoveIcon(
const extensions::MenuItem::ExtensionKey& extension_key) {
size_t count = icons_.erase(extension_key);
count += pending_icons_.erase(extension_key);
// The icon should only be present in one of these two containers.
CHECK_EQ(1u, count);
}
void ControlledFrameMenuIconLoader::SetNotifyOnLoadedCallbackForTesting(
base::RepeatingClosure callback) {
on_loaded_callback_ = callback;
}
void ControlledFrameMenuIconLoader::OnIconLoaded(
const extensions::MenuItem::ExtensionKey& extension_key,
web_app::IconPurpose purpose,
SkBitmap bitmap) {
gfx::Image image = gfx::Image::CreateFrom1xBitmap(bitmap);
CHECK(!image.IsEmpty());
// The icon may have been removed while waiting for it to load. Do nothing in
// that case.
if (pending_icons_.erase(extension_key) == 0) {
return;
}
if (image.Height() != gfx::kFaviconSize ||
image.Width() != gfx::kFaviconSize) {
gfx::Size favicon_size(gfx::kFaviconSize, gfx::kFaviconSize);
gfx::ImageSkia scaled_image = gfx::ImageSkiaOperations::CreateResizedImage(
image.AsImageSkia(), skia::ImageOperations::ResizeMethod::RESIZE_GOOD,
favicon_size);
image = gfx::Image(scaled_image);
}
CHECK_EQ(gfx::kFaviconSize, image.Width());
CHECK_EQ(gfx::kFaviconSize, image.Height());
icons_[extension_key] = gfx::Image(image);
if (on_loaded_callback_) {
on_loaded_callback_.Run();
}
}
} // namespace controlled_frame
|
