1 files changed, 92 insertions, 0 deletions
diff --git a/chrome/browser/automation/automation_tab_tracker.h b/chrome/browser/automation/automation_tab_tracker.h
new file mode 100644
index 0000000..02b40be
--- /dev/null
+++ b/chrome/browser/automation/automation_tab_tracker.h
@@ -0,0 +1,92 @@
+// Copyright (c) 2006-2008 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.
+
+#ifndef CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_TRACKER_H_
+#define CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_TRACKER_H_
+
+#include <map>
+
+#include "base/time.h"
+#include "chrome/browser/automation/automation_resource_tracker.h"
+#include "chrome/browser/tab_contents/navigation_controller.h"
+#include "chrome/common/notification_registrar.h"
+#include "chrome/common/notification_type.h"
+
+class AutomationTabTracker
+  : public AutomationResourceTracker<NavigationController*> {
+ public:
+  explicit AutomationTabTracker(IPC::Message::Sender* automation)
+      : AutomationResourceTracker<NavigationController*>(automation) {}
+
+  virtual ~AutomationTabTracker() {
+  }
+
+  virtual void AddObserver(NavigationController* resource) {
+    // This tab could either be a regular tab or an external tab
+    // Register for both notifications.
+    registrar_.Add(this, NotificationType::TAB_CLOSING,
+                   Source<NavigationController>(resource));
+    registrar_.Add(this, NotificationType::EXTERNAL_TAB_CLOSED,
+                   Source<NavigationController>(resource));
+    // We also want to know about navigations so we can keep track of the last
+    // navigation time.
+    registrar_.Add(this, NotificationType::LOAD_STOP,
+                   Source<NavigationController>(resource));
+  }
+
+  virtual void RemoveObserver(NavigationController* resource) {
+    registrar_.Remove(this, NotificationType::TAB_CLOSING,
+                      Source<NavigationController>(resource));
+    registrar_.Remove(this, NotificationType::EXTERNAL_TAB_CLOSED,
+                      Source<NavigationController>(resource));
+    registrar_.Remove(this, NotificationType::LOAD_STOP,
+                      Source<NavigationController>(resource));
+  }
+
+  virtual void Observe(NotificationType type,
+                       const NotificationSource& source,
+                       const NotificationDetails& details) {
+    switch (type.value) {
+      case NotificationType::LOAD_STOP:
+        last_navigation_times_[Source<NavigationController>(source).ptr()] =
+            base::Time::Now();
+        return;
+      case NotificationType::EXTERNAL_TAB_CLOSED:
+      case NotificationType::TAB_CLOSING:
+        {
+          std::map<NavigationController*, base::Time>::iterator iter =
+              last_navigation_times_.find(
+              Source<NavigationController>(source).ptr());
+          if (iter != last_navigation_times_.end())
+            last_navigation_times_.erase(iter);
+        }
+        break;
+      default:
+        NOTREACHED();
+    }
+    AutomationResourceTracker<NavigationController*>::Observe(type, source,
+                                                              details);
+  }
+
+  base::Time GetLastNavigationTime(int handle) {
+    if (ContainsHandle(handle)) {
+      NavigationController* controller = GetResource(handle);
+      if (controller) {
+        std::map<NavigationController*, base::Time>::const_iterator iter =
+            last_navigation_times_.find(controller);
+        if (iter != last_navigation_times_.end())
+          return iter->second;
+      }
+    }
+    return base::Time();
+  }
+
+ private:
+  // Last time a navigation occurred.
+  std::map<NavigationController*, base::Time> last_navigation_times_;
+
+  DISALLOW_COPY_AND_ASSIGN(AutomationTabTracker);
+};
+
+#endif  // CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_TRACKER_H_