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 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
|
// Copyright 2012 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/ui/webui/fileicon_source.h"
#include <string_view>
#include "base/files/file_path.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/memory/ref_counted_memory.h"
#include "base/memory/scoped_refptr.h"
#include "base/strings/escape.h"
#include "base/strings/string_split.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/browser_process.h"
#include "chrome/common/webui_url_constants.h"
#include "net/base/url_util.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/base/webui/web_ui_util.h"
#include "ui/gfx/codec/png_codec.h"
#include "ui/gfx/image/image.h"
#include "ui/gfx/image/image_skia.h"
#include "ui/gfx/image/image_skia_rep.h"
#include "url/gurl.h"
namespace {
typedef std::map<std::string, IconLoader::IconSize> QueryIconSizeMap;
// The path used in internal URLs to file icon data.
const char kFileIconPath[] = "fileicon";
// URL parameter specifying icon size.
const char kIconSizeParameter[] = "iconsize";
// URL parameter specifying the file path for which to get an icon.
const char kPathParameter[] = "path";
// URL parameter specifying scale factor.
const char kScaleFactorParameter[] = "scale";
IconLoader::IconSize SizeStringToIconSize(std::string_view size_string) {
if (size_string == "small") {
return IconLoader::SMALL;
}
if (size_string == "large") {
return IconLoader::LARGE;
}
// We default to NORMAL if we don't recognize the size_string. Including
// size_string=="normal".
return IconLoader::NORMAL;
}
void ParseQueryParams(const std::string& path,
base::FilePath* file_path,
float* scale_factor,
IconLoader::IconSize* icon_size) {
GURL request = GURL(chrome::kChromeUIFileiconURL).Resolve(path);
for (net::QueryIterator it(request); !it.IsAtEnd(); it.Advance()) {
const std::string_view key = it.GetKey();
if (key == kPathParameter) {
*file_path = base::FilePath::FromUTF8Unsafe(it.GetUnescapedValue())
.NormalizePathSeparators();
} else if (key == kIconSizeParameter) {
*icon_size = SizeStringToIconSize(it.GetValue());
} else if (key == kScaleFactorParameter) {
webui::ParseScaleFactor(it.GetValue(), scale_factor);
}
}
}
} // namespace
FileIconSource::IconRequestDetails::IconRequestDetails() = default;
FileIconSource::IconRequestDetails::IconRequestDetails(
IconRequestDetails&& other) = default;
FileIconSource::IconRequestDetails&
FileIconSource::IconRequestDetails::operator=(IconRequestDetails&& other) =
default;
FileIconSource::IconRequestDetails::~IconRequestDetails() = default;
FileIconSource::FileIconSource() = default;
FileIconSource::~FileIconSource() = default;
void FileIconSource::FetchFileIcon(
const base::FilePath& path,
float scale_factor,
IconLoader::IconSize icon_size,
content::URLDataSource::GotDataCallback callback) {
IconManager* im = g_browser_process->icon_manager();
gfx::Image* icon = im->LookupIconFromFilepath(path, icon_size, scale_factor);
if (icon) {
scoped_refptr<base::RefCountedBytes> icon_data(new base::RefCountedBytes);
std::optional<std::vector<uint8_t>> data =
gfx::PNGCodec::EncodeBGRASkBitmap(
icon->ToImageSkia()->GetRepresentation(scale_factor).GetBitmap(),
/*discard_transparency=*/false);
if (data) {
icon_data->as_vector() = std::move(data.value());
}
std::move(callback).Run(icon_data.get());
} else {
// Attach the ChromeURLDataManager request ID to the history request.
IconRequestDetails details;
details.callback = std::move(callback);
details.scale_factor = scale_factor;
// Icon was not in cache, go fetch it slowly.
im->LoadIcon(path, icon_size, scale_factor,
base::BindOnce(&FileIconSource::OnFileIconDataAvailable,
base::Unretained(this), std::move(details)),
&cancelable_task_tracker_);
}
}
std::string FileIconSource::GetSource() {
return kFileIconPath;
}
void FileIconSource::StartDataRequest(
const GURL& url,
const content::WebContents::Getter& wc_getter,
content::URLDataSource::GotDataCallback callback) {
const std::string path = content::URLDataSource::URLToRequestPath(url);
base::FilePath file_path;
IconLoader::IconSize icon_size = IconLoader::NORMAL;
float scale_factor = 1.0f;
// TODO(crbug.com/40050262): Make ParseQueryParams take GURL.
ParseQueryParams(path, &file_path, &scale_factor, &icon_size);
FetchFileIcon(file_path, scale_factor, icon_size, std::move(callback));
}
std::string FileIconSource::GetMimeType(const GURL&) {
// Rely on image decoder inferring the correct type.
return std::string();
}
bool FileIconSource::AllowCaching() {
return false;
}
void FileIconSource::OnFileIconDataAvailable(IconRequestDetails details,
gfx::Image icon) {
if (!icon.IsEmpty()) {
std::optional<std::vector<uint8_t>> data =
gfx::PNGCodec::EncodeBGRASkBitmap(
icon.ToImageSkia()
->GetRepresentation(details.scale_factor)
.GetBitmap(),
/*discard_transparency=*/false);
if (data) {
std::move(details.callback)
.Run(base::MakeRefCounted<base::RefCountedBytes>(
std::move(data.value())));
return;
}
}
// TODO(glen): send a dummy icon.
std::move(details.callback).Run(nullptr);
}
|