[Offline Prefetech] Send GetOperationReqest to the server

components/offline_pages/core/prefetch/prefetch_request_fetcher.cc

 // Copyright 2017 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/offline_pages/core/prefetch/prefetch_request_fetcher.h"
 
 #include "base/logging.h"
 #include "net/base/load_flags.h"
+#include "net/http/http_request_headers.h"
 #include "net/http/http_status_code.h"
 #include "net/traffic_annotation/network_traffic_annotation.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_status.h"
 #include "url/gurl.h"
 
 namespace offline_pages {
 
 namespace {
+
+const char kPrefetchServer[] =
+    "http://staging-offlinepages-pa.sandbox.googleapis.com/";
+
+// Content type needed in order to communicate with the server in binary
+// proto format.
 const char kRequestContentType[] = "application/x-protobuf";
+
 }  // namespace
 
+// static
+std::unique_ptr<PrefetchRequestFetcher> PrefetchRequestFetcher::CreateForGet(
+    const std::string& url_path,
+    net::URLRequestContextGetter* request_context_getter,
+    const FinishedCallback& callback) {
+  return std::unique_ptr<PrefetchRequestFetcher>(new PrefetchRequestFetcher(

[dewittj, 2017/05/16 23:06:22]

if you want: base::WrapUnique(...) is shorter

[jianli, 2017/05/16 23:52:33]

Done.

+      url_path, std::string(), request_context_getter, callback));
+}
+
+// static
+std::unique_ptr<PrefetchRequestFetcher> PrefetchRequestFetcher::CreateForPost(
+    const std::string& url_path,
+    const std::string& message,
+    net::URLRequestContextGetter* request_context_getter,
+    const FinishedCallback& callback) {
+  return std::unique_ptr<PrefetchRequestFetcher>(new PrefetchRequestFetcher(
+      url_path, message, request_context_getter, callback));
+}
+
 PrefetchRequestFetcher::PrefetchRequestFetcher(
-    const GURL& url,
+    const std::string& url_path,
     const std::string& message,
     net::URLRequestContextGetter* request_context_getter,
     const FinishedCallback& callback)
     : request_context_getter_(request_context_getter), callback_(callback) {
   net::NetworkTrafficAnnotationTag traffic_annotation =
       net::DefineNetworkTrafficAnnotation("offline_prefetch", R"(
         semantics {
           sender: "Offline Prefetch"
           description:
             "Chromium interacts with Offline Page Service to prefetch "
             "suggested website resources."
           trigger:
             "When there are suggested website resources to fetch."
           data:
             "URLs of the suggested website resources to fetch."
           destination: GOOGLE_OWNED_SERVICE
         }
         policy {
           cookies_allowed: false
           setting:
             "Users can enable or disable the offline prefetch by toggling"
             "chrome://flags#offline-prefetch in Chromium on Android."
           policy_exception_justification:
             "Not implemented, considered not useful."
         })");
-  url_fetcher_ = net::URLFetcher::Create(url, net::URLFetcher::POST, this,
-                                         traffic_annotation);
+  url_fetcher_ = net::URLFetcher::Create(
+      GURL(kPrefetchServer + url_path),
+      message.empty() ? net::URLFetcher::GET : net::URLFetcher::POST, this,
+      traffic_annotation);
   url_fetcher_->SetRequestContext(request_context_getter_.get());
   url_fetcher_->SetAutomaticallyRetryOn5xx(false);
   url_fetcher_->SetAutomaticallyRetryOnNetworkChanges(0);
-  url_fetcher_->SetUploadData(kRequestContentType, message);
+  if (message.empty()) {
+    std::string extra_header(net::HttpRequestHeaders::kContentType);
+    extra_header += ": ";
+    extra_header += kRequestContentType;
+    url_fetcher_->AddExtraRequestHeader(extra_header);
+  } else {
+    url_fetcher_->SetUploadData(kRequestContentType, message);
+  }
   url_fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
                              net::LOAD_DO_NOT_SAVE_COOKIES);
   url_fetcher_->Start();
 }
 
 PrefetchRequestFetcher::~PrefetchRequestFetcher() {}
 
 void PrefetchRequestFetcher::OnURLFetchComplete(const net::URLFetcher* source) {
   std::string data;
   PrefetchRequestStatus status = ParseResponse(source, &data);
@@ skipping 25 matching lines @@
 
   if (!source->GetResponseAsString(data) || data->empty()) {
     DVLOG(1) << "Failed to get response or empty response";
     return PrefetchRequestStatus::SHOULD_RETRY_WITH_BACKOFF;
   }
 
   return PrefetchRequestStatus::SUCCESS;
 }
 
 }  // offline_pages