components/suggestions/webui/suggestions_source.cc - chromium/src - Git at Google

 // Copyright 2017 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 "components/suggestions/webui/suggestions_source.h"

 #include "base/barrier_closure.h"
 #include "base/base64.h"
 #include "base/bind.h"
 #include "base/strings/strcat.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "components/suggestions/proto/suggestions.pb.h"
 #include "net/base/escape.h"
 #include "ui/base/l10n/time_format.h"
 #include "ui/gfx/codec/png_codec.h"
 #include "ui/gfx/image/image_skia.h"

 namespace suggestions {

 namespace {

 const char kHtmlHeader[] =
     "<!DOCTYPE html>\n<html>\n<head>\n<title>Suggestions</title>\n"
     "<meta charset=\"utf-8\">\n"
     "<style type=\"text/css\">\nli {white-space: nowrap;}\n</style>\n";
 const char kHtmlBody[] = "</head>\n<body>\n";
 const char kHtmlFooter[] = "</body>\n</html>\n";

 const char kRefreshPath[] = "refresh";

 std::string GetRefreshHtml(const std::string& base_url, bool is_refresh) {
   if (is_refresh)
     return "<p>Refreshing in the background, reload to see new data.</p>\n";
   return std::string("<p><a href=\"") + base_url + kRefreshPath +
          "\">Refresh</a></p>\n";
 }
 // Returns the HTML needed to display the suggestions.
 std::string RenderOutputHtml(const std::string& base_url,
                              bool is_refresh,
                              const SuggestionsProfile& profile) {
   std::vector<std::string> out;
   out.push_back(kHtmlHeader);
   out.push_back(kHtmlBody);
   out.push_back("<h1>Suggestions</h1>\n");
   out.push_back(GetRefreshHtml(base_url, is_refresh));
   out.push_back("<ul>");
   int64_t now = (base::Time::NowFromSystemTime() - base::Time::UnixEpoch())
                     .ToInternalValue();
   size_t size = profile.suggestions_size();
   for (size_t i = 0; i < size; ++i) {
     const ChromeSuggestion& suggestion = profile.suggestions(i);
     base::TimeDelta remaining_time =
         base::TimeDelta::FromMicroseconds(suggestion.expiry_ts() - now);
     base::string16 remaining_time_formatted = ui::TimeFormat::Detailed(
         ui::TimeFormat::Format::FORMAT_DURATION,
         ui::TimeFormat::Length::LENGTH_LONG, -1, remaining_time);
     std::string line;
     line += "<li><a href=\"";
     line += net::EscapeForHTML(suggestion.url());
     line += "\" target=\"_blank\">";
     line += net::EscapeForHTML(suggestion.title());
     line += "</a> Expires in ";
     line += base::UTF16ToUTF8(remaining_time_formatted);
     std::vector<std::string> providers;
     for (int p = 0; p < suggestion.providers_size(); ++p)
       providers.push_back(base::NumberToString(suggestion.providers(p)));
     line += ". Provider IDs: " + base::JoinString(providers, ", ");
     line += "</li>\n";
     out.push_back(line);
   }
   out.push_back("</ul>");
   out.push_back(kHtmlFooter);
   return base::StrCat(out);
 }

 // Returns the HTML needed to display that no suggestions are available.
 std::string RenderOutputHtmlNoSuggestions(const std::string& base_url,
                                           bool is_refresh) {
   std::vector<std::string> out;
   out.push_back(kHtmlHeader);
   out.push_back(kHtmlBody);
   out.push_back("<h1>Suggestions</h1>\n");
   out.push_back("<p>You have no suggestions.</p>\n");
   out.push_back(GetRefreshHtml(base_url, is_refresh));
   out.push_back(kHtmlFooter);
   return base::StrCat(out);
 }

 }  // namespace

 SuggestionsSource::SuggestionsSource(SuggestionsService* suggestions_service,
                                      const std::string& base_url)
     : suggestions_service_(suggestions_service),
       base_url_(base_url),
       weak_ptr_factory_(this) {}

 SuggestionsSource::~SuggestionsSource() {}

 void SuggestionsSource::StartDataRequest(const std::string& path,
                                          const GotDataCallback& callback) {
   // If this was called as "chrome://suggestions/refresh", we also trigger an
   // async update of the suggestions.
   bool is_refresh = (path == kRefreshPath);

   // |suggestions_service| is null for guest profiles.
   if (!suggestions_service_) {
     std::string output = RenderOutputHtmlNoSuggestions(base_url_, is_refresh);
     callback.Run(base::RefCountedString::TakeString(&output));
     return;
   }

   if (is_refresh)
     suggestions_service_->FetchSuggestionsData();

   SuggestionsProfile suggestions_profile =
       suggestions_service_->GetSuggestionsDataFromCache().value_or(
           SuggestionsProfile());
   size_t size = suggestions_profile.suggestions_size();

   std::string output =
       !size ? RenderOutputHtmlNoSuggestions(base_url_, is_refresh)
             : RenderOutputHtml(base_url_, is_refresh, suggestions_profile);
   callback.Run(base::RefCountedString::TakeString(&output));
 }

 std::string SuggestionsSource::GetMimeType(const std::string& path) const {
   return "text/html";
 }

 }  // namespace suggestions
	// Copyright 2017 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 "components/suggestions/webui/suggestions_source.h"

	#include "base/barrier_closure.h"
	#include "base/base64.h"
	#include "base/bind.h"
	#include "base/strings/strcat.h"
	#include "base/strings/string16.h"
	#include "base/strings/string_number_conversions.h"
	#include "base/strings/string_piece.h"
	#include "base/strings/string_util.h"
	#include "base/strings/utf_string_conversions.h"
	#include "base/time/time.h"
	#include "components/suggestions/proto/suggestions.pb.h"
	#include "net/base/escape.h"
	#include "ui/base/l10n/time_format.h"
	#include "ui/gfx/codec/png_codec.h"
	#include "ui/gfx/image/image_skia.h"

	namespace suggestions {

	namespace {

	const char kHtmlHeader[] =
	"<!DOCTYPE html>\n<html>\n<head>\n<title>Suggestions</title>\n"
	"<meta charset=\"utf-8\">\n"
	"<style type=\"text/css\">\nli {white-space: nowrap;}\n</style>\n";
	const char kHtmlBody[] = "</head>\n<body>\n";
	const char kHtmlFooter[] = "</body>\n</html>\n";

	const char kRefreshPath[] = "refresh";

	std::string GetRefreshHtml(const std::string& base_url, bool is_refresh) {
	if (is_refresh)
	return "<p>Refreshing in the background, reload to see new data.</p>\n";
	return std::string("<p><a href=\"") + base_url + kRefreshPath +
	"\">Refresh</a></p>\n";
	}
	// Returns the HTML needed to display the suggestions.
	std::string RenderOutputHtml(const std::string& base_url,
	bool is_refresh,
	const SuggestionsProfile& profile) {
	std::vector<std::string> out;
	out.push_back(kHtmlHeader);
	out.push_back(kHtmlBody);
	out.push_back("<h1>Suggestions</h1>\n");
	out.push_back(GetRefreshHtml(base_url, is_refresh));
	out.push_back("<ul>");
	int64_t now = (base::Time::NowFromSystemTime() - base::Time::UnixEpoch())
	.ToInternalValue();
	size_t size = profile.suggestions_size();
	for (size_t i = 0; i < size; ++i) {
	const ChromeSuggestion& suggestion = profile.suggestions(i);
	base::TimeDelta remaining_time =
	base::TimeDelta::FromMicroseconds(suggestion.expiry_ts() - now);
	base::string16 remaining_time_formatted = ui::TimeFormat::Detailed(
	ui::TimeFormat::Format::FORMAT_DURATION,
	ui::TimeFormat::Length::LENGTH_LONG, -1, remaining_time);
	std::string line;
	line += "<li><a href=\"";
	line += net::EscapeForHTML(suggestion.url());
	line += "\" target=\"_blank\">";
	line += net::EscapeForHTML(suggestion.title());
	line += "</a> Expires in ";
	line += base::UTF16ToUTF8(remaining_time_formatted);
	std::vector<std::string> providers;
	for (int p = 0; p < suggestion.providers_size(); ++p)
	providers.push_back(base::NumberToString(suggestion.providers(p)));
	line += ". Provider IDs: " + base::JoinString(providers, ", ");
	line += "</li>\n";
	out.push_back(line);
	}
	out.push_back("</ul>");
	out.push_back(kHtmlFooter);
	return base::StrCat(out);
	}

	// Returns the HTML needed to display that no suggestions are available.
	std::string RenderOutputHtmlNoSuggestions(const std::string& base_url,
	bool is_refresh) {
	std::vector<std::string> out;
	out.push_back(kHtmlHeader);
	out.push_back(kHtmlBody);
	out.push_back("<h1>Suggestions</h1>\n");
	out.push_back("<p>You have no suggestions.</p>\n");
	out.push_back(GetRefreshHtml(base_url, is_refresh));
	out.push_back(kHtmlFooter);
	return base::StrCat(out);
	}

	} // namespace

	SuggestionsSource::SuggestionsSource(SuggestionsService* suggestions_service,
	const std::string& base_url)
	: suggestions_service_(suggestions_service),
	base_url_(base_url),
	weak_ptr_factory_(this) {}

	SuggestionsSource::~SuggestionsSource() {}

	void SuggestionsSource::StartDataRequest(const std::string& path,
	const GotDataCallback& callback) {
	// If this was called as "chrome://suggestions/refresh", we also trigger an
	// async update of the suggestions.
	bool is_refresh = (path == kRefreshPath);

	// \|suggestions_service\| is null for guest profiles.
	if (!suggestions_service_) {
	std::string output = RenderOutputHtmlNoSuggestions(base_url_, is_refresh);
	callback.Run(base::RefCountedString::TakeString(&output));
	return;
	}

	if (is_refresh)
	suggestions_service_->FetchSuggestionsData();

	SuggestionsProfile suggestions_profile =
	suggestions_service_->GetSuggestionsDataFromCache().value_or(
	SuggestionsProfile());
	size_t size = suggestions_profile.suggestions_size();

	std::string output =
	!size ? RenderOutputHtmlNoSuggestions(base_url_, is_refresh)
	: RenderOutputHtml(base_url_, is_refresh, suggestions_profile);
	callback.Run(base::RefCountedString::TakeString(&output));
	}

	std::string SuggestionsSource::GetMimeType(const std::string& path) const {
	return "text/html";
	}

	} // namespace suggestions