OLD | NEW |
| (Empty) |
1 // Copyright (C) 2014 Google Inc. | |
2 // | |
3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
4 // you may not use this file except in compliance with the License. | |
5 // You may obtain a copy of the License at | |
6 // | |
7 // http://www.apache.org/licenses/LICENSE-2.0 | |
8 // | |
9 // Unless required by applicable law or agreed to in writing, software | |
10 // distributed under the License is distributed on an "AS IS" BASIS, | |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
12 // See the License for the specific language governing permissions and | |
13 // limitations under the License. | |
14 | |
15 #include "country_rules_aggregator.h" | |
16 | |
17 #include <libaddressinput/address_field.h> | |
18 #include <libaddressinput/callback.h> | |
19 #include <libaddressinput/util/scoped_ptr.h> | |
20 | |
21 #include <algorithm> | |
22 #include <cassert> | |
23 #include <cstddef> | |
24 #include <map> | |
25 #include <string> | |
26 #include <vector> | |
27 | |
28 #include "retriever.h" | |
29 #include "rule.h" | |
30 #include "ruleset.h" | |
31 #include "util/json.h" | |
32 | |
33 namespace i18n { | |
34 namespace addressinput { | |
35 | |
36 CountryRulesAggregator::CountryRulesAggregator(scoped_ptr<Retriever> retriever) | |
37 : retriever_(retriever.Pass()), | |
38 country_code_(), | |
39 key_(), | |
40 rules_ready_(), | |
41 json_(), | |
42 non_default_languages_() { | |
43 assert(retriever_ != NULL); | |
44 } | |
45 | |
46 CountryRulesAggregator::~CountryRulesAggregator() {} | |
47 | |
48 void CountryRulesAggregator::AggregateRules(const std::string& country_code, | |
49 scoped_ptr<Callback> rules_ready) { | |
50 Reset(); | |
51 country_code_ = country_code; | |
52 rules_ready_ = rules_ready.Pass(); | |
53 | |
54 // Key construction: | |
55 // https://code.google.com/p/libaddressinput/wiki/AddressValidationMetadata | |
56 // Example of a country-level key: "data/CA". | |
57 key_ = "data/" + country_code_; | |
58 retriever_->Retrieve( | |
59 key_, BuildCallback(this, &CountryRulesAggregator::OnDataReady)); | |
60 } | |
61 | |
62 bool CountryRulesAggregator::OnDataReady(bool success, | |
63 const std::string& key, | |
64 const std::string& data) { | |
65 if (key != key_) { | |
66 return true; // An abandoned request. | |
67 } | |
68 | |
69 json_ = Json::Build().Pass(); | |
70 if (!success || !json_->ParseObject(data)) { | |
71 (*rules_ready_)(false, country_code_, scoped_ptr<Ruleset>()); | |
72 Reset(); | |
73 return false; | |
74 } | |
75 | |
76 std::map<std::string, std::string> language_keys; | |
77 scoped_ptr<Ruleset> ruleset = Build(key_, COUNTRY, language_keys); | |
78 const bool parse_success = ruleset != NULL; | |
79 (*rules_ready_)(parse_success, country_code_, ruleset.Pass()); | |
80 Reset(); | |
81 return parse_success; | |
82 } | |
83 | |
84 scoped_ptr<Ruleset> CountryRulesAggregator::Build( | |
85 const std::string& key, | |
86 AddressField field, | |
87 const std::map<std::string, std::string>& language_specific_keys) { | |
88 scoped_ptr<Rule> rule = ParseRule(key, field); | |
89 if (rule == NULL) { | |
90 return scoped_ptr<Ruleset>(); | |
91 } | |
92 | |
93 // Determine the languages that have language-specific rules. For example, | |
94 // the default language in Switzerland is "de", but "fr" and "it" have | |
95 // language specific rules. | |
96 if (field == COUNTRY) { | |
97 non_default_languages_ = rule->GetLanguages(); | |
98 std::vector<std::string>::iterator default_language_it = | |
99 std::find(non_default_languages_.begin(), | |
100 non_default_languages_.end(), | |
101 rule->GetLanguage()); | |
102 if (default_language_it != non_default_languages_.end()) { | |
103 non_default_languages_.erase(default_language_it); | |
104 } | |
105 } | |
106 | |
107 scoped_ptr<Ruleset> ruleset(new Ruleset(field, rule.Pass())); | |
108 std::map<std::string, std::map<std::string, std::string> > | |
109 language_specific_subkeys; | |
110 | |
111 // Parse the language-specific rules. For example, parse the rules for "fr" | |
112 // and "it" languages in Switzerland. | |
113 for (std::vector<std::string>::const_iterator | |
114 non_default_language_it = non_default_languages_.begin(); | |
115 non_default_language_it != non_default_languages_.end(); | |
116 ++non_default_language_it) { | |
117 std::map<std::string, std::string>::const_iterator | |
118 language_specific_key_it = | |
119 language_specific_keys.find(*non_default_language_it); | |
120 std::string language_specific_key = | |
121 language_specific_key_it != language_specific_keys.end() | |
122 ? language_specific_key_it->second | |
123 : key; | |
124 scoped_ptr<Rule> language_specific_rule = | |
125 ParseRule(language_specific_key + "--" + *non_default_language_it, | |
126 field); | |
127 | |
128 if (language_specific_rule == NULL || | |
129 language_specific_rule->GetSubKeys().size() != | |
130 ruleset->rule().GetSubKeys().size()) { | |
131 return scoped_ptr<Ruleset>(); | |
132 } | |
133 | |
134 // Build the language specific subkeys for the next level of | |
135 // rules. Examples: | |
136 // ["data/CA/AB"]["fr"] <- "data/CA/AB" | |
137 // ["data/HK/香港島"]["en"] <- "data/HK/Kowloon" | |
138 for (size_t i = 0; i < ruleset->rule().GetSubKeys().size(); ++i) { | |
139 const std::string& subkey = key + "/" + ruleset->rule().GetSubKeys()[i]; | |
140 const std::string& language_specific_subkey = | |
141 key + "/" + language_specific_rule->GetSubKeys()[i]; | |
142 language_specific_subkeys[subkey][*non_default_language_it] = | |
143 language_specific_subkey; | |
144 } | |
145 | |
146 ruleset->AddLanguageCodeRule( | |
147 *non_default_language_it, language_specific_rule.Pass()); | |
148 } | |
149 | |
150 // Parse the sub-keys recursively. For example, parse the rules for all of the | |
151 // states in US: "CA", "TX", "NY", etc. | |
152 for (std::vector<std::string>::const_iterator | |
153 subkey_it = ruleset->rule().GetSubKeys().begin(); | |
154 subkey_it != ruleset->rule().GetSubKeys().end(); ++subkey_it) { | |
155 std::string subkey = key + "/" + *subkey_it; | |
156 scoped_ptr<Ruleset> sub_ruleset = | |
157 Build(subkey, | |
158 static_cast<AddressField>(field + 1), | |
159 language_specific_subkeys[subkey]); | |
160 | |
161 if (sub_ruleset == NULL) { | |
162 return scoped_ptr<Ruleset>(); | |
163 } | |
164 | |
165 ruleset->AddSubRegionRuleset(*subkey_it, sub_ruleset.Pass()); | |
166 } | |
167 | |
168 return ruleset.Pass(); | |
169 } | |
170 | |
171 scoped_ptr<Rule> CountryRulesAggregator::ParseRule(const std::string& key, | |
172 AddressField field) const { | |
173 scoped_ptr<Json> value; | |
174 if (!json_->GetJsonValueForKey(key, &value) || value == NULL) { | |
175 return scoped_ptr<Rule>(); | |
176 } | |
177 scoped_ptr<Rule> rule(new Rule); | |
178 if (field == COUNTRY) { | |
179 rule->CopyFrom(Rule::GetDefault()); | |
180 } | |
181 rule->ParseJsonRule(*value); | |
182 return rule.Pass(); | |
183 } | |
184 | |
185 void CountryRulesAggregator::Reset() { | |
186 country_code_.clear(); | |
187 key_.clear(); | |
188 rules_ready_.reset(); | |
189 json_.reset(); | |
190 non_default_languages_.clear(); | |
191 } | |
192 | |
193 } // namespace addressinput | |
194 } // namespace i18n | |
OLD | NEW |