Refactor GlobalShortcutListener.

chrome/browser/extensions/global_shortcut_listener.cc

Index: chrome/browser/extensions/global_shortcut_listener.cc
diff --git a/chrome/browser/extensions/global_shortcut_listener.cc b/chrome/browser/extensions/global_shortcut_listener.cc
index aad5453eca527b92e1bc426da2f8bbef700db1d7..a4ab66486909359041f4b22d9587240c569b3b8a 100644
--- a/chrome/browser/extensions/global_shortcut_listener.cc
+++ b/chrome/browser/extensions/global_shortcut_listener.cc
@@ -15,33 +15,37 @@ GlobalShortcutListener::~GlobalShortcutListener() {
   DCHECK(accelerator_map_.empty());  // Make sure we've cleaned up.
 }
 
-void GlobalShortcutListener::RegisterAccelerator(
+bool GlobalShortcutListener::RegisterAccelerator(
     const ui::Accelerator& accelerator, Observer* observer) {
   AcceleratorMap::const_iterator it = accelerator_map_.find(accelerator);
-  if (it == accelerator_map_.end()) {
-    if (accelerator_map_.empty())
-      GlobalShortcutListener::GetInstance()->StartListening();
-    Observers* observers = new Observers;
-    observers->AddObserver(observer);
-    accelerator_map_[accelerator] = observers;
-  } else {
-    // Make sure we don't register the same accelerator twice.
-    DCHECK(!accelerator_map_[accelerator]->HasObserver(observer));
-    accelerator_map_[accelerator]->AddObserver(observer);
-  }
+  // The accelerator has been registered.
+  if (it != accelerator_map_.end())
+    return false;
+
+  // If the platform-specific registration fails, mostly likely the shortcut has
+  // been registered by other native applications.
+  if (!RegisterAcceleratorImpl(accelerator))
+    return false;
+
+  if (accelerator_map_.empty())
+    GlobalShortcutListener::GetInstance()->StartListening();
+
+  accelerator_map_[accelerator] = observer;
+  return true;
 }
 
 void GlobalShortcutListener::UnregisterAccelerator(
     const ui::Accelerator& accelerator, Observer* observer) {
   AcceleratorMap::iterator it = accelerator_map_.find(accelerator);
+  // We should never get asked to unregister something that we didn't register.
   DCHECK(it != accelerator_map_.end());
-  DCHECK(it->second->HasObserver(observer));
-  it->second->RemoveObserver(observer);
-  if (!it->second->might_have_observers()) {
-    accelerator_map_.erase(it);
-    if (accelerator_map_.empty())
-      GlobalShortcutListener::GetInstance()->StopListening();
-  }
+  // The caller should call this function with the right observer.
+  DCHECK(it->second == observer);
+
+  UnregisterAcceleratorImpl(accelerator);
+  accelerator_map_.erase(it);
+  if (accelerator_map_.empty())
+    GlobalShortcutListener::GetInstance()->StopListening();
 }
 
 void GlobalShortcutListener::NotifyKeyPressed(
@@ -53,10 +57,8 @@ void GlobalShortcutListener::NotifyKeyPressed(
     NOTREACHED();
     return;  // No-one is listening to this key.
   }
-  // The observer list should not be empty.
-  DCHECK(iter->second->might_have_observers());
 
-  FOR_EACH_OBSERVER(Observer, *(iter->second), OnKeyPressed(accelerator));
+  iter->second->OnKeyPressed(accelerator);
 }
 
 }  // namespace extensions