Add GetMessageHandlers to HtmlDialogUIDelegate to allow...

chrome/browser/dom_ui/html_dialog_ui.cc

 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/dom_ui/html_dialog_ui.h"
 
 #include "base/singleton.h"
 #include "base/values.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
@@ skipping 43 matching lines @@
 }
 
 void HtmlDialogUI::RenderViewCreated(RenderViewHost* render_view_host) {
   // Hook up the javascript function calls, also known as chrome.send("foo")
   // calls in the HTML, to the actual C++ functions.
   RegisterMessageCallback("DialogClose",
                           NewCallback(this, &HtmlDialogUI::OnDialogClosed));
 
   // Pass the arguments to the renderer supplied by the delegate.
   std::string dialog_args;
+  std::vector<DOMMessageHandler*> handlers;
   HtmlDialogUIDelegate** delegate = GetPropertyAccessor().GetProperty(
       tab_contents()->property_bag());
-  if (delegate)
+  if (delegate) {
     dialog_args = (*delegate)->GetDialogArgs();
+    (*delegate)->GetDOMMessageHandlers(&handlers);
+  }
+
   render_view_host->SetDOMUIProperty("dialogArguments", dialog_args);
+  for (std::vector<DOMMessageHandler*>::iterator it = handlers.begin();
+       it != handlers.end(); ++it) {
+    (*it)->Attach(this);
+    AddMessageHandler(*it);
+  }
 }
 
 void HtmlDialogUI::OnDialogClosed(const Value* content) {
   HtmlDialogUIDelegate** delegate = GetPropertyAccessor().GetProperty(
       tab_contents()->property_bag());
   if (delegate)
     (*delegate)->OnDialogClosed(GetJsonResponse(content));
 }