OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/automation/automation_tab_tracker.h" |
| 6 |
| 7 #include "chrome/browser/tab_contents/navigation_controller.h" |
| 8 #include "chrome/common/notification_type.h" |
| 9 #include "chrome/common/notification_source.h" |
| 10 |
| 11 AutomationTabTracker::AutomationTabTracker(IPC::Message::Sender* automation) |
| 12 : AutomationResourceTracker<NavigationController*>(automation) { |
| 13 } |
| 14 |
| 15 AutomationTabTracker::~AutomationTabTracker() { |
| 16 } |
| 17 |
| 18 void AutomationTabTracker::AddObserver(NavigationController* resource) { |
| 19 // This tab could either be a regular tab or an external tab |
| 20 // Register for both notifications. |
| 21 registrar_.Add(this, NotificationType::TAB_CLOSING, |
| 22 Source<NavigationController>(resource)); |
| 23 registrar_.Add(this, NotificationType::EXTERNAL_TAB_CLOSED, |
| 24 Source<NavigationController>(resource)); |
| 25 // We also want to know about navigations so we can keep track of the last |
| 26 // navigation time. |
| 27 registrar_.Add(this, NotificationType::LOAD_STOP, |
| 28 Source<NavigationController>(resource)); |
| 29 } |
| 30 |
| 31 void AutomationTabTracker::RemoveObserver(NavigationController* resource) { |
| 32 registrar_.Remove(this, NotificationType::TAB_CLOSING, |
| 33 Source<NavigationController>(resource)); |
| 34 registrar_.Remove(this, NotificationType::EXTERNAL_TAB_CLOSED, |
| 35 Source<NavigationController>(resource)); |
| 36 registrar_.Remove(this, NotificationType::LOAD_STOP, |
| 37 Source<NavigationController>(resource)); |
| 38 } |
| 39 |
| 40 void AutomationTabTracker::Observe(NotificationType type, |
| 41 const NotificationSource& source, |
| 42 const NotificationDetails& details) { |
| 43 switch (type.value) { |
| 44 case NotificationType::LOAD_STOP: |
| 45 last_navigation_times_[Source<NavigationController>(source).ptr()] = |
| 46 base::Time::Now(); |
| 47 return; |
| 48 case NotificationType::EXTERNAL_TAB_CLOSED: |
| 49 case NotificationType::TAB_CLOSING: |
| 50 { |
| 51 std::map<NavigationController*, base::Time>::iterator iter = |
| 52 last_navigation_times_.find( |
| 53 Source<NavigationController>(source).ptr()); |
| 54 if (iter != last_navigation_times_.end()) |
| 55 last_navigation_times_.erase(iter); |
| 56 } |
| 57 break; |
| 58 default: |
| 59 NOTREACHED(); |
| 60 } |
| 61 AutomationResourceTracker<NavigationController*>::Observe(type, source, |
| 62 details); |
| 63 } |
| 64 |
| 65 base::Time AutomationTabTracker::GetLastNavigationTime(int handle) { |
| 66 if (ContainsHandle(handle)) { |
| 67 NavigationController* controller = GetResource(handle); |
| 68 if (controller) { |
| 69 std::map<NavigationController*, base::Time>::const_iterator iter = |
| 70 last_navigation_times_.find(controller); |
| 71 if (iter != last_navigation_times_.end()) |
| 72 return iter->second; |
| 73 } |
| 74 } |
| 75 return base::Time(); |
| 76 } |
OLD | NEW |