chrome/browser/automation/automation_tab_tracker.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// 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 "chrome/browser/automation/automation_resource_tracker.h"
#include "chrome/browser/browser.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/navigation_controller.h"

class AutomationTabTracker
  : public AutomationResourceTracker<NavigationController*> {
public:
  AutomationTabTracker(IPC::Message::Sender* automation)
    : AutomationResourceTracker(automation) {}

  virtual ~AutomationTabTracker() {
    ClearAllMappings();
  }

  virtual void AddObserver(NavigationController* resource) {
    // This tab could either be a regular tab or an external tab
    // Register for both notifications
    NotificationService::current()->AddObserver(
        this, NOTIFY_TAB_CLOSING, Source<NavigationController>(resource));
    NotificationService::current()->AddObserver(
        this, NOTIFY_EXTERNAL_TAB_CLOSED,
        Source<NavigationController>(resource));
  }

  virtual void RemoveObserver(NavigationController* resource) {
    NotificationService::current()->RemoveObserver(
        this, NOTIFY_TAB_CLOSING, Source<NavigationController>(resource));
    NotificationService::current()->RemoveObserver(
        this, NOTIFY_EXTERNAL_TAB_CLOSED,
        Source<NavigationController>(resource));
  }
};

#endif // CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_TRACKER_H__