1 // Copyright (c) 2012 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 "content/public/browser/web_ui_message_handler.h"
6
7 #include "base/logging.h"
8 #include "base/strings/string_number_conversions.h"
9 #include "base/strings/utf_string_conversions.h"
10 #include "base/values.h"
11
12 namespace content {
13
AllowJavascriptForTesting()14 void WebUIMessageHandler::AllowJavascriptForTesting() {
15 AllowJavascript();
16 }
17
AllowJavascript()18 void WebUIMessageHandler::AllowJavascript() {
19 if (javascript_allowed_)
20 return;
21
22 javascript_allowed_ = true;
23 CHECK(IsJavascriptAllowed());
24
25 OnJavascriptAllowed();
26 }
27
DisallowJavascript()28 void WebUIMessageHandler::DisallowJavascript() {
29 if (!javascript_allowed_)
30 return;
31
32 javascript_allowed_ = false;
33 DCHECK(!IsJavascriptAllowed());
34
35 OnJavascriptDisallowed();
36 }
37
IsJavascriptAllowed() const38 bool WebUIMessageHandler::IsJavascriptAllowed() const {
39 return javascript_allowed_ && web_ui() && web_ui()->CanCallJavascript();
40 }
41
ExtractIntegerValue(const base::ListValue * value,int * out_int)42 bool WebUIMessageHandler::ExtractIntegerValue(const base::ListValue* value,
43 int* out_int) {
44 std::string string_value;
45 if (value->GetString(0, &string_value))
46 return base::StringToInt(string_value, out_int);
47 double double_value;
48 if (value->GetDouble(0, &double_value)) {
49 *out_int = static_cast<int>(double_value);
50 return true;
51 }
52 NOTREACHED();
53 return false;
54 }
55
ExtractDoubleValue(const base::ListValue * value,double * out_value)56 bool WebUIMessageHandler::ExtractDoubleValue(const base::ListValue* value,
57 double* out_value) {
58 std::string string_value;
59 if (value->GetString(0, &string_value))
60 return base::StringToDouble(string_value, out_value);
61 if (value->GetDouble(0, out_value))
62 return true;
63 NOTREACHED();
64 return false;
65 }
66
ExtractStringValue(const base::ListValue * value)67 base::string16 WebUIMessageHandler::ExtractStringValue(
68 const base::ListValue* value) {
69 base::string16 string16_value;
70 if (value->GetString(0, &string16_value))
71 return string16_value;
72 NOTREACHED();
73 return base::string16();
74 }
75
ResolveJavascriptCallback(const base::Value & callback_id,const base::Value & response)76 void WebUIMessageHandler::ResolveJavascriptCallback(
77 const base::Value& callback_id,
78 const base::Value& response) {
79 // cr.webUIResponse is a global JS function exposed from cr.js.
80 CallJavascriptFunction("cr.webUIResponse", callback_id, base::Value(true),
81 response);
82 }
83
RejectJavascriptCallback(const base::Value & callback_id,const base::Value & response)84 void WebUIMessageHandler::RejectJavascriptCallback(
85 const base::Value& callback_id,
86 const base::Value& response) {
87 // cr.webUIResponse is a global JS function exposed from cr.js.
88 CallJavascriptFunction("cr.webUIResponse", callback_id, base::Value(false),
89 response);
90 }
91
92 } // namespace content
93