blob: 1cba5a6b06d9a6d653fc43fbeb17181892c0993d [file] [log] [blame]
// 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 <string>
#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 "net/base/escape.h"
#include "net/url_request/url_request.h"
namespace {
const char kHtmlHeader[] =
"<!DOCTYPE html>\n<html>\n<head>\n<title>TopSites Thumbnails</title>\n"
"<meta charset=\"utf-8\">\n"
"<style type=\"text/css\">\nimg.thumb {border: 1px solid black;}\n"
"li {white-space: nowrap;}\n</style>\n";
const char kHtmlBody[] = "</head>\n<body>\n";
const char kHtmlFooter[] = "</body>\n</html>\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<std::string>& base64_encoded_pngs,
bool want_thumbnails,
std::vector<std::string>* out) {
DCHECK_EQ(mvurl_list.size(), base64_encoded_pngs.size());
bool doing_forced_urls = true;
out->push_back("<div><b>Forced URLs:</b></div>\n"
"<div><ul>\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("</ul></div>\n"
"<div><b>Non-forced URLs:</b></div>\n"
"<div><ul>\n");
doing_forced_urls = false;
}
bool has_thumbnail = !base64_encoded_pngs[i].empty();
if (has_thumbnail == want_thumbnails) {
out->push_back("<li>\n");
out->push_back(net::EscapeForHTML(mvurl.url.spec()) + "\n");
if (want_thumbnails) {
out->push_back("<div><img class=\"thumb\" "
"src=\"data:image/png;base64," +
base64_encoded_pngs[i] + "\"/></div>\n");
}
if (!mvurl.redirects.empty()) {
out->push_back("<ul>\n");
history::RedirectList::const_iterator jt;
for (jt = mvurl.redirects.begin();
jt != mvurl.redirects.end(); ++jt) {
out->push_back("<li>" + net::EscapeForHTML(jt->spec()) + "</li>\n");
}
out->push_back("</ul>\n");
}
out->push_back("</li>\n");
}
}
out->push_back("</ul></div>\n");
}
} // namespace
ThumbnailListSource::ThumbnailListSource(Profile* profile)
: thumbnail_service_(ThumbnailServiceFactory::GetForProfile(profile)),
profile_(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) {
scoped_refptr<history::TopSites> top_sites =
TopSitesFactory::GetForProfile(profile_);
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) {
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<std::string> base64_encoded_pngs(num_mv);
for (size_t i = 0; i < num_mv; ++i) {
scoped_refptr<base::RefCountedMemory> data;
if (thumbnail_service_->GetPageThumbnail(mvurl_list[i].url, false, &data)) {
base::Base64Encode(std::string(data->front_as<char>(), data->size()),
&base64_encoded_pngs[i]);
++num_mv_with_thumb;
}
}
// Render HTML to embed URLs and thumbnails.
std::vector<std::string> out;
out.push_back(kHtmlHeader);
out.push_back(kHtmlBody);
if (num_mv_with_thumb > 0) {
out.push_back("<h2>TopSites URLs with Thumbnails</h2>\n");
RenderMostVisitedURLList(mvurl_list, base64_encoded_pngs, true, &out);
}
if (num_mv_with_thumb < num_mv) {
out.push_back("<h2>TopSites URLs without Thumbnails</h2>\n");
RenderMostVisitedURLList(mvurl_list, base64_encoded_pngs, false, &out);
}
out.push_back(kHtmlFooter);
std::string out_html = JoinString(out, "");
callback.Run(base::RefCountedString::TakeString(&out_html));
}