ui/web_dialogs/web_dialog_ui.cc

   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:
  29   explicit WebDialogDelegateUserData(WebDialogDelegate* delegate)
  30       : delegate_(delegate) {}
  31   ~WebDialogDelegateUserData() override {}
  32   WebDialogDelegate* delegate() { return delegate_; }
  33
  34  private:
  35   WebDialogDelegate* delegate_;  // unowned
  36 };
  37
  38 }  // namespace
  39
  40 WebDialogUI::WebDialogUI(content::WebUI* web_ui)
  41     : WebUIController(web_ui) {
  42 }
  43
  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
  55 void WebDialogUI::CloseDialog(const base::ListValue* args) {
  56   OnDialogClosed(args);
  57 }
  58
  59 // static
  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
  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
  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
 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
 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
 125 ExternalWebDialogUI::~ExternalWebDialogUI() {
 126 }
 127
 128 }  // namespace ui