| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "chrome/browser/prerender/prerender_render_view_host_observer.h" | 5 #include "chrome/browser/prerender/prerender_render_view_host_observer.h" |
| 6 | 6 |
| 7 #include "chrome/browser/prerender/prerender_contents.h" | 7 #include "chrome/browser/prerender/prerender_contents.h" |
| 8 #include "chrome/common/icon_messages.h" | 8 #include "chrome/common/icon_messages.h" |
| 9 #include "chrome/common/render_messages.h" | 9 #include "chrome/common/render_messages.h" |
| 10 | 10 |
| 11 namespace prerender { | 11 namespace prerender { |
| 12 | 12 |
| 13 PrerenderRenderViewHostObserver::PrerenderRenderViewHostObserver( | 13 PrerenderRenderViewHostObserver::PrerenderRenderViewHostObserver( |
| 14 PrerenderContents* prerender_contents, | 14 PrerenderContents* prerender_contents, |
| 15 RenderViewHost* render_view_host) | 15 RenderViewHost* render_view_host) |
| 16 : RenderViewHostObserver(render_view_host), | 16 : content::RenderViewHostObserver(render_view_host), |
| 17 prerender_contents_(prerender_contents) { | 17 prerender_contents_(prerender_contents) { |
| 18 } | 18 } |
| 19 | 19 |
| 20 void PrerenderRenderViewHostObserver::RenderViewHostDestroyed( | 20 void PrerenderRenderViewHostObserver::RenderViewHostDestroyed( |
| 21 RenderViewHost* rvh) { | 21 RenderViewHost* rvh) { |
| 22 // The base class deletes |this| on RenderViewHost destruction but we want the | 22 // The base class deletes |this| on RenderViewHost destruction but we want the |
| 23 // lifetime to be tied to the PrerenderContents instead, so we'll do nothing | 23 // lifetime to be tied to the PrerenderContents instead, so we'll do nothing |
| 24 // here. | 24 // here. |
| 25 } | 25 } |
| 26 | 26 |
| 27 bool PrerenderRenderViewHostObserver::OnMessageReceived( | 27 bool PrerenderRenderViewHostObserver::OnMessageReceived( |
| 28 const IPC::Message& message) { | 28 const IPC::Message& message) { |
| 29 if (!prerender_contents_) | 29 if (!prerender_contents_) |
| 30 return RenderViewHostObserver::OnMessageReceived(message); | 30 return content::RenderViewHostObserver::OnMessageReceived(message); |
| 31 | 31 |
| 32 bool handled = true; | 32 bool handled = true; |
| 33 // The following messages we do want to consume. | 33 // The following messages we do want to consume. |
| 34 IPC_BEGIN_MESSAGE_MAP(PrerenderRenderViewHostObserver, message) | 34 IPC_BEGIN_MESSAGE_MAP(PrerenderRenderViewHostObserver, message) |
| 35 IPC_MESSAGE_HANDLER(IconHostMsg_UpdateFaviconURL, OnUpdateFaviconURL) | 35 IPC_MESSAGE_HANDLER(IconHostMsg_UpdateFaviconURL, OnUpdateFaviconURL) |
| 36 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_MaybeCancelPrerenderForHTML5Media, | 36 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_MaybeCancelPrerenderForHTML5Media, |
| 37 OnMaybeCancelPrerenderForHTML5Media) | 37 OnMaybeCancelPrerenderForHTML5Media) |
| 38 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_CancelPrerenderForPrinting, | 38 IPC_MESSAGE_HANDLER(ChromeViewHostMsg_CancelPrerenderForPrinting, |
| 39 OnCancelPrerenderForPrinting) | 39 OnCancelPrerenderForPrinting) |
| 40 IPC_MESSAGE_UNHANDLED(handled = false) | 40 IPC_MESSAGE_UNHANDLED(handled = false) |
| 41 IPC_END_MESSAGE_MAP() | 41 IPC_END_MESSAGE_MAP() |
| 42 | 42 |
| 43 // Pass the message through. | 43 // Pass the message through. |
| 44 if (!handled) | 44 if (!handled) |
| 45 handled = RenderViewHostObserver::OnMessageReceived(message); | 45 handled = content::RenderViewHostObserver::OnMessageReceived(message); |
| 46 | 46 |
| 47 return handled; | 47 return handled; |
| 48 } | 48 } |
| 49 | 49 |
| 50 // The base class declares this as protected so this is just here to make it | 50 // The base class declares this as protected so this is just here to make it |
| 51 // public so it is visible to the caller. | 51 // public so it is visible to the caller. |
| 52 bool PrerenderRenderViewHostObserver::Send(IPC::Message* message) { | 52 bool PrerenderRenderViewHostObserver::Send(IPC::Message* message) { |
| 53 return RenderViewHostObserver::Send(message); | 53 return content::RenderViewHostObserver::Send(message); |
| 54 } | 54 } |
| 55 | 55 |
| 56 void PrerenderRenderViewHostObserver::OnUpdateFaviconURL( | 56 void PrerenderRenderViewHostObserver::OnUpdateFaviconURL( |
| 57 int32 page_id, | 57 int32 page_id, |
| 58 const std::vector<FaviconURL>& urls) { | 58 const std::vector<FaviconURL>& urls) { |
| 59 prerender_contents_->OnUpdateFaviconURL(page_id, urls); | 59 prerender_contents_->OnUpdateFaviconURL(page_id, urls); |
| 60 } | 60 } |
| 61 | 61 |
| 62 void PrerenderRenderViewHostObserver::OnMaybeCancelPrerenderForHTML5Media() { | 62 void PrerenderRenderViewHostObserver::OnMaybeCancelPrerenderForHTML5Media() { |
| 63 prerender_contents_->Destroy(FINAL_STATUS_HTML5_MEDIA); | 63 prerender_contents_->Destroy(FINAL_STATUS_HTML5_MEDIA); |
| 64 } | 64 } |
| 65 | 65 |
| 66 void PrerenderRenderViewHostObserver::OnCancelPrerenderForPrinting() { | 66 void PrerenderRenderViewHostObserver::OnCancelPrerenderForPrinting() { |
| 67 prerender_contents_->Destroy(FINAL_STATUS_WINDOW_PRINT); | 67 prerender_contents_->Destroy(FINAL_STATUS_WINDOW_PRINT); |
| 68 } | 68 } |
| 69 | 69 |
| 70 } | 70 } |
| OLD | NEW |