Implementation of AutomationEventQueue and associated framework to support generic non-blocking aut…

chrome/browser/automation/automation_event_queue.cc

+// Copyright (c) 2012 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.
+
+#include "chrome/browser/automation/automation_event_observer.h"
+#include "chrome/browser/automation/automation_event_queue.h"
+
+#include "chrome/browser/automation/automation_provider.h"
+#include "chrome/browser/automation/automation_provider_json.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/browser/notification_types.h"
+
+AutomationEventQueue::AutomationEventQueue()
+    : observer_id_count_(0),
+      wait_reply_message_(NULL),
+      wait_observer_id_(-1) {}
+
+AutomationEventQueue::~AutomationEventQueue() {
+  ClearObservers();
+  Clear();
+}
+
+AutomationEventQueue::AutomationEvent::AutomationEvent(
+    int observer_id, DictionaryValue* event_value)
+    : event_value_(event_value), observer_id_(observer_id) {}
+
+void AutomationEventQueue::AutomationEvent::ReturnValue(
+    AutomationProvider* automation,
+    IPC::Message* reply_message) {
+  AutomationJSONReply(automation, reply_message)

[Nirnimesh, 2012/02/28 09:13:09]

This class should not have to respond back to the client directly. It doesn't
even need to know about the json interface. WDYT?

[craigdh, 2012/02/28 22:42:56]

I agree. Changed to GetValue().

+      .SendSuccess(event_value_.get());
+}
+
+void AutomationEventQueue::GetEvent(AutomationProvider* automation,
+                                    IPC::Message* reply_message,
+                                    int observer_id,
+                                    bool blocking) {
+  SetWait(automation, reply_message, observer_id);
+  if (!CheckReturnEvent() && !blocking) {
+    ClearWait();
+    AutomationJSONReply(automation, reply_message).SendSuccess(NULL);
+  }
+}
+
+void AutomationEventQueue::Clear() {
+  std::list<AutomationEvent*>::iterator i;
+  for (i = event_queue_.begin(); i != event_queue_.end(); i++) {
+    delete *i;
+  }
+  event_queue_.clear();
+}
+
+bool AutomationEventQueue::IsEmpty() {
+  return event_queue_.empty();
+}
+
+AutomationEventQueue::AutomationEvent* AutomationEventQueue::PopEvent() {
+  if (event_queue_.empty()) {
+    return NULL;
+  }
+  AutomationEvent* event = event_queue_.back();
+  event_queue_.pop_back();
+  return event;
+}
+
+AutomationEventQueue::AutomationEvent* AutomationEventQueue::PopEvent(
+    int observer_id) {
+  AutomationEvent* event = NULL;
+  std::list<AutomationEvent*>::reverse_iterator i;
+  for (i = event_queue_.rbegin(); i != event_queue_.rend(); i++) {
+    if (observer_id < 0 || (*i)->GetId() == observer_id) {
+      event = *i;
+      break;
+    }
+  }
+  if (event) {
+    event_queue_.remove(event);
+    return event;
+  }
+  return NULL;
+}
+
+void AutomationEventQueue::NotifyEvent(
+    AutomationEventQueue::AutomationEvent* event) {
+  event_queue_.push_front(event);
+  CheckReturnEvent();
+}
+
+int AutomationEventQueue::AddObserver(AutomationEventObserver* observer) {
+  int id = observer_id_count_++;
+  observer->Init(id);
+  observers_[id] = observer;
+  return id;
+}
+
+void AutomationEventQueue::RemoveObserver(int observer_id) {
+  if (observers_.count(observer_id)) {
+    delete observers_[observer_id];
+    observers_.erase(observer_id);
+  }
+}
+
+void AutomationEventQueue::ClearObservers() {
+  std::map<int, AutomationEventObserver*>::iterator i;

[Nirnimesh, 2012/02/28 09:13:09]

s/i/it/

[craigdh, 2012/02/28 22:42:56]

Done.

+  for (i = observers_.begin(); i != observers_.end(); i++) {
+    delete (*i).second;

[Nirnimesh, 2012/02/28 09:13:09]

i->second

[craigdh, 2012/02/28 22:42:56]

Done.

+  }
+  observers_.clear();
+}
+
+bool AutomationEventQueue::CheckReturnEvent() {
+  if (wait_automation_ && wait_reply_message_.get()) {
+    AutomationEventQueue::AutomationEvent* event = wait_observer_id_ < 0 ?
+                             PopEvent() :
+                             PopEvent(wait_observer_id_);
+    if (event) {
+      event->ReturnValue(wait_automation_, wait_reply_message_.release());
+      delete event;
+      ClearWait();
+      return true;
+    }
+  }
+  return false;
+}
+
+void AutomationEventQueue::SetWait(AutomationProvider* automation,
+                                   IPC::Message* reply_message,
+                                   int observer_id) {
+  wait_automation_ = automation;
+  wait_reply_message_.reset(reply_message);
+  wait_observer_id_ = observer_id;
+}
+
+void AutomationEventQueue::ClearWait() {
+  SetWait(NULL, NULL, -1);
+}