Add common code for extensions event filtering.

chrome/common/extensions/event_filter.cc

Index: chrome/common/extensions/event_filter.cc
diff --git a/chrome/common/extensions/event_filter.cc b/chrome/common/extensions/event_filter.cc
new file mode 100644
index 0000000000000000000000000000000000000000..ec183ac6c892895ef3478c24c144d63c86005b3d
--- /dev/null
+++ b/chrome/common/extensions/event_filter.cc
@@ -0,0 +1,139 @@
+// 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/common/extensions/event_filter.h"
+
+#include "chrome/common/extensions/matcher/url_matcher_factory.h"
+
+namespace extensions {
+
+EventFilter::EventMatcherEntry::EventMatcherEntry(
+    scoped_ptr<EventMatcher> event_matcher,
+    URLMatcher* url_matcher,
+    const URLMatcherConditionSet::Vector& condition_sets)
+    : event_matcher_(event_matcher.Pass()),
+      url_matcher_(url_matcher) {
+  for (URLMatcherConditionSet::Vector::const_iterator it =
+       condition_sets.begin(); it != condition_sets.end(); it++)
+    condition_set_ids_.push_back((*it)->id());
+  url_matcher_->AddConditionSets(condition_sets);
+}
+
+EventFilter::EventMatcherEntry::~EventMatcherEntry() {
+  url_matcher_->RemoveConditionSets(condition_set_ids_);

[battre, 2012/05/17 08:50:51]

This architecture makes the destruction of the EventFilter at browser shutdown
somewhat expensive. Maybe add a function to clear condition_set_ids_ and call it
from ~EventFilter(), because they are cleared when the URLMatcher dies anyway? 

[koz (OOO until 15th September), 2012/05/18 01:13:50]

Good point. Done.

+}
+
+EventFilter::EventFilter()
+    : next_id_(0),
+      next_condition_set_id_(0) {
+}
+
+EventFilter::~EventFilter() {
+}
+
+EventFilter::MatcherID
+EventFilter::AddEventMatcher(const std::string& event_name,
+                            scoped_ptr<EventMatcher> matcher) {

[battre, 2012/05/17 08:50:51]

nit: indent +1

[koz (OOO until 15th September), 2012/05/18 01:13:50]

Done.

+  MatcherID id = next_id_++;
+  URLMatcherConditionSet::Vector condition_sets;
+  if (!CreateConditionSets(id, matcher->url_filters(), &condition_sets))
+    return -1;
+
+  for (URLMatcherConditionSet::Vector::iterator it = condition_sets.begin();
+       it != condition_sets.end(); it++) {
+    condition_set_id_to_event_matcher_id_.insert(
+        std::make_pair((*it)->id(), id));
+  }
+  id_to_event_name_[id] = event_name;
+  event_matchers_[event_name][id] = linked_ptr<EventMatcherEntry>(
+      new EventMatcherEntry(matcher.Pass(), &url_matcher_, condition_sets));
+  return id;
+}
+
+bool EventFilter::CreateConditionSets(
+    MatcherID id,
+    base::ListValue* url_filters,
+    URLMatcherConditionSet::Vector* condition_sets) {
+  if (!url_filters || url_filters->GetSize() == 0) {
+    // If there are no url_filters then we want to match all events, so create a
+    // URLFilter from an empty dictionary.
+    base::DictionaryValue empty_dict;
+    return AddDictionaryAsConditionSet(&empty_dict, condition_sets);
+  }
+  for (size_t i = 0; i < url_filters->GetSize(); i++) {
+    base::DictionaryValue* url_filter;
+    if (!url_filters->GetDictionary(i, &url_filter))
+      return false;
+    if (!AddDictionaryAsConditionSet(url_filter, condition_sets))
+      return false;
+  }
+  return true;
+}
+
+bool EventFilter::AddDictionaryAsConditionSet(
+    base::DictionaryValue* url_filter,
+    URLMatcherConditionSet::Vector* condition_sets) {
+  std::string error;
+  URLMatcherConditionSet::ID condition_set_id = next_condition_set_id_++;
+  condition_sets->push_back(URLMatcherFactory::CreateFromURLFilterDictionary(
+      url_matcher_.condition_factory(),
+      url_filter,
+      condition_set_id,
+      &error));
+  if (!error.empty()) {
+    LOG(ERROR) << "CreateFromURLFilterDictionary failed: " << error;
+    url_matcher_.ClearUnusedConditionSets();
+    condition_sets->clear();
+    return false;
+  }
+  return true;
+}
+
+std::string EventFilter::RemoveEventMatcher(MatcherID id) {
+  std::map<int, std::string>::iterator it = id_to_event_name_.find(id);
+  event_matchers_[it->second].erase(id);
+  std::string event_name = it->second;
+  id_to_event_name_.erase(it);
+  return event_name;
+}
+
+std::set<EventFilter::MatcherID> EventFilter::MatchEvent(
+    const std::string& event_name, const EventFilteringInfo& event_info) {
+  std::set<MatcherID> matchers;
+  EventMatcherMultiMap::iterator it = event_matchers_.find(event_name);
+  if (it == event_matchers_.end())
+    return matchers;
+
+  EventMatcherMap& matcher_map = it->second;
+  std::set<MatcherID> url_matched_matchers;
+  GURL url_to_match_against = event_info.has_url() ? event_info.url() : GURL();
+  std::set<URLMatcherConditionSet::ID> matching_condition_set_ids =
+      url_matcher_.MatchURL(url_to_match_against);
+  for (std::set<URLMatcherConditionSet::ID>::iterator it =

[battre, 2012/05/17 08:50:51]

I would recommend to combine these two loop. I don't think it is necessary to
store everything into url_matched_matchers first. I would add a
matchers.reserve(url_matched_matchers.size()) infront of this loop.

[koz (OOO until 15th September), 2012/05/18 01:13:50]

I merged the loops but there is no reserve method on std::set.

+       matching_condition_set_ids.begin();
+       it != matching_condition_set_ids.end(); it++) {
+    url_matched_matchers.insert(condition_set_id_to_event_matcher_id_[*it]);
+  }
+
+  for (std::set<MatcherID>::const_iterator it = url_matched_matchers.begin();
+       it != url_matched_matchers.end(); it++) {
+    const EventMatcher& event_matcher = matcher_map[*it]->event_matcher();
+    if (event_matcher.MatchNonURLCriteria(event_info)) {
+      CHECK(!event_matcher.url_filters() || event_info.has_url());
+      matchers.insert(*it);
+    }
+  }
+
+  return matchers;
+}
+
+int EventFilter::GetMatcherCountForEvent(const std::string& name) {
+  EventMatcherMultiMap::const_iterator it = event_matchers_.find(name);
+  if (it == event_matchers_.end())
+    return 0;
+
+  return it->second.size();
+}
+
+}  // namespace extensions