Index: chrome/browser/interests/interests_retriever.cc |
diff --git a/chrome/browser/interests/interests_retriever.cc b/chrome/browser/interests/interests_retriever.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..28469d3cc788f3770a9243b66907e25e44931499 |
--- /dev/null |
+++ b/chrome/browser/interests/interests_retriever.cc |
@@ -0,0 +1,194 @@ |
+// Copyright 2015 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/interests/interests_retriever.h" |
+ |
+#include "base/json/json_reader.h" |
+#include "base/logging.h" |
+#include "base/values.h" |
+#include "net/base/load_flags.h" |
+#include "net/http/http_status_code.h" |
+#include "net/url_request/test_url_fetcher_factory.h" |
+#include "net/url_request/url_fetcher.h" |
+#include "net/url_request/url_fetcher_factory.h" |
+#include "net/url_request/url_request_context_getter.h" |
+#include "net/url_request/url_request_status.h" |
+ |
+using net::FakeURLFetcher; |
+using net::URLFetcher; |
+using net::URLFetcherDelegate; |
+using net::URLRequestContextGetter; |
+using net::URLRequestStatus; |
+using net::URLFetcherFactory; |
+ |
+namespace { |
+ |
+const int kNumRetries = 1; |
+const char kIdInterests[] = "interest"; |
+const char kIdInterestName[] = "name"; |
+const char kIdInterestImageUrl[] = "imageUrl"; |
+const char kIdInterestRelevance[] = "relevance"; |
+ |
+const char kInterestsUrl[] = "https://www.googleapis.com/TBD/v1/interests"; |
+ |
+const char kFakeResponse[] = |
+ "{" |
+ " \"interest\": [" |
+ " {\n" |
+ " \"name\": \"Munich\",\n" |
+ " \"imageUrl\": " |
+ "\"https://usercontent.googleapis.com/freebase/v1/image/m/" |
+ "02h6_6p?maxwidth=200&maxheight=200&mode=fill\"," |
+ " \"relevance\": 0.85" |
+ " }," |
+ " {" |
+ " \"name\": \"Defense of the Ancients\"," |
+ " \"imageUrl\": " |
+ "\"https://usercontent.googleapis.com/freebase/v1/image/m/" |
+ "073pnt?maxwidth=200&maxheight=200&mode=fill\"," |
+ " \"relevance\": 1" |
+ " }," |
+ " {" |
+ " \"name\": \"Google\"," |
+ " \"imageUrl\": " |
+ "\"https://usercontent.googleapis.com/freebase/v1/image/m/" |
+ "045c7b?maxwidth=200&maxheight=200&mode=fill&A\"," |
+ " \"relevance\": 0.9" |
+ " }," |
+ " {" |
+ " \"name\": \"Google Chrome\"," |
+ " \"imageUrl\": " |
+ "\"https://usercontent.googleapis.com/freebase/v1/image/m/" |
+ "04j7cyf?maxwidth=200&maxheight=200&mode=fill\"," |
+ " \"relevance\": 0.98" |
+ " }" |
+ " ]" |
+ "}" |
+ ""; |
+ |
+std::vector<InterestsRetriever::Interest> EmptyResponse() { |
+ return std::vector<InterestsRetriever::Interest>(); |
+} |
+ |
+} // namespace |
+ |
+InterestsRetriever::InterestsRetriever( |
+ net::URLRequestContextGetter* url_request_context, |
+ const std::string& access_token, |
+ const InterestsRetriever::InterestsCallback& callback, |
+ URLFetcherFactory* url_fetcher_factory) |
+ : url_request_context_(url_request_context), |
+ access_token_(access_token), |
+ callback_(callback), |
+ url_fetcher_factory_(url_fetcher_factory) { |
+ fetcher_ = CreateFetcher(); |
+ |
+ // Setup fetcher |
+ fetcher_->SetRequestContext(url_request_context_); |
+ fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
+ net::LOAD_DO_NOT_SAVE_COOKIES); |
+ fetcher_->SetAutomaticallyRetryOnNetworkChanges(kNumRetries); |
+ |
+ // Add oauth access token |
+ fetcher_->AddExtraRequestHeader(std::string("Authorization: Bearer ") + |
+ access_token_); |
+ |
+ fetcher_->Start(); |
+} |
+ |
+scoped_ptr<URLFetcher> InterestsRetriever::CreateFetcher() { |
+ // The API is not accessible yet. So a static response is provided. |
+ // return URLFetcher::Create(GURL(kInterestsUrl), URLFetcher::GET, this); |
+ |
+ // If a Factory is provided use it. This is needed for testing. |
+ if (url_fetcher_factory_) { |
+ return url_fetcher_factory_->CreateURLFetcher(0, GURL(kInterestsUrl), |
+ URLFetcher::GET, this); |
+ |
+ } else { |
+ return scoped_ptr<URLFetcher>(new FakeURLFetcher( |
+ GURL(kInterestsUrl), this, std::string(kFakeResponse), net::HTTP_OK, |
+ net::URLRequestStatus::SUCCESS)); |
+ } |
+} |
+ |
+void InterestsRetriever::OnURLFetchComplete(const net::URLFetcher* source) { |
+ const URLRequestStatus& status = source->GetStatus(); |
+ |
+ if (!status.is_success()) { |
+ DLOG(WARNING) << "URL request failed!"; |
+ callback_.Run(EmptyResponse()); |
+ return; |
+ } |
+ |
+ std::string response_body; |
+ source->GetResponseAsString(&response_body); |
+ |
+ callback_.Run(ExtractInterests(response_body)); |
+} |
+ |
+std::vector<InterestsRetriever::Interest> InterestsRetriever::ExtractInterests( |
+ const std::string& response) { |
+ scoped_ptr<base::Value> value = base::JSONReader::Read(response); |
+ |
+ const base::DictionaryValue* dict = nullptr; |
+ if (!value || !value->GetAsDictionary(&dict)) { |
+ DLOG(WARNING) << "ExtractInterests failed to parse global dictionary"; |
+ return EmptyResponse(); |
+ } |
+ |
+ const base::ListValue* interests_list = nullptr; |
+ std::vector<Interest> res; |
+ |
+ if (!dict->GetList(kIdInterests, &interests_list)) { |
+ DLOG(WARNING) << "ExtractInterests failed to parse interests list"; |
+ return EmptyResponse(); |
+ } |
+ |
+ for (const base::Value* entry : *interests_list) { |
+ const base::DictionaryValue* interest_dict = nullptr; |
+ if (!entry->GetAsDictionary(&interest_dict)) { |
+ DLOG(WARNING) << "ExtractInterests failed to parse interest dictionary"; |
+ return EmptyResponse(); |
+ } |
+ |
+ std::string name; |
+ if (!interest_dict->GetString(kIdInterestName, &name)) { |
+ DLOG(WARNING) << "ExtractInterests failed to parse interest name"; |
+ return EmptyResponse(); |
+ } |
+ |
+ std::string image_url; |
+ if (!interest_dict->GetString(kIdInterestImageUrl, &image_url)) { |
+ // image_url is allowed to be missing. |
+ // |
+ // However this is still logged as a warning, since, currently, the server |
+ // should always provide an image_url. |
+ // |
+ DLOG(WARNING) << "ExtractInterests failed to parse interest image URL"; |
+ } |
+ |
+ double relevance; |
+ if (!interest_dict->GetDouble(kIdInterestRelevance, &relevance)) { |
+ DLOG(WARNING) << "ExtractInterests failed to parse interest relevance"; |
+ return EmptyResponse(); |
+ } |
+ |
+ res.push_back(Interest{name, image_url, relevance}); |
+ } |
+ |
+ return res; |
+} |
+ |
+InterestsRetriever::~InterestsRetriever() {} |
+ |
+std::string InterestsRetriever::GetFakeResponseForTest() { |
+ return kFakeResponse; |
+} |
+ |
+bool InterestsRetriever::Interest::operator==(const Interest& interest) const { |
+ return name == interest.name && |
+ image_url == interest.image_url && |
+ relevance == interest.relevance; |
+} |