Move dispatching and sending of the last extension specific messages out of TabContents and Rende...

chrome/browser/notifications/balloon_host.h

 // Copyright (c) 2011 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_NOTIFICATIONS_BALLOON_HOST_H_
 #define CHROME_BROWSER_NOTIFICATIONS_BALLOON_HOST_H_
 #pragma once
 
 #include <string>
 #include <vector>
 
 #include "base/scoped_ptr.h"
 #include "chrome/browser/extensions/extension_function_dispatcher.h"
 #include "chrome/browser/tab_contents/render_view_host_delegate_helper.h"
 #include "content/browser/renderer_host/render_view_host_delegate.h"
 #include "content/common/notification_observer.h"
 #include "content/common/notification_registrar.h"
 
 class Balloon;
 class Browser;
+class ExtensionMessageHandler;
 class Profile;
 class SiteInstance;
 struct RendererPreferences;
 struct WebPreferences;
 
 class BalloonHost : public RenderViewHostDelegate,
                     public RenderViewHostDelegate::View,
                     public ExtensionFunctionDispatcher::Delegate,
                     public NotificationObserver {
  public:
   explicit BalloonHost(Balloon* balloon);
 
   // Initialize the view.
   void Init();
 
   // Stops showing the balloon.
   void Shutdown();
 
   // ExtensionFunctionDispatcher::Delegate overrides.
   virtual Browser* GetBrowser();
   virtual gfx::NativeView GetNativeViewOfHost();
   virtual TabContents* associated_tab_contents() const;
 
   RenderViewHost* render_view_host() const { return render_view_host_; }
 
   const string16& GetSource() const;
 
   // RenderViewHostDelegate overrides.
+  virtual bool OnMessageReceived(const IPC::Message& message);
   virtual WebPreferences GetWebkitPrefs();
   virtual SiteInstance* GetSiteInstance() const;
   virtual Profile* GetProfile() const;
   virtual const GURL& GetURL() const;
   virtual void Close(RenderViewHost* render_view_host);
   virtual void RenderViewCreated(RenderViewHost* render_view_host);
   virtual void RenderViewReady(RenderViewHost* render_view_host);
   virtual void RenderViewGone(RenderViewHost* render_view_host,
                               base::TerminationStatus status,
                               int error_code);
   virtual void UpdateTitle(RenderViewHost* render_view_host,
                            int32 page_id, const std::wstring& title) {}
   virtual int GetBrowserWindowID() const;
   virtual ViewType::Type GetRenderViewType() const;
   virtual RenderViewHostDelegate::View* GetViewDelegate();
-  virtual void ProcessWebUIMessage(
-      const ExtensionHostMsg_DomMessage_Params& params);
 
   // NotificationObserver override.
   virtual void Observe(NotificationType type,
                        const NotificationSource& source,
                        const NotificationDetails& details);
 
 
   // RenderViewHostDelegate::View methods. Only the ones for opening new
   // windows are currently implemented.
   virtual void CreateNewWindow(
@@ skipping 77 matching lines @@
   RenderViewHostDelegateViewHelper delegate_view_helper_;
 
   // Handles requests to extension APIs. Will only be non-NULL if we are
   // rendering a page from an extension.
   scoped_ptr<ExtensionFunctionDispatcher> extension_function_dispatcher_;
 
   // A flag to enable Web UI.
   bool enable_web_ui_;
 
   NotificationRegistrar registrar_;
+
+  // Handles extension IPCs.
+  scoped_ptr<ExtensionMessageHandler> extension_message_handler_;
 };
 
 #endif  // CHROME_BROWSER_NOTIFICATIONS_BALLOON_HOST_H_