OLD | NEW |
| (Empty) |
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 "chrome/browser/ui/app_modal_dialogs/javascript_dialog_creator.h" | |
6 | |
7 #include <map> | |
8 | |
9 #include "base/compiler_specific.h" | |
10 #include "base/i18n/rtl.h" | |
11 #include "base/memory/singleton.h" | |
12 #include "base/utf_string_conversions.h" | |
13 #include "chrome/browser/extensions/extension_host.h" | |
14 #include "chrome/browser/extensions/extension_service.h" | |
15 #include "chrome/browser/ui/app_modal_dialogs/app_modal_dialog_queue.h" | |
16 #include "chrome/browser/ui/app_modal_dialogs/javascript_app_modal_dialog.h" | |
17 #include "chrome/common/chrome_constants.h" | |
18 #include "chrome/common/chrome_notification_types.h" | |
19 #include "content/public/browser/notification_observer.h" | |
20 #include "content/public/browser/notification_registrar.h" | |
21 #include "content/public/browser/notification_service.h" | |
22 #include "content/public/common/content_client.h" | |
23 #include "content/public/common/javascript_message_type.h" | |
24 #include "grit/generated_resources.h" | |
25 #include "net/base/net_util.h" | |
26 #include "ui/base/l10n/l10n_util.h" | |
27 | |
28 using content::JavaScriptDialogCreator; | |
29 using content::WebContents; | |
30 | |
31 namespace { | |
32 | |
33 class ChromeJavaScriptDialogCreator : public JavaScriptDialogCreator, | |
34 public content::NotificationObserver { | |
35 public: | |
36 static ChromeJavaScriptDialogCreator* GetInstance(); | |
37 | |
38 explicit ChromeJavaScriptDialogCreator( | |
39 extensions::ExtensionHost* extension_host); | |
40 virtual ~ChromeJavaScriptDialogCreator(); | |
41 | |
42 virtual void RunJavaScriptDialog( | |
43 WebContents* web_contents, | |
44 const GURL& origin_url, | |
45 const std::string& accept_lang, | |
46 content::JavaScriptMessageType message_type, | |
47 const string16& message_text, | |
48 const string16& default_prompt_text, | |
49 const DialogClosedCallback& callback, | |
50 bool* did_suppress_message) OVERRIDE; | |
51 | |
52 virtual void RunBeforeUnloadDialog( | |
53 WebContents* web_contents, | |
54 const string16& message_text, | |
55 bool is_reload, | |
56 const DialogClosedCallback& callback) OVERRIDE; | |
57 | |
58 virtual void ResetJavaScriptState(WebContents* web_contents) OVERRIDE; | |
59 | |
60 private: | |
61 ChromeJavaScriptDialogCreator(); | |
62 | |
63 friend struct DefaultSingletonTraits<ChromeJavaScriptDialogCreator>; | |
64 | |
65 // Overridden from content::NotificationObserver: | |
66 virtual void Observe(int type, | |
67 const content::NotificationSource& source, | |
68 const content::NotificationDetails& details) OVERRIDE; | |
69 | |
70 string16 GetTitle(const GURL& origin_url, | |
71 const std::string& accept_lang, | |
72 bool is_alert); | |
73 | |
74 void CancelPendingDialogs(WebContents* web_contents); | |
75 | |
76 // Wrapper around a DialogClosedCallback so that we can intercept it before | |
77 // passing it onto the original callback. | |
78 void OnDialogClosed(DialogClosedCallback callback, | |
79 bool success, | |
80 const string16& user_input); | |
81 | |
82 // Mapping between the WebContents and their extra data. The key | |
83 // is a void* because the pointer is just a cookie and is never dereferenced. | |
84 typedef std::map<void*, ChromeJavaScriptDialogExtraData> | |
85 JavaScriptDialogExtraDataMap; | |
86 JavaScriptDialogExtraDataMap javascript_dialog_extra_data_; | |
87 | |
88 // Extension Host which owns the ChromeJavaScriptDialogCreator instance. | |
89 // It's used to get a extension name from a URL. | |
90 // If it's not owned by any Extension, it should be NULL. | |
91 extensions::ExtensionHost* extension_host_; | |
92 | |
93 content::NotificationRegistrar registrar_; | |
94 | |
95 DISALLOW_COPY_AND_ASSIGN(ChromeJavaScriptDialogCreator); | |
96 }; | |
97 | |
98 //////////////////////////////////////////////////////////////////////////////// | |
99 // ChromeJavaScriptDialogCreator, public: | |
100 | |
101 ChromeJavaScriptDialogCreator::ChromeJavaScriptDialogCreator() | |
102 : extension_host_(NULL) { | |
103 } | |
104 | |
105 ChromeJavaScriptDialogCreator::~ChromeJavaScriptDialogCreator() { | |
106 extension_host_ = NULL; | |
107 } | |
108 | |
109 ChromeJavaScriptDialogCreator::ChromeJavaScriptDialogCreator( | |
110 extensions::ExtensionHost* extension_host) | |
111 : extension_host_(extension_host) { | |
112 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED, | |
113 content::Source<Profile>(extension_host_->profile())); | |
114 } | |
115 | |
116 // static | |
117 ChromeJavaScriptDialogCreator* ChromeJavaScriptDialogCreator::GetInstance() { | |
118 return Singleton<ChromeJavaScriptDialogCreator>::get(); | |
119 } | |
120 | |
121 void ChromeJavaScriptDialogCreator::RunJavaScriptDialog( | |
122 WebContents* web_contents, | |
123 const GURL& origin_url, | |
124 const std::string& accept_lang, | |
125 content::JavaScriptMessageType message_type, | |
126 const string16& message_text, | |
127 const string16& default_prompt_text, | |
128 const DialogClosedCallback& callback, | |
129 bool* did_suppress_message) { | |
130 *did_suppress_message = false; | |
131 | |
132 ChromeJavaScriptDialogExtraData* extra_data = | |
133 &javascript_dialog_extra_data_[web_contents]; | |
134 | |
135 if (extra_data->suppress_javascript_messages_) { | |
136 *did_suppress_message = true; | |
137 return; | |
138 } | |
139 | |
140 base::TimeDelta time_since_last_message = base::TimeTicks::Now() - | |
141 extra_data->last_javascript_message_dismissal_; | |
142 bool display_suppress_checkbox = false; | |
143 // Show a checkbox offering to suppress further messages if this message is | |
144 // being displayed within kJavascriptMessageExpectedDelay of the last one. | |
145 if (time_since_last_message < | |
146 base::TimeDelta::FromMilliseconds( | |
147 chrome::kJavascriptMessageExpectedDelay)) { | |
148 display_suppress_checkbox = true; | |
149 } | |
150 | |
151 bool is_alert = message_type == content::JAVASCRIPT_MESSAGE_TYPE_ALERT; | |
152 string16 dialog_title = GetTitle(origin_url, accept_lang, is_alert); | |
153 | |
154 if (extension_host_) | |
155 extension_host_->WillRunJavaScriptDialog(); | |
156 | |
157 AppModalDialogQueue::GetInstance()->AddDialog(new JavaScriptAppModalDialog( | |
158 web_contents, | |
159 extra_data, | |
160 dialog_title, | |
161 message_type, | |
162 message_text, | |
163 default_prompt_text, | |
164 display_suppress_checkbox, | |
165 false, // is_before_unload_dialog | |
166 false, // is_reload | |
167 base::Bind(&ChromeJavaScriptDialogCreator::OnDialogClosed, | |
168 base::Unretained(this), callback))); | |
169 } | |
170 | |
171 void ChromeJavaScriptDialogCreator::RunBeforeUnloadDialog( | |
172 WebContents* web_contents, | |
173 const string16& message_text, | |
174 bool is_reload, | |
175 const DialogClosedCallback& callback) { | |
176 ChromeJavaScriptDialogExtraData* extra_data = | |
177 &javascript_dialog_extra_data_[web_contents]; | |
178 | |
179 const string16 title = l10n_util::GetStringUTF16(is_reload ? | |
180 IDS_BEFORERELOAD_MESSAGEBOX_TITLE : IDS_BEFOREUNLOAD_MESSAGEBOX_TITLE); | |
181 const string16 footer = l10n_util::GetStringUTF16(is_reload ? | |
182 IDS_BEFORERELOAD_MESSAGEBOX_FOOTER : IDS_BEFOREUNLOAD_MESSAGEBOX_FOOTER); | |
183 | |
184 string16 full_message = message_text + ASCIIToUTF16("\n\n") + footer; | |
185 | |
186 if (extension_host_) | |
187 extension_host_->WillRunJavaScriptDialog(); | |
188 | |
189 AppModalDialogQueue::GetInstance()->AddDialog(new JavaScriptAppModalDialog( | |
190 web_contents, | |
191 extra_data, | |
192 title, | |
193 content::JAVASCRIPT_MESSAGE_TYPE_CONFIRM, | |
194 full_message, | |
195 string16(), // default_prompt_text | |
196 false, // display_suppress_checkbox | |
197 true, // is_before_unload_dialog | |
198 is_reload, | |
199 base::Bind(&ChromeJavaScriptDialogCreator::OnDialogClosed, | |
200 base::Unretained(this), callback))); | |
201 } | |
202 | |
203 void ChromeJavaScriptDialogCreator::ResetJavaScriptState( | |
204 WebContents* web_contents) { | |
205 CancelPendingDialogs(web_contents); | |
206 javascript_dialog_extra_data_.erase(web_contents); | |
207 } | |
208 | |
209 void ChromeJavaScriptDialogCreator::Observe( | |
210 int type, | |
211 const content::NotificationSource& source, | |
212 const content::NotificationDetails& details) { | |
213 DCHECK_EQ(type, chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED); | |
214 extension_host_ = NULL; | |
215 } | |
216 | |
217 string16 ChromeJavaScriptDialogCreator::GetTitle(const GURL& origin_url, | |
218 const std::string& accept_lang, | |
219 bool is_alert) { | |
220 // If the URL hasn't any host, return the default string. | |
221 if (!origin_url.has_host()) { | |
222 return l10n_util::GetStringUTF16( | |
223 is_alert ? IDS_JAVASCRIPT_ALERT_DEFAULT_TITLE | |
224 : IDS_JAVASCRIPT_MESSAGEBOX_DEFAULT_TITLE); | |
225 } | |
226 | |
227 // If the URL is a chrome extension one, return the extension name. | |
228 if (extension_host_) { | |
229 const extensions::Extension* extension = extension_host_-> | |
230 profile()->GetExtensionService()->extensions()-> | |
231 GetExtensionOrAppByURL(ExtensionURLInfo(origin_url)); | |
232 if (extension) { | |
233 return UTF8ToUTF16(base::StringPiece(extension->name())); | |
234 } | |
235 } | |
236 | |
237 // Otherwise, return the formatted URL. | |
238 // In this case, force URL to have LTR directionality. | |
239 string16 url_string = net::FormatUrl(origin_url, accept_lang); | |
240 return l10n_util::GetStringFUTF16( | |
241 is_alert ? IDS_JAVASCRIPT_ALERT_TITLE | |
242 : IDS_JAVASCRIPT_MESSAGEBOX_TITLE, | |
243 base::i18n::GetDisplayStringInLTRDirectionality(url_string)); | |
244 } | |
245 | |
246 void ChromeJavaScriptDialogCreator::CancelPendingDialogs( | |
247 WebContents* web_contents) { | |
248 AppModalDialogQueue* queue = AppModalDialogQueue::GetInstance(); | |
249 AppModalDialog* active_dialog = queue->active_dialog(); | |
250 if (active_dialog && active_dialog->web_contents() == web_contents) | |
251 active_dialog->Invalidate(); | |
252 for (AppModalDialogQueue::iterator i = queue->begin(); | |
253 i != queue->end(); ++i) { | |
254 if ((*i)->web_contents() == web_contents) | |
255 (*i)->Invalidate(); | |
256 } | |
257 } | |
258 | |
259 void ChromeJavaScriptDialogCreator::OnDialogClosed( | |
260 DialogClosedCallback callback, | |
261 bool success, | |
262 const string16& user_input) { | |
263 if (extension_host_) | |
264 extension_host_->DidCloseJavaScriptDialog(); | |
265 callback.Run(success, user_input); | |
266 } | |
267 | |
268 } // namespace | |
269 | |
270 content::JavaScriptDialogCreator* GetJavaScriptDialogCreatorInstance() { | |
271 return ChromeJavaScriptDialogCreator::GetInstance(); | |
272 } | |
273 | |
274 content::JavaScriptDialogCreator* CreateJavaScriptDialogCreatorInstance( | |
275 extensions::ExtensionHost* extension_host) { | |
276 return new ChromeJavaScriptDialogCreator(extension_host); | |
277 } | |
OLD | NEW |