| // 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.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 "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) { |
| if (!profile_->GetTopSites()) { |
| callback.Run(NULL); |
| return; |
| } |
| |
| profile_->GetTopSites()->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)); |
| } |