RulesRegistry for declarative APIs.

chrome/browser/extensions/api/declarative/test_rules_registry.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/extensions/api/declarative/test_rules_registry.h"
+
+#include <algorithm>
+
+#include "base/logging.h"
+#include "base/stl_util.h"
+#include "chrome/browser/extensions/api/declarative/declarative_api_constants.h"
+
+namespace keys = extensions::declarative_api_constants;
+
+namespace {
+
+std::string GetRuleId(DictionaryValue* rule) {
+  std::string rule_id;
+  CHECK(rule->GetString(keys::kId, &rule_id));
+  return rule_id;
+}
+
+}  // namespace
+
+namespace extensions {
+
+TestRulesRegistry::TestRulesRegistry() {
+}
+
+TestRulesRegistry::~TestRulesRegistry() {
+  STLDeleteValues(&rules_);
+}
+
+bool TestRulesRegistry::AddRules(
+    const std::string& extension_id,
+    const std::vector<DictionaryValue*>& rules,
+    std::string* error) {
+  std::vector<DictionaryValue*>::const_iterator i;
+  for (i = rules.begin(); i != rules.end(); ++i) {
+    std::string rule_id = GetRuleId(*i);
+    // TODO: relax this (check first and abort with returning false).
+    CHECK(rules_.find(rule_id) == rules_.end());
+    rules_[rule_id] = (*i)->DeepCopy();
+  }
+  return true;
+}
+
+bool TestRulesRegistry::RemoveRules(
+    const std::string& extension_id,
+    const std::vector<std::string>& rule_identifiers,
+    std::string* error) {
+  if (rule_identifiers.empty()) {
+    STLDeleteValues(&rules_);
+  } else {
+    std::vector<std::string>::const_iterator i;
+    for (i = rule_identifiers.begin(); i != rule_identifiers.end(); ++i) {
+      std::map<std::string, DictionaryValue*>::iterator entry = rules_.find(*i);
+      // TODO: relax this (check first and abort with returning false).
+      CHECK(entry != rules_.end());
+      delete entry->second;
+      rules_.erase(entry);
+    }
+  }
+  return true;
+}
+
+void TestRulesRegistry::GetRules(
+    const std::string& extension_id,
+    const std::vector<std::string>& rule_identifiers,
+    std::vector<DictionaryValue*>* out) {
+  if (rule_identifiers.empty()) {
+    std::map<std::string, DictionaryValue*>::const_iterator i;
+    for (i = rules_.begin(); i != rules_.end(); ++i)
+      out->push_back(i->second->DeepCopy());
+  } else {
+    std::vector<std::string>::const_iterator i;
+    for (i = rule_identifiers.begin(); i != rule_identifiers.end(); ++i) {
+      std::map<std::string, DictionaryValue*>::iterator entry = rules_.find(*i);
+      // TODO: relax this (check first and abort with returning false).
+      CHECK(entry != rules_.end());
+      out->push_back(entry->second->DeepCopy());
+    }
+  }
+}
+
+void TestRulesRegistry::OnExtensionUnloaded(const std::string& extension_id) {
+  std::vector<std::string> no_rule_identifiers;
+  std::string error;
+  if (!RemoveRules(extension_id, no_rule_identifiers, &error))
+    LOG(ERROR) << error;
+}
+
+}  // namespace extensions