OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "extensions/common/features/base_feature_provider.h" | 5 #include "extensions/common/features/json_feature_provider.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include <stack> | 9 #include <stack> |
10 #include <utility> | 10 #include <utility> |
11 | 11 |
12 #include "base/strings/string_split.h" | 12 #include "base/strings/string_split.h" |
13 #include "base/strings/string_util.h" | 13 #include "base/strings/string_util.h" |
14 #include "base/values.h" | 14 #include "base/values.h" |
15 #include "extensions/common/extensions_client.h" | 15 #include "extensions/common/extensions_client.h" |
(...skipping 20 matching lines...) Expand all Loading... |
36 SimpleFeature* feature) { | 36 SimpleFeature* feature) { |
37 feature->set_name(name); | 37 feature->set_name(name); |
38 std::string error = feature->Parse(value); | 38 std::string error = feature->Parse(value); |
39 if (!error.empty()) | 39 if (!error.empty()) |
40 LOG(ERROR) << error; | 40 LOG(ERROR) << error; |
41 return error.empty(); | 41 return error.empty(); |
42 } | 42 } |
43 | 43 |
44 } // namespace | 44 } // namespace |
45 | 45 |
46 BaseFeatureProvider::BaseFeatureProvider(const base::DictionaryValue& root, | 46 JSONFeatureProvider::JSONFeatureProvider(const base::DictionaryValue& root, |
47 FeatureFactory factory) | 47 FeatureFactory factory) |
48 : factory_(factory) { | 48 : factory_(factory) { |
49 for (base::DictionaryValue::Iterator iter(root); !iter.IsAtEnd(); | 49 for (base::DictionaryValue::Iterator iter(root); !iter.IsAtEnd(); |
50 iter.Advance()) { | 50 iter.Advance()) { |
51 if (IsNocompile(iter.value())) { | 51 if (IsNocompile(iter.value())) { |
52 continue; | 52 continue; |
53 } | 53 } |
54 | 54 |
55 if (iter.value().GetType() == base::Value::TYPE_DICTIONARY) { | 55 if (iter.value().GetType() == base::Value::TYPE_DICTIONARY) { |
56 std::unique_ptr<SimpleFeature> feature((*factory_)()); | 56 std::unique_ptr<SimpleFeature> feature((*factory_)()); |
57 | 57 |
58 std::vector<std::string> split = base::SplitString( | 58 std::vector<std::string> split = base::SplitString( |
59 iter.key(), ".", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL); | 59 iter.key(), ".", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL); |
60 | 60 |
61 // Push parent features on the stack, starting with the current feature. | 61 // Push parent features on the stack, starting with the current feature. |
62 // If one of the features has "noparent" set, stop pushing features on | 62 // If one of the features has "noparent" set, stop pushing features on |
63 // the stack. The features will then be parsed in order, starting with | 63 // the stack. The features will then be parsed in order, starting with |
64 // the farthest parent that is either top level or has "noparent" set. | 64 // the farthest parent that is either top level or has "noparent" set. |
65 std::stack<std::pair<std::string, const base::DictionaryValue*> > | 65 std::stack<std::pair<std::string, const base::DictionaryValue*>> |
66 parse_stack; | 66 parse_stack; |
67 while (!split.empty()) { | 67 while (!split.empty()) { |
68 std::string parent_name = base::JoinString(split, "."); | 68 std::string parent_name = base::JoinString(split, "."); |
69 split.pop_back(); | 69 split.pop_back(); |
70 if (root.HasKey(parent_name)) { | 70 if (root.HasKey(parent_name)) { |
71 const base::DictionaryValue* parent = nullptr; | 71 const base::DictionaryValue* parent = nullptr; |
72 if (!root.GetDictionaryWithoutPathExpansion(parent_name, &parent)) { | 72 if (!root.GetDictionaryWithoutPathExpansion(parent_name, &parent)) { |
73 // If the parent is a complex feature, find the parent with the | 73 // If the parent is a complex feature, find the parent with the |
74 // 'default_parent' flag. | 74 // 'default_parent' flag. |
75 const base::ListValue* parent_list = nullptr; | 75 const base::ListValue* parent_list = nullptr; |
76 CHECK(root.GetListWithoutPathExpansion(parent_name, &parent_list)); | 76 CHECK(root.GetListWithoutPathExpansion(parent_name, &parent_list)); |
77 for (size_t i = 0; i < parent_list->GetSize(); ++i) { | 77 for (size_t i = 0; i < parent_list->GetSize(); ++i) { |
78 CHECK(parent_list->GetDictionary(i, &parent)); | 78 CHECK(parent_list->GetDictionary(i, &parent)); |
79 if (parent->HasKey("default_parent")) | 79 if (parent->HasKey("default_parent")) |
80 break; | 80 break; |
81 parent = nullptr; | 81 parent = nullptr; |
82 } | 82 } |
83 CHECK(parent) << parent_name << " must declare one of its features" | 83 CHECK(parent) |
| 84 << parent_name << " must declare one of its features" |
84 << " the default parent, with {\"default_parent\": true}."; | 85 << " the default parent, with {\"default_parent\": true}."; |
85 } | 86 } |
86 parse_stack.push(std::make_pair(parent_name, parent)); | 87 parse_stack.push(std::make_pair(parent_name, parent)); |
87 bool no_parent = false; | 88 bool no_parent = false; |
88 parent->GetBoolean("noparent", &no_parent); | 89 parent->GetBoolean("noparent", &no_parent); |
89 if (no_parent) | 90 if (no_parent) |
90 break; | 91 break; |
91 } | 92 } |
92 } | 93 } |
93 | 94 |
94 CHECK(!parse_stack.empty()); | 95 CHECK(!parse_stack.empty()); |
95 // Parse all parent features. | 96 // Parse all parent features. |
96 bool parse_error = false; | 97 bool parse_error = false; |
97 while (!parse_stack.empty()) { | 98 while (!parse_stack.empty()) { |
98 if (!ParseFeature(parse_stack.top().second, | 99 if (!ParseFeature(parse_stack.top().second, parse_stack.top().first, |
99 parse_stack.top().first, | |
100 feature.get())) { | 100 feature.get())) { |
101 parse_error = true; | 101 parse_error = true; |
102 break; | 102 break; |
103 } | 103 } |
104 parse_stack.pop(); | 104 parse_stack.pop(); |
105 } | 105 } |
106 | 106 |
107 if (parse_error) | 107 if (parse_error) |
108 continue; | 108 continue; |
109 | 109 |
(...skipping 27 matching lines...) Expand all Loading... |
137 feature->set_name(iter.key()); | 137 feature->set_name(iter.key()); |
138 | 138 |
139 features_[iter.key()] = std::move(feature); | 139 features_[iter.key()] = std::move(feature); |
140 } else { | 140 } else { |
141 LOG(ERROR) << iter.key() << ": Feature description must be dictionary or" | 141 LOG(ERROR) << iter.key() << ": Feature description must be dictionary or" |
142 << " list of dictionaries."; | 142 << " list of dictionaries."; |
143 } | 143 } |
144 } | 144 } |
145 } | 145 } |
146 | 146 |
147 BaseFeatureProvider::~BaseFeatureProvider() { | 147 JSONFeatureProvider::~JSONFeatureProvider() {} |
148 } | |
149 | |
150 const FeatureMap& BaseFeatureProvider::GetAllFeatures() const { | |
151 return features_; | |
152 } | |
153 | |
154 Feature* BaseFeatureProvider::GetFeature(const std::string& name) const { | |
155 FeatureMap::const_iterator iter = features_.find(name); | |
156 if (iter != features_.end()) | |
157 return iter->second.get(); | |
158 else | |
159 return nullptr; | |
160 } | |
161 | |
162 Feature* BaseFeatureProvider::GetParent(Feature* feature) const { | |
163 CHECK(feature); | |
164 if (feature->no_parent()) | |
165 return nullptr; | |
166 | |
167 std::vector<std::string> split = base::SplitString( | |
168 feature->name(), ".", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL); | |
169 if (split.size() < 2) | |
170 return nullptr; | |
171 split.pop_back(); | |
172 return GetFeature(base::JoinString(split, ".")); | |
173 } | |
174 | |
175 // Children of a given API are named starting with parent.name()+".", which | |
176 // means they'll be contiguous in the features_ std::map. | |
177 std::vector<Feature*> BaseFeatureProvider::GetChildren(const Feature& parent) | |
178 const { | |
179 std::string prefix = parent.name() + "."; | |
180 const FeatureMap::const_iterator first_child = features_.lower_bound(prefix); | |
181 | |
182 // All children have names before (parent.name() + ('.'+1)). | |
183 ++prefix.back(); | |
184 const FeatureMap::const_iterator after_children = | |
185 features_.lower_bound(prefix); | |
186 | |
187 std::vector<Feature*> result; | |
188 result.reserve(std::distance(first_child, after_children)); | |
189 for (FeatureMap::const_iterator it = first_child; it != after_children; | |
190 ++it) { | |
191 result.push_back(it->second.get()); | |
192 } | |
193 return result; | |
194 } | |
195 | 148 |
196 } // namespace extensions | 149 } // namespace extensions |
OLD | NEW |