| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ | 5 #ifndef EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ |
| 6 #define EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ | 6 #define EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ |
| 7 | 7 |
| 8 #include <set> |
| 8 #include <string> | 9 #include <string> |
| 9 | 10 |
| 10 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
| 11 #include "base/macros.h" | 12 #include "base/macros.h" |
| 12 #include "content/public/browser/web_contents_observer.h" | 13 #include "content/public/browser/web_contents_observer.h" |
| 13 | 14 |
| 14 namespace content { | 15 namespace content { |
| 15 class BrowserContext; | 16 class BrowserContext; |
| 17 class RenderFrameHost; |
| 16 class RenderViewHost; | 18 class RenderViewHost; |
| 17 class WebContents; | 19 class WebContents; |
| 18 } | 20 } |
| 19 | 21 |
| 20 namespace extensions { | 22 namespace extensions { |
| 21 class Extension; | 23 class Extension; |
| 22 | 24 |
| 23 // A web contents observer used for renderer and extension processes. Grants the | 25 // A web contents observer used for renderer and extension processes. Grants the |
| 24 // renderer access to certain URL scheme patterns for extensions and notifies | 26 // renderer access to certain URL scheme patterns for extensions and notifies |
| 25 // the renderer that the extension was loaded. | 27 // the renderer that the extension was loaded. |
| 26 // | 28 // |
| 27 // Extension system embedders must create an instance for every extension | 29 // Extension system embedders must create an instance for every extension |
| 28 // WebContents. It must be a subclass so that creating an instance via | 30 // WebContents. It must be a subclass so that creating an instance via |
| 29 // content::WebContentsUserData::CreateForWebContents() provides an object of | 31 // content::WebContentsUserData::CreateForWebContents() provides an object of |
| 30 // the correct type. For an example, see ChromeExtensionWebContentsObserver. | 32 // the correct type. For an example, see ChromeExtensionWebContentsObserver. |
| 31 class ExtensionWebContentsObserver : public content::WebContentsObserver { | 33 class ExtensionWebContentsObserver : public content::WebContentsObserver { |
| 32 protected: | 34 protected: |
| 33 explicit ExtensionWebContentsObserver(content::WebContents* web_contents); | 35 explicit ExtensionWebContentsObserver(content::WebContents* web_contents); |
| 34 virtual ~ExtensionWebContentsObserver(); | 36 virtual ~ExtensionWebContentsObserver(); |
| 35 | 37 |
| 36 content::BrowserContext* browser_context() { return browser_context_; } | 38 content::BrowserContext* browser_context() { return browser_context_; } |
| 37 | 39 |
| 38 // content::WebContentsObserver overrides. | 40 // content::WebContentsObserver overrides. |
| 39 | 41 |
| 42 virtual void RenderFrameCreated( |
| 43 content::RenderFrameHost* render_frame_host) override; |
| 44 |
| 45 virtual void RenderFrameDeleted( |
| 46 content::RenderFrameHost* render_frame_host) override; |
| 47 |
| 40 // A subclass should invoke this method to finish extension process setup. | 48 // A subclass should invoke this method to finish extension process setup. |
| 41 virtual void RenderViewCreated(content::RenderViewHost* render_view_host) | 49 virtual void RenderViewCreated(content::RenderViewHost* render_view_host) |
| 42 override; | 50 override; |
| 43 | 51 |
| 44 // Returns the extension or app associated with a render view host. Returns | 52 // Returns the extension or app associated with a render view host. Returns |
| 45 // NULL if the render view host is not for a valid extension. | 53 // NULL if the render view host is not for a valid extension. |
| 46 const Extension* GetExtension(content::RenderViewHost* render_view_host); | 54 const Extension* GetExtension(content::RenderViewHost* render_view_host); |
| 47 | 55 |
| 48 // Updates ViewType for RenderViewHost based on GetViewType(web_contents()). | 56 // Updates ViewType for RenderViewHost based on GetViewType(web_contents()). |
| 49 void NotifyRenderViewType(content::RenderViewHost* render_view_host); | 57 void NotifyRenderViewType(content::RenderViewHost* render_view_host); |
| 50 | 58 |
| 51 // Returns the extension or app ID associated with a render view host. Returns | 59 // Returns the extension or app ID associated with a render view host. Returns |
| 52 // the empty string if the render view host is not for a valid extension. | 60 // the empty string if the render view host is not for a valid extension. |
| 53 static std::string GetExtensionId(content::RenderViewHost* render_view_host); | 61 static std::string GetExtensionId(content::RenderViewHost* render_view_host); |
| 54 | 62 |
| 55 private: | 63 private: |
| 56 // The BrowserContext associated with the WebContents being observed. | 64 // The BrowserContext associated with the WebContents being observed. |
| 57 content::BrowserContext* browser_context_; | 65 content::BrowserContext* browser_context_; |
| 58 | 66 |
| 67 // The set of RenderFrameHosts that we have seen be created, but not deleted. |
| 68 std::set<content::RenderFrameHost*> known_render_frame_hosts_; |
| 69 |
| 59 DISALLOW_COPY_AND_ASSIGN(ExtensionWebContentsObserver); | 70 DISALLOW_COPY_AND_ASSIGN(ExtensionWebContentsObserver); |
| 60 }; | 71 }; |
| 61 | 72 |
| 62 } // namespace extensions | 73 } // namespace extensions |
| 63 | 74 |
| 64 #endif // EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ | 75 #endif // EXTENSIONS_BROWSER_EXTENSION_WEB_CONTENTS_OBSERVER_H_ |
| OLD | NEW |