Index: components/omnibox/suggestion_answer.cc |
diff --git a/components/omnibox/suggestion_answer.cc b/components/omnibox/suggestion_answer.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..3d545e166f872e49fa9b535a73885d943ce36a79 |
--- /dev/null |
+++ b/components/omnibox/suggestion_answer.cc |
@@ -0,0 +1,209 @@ |
+// Copyright 2014 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 "components/omnibox/suggestion_answer.h" |
+ |
+#include "base/json/json_reader.h" |
+#include "base/strings/string_number_conversions.h" |
+#include "base/strings/string_util.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "base/values.h" |
+ |
+namespace { |
+ |
+static const char* kAnswerJsonLines = "l"; |
+static const char* kAnswerJsonImageLine = "il"; |
+static const char* kAnswerJsonText = "t"; |
+static const char* kAnswerJsonAdditionalText = "at"; |
+static const char* kAnswerJsonStatusText = "st"; |
+static const char* kAnswerJsonTextType = "tt"; |
+static const char* kAnswerJsonImage = "i"; |
+static const char* kAnswerJsonImageData = "i.d"; |
Peter Kasting
2014/10/24 22:38:10
Nit: I tend to interpret the style guide's "Place
Justin Donnelly
2014/10/27 21:41:15
Makes sense, but these are a bit of a special case
Peter Kasting
2014/10/27 21:50:09
I don't care much. If you think there's a win in
Justin Donnelly
2014/10/28 00:39:34
Done.
|
+ |
+} |
+ |
+// SuggestionAnswer::TextField ------------------------------------------------- |
+ |
+SuggestionAnswer::TextField::TextField() : type_(-1) {} |
+SuggestionAnswer::TextField::~TextField() {} |
+ |
+// static |
+void SuggestionAnswer::TextField::ParseTextField( |
Peter Kasting
2014/10/24 22:38:10
This function returns void and the caller checks v
Justin Donnelly
2014/10/27 21:41:15
The issue here is that the is_valid() functions ar
Peter Kasting
2014/10/27 21:50:09
Would passing by pointer muck things up too much?
Justin Donnelly
2014/10/28 00:39:34
Yeah, it creates a lot of slightly more complicate
Peter Kasting
2014/10/29 17:56:17
OK, I read through patch set 1 and looked at the d
|
+ const base::DictionaryValue* field_json, TextField* text_field) { |
+ if (!field_json->GetString(kAnswerJsonText, &text_field->text_) || |
+ text_field->text_.empty()) |
+ return; |
+ field_json->GetInteger(kAnswerJsonTextType, &text_field->type_); |
+} |
+ |
+bool SuggestionAnswer::TextField::operator==(const TextField& field) const { |
+ return type_ == field.type_ && text_ == field.text_; |
+} |
+ |
+bool SuggestionAnswer::TextField::operator!=(const TextField& field) const { |
+ return !operator==(field); |
+} |
+ |
+void SuggestionAnswer::TextField::Clear() { |
+ text_.clear(); |
+ type_ = -1; |
+} |
+ |
+// SuggestionAnswer::ImageLine ------------------------------------------------- |
+ |
+SuggestionAnswer::ImageLine::ImageLine() : is_valid_(false) {} |
+SuggestionAnswer::ImageLine::~ImageLine() {} |
+ |
+// static |
+void SuggestionAnswer::ImageLine::ParseImageLine( |
+ const base::DictionaryValue* line_json, ImageLine* image_line) { |
+ image_line->is_valid_ = false; |
+ |
+ const base::DictionaryValue* inner_json; |
+ if (!line_json->GetDictionary(kAnswerJsonImageLine, &inner_json)) |
+ return; |
+ |
+ const base::ListValue* fields_json; |
+ if (!inner_json->GetList(kAnswerJsonText, &fields_json) || |
+ fields_json->GetSize() == 0) |
+ return; |
+ |
+ for (size_t i = 0; i < fields_json->GetSize(); ++i) { |
+ const base::DictionaryValue* field_json; |
+ TextField text_field; |
+ if (!fields_json->GetDictionary(i, &field_json)) |
+ return; |
+ TextField::ParseTextField(field_json, &text_field); |
+ if (!text_field.is_valid()) |
+ return; |
+ image_line->text_fields_.push_back(text_field); |
+ } |
+ |
+ if (inner_json->HasKey(kAnswerJsonAdditionalText)) { |
+ const base::DictionaryValue* field_json; |
+ if (!inner_json->GetDictionary(kAnswerJsonAdditionalText, &field_json)) |
+ return; |
+ TextField::ParseTextField(field_json, &image_line->additional_text_); |
+ if (!image_line->additional_text_.is_valid()) |
+ return; |
+ } |
+ |
+ if (inner_json->HasKey(kAnswerJsonStatusText)) { |
+ const base::DictionaryValue* field_json; |
+ if (!inner_json->GetDictionary(kAnswerJsonStatusText, &field_json)) |
+ return; |
+ TextField::ParseTextField(field_json, &image_line->status_text_); |
+ if (!image_line->status_text_.is_valid()) |
+ return; |
+ } |
+ |
+ if (inner_json->HasKey(kAnswerJsonImage)) { |
+ base::string16 url_string; |
+ if (!inner_json->GetString(kAnswerJsonImageData, &url_string) || |
+ url_string.empty()) |
+ return; |
+ // If necessary, concatenate scheme and host/path using only ':' as |
+ // separator. This is due to the results delivering strings of the form |
+ // "//host/path", which is web-speak for "use the enclosing page's scheme", |
+ // but not a valid path of an URL. |
+ if (StartsWith(url_string, base::ASCIIToUTF16("//"), false)) |
+ image_line->image_url_ = GURL(base::ASCIIToUTF16("https:") + url_string); |
Peter Kasting
2014/10/24 22:38:10
Nit: Maybe we should use kHttpsScheme from url/url
Justin Donnelly
2014/10/27 21:41:15
Done.
|
+ else |
+ image_line->image_url_ = GURL(url_string); |
Peter Kasting
2014/10/24 22:38:10
Nit: Shorter:
image_line->image_url_ = GURL(
Justin Donnelly
2014/10/27 21:41:15
Done.
|
+ |
+ if (!image_line->image_url_.is_valid()) |
+ return; |
+ } |
+ |
+ image_line->is_valid_ = true; |
+} |
+ |
+bool SuggestionAnswer::ImageLine::operator==(const ImageLine& line) const { |
+ return text_fields_ == line.text_fields_ && |
+ additional_text_ == line.additional_text_ && |
+ status_text_ == line.status_text_ && |
+ image_url_ == line.image_url_; |
+} |
+ |
+bool SuggestionAnswer::ImageLine::operator!=(const ImageLine& line) const { |
+ return !operator==(line); |
+} |
+ |
+void SuggestionAnswer::ImageLine::Clear() { |
+ text_fields_.clear(); |
+ additional_text_.Clear(); |
+ status_text_.Clear(); |
+ image_url_ = GURL(); |
+ is_valid_ = false; |
+} |
+ |
+// SuggestionAnswer ------------------------------------------------------------ |
+ |
+SuggestionAnswer::SuggestionAnswer() : type_(-1) {} |
+SuggestionAnswer::~SuggestionAnswer() {} |
+ |
+// static |
+void SuggestionAnswer::ParseAnswer( |
+ const std::string& answer_json, SuggestionAnswer* answer) { |
+ scoped_ptr<base::Value> value(base::JSONReader::Read(answer_json)); |
+ base::DictionaryValue* dict; |
+ if (!value.get() || !value->GetAsDictionary(&dict)) |
+ return; |
+ |
+ ParseAnswer(dict, answer); |
+} |
+ |
+// static |
+void SuggestionAnswer::ParseAnswer( |
+ const base::DictionaryValue* answer_json, SuggestionAnswer* answer) { |
+ const base::ListValue* lines_json; |
+ if (!answer_json->GetList(kAnswerJsonLines, &lines_json) || |
+ lines_json->GetSize() != 2) |
+ return; |
+ |
+ const base::DictionaryValue* first_line_json; |
+ if (!lines_json->GetDictionary(0, &first_line_json)) |
+ return; |
+ ImageLine::ParseImageLine(first_line_json, &answer->first_line_); |
+ if (!answer->first_line_.is_valid()) |
+ return; |
+ |
+ const base::DictionaryValue* second_line_json; |
+ if (!lines_json->GetDictionary(1, &second_line_json)) |
+ return; |
+ ImageLine::ParseImageLine(second_line_json, &answer->second_line_); |
+} |
+ |
+bool SuggestionAnswer::operator==(const SuggestionAnswer& answer) const { |
+ return type_ == answer.type_ && |
+ first_line_ == answer.first_line_ && |
+ second_line_ == answer.second_line_; |
+} |
+ |
+bool SuggestionAnswer::operator!=(const SuggestionAnswer& answer) const { |
+ return !operator==(answer); |
+} |
+ |
+void SuggestionAnswer::SetType(const base::string16& type) { |
+ if (!base::StringToInt(type, &type_)) |
+ type_ = -1; |
+} |
+ |
+void SuggestionAnswer::Clear() { |
+ first_line_.Clear(); |
+ second_line_.Clear(); |
+ type_ = -1; |
+} |
+ |
+void SuggestionAnswer::GetImageURLs(std::vector<GURL>* urls) const { |
+ urls->clear(); |
+ |
+ if(!is_valid()) |
+ return; |
+ |
+ if (first_line_.has_image_url()) |
+ urls->push_back(first_line_.image_url()); |
+ if (second_line_.has_image_url()) |
+ urls->push_back(second_line_.image_url()); |
+} |