Add GetMessageHandlers to HtmlDialogUIDelegate to allow...

chrome/browser/dom_ui/dom_ui.h

 // 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.
 
 #ifndef CHROME_BROWSER_DOM_UI_DOM_UI_H_
 #define CHROME_BROWSER_DOM_UI_DOM_UI_H_
 
 #include <map>
 #include <string>
 #include <vector>
@@ skipping 111 matching lines @@
   MessageCallbackMap message_callbacks_;
 
   DISALLOW_COPY_AND_ASSIGN(DOMUI);
 };
 
 // Messages sent from the DOM are forwarded via the DOMUI to handler
 // classes. These objects are owned by DOMUI and destroyed when the
 // host is destroyed.
 class DOMMessageHandler {
  public:
-  explicit DOMMessageHandler(DOMUI* dom_ui);
+  DOMMessageHandler() : dom_ui_(NULL) { }
   virtual ~DOMMessageHandler() {};
 
+  // Attaches |this| to |dom_ui| in order to handle messages from it.  Declared
+  // virtual so that subclasses can do special init work as soon as the dom_ui
+  // is provided.  Returns |this| for convenience.
+  virtual DOMMessageHandler* Attach(DOMUI* dom_ui);
  protected:
   // Adds "url" and "title" keys on incoming dictionary, setting title
   // as the url as a fallback on empty title.
   static void SetURLAndTitle(DictionaryValue* dictionary,
                              std::wstring title,
                              const GURL& gurl);
 
+  // This is where subclasses specify which messages they'd like to handle.
+  virtual void RegisterMessages() = 0;
+
   // Extract an integer value from a Value.
   bool ExtractIntegerValue(const Value* value, int* out_int);
 
   // Extract a string value from a Value.
   std::wstring ExtractStringValue(const Value* value);
 
-  DOMUI* const dom_ui_;
+  DOMUI* dom_ui_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(DOMMessageHandler);
 };
 
 #endif  // CHROME_BROWSER_DOM_UI_DOM_UI_H_