| Index: chrome/browser/prerender/prerender_tab_helper.cc
|
| diff --git a/chrome/browser/prerender/prerender_tab_helper.cc b/chrome/browser/prerender/prerender_tab_helper.cc
|
| index 2a783048e2b2dd8df1dff08e0b3af1284efd1c29..afff839a8b65be2156a6c557f5d4cf7e1d8830bf 100644
|
| --- a/chrome/browser/prerender/prerender_tab_helper.cc
|
| +++ b/chrome/browser/prerender/prerender_tab_helper.cc
|
| @@ -13,7 +13,10 @@
|
| #include "chrome/browser/profiles/profile.h"
|
| #include "chrome/browser/ui/tab_contents/core_tab_helper.h"
|
| #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
|
| +#include "content/public/browser/notification_service.h"
|
| +#include "content/public/browser/notification_types.h"
|
| #include "content/public/browser/render_view_host.h"
|
| +#include "content/public/browser/resource_request_details.h"
|
| #include "content/public/browser/web_contents.h"
|
| #include "content/public/browser/web_contents_view.h"
|
| #include "skia/ext/platform_canvas.h"
|
| @@ -149,27 +152,14 @@ PrerenderTabHelper::PrerenderTabHelper(TabContentsWrapper* tab)
|
| : content::WebContentsObserver(tab->web_contents()),
|
| pixel_stats_(new PixelStats(this)),
|
| tab_(tab) {
|
| + notification_registrar_.Add(
|
| + this, content::NOTIFICATION_RESOURCE_RECEIVED_REDIRECT,
|
| + content::Source<content::WebContents>(web_contents()));
|
| }
|
|
|
| PrerenderTabHelper::~PrerenderTabHelper() {
|
| }
|
|
|
| -void PrerenderTabHelper::ProvisionalChangeToMainFrameUrl(
|
| - const GURL& url,
|
| - const GURL& opener_url,
|
| - content::RenderViewHost* render_view_host) {
|
| - url_ = url;
|
| - RecordPageviewEvent(PAGEVIEW_EVENT_NEW_URL);
|
| - if (IsTopSite(url))
|
| - RecordPageviewEvent(PAGEVIEW_EVENT_TOP_SITE_NEW_URL);
|
| - PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
|
| - if (!prerender_manager)
|
| - return;
|
| - if (prerender_manager->IsWebContentsPrerendering(web_contents()))
|
| - return;
|
| - prerender_manager->MarkWebContentsAsNotPrerendered(web_contents());
|
| -}
|
| -
|
| void PrerenderTabHelper::DidCommitProvisionalLoadForFrame(
|
| int64 frame_id,
|
| bool is_main_frame,
|
| @@ -278,4 +268,38 @@ bool PrerenderTabHelper::IsTopSite(const GURL& url) {
|
| return (pm && pm->IsTopSite(url));
|
| }
|
|
|
| +void PrerenderTabHelper::HandleResourceReceivedRedirect(const GURL& new_url) {
|
| + url_ = new_url;
|
| + RecordPageviewEvent(PAGEVIEW_EVENT_NEW_URL);
|
| + if (IsTopSite(url_))
|
| + RecordPageviewEvent(PAGEVIEW_EVENT_TOP_SITE_NEW_URL);
|
| + PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
|
| + if (!prerender_manager)
|
| + return;
|
| + if (prerender_manager->IsWebContentsPrerendering(web_contents()))
|
| + return;
|
| + prerender_manager->MarkWebContentsAsNotPrerendered(web_contents());
|
| +}
|
| +
|
| +void PrerenderTabHelper::Observe(
|
| + int type,
|
| + const content::NotificationSource& source,
|
| + const content::NotificationDetails& details) {
|
| + switch (type) {
|
| + case content::NOTIFICATION_RESOURCE_RECEIVED_REDIRECT: {
|
| + DCHECK(content::Source<content::WebContents>(source).ptr() ==
|
| + web_contents());
|
| + content::ResourceRedirectDetails* resource_redirect_details =
|
| + content::Details<content::ResourceRedirectDetails>(details).ptr();
|
| + CHECK(resource_redirect_details);
|
| + HandleResourceReceivedRedirect(resource_redirect_details->new_url);
|
| + break;
|
| + }
|
| +
|
| + default:
|
| + NOTREACHED() << "Unexpected notification sent.";
|
| + break;
|
| + }
|
| +}
|
| +
|
| } // namespace prerender
|
|
|