// Copyright 2014 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/thumbnails/thumbnail_list_source.h" #include #include "base/base64.h" #include "base/bind.h" #include "base/logging.h" #include "base/memory/ref_counted_memory.h" #include "base/strings/string_util.h" #include "chrome/browser/history/top_sites_factory.h" #include "chrome/browser/profiles/profile.h" #include "chrome/browser/search/instant_io_context.h" #include "chrome/browser/thumbnails/thumbnail_service.h" #include "chrome/browser/thumbnails/thumbnail_service_factory.h" #include "chrome/common/url_constants.h" #include "components/history/core/browser/top_sites.h" #include "content/public/browser/browser_thread.h" #include "net/base/escape.h" #include "net/url_request/url_request.h" using content::BrowserThread; namespace { const char kHtmlHeader[] = "\n\n\nTopSites Thumbnails\n" "\n" "\n"; const char kHtmlBody[] = "\n\n"; const char kHtmlFooter[] = "\n\n"; // If |want_thumbnails| == true, then renders elements in |mvurl_list| that have // thumbnails, with their thumbnails. Otherwise renders elements in |mvurl_list| // that have no thumbnails. void RenderMostVisitedURLList( const history::MostVisitedURLList& mvurl_list, const std::vector& base64_encoded_pngs, bool want_thumbnails, std::vector* out) { DCHECK_EQ(mvurl_list.size(), base64_encoded_pngs.size()); bool doing_forced_urls = true; out->push_back("
Forced URLs:
\n" "
    \n"); for (size_t i = 0; i < mvurl_list.size(); ++i) { const history::MostVisitedURL& mvurl = mvurl_list[i]; if (doing_forced_urls && mvurl.last_forced_time.is_null()) { out->push_back("
\n" "
Non-forced URLs:
\n" "
    \n"); doing_forced_urls = false; } bool has_thumbnail = !base64_encoded_pngs[i].empty(); if (has_thumbnail == want_thumbnails) { out->push_back("
  • \n"); out->push_back(net::EscapeForHTML(mvurl.url.spec()) + "\n"); if (want_thumbnails) { out->push_back("
    \n"); } if (!mvurl.redirects.empty()) { out->push_back("
      \n"); history::RedirectList::const_iterator jt; for (jt = mvurl.redirects.begin(); jt != mvurl.redirects.end(); ++jt) { out->push_back("
    • " + net::EscapeForHTML(jt->spec()) + "
      • \n"); } out->push_back("
    \n"); } out->push_back("
    • \n"); } } out->push_back("
\n"); } } // namespace ThumbnailListSource::ThumbnailListSource(Profile* profile) : thumbnail_service_(ThumbnailServiceFactory::GetForProfile(profile)), top_sites_(TopSitesFactory::GetForProfile(profile)), weak_ptr_factory_(this) { } ThumbnailListSource::~ThumbnailListSource() { } std::string ThumbnailListSource::GetSource() const { return chrome::kChromeUIThumbnailListHost; } void ThumbnailListSource::StartDataRequest( const std::string& path, int render_process_id, int render_frame_id, const content::URLDataSource::GotDataCallback& callback) { DCHECK_CURRENTLY_ON(BrowserThread::IO); if (!top_sites_) { callback.Run(NULL); return; } top_sites_->GetMostVisitedURLs( base::Bind(&ThumbnailListSource::OnMostVisitedURLsAvailable, weak_ptr_factory_.GetWeakPtr(), callback), true); } std::string ThumbnailListSource::GetMimeType(const std::string& path) const { return "text/html"; } base::MessageLoop* ThumbnailListSource::MessageLoopForRequestPath( const std::string& path) const { // TopSites can be accessed from the IO thread. return thumbnail_service_.get() ? NULL : content::URLDataSource::MessageLoopForRequestPath(path); } bool ThumbnailListSource::ShouldServiceRequest( const net::URLRequest* request) const { if (request->url().SchemeIs(chrome::kChromeSearchScheme)) return InstantIOContext::ShouldServiceRequest(request); return URLDataSource::ShouldServiceRequest(request); } bool ThumbnailListSource::ShouldReplaceExistingSource() const { return false; } void ThumbnailListSource::OnMostVisitedURLsAvailable( const content::URLDataSource::GotDataCallback& callback, const history::MostVisitedURLList& mvurl_list) { DCHECK_CURRENTLY_ON(BrowserThread::IO); const size_t num_mv = mvurl_list.size(); size_t num_mv_with_thumb = 0; // Encode all available thumbnails and store into |base64_encoded_pngs|. std::vector base64_encoded_pngs(num_mv); for (size_t i = 0; i < num_mv; ++i) { scoped_refptr data; if (thumbnail_service_->GetPageThumbnail(mvurl_list[i].url, false, &data)) { base::Base64Encode(std::string(data->front_as(), data->size()), &base64_encoded_pngs[i]); ++num_mv_with_thumb; } } // Render HTML to embed URLs and thumbnails. std::vector out; out.push_back(kHtmlHeader); out.push_back(kHtmlBody); if (num_mv_with_thumb > 0) { out.push_back("

TopSites URLs with Thumbnails

\n"); RenderMostVisitedURLList(mvurl_list, base64_encoded_pngs, true, &out); } if (num_mv_with_thumb < num_mv) { out.push_back("

TopSites URLs without Thumbnails

\n"); RenderMostVisitedURLList(mvurl_list, base64_encoded_pngs, false, &out); } out.push_back(kHtmlFooter); std::string out_html = base::JoinString(out, base::StringPiece()); callback.Run(base::RefCountedString::TakeString(&out_html)); }