1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/ui/webui/thumbnail_source.h"
6
7 #include "base/callback.h"
8 #include "chrome/browser/history/top_sites.h"
9 #include "chrome/browser/profiles/profile.h"
10 #include "chrome/common/url_constants.h"
11 #include "googleurl/src/gurl.h"
12 #include "grit/theme_resources.h"
13 #include "ui/base/resource/resource_bundle.h"
14
ThumbnailSource(Profile * profile)15 ThumbnailSource::ThumbnailSource(Profile* profile)
16 : DataSource(chrome::kChromeUIThumbnailPath, MessageLoop::current()),
17 // Set TopSites now as Profile isn't thread safe.
18 top_sites_(profile->GetTopSites()) {
19 }
20
~ThumbnailSource()21 ThumbnailSource::~ThumbnailSource() {
22 }
23
StartDataRequest(const std::string & path,bool is_incognito,int request_id)24 void ThumbnailSource::StartDataRequest(const std::string& path,
25 bool is_incognito,
26 int request_id) {
27 scoped_refptr<RefCountedBytes> data;
28 if (top_sites_->GetPageThumbnail(GURL(path), &data)) {
29 // We have the thumbnail.
30 SendResponse(request_id, data.get());
31 } else {
32 SendDefaultThumbnail(request_id);
33 }
34 }
35
GetMimeType(const std::string &) const36 std::string ThumbnailSource::GetMimeType(const std::string&) const {
37 // We need to explicitly return a mime type, otherwise if the user tries to
38 // drag the image they get no extension.
39 return "image/png";
40 }
41
MessageLoopForRequestPath(const std::string & path) const42 MessageLoop* ThumbnailSource::MessageLoopForRequestPath(
43 const std::string& path) const {
44 // TopSites can be accessed from the IO thread.
45 return top_sites_.get() ? NULL : DataSource::MessageLoopForRequestPath(path);
46 }
47
SendDefaultThumbnail(int request_id)48 void ThumbnailSource::SendDefaultThumbnail(int request_id) {
49 // Use placeholder thumbnail.
50 if (!default_thumbnail_.get()) {
51 default_thumbnail_ =
52 ResourceBundle::GetSharedInstance().LoadDataResourceBytes(
53 IDR_DEFAULT_THUMBNAIL);
54 }
55 SendResponse(request_id, default_thumbnail_);
56 }
57