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
|
// Copyright (c) 2012 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/search/thumbnail_source.h"
#include <stddef.h>
#include "base/callback.h"
#include "base/memory/ref_counted_memory.h"
#include "base/message_loop/message_loop.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/search/instant_io_context.h"
#include "chrome/browser/search/suggestions/image_fetcher_impl.h"
#include "chrome/browser/thumbnails/thumbnail_service.h"
#include "chrome/browser/thumbnails/thumbnail_service_factory.h"
#include "chrome/common/url_constants.h"
#include "components/suggestions/image_encoder.h"
#include "net/url_request/url_request.h"
#include "ui/gfx/image/image_skia.h"
// The delimiter between the first url and the fallback url passed to
// StartDataRequest.
const char kUrlDelimiter[] = "?fb=";
// Set ThumbnailService now as Profile isn't thread safe.
ThumbnailSource::ThumbnailSource(Profile* profile, bool capture_thumbnails)
: thumbnail_service_(ThumbnailServiceFactory::GetForProfile(profile)),
capture_thumbnails_(capture_thumbnails),
weak_ptr_factory_(this) {
image_fetcher_.reset(
new suggestions::ImageFetcherImpl(profile->GetRequestContext()));
}
ThumbnailSource::~ThumbnailSource() {
}
std::string ThumbnailSource::GetSource() const {
return capture_thumbnails_ ?
chrome::kChromeUIThumbnailHost2 : chrome::kChromeUIThumbnailHost;
}
void ThumbnailSource::StartDataRequest(
const std::string& path,
int render_process_id,
int render_frame_id,
const content::URLDataSource::GotDataCallback& callback) {
GURL page_url;
GURL fallback_thumbnail_url;
ExtractPageAndThumbnailUrls(path, &page_url, &fallback_thumbnail_url);
scoped_refptr<base::RefCountedMemory> data;
if (page_url.is_valid() &&
thumbnail_service_->GetPageThumbnail(page_url, capture_thumbnails_,
&data)) {
// If a local thumbnail is available for the page's URL, provide it.
callback.Run(data.get());
} else if (fallback_thumbnail_url.is_valid()) {
// Otherwise, if a fallback thumbnail URL was provided, fetch it and
// eventually return it.
image_fetcher_->StartOrQueueNetworkRequest(
page_url, fallback_thumbnail_url,
base::Bind(&ThumbnailSource::SendFetchedUrlImage,
weak_ptr_factory_.GetWeakPtr(), callback));
} else {
callback.Run(default_thumbnail_.get());
}
if (page_url.is_valid() && capture_thumbnails_)
thumbnail_service_->AddForcedURL(page_url);
}
std::string ThumbnailSource::GetMimeType(const std::string&) const {
// We need to explicitly return a mime type, otherwise if the user tries to
// drag the image they get no extension.
return "image/png";
}
base::MessageLoop* ThumbnailSource::MessageLoopForRequestPath(
const std::string& path) const {
// TopSites can be accessed from the IO thread. Otherwise, the URLs should be
// accessed on the UI thread.
return thumbnail_service_.get()
? NULL
: content::URLDataSource::MessageLoopForRequestPath(path);
}
bool ThumbnailSource::ShouldServiceRequest(
const net::URLRequest* request) const {
if (request->url().SchemeIs(chrome::kChromeSearchScheme))
return InstantIOContext::ShouldServiceRequest(request);
return URLDataSource::ShouldServiceRequest(request);
}
void ThumbnailSource::ExtractPageAndThumbnailUrls(
const std::string& path,
GURL* page_url,
GURL* fallback_thumbnail_url) {
std::string page_url_str(path);
std::string fallback_thumbnail_url_str;
size_t pos = path.find(kUrlDelimiter);
if (pos != std::string::npos) {
page_url_str = path.substr(0, pos);
fallback_thumbnail_url_str = path.substr(pos + strlen(kUrlDelimiter));
}
*page_url = GURL(page_url_str);
*fallback_thumbnail_url = GURL(fallback_thumbnail_url_str);
}
void ThumbnailSource::SendFetchedUrlImage(
const content::URLDataSource::GotDataCallback& callback,
const GURL& url,
const SkBitmap* bitmap) {
if (!bitmap) {
callback.Run(default_thumbnail_.get());
return;
}
scoped_refptr<base::RefCountedBytes> encoded_data(
new base::RefCountedBytes());
if (!suggestions::EncodeSkBitmapToJPEG(*bitmap, &encoded_data->data())) {
callback.Run(default_thumbnail_.get());
return;
}
callback.Run(encoded_data.get());
}
|
