// 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_