Index: chrome/browser/ui/webui/ntp/thumbnail_list_source.cc |
diff --git a/chrome/browser/ui/webui/ntp/thumbnail_list_source.cc b/chrome/browser/ui/webui/ntp/thumbnail_list_source.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..482ab0e0f081af0460f62fc28c077b6ae1f7a1b3 |
--- /dev/null |
+++ b/chrome/browser/ui/webui/ntp/thumbnail_list_source.cc |
@@ -0,0 +1,148 @@ |
+// Copyright (c) 2013 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/ui/webui/ntp/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* html_header = |
+ "<!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* html_body = "</head>\n<body>\n"; |
+const char* html_footer = "</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()); |
+ out->push_back("<div><ul>\n"); |
+ for (size_t i = 0; i < mvurl_list.size(); ++i) { |
+ const history::MostVisitedURL& mvurl = mvurl_list[i]; |
+ 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_view_id, |
+ const content::URLDataSource::GotDataCallback& callback) { |
+ profile_->GetTopSites()->GetMostVisitedURLs( |
+ base::Bind(&ThumbnailListSource::OnMostVisitedURLsAvailable, |
+ weak_ptr_factory_.GetWeakPtr(), |
+ callback)); |
+} |
+ |
+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); |
+} |
+ |
+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)) { |
+ std::string data_str; |
+ data_str.assign(reinterpret_cast<const char*>(data->front()), |
+ data->size()); |
+ base::Base64Encode(data_str, &base64_encoded_pngs[i]); |
+ ++num_mv_with_thumb; |
+ } |
+ } |
+ |
+ // Render HTML to embed URLs and thumbnails. |
+ std::vector<std::string> out; |
+ out.push_back(html_header); |
+ out.push_back(html_body); |
+ 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(html_footer); |
+ |
+ std::string out_html = JoinString(out, ""); |
+ callback.Run(base::RefCountedString::TakeString(&out_html)); |
+} |