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 "ui/web_dialogs/web_dialog_ui.h"
6
7 #include "base/bind.h"
8 #include "base/bind_helpers.h"
9 #include "base/lazy_instance.h"
10 #include "base/values.h"
11 #include "content/public/browser/render_view_host.h"
12 #include "content/public/browser/web_contents.h"
13 #include "content/public/browser/web_ui.h"
14 #include "content/public/browser/web_ui_message_handler.h"
15 #include "content/public/common/bindings_policy.h"
16 #include "ui/web_dialogs/web_dialog_delegate.h"
17
18 using content::RenderViewHost;
19 using content::WebUIMessageHandler;
20
21 namespace ui {
22
23 namespace {
24
25 const char kWebDialogDelegateUserDataKey[] = "WebDialogDelegateUserData";
26
27 class WebDialogDelegateUserData : public base::SupportsUserData::Data {
28 public:
WebDialogDelegateUserData(WebDialogDelegate * delegate)29 explicit WebDialogDelegateUserData(WebDialogDelegate* delegate)
30 : delegate_(delegate) {}
~WebDialogDelegateUserData()31 virtual ~WebDialogDelegateUserData() {}
delegate()32 WebDialogDelegate* delegate() { return delegate_; }
33
34 private:
35 WebDialogDelegate* delegate_; // unowned
36 };
37
38 } // namespace
39
WebDialogUI(content::WebUI * web_ui)40 WebDialogUI::WebDialogUI(content::WebUI* web_ui)
41 : WebUIController(web_ui) {
42 }
43
~WebDialogUI()44 WebDialogUI::~WebDialogUI() {
45 // Don't unregister our user data. During the teardown of the WebContents,
46 // this will be deleted, but the WebContents will already be destroyed.
47 //
48 // This object is owned indirectly by the WebContents. WebUIs can change, so
49 // it's scary if this WebUI is changed out and replaced with something else,
50 // since the user data will still point to the old delegate. But the delegate
51 // is itself the owner of the WebContents for a dialog so will be in scope,
52 // and the HTML dialogs won't swap WebUIs anyway since they don't navigate.
53 }
54
CloseDialog(const base::ListValue * args)55 void WebDialogUI::CloseDialog(const base::ListValue* args) {
56 OnDialogClosed(args);
57 }
58
59 // static
SetDelegate(content::WebContents * web_contents,WebDialogDelegate * delegate)60 void WebDialogUI::SetDelegate(content::WebContents* web_contents,
61 WebDialogDelegate* delegate) {
62 web_contents->SetUserData(&kWebDialogDelegateUserDataKey,
63 new WebDialogDelegateUserData(delegate));
64 }
65
66 ////////////////////////////////////////////////////////////////////////////////
67 // Private:
68
GetDelegate(content::WebContents * web_contents)69 WebDialogDelegate* WebDialogUI::GetDelegate(
70 content::WebContents* web_contents) {
71 WebDialogDelegateUserData* user_data =
72 static_cast<WebDialogDelegateUserData*>(
73 web_contents->GetUserData(&kWebDialogDelegateUserDataKey));
74
75 return user_data ? user_data->delegate() : NULL;
76 }
77
78
RenderViewCreated(RenderViewHost * render_view_host)79 void WebDialogUI::RenderViewCreated(RenderViewHost* render_view_host) {
80 // Hook up the javascript function calls, also known as chrome.send("foo")
81 // calls in the HTML, to the actual C++ functions.
82 web_ui()->RegisterMessageCallback("dialogClose",
83 base::Bind(&WebDialogUI::OnDialogClosed, base::Unretained(this)));
84
85 // Pass the arguments to the renderer supplied by the delegate.
86 std::string dialog_args;
87 std::vector<WebUIMessageHandler*> handlers;
88 WebDialogDelegate* delegate = GetDelegate(web_ui()->GetWebContents());
89 if (delegate) {
90 dialog_args = delegate->GetDialogArgs();
91 delegate->GetWebUIMessageHandlers(&handlers);
92 }
93
94 if (0 != (web_ui()->GetBindings() & content::BINDINGS_POLICY_WEB_UI))
95 render_view_host->SetWebUIProperty("dialogArguments", dialog_args);
96 for (std::vector<WebUIMessageHandler*>::iterator it = handlers.begin();
97 it != handlers.end(); ++it) {
98 web_ui()->AddMessageHandler(*it);
99 }
100
101 if (delegate)
102 delegate->OnDialogShown(web_ui(), render_view_host);
103 }
104
OnDialogClosed(const base::ListValue * args)105 void WebDialogUI::OnDialogClosed(const base::ListValue* args) {
106 WebDialogDelegate* delegate = GetDelegate(web_ui()->GetWebContents());
107 if (delegate) {
108 std::string json_retval;
109 if (args && !args->empty() && !args->GetString(0, &json_retval))
110 NOTREACHED() << "Could not read JSON argument";
111
112 delegate->OnDialogCloseFromWebUI(json_retval);
113 }
114 }
115
ExternalWebDialogUI(content::WebUI * web_ui)116 ExternalWebDialogUI::ExternalWebDialogUI(content::WebUI* web_ui)
117 : WebDialogUI(web_ui) {
118 // Non-file based UI needs to not have access to the Web UI bindings
119 // for security reasons. The code hosting the dialog should provide
120 // dialog specific functionality through other bindings and methods
121 // that are scoped in duration to the dialogs existence.
122 web_ui->SetBindings(web_ui->GetBindings() & ~content::BINDINGS_POLICY_WEB_UI);
123 }
124
~ExternalWebDialogUI()125 ExternalWebDialogUI::~ExternalWebDialogUI() {
126 }
127
128 } // namespace ui
129