1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "components/dom_distiller/webui/dom_distiller_handler.h"
6
7 #include <vector>
8
9 #include "base/bind.h"
10 #include "base/values.h"
11 #include "components/dom_distiller/core/dom_distiller_service.h"
12 #include "components/dom_distiller/core/proto/distilled_page.pb.h"
13 #include "components/dom_distiller/core/url_utils.h"
14 #include "content/public/browser/web_contents.h"
15 #include "content/public/browser/web_ui.h"
16 #include "net/base/escape.h"
17 #include "url/gurl.h"
18
19 namespace dom_distiller {
20
21 namespace {
22
GetViewUrlFromArgs(const std::string & scheme,const base::ListValue * args)23 GURL GetViewUrlFromArgs(const std::string& scheme,
24 const base::ListValue* args) {
25 std::string url;
26 if (args->GetString(0, &url)) {
27 const GURL gurl(url);
28 if (url_utils::IsUrlDistillable(gurl)) {
29 return url_utils::GetDistillerViewUrlFromUrl(scheme, gurl);
30 }
31 }
32 return GURL();
33 }
34
35 } // namespace
36
DomDistillerHandler(DomDistillerService * service,const std::string & scheme)37 DomDistillerHandler::DomDistillerHandler(DomDistillerService* service,
38 const std::string& scheme)
39 : service_(service), article_scheme_(scheme), weak_ptr_factory_(this) {}
40
~DomDistillerHandler()41 DomDistillerHandler::~DomDistillerHandler() {}
42
RegisterMessages()43 void DomDistillerHandler::RegisterMessages() {
44 web_ui()->RegisterMessageCallback(
45 "requestEntries",
46 base::Bind(&DomDistillerHandler::HandleRequestEntries,
47 base::Unretained(this)));
48 web_ui()->RegisterMessageCallback(
49 "addArticle",
50 base::Bind(&DomDistillerHandler::HandleAddArticle,
51 base::Unretained(this)));
52 web_ui()->RegisterMessageCallback(
53 "selectArticle",
54 base::Bind(&DomDistillerHandler::HandleSelectArticle,
55 base::Unretained(this)));
56 web_ui()->RegisterMessageCallback(
57 "viewUrl",
58 base::Bind(&DomDistillerHandler::HandleViewUrl, base::Unretained(this)));
59 }
60
HandleAddArticle(const base::ListValue * args)61 void DomDistillerHandler::HandleAddArticle(const base::ListValue* args) {
62 std::string url;
63 args->GetString(0, &url);
64 GURL gurl(url);
65 if (gurl.is_valid()) {
66 service_->AddToList(
67 gurl,
68 service_->CreateDefaultDistillerPage(
69 web_ui()->GetWebContents()->GetContainerBounds().size()),
70 base::Bind(base::Bind(&DomDistillerHandler::OnArticleAdded,
71 base::Unretained(this))));
72 } else {
73 web_ui()->CallJavascriptFunction("domDistiller.onArticleAddFailed");
74 }
75 }
76
HandleViewUrl(const base::ListValue * args)77 void DomDistillerHandler::HandleViewUrl(const base::ListValue* args) {
78 GURL view_url = GetViewUrlFromArgs(article_scheme_, args);
79 if (view_url.is_valid()) {
80 web_ui()->GetWebContents()->GetController().LoadURL(
81 view_url,
82 content::Referrer(),
83 ui::PAGE_TRANSITION_GENERATED,
84 std::string());
85 } else {
86 web_ui()->CallJavascriptFunction("domDistiller.onViewUrlFailed");
87 }
88 }
89
HandleSelectArticle(const base::ListValue * args)90 void DomDistillerHandler::HandleSelectArticle(const base::ListValue* args) {
91 std::string entry_id;
92 args->GetString(0, &entry_id);
93 GURL url =
94 url_utils::GetDistillerViewUrlFromEntryId(article_scheme_, entry_id);
95 DCHECK(url.is_valid());
96 web_ui()->GetWebContents()->GetController().LoadURL(
97 url,
98 content::Referrer(),
99 ui::PAGE_TRANSITION_GENERATED,
100 std::string());
101 }
102
HandleRequestEntries(const base::ListValue * args)103 void DomDistillerHandler::HandleRequestEntries(const base::ListValue* args) {
104 base::ListValue entries;
105 const std::vector<ArticleEntry>& entries_specifics = service_->GetEntries();
106 for (std::vector<ArticleEntry>::const_iterator it = entries_specifics.begin();
107 it != entries_specifics.end();
108 ++it) {
109 const ArticleEntry& article = *it;
110 DCHECK(IsEntryValid(article));
111 scoped_ptr<base::DictionaryValue> entry(new base::DictionaryValue());
112 entry->SetString("entry_id", article.entry_id());
113 std::string title = (!article.has_title() || article.title().empty())
114 ? article.entry_id()
115 : article.title();
116 entry->SetString("title", net::EscapeForHTML(title));
117 entries.Append(entry.release());
118 }
119 // TODO(nyquist): Write a test that ensures we sanitize the data we send.
120 web_ui()->CallJavascriptFunction("domDistiller.onReceivedEntries", entries);
121 }
122
OnArticleAdded(bool article_available)123 void DomDistillerHandler::OnArticleAdded(bool article_available) {
124 // TODO(nyquist): Update this function.
125 if (article_available) {
126 HandleRequestEntries(NULL);
127 } else {
128 web_ui()->CallJavascriptFunction("domDistiller.onArticleAddFailed");
129 }
130 }
131
132 } // namespace dom_distiller
133