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
15ThumbnailSource::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
21ThumbnailSource::~ThumbnailSource() {
22}
23
24void 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
36std::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
42MessageLoop* 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
48void 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