| // Copyright (c) 2012 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. |
| |
| #include "chrome/browser/google/google_url_tracker_navigation_helper_impl.h" |
| |
| #include "chrome/browser/chrome_notification_types.h" |
| #include "chrome/browser/google/google_url_tracker.h" |
| #include "chrome/browser/infobars/infobar_service.h" |
| #include "content/public/browser/navigation_controller.h" |
| #include "content/public/browser/navigation_entry.h" |
| #include "content/public/browser/notification_service.h" |
| #include "content/public/browser/web_contents.h" |
| |
| GoogleURLTrackerNavigationHelperImpl:: |
| GoogleURLTrackerNavigationHelperImpl() : tracker_(NULL) { |
| } |
| |
| GoogleURLTrackerNavigationHelperImpl:: |
| ~GoogleURLTrackerNavigationHelperImpl() { |
| } |
| |
| void GoogleURLTrackerNavigationHelperImpl::SetGoogleURLTracker( |
| GoogleURLTracker* tracker) { |
| DCHECK(tracker); |
| tracker_ = tracker; |
| } |
| |
| void GoogleURLTrackerNavigationHelperImpl::SetListeningForNavigationStart( |
| bool listen) { |
| if (listen) { |
| registrar_.Add(this, content::NOTIFICATION_NAV_ENTRY_PENDING, |
| content::NotificationService::AllBrowserContextsAndSources()); |
| } else { |
| registrar_.Remove(this, content::NOTIFICATION_NAV_ENTRY_PENDING, |
| content::NotificationService::AllBrowserContextsAndSources()); |
| } |
| } |
| |
| bool GoogleURLTrackerNavigationHelperImpl::IsListeningForNavigationStart() { |
| return registrar_.IsRegistered(this, content::NOTIFICATION_NAV_ENTRY_PENDING, |
| content::NotificationService::AllBrowserContextsAndSources()); |
| } |
| |
| void GoogleURLTrackerNavigationHelperImpl::SetListeningForNavigationCommit( |
| const content::NavigationController* nav_controller, |
| bool listen) { |
| content::NotificationSource navigation_controller_source = |
| content::Source<content::NavigationController>(nav_controller); |
| if (listen) { |
| registrar_.Add(this, content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
| navigation_controller_source); |
| } else { |
| registrar_.Remove(this, content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
| navigation_controller_source); |
| } |
| } |
| |
| bool GoogleURLTrackerNavigationHelperImpl::IsListeningForNavigationCommit( |
| const content::NavigationController* nav_controller) { |
| content::NotificationSource navigation_controller_source = |
| content::Source<content::NavigationController>(nav_controller); |
| return registrar_.IsRegistered( |
| this, content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
| navigation_controller_source); |
| } |
| |
| void GoogleURLTrackerNavigationHelperImpl::SetListeningForTabDestruction( |
| const content::NavigationController* nav_controller, |
| bool listen) { |
| content::NotificationSource navigation_controller_source = |
| content::Source<content::NavigationController>(nav_controller); |
| if (listen) { |
| registrar_.Add(this, content::NOTIFICATION_WEB_CONTENTS_DESTROYED, |
| GetWebContentsSource(navigation_controller_source)); |
| } else { |
| registrar_.Remove(this, content::NOTIFICATION_WEB_CONTENTS_DESTROYED, |
| GetWebContentsSource(navigation_controller_source)); |
| } |
| } |
| |
| bool GoogleURLTrackerNavigationHelperImpl::IsListeningForTabDestruction( |
| const content::NavigationController* nav_controller) { |
| content::NotificationSource navigation_controller_source = |
| content::Source<content::NavigationController>(nav_controller); |
| return registrar_.IsRegistered( |
| this, |
| content::NOTIFICATION_WEB_CONTENTS_DESTROYED, |
| GetWebContentsSource(navigation_controller_source)); |
| } |
| |
| content::NotificationSource |
| GoogleURLTrackerNavigationHelperImpl::GetWebContentsSource( |
| const content::NotificationSource& nav_controller_source) { |
| content::NavigationController* controller = |
| content::Source<content::NavigationController>( |
| nav_controller_source).ptr(); |
| content::WebContents* web_contents = controller->GetWebContents(); |
| return content::Source<content::WebContents>(web_contents); |
| } |
| |
| void GoogleURLTrackerNavigationHelperImpl::Observe( |
| int type, |
| const content::NotificationSource& source, |
| const content::NotificationDetails& details) { |
| switch (type) { |
| case content::NOTIFICATION_NAV_ENTRY_PENDING: { |
| content::NavigationController* controller = |
| content::Source<content::NavigationController>(source).ptr(); |
| content::WebContents* web_contents = controller->GetWebContents(); |
| InfoBarService* infobar_service = |
| InfoBarService::FromWebContents(web_contents); |
| // Because we're listening to all sources, there may be no |
| // InfoBarService for some notifications, e.g. navigations in |
| // bubbles/balloons etc. |
| if (infobar_service) { |
| tracker_->OnNavigationPending( |
| controller, infobar_service, |
| controller->GetPendingEntry()->GetUniqueID()); |
| } |
| break; |
| } |
| |
| case content::NOTIFICATION_NAV_ENTRY_COMMITTED: { |
| content::NavigationController* controller = |
| content::Source<content::NavigationController>(source).ptr(); |
| // Here we're only listening to notifications where we already know |
| // there's an associated InfoBarService. |
| content::WebContents* web_contents = controller->GetWebContents(); |
| InfoBarService* infobar_service = |
| InfoBarService::FromWebContents(web_contents); |
| DCHECK(infobar_service); |
| const GURL& search_url = controller->GetActiveEntry()->GetURL(); |
| if (!search_url.is_valid()) // Not clear if this can happen. |
| tracker_->OnTabClosed(controller); |
| tracker_->OnNavigationCommitted(infobar_service, search_url); |
| break; |
| } |
| |
| case content::NOTIFICATION_WEB_CONTENTS_DESTROYED: { |
| content::WebContents* web_contents = |
| content::Source<content::WebContents>(source).ptr(); |
| tracker_->OnTabClosed(&web_contents->GetController()); |
| break; |
| } |
| |
| default: |
| NOTREACHED() << "Unknown notification received:" << type; |
| } |
| } |