Integrating Online Wallet into Chrome.

chrome/browser/autofill/wallet/wallet_client.cc

+// Copyright (c) 2012 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/autofill/wallet/wallet_client.h"
+
+#include "base/json/json_reader.h"
+#include "base/json/json_writer.h"
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/string_number_conversions.h"
+#include "base/string_split.h"
+#include "base/stringprintf.h"
+#include "base/values.h"
+#include "chrome/browser/autofill/wallet/cart.h"
+#include "chrome/browser/autofill/wallet/full_wallet.h"
+#include "chrome/browser/autofill/wallet/wallet_address.h"
+#include "chrome/browser/autofill/wallet/wallet_items.h"
+#include "chrome/browser/autofill/wallet/wallet_service_url.h"
+#include "googleurl/src/gurl.h"
+#include "net/http/http_status_code.h"
+#include "net/url_request/url_fetcher.h"
+#include "net/url_request/url_fetcher_delegate.h"
+#include "net/url_request/url_request_context_getter.h"
+
+namespace {
+
+const char kEncryptOtpBodyFormat[] = "cvv=%s:%s";
+const char kJsonMimeType[] = "application/json";
+const char kApplicationMimeType[] = "application/x-www-form-urlencoded";
+const size_t kMaxBits = 63;
+
+}  // end anonymous namespace
+
+namespace wallet {
+
+class WalletClient::Core

[Evan Stade, 2012/12/07 19:57:44]

class level comment.

[ahutter, 2012/12/15 01:06:31]

Done.

+    : public base::RefCountedThreadSafe<WalletClient::Core>,
+      public net::URLFetcherDelegate {
+ public:
+  explicit Core(net::URLRequestContextGetter* context_getter)

[Evan Stade, 2012/12/07 19:57:44]

i believe this should take a scoped_refptr

[ahutter, 2012/12/15 01:06:31]

I'm not seeing any other users of RequestContextGetter that do so.

+      : context_getter_(context_getter),
+        observer_(NULL),
+        request_type_(NO_PENDING_REQUEST),
+        num_retries_(0) {}
+  void AcceptLegalDocuments(const std::vector<std::string>& document_ids,
+                            const std::string& google_transaction_id,
+                            WalletClientObserver* observer);
+  void EncryptOtp(const void* otp,
+                  size_t length,
+                  WalletClientObserver* observer);
+  void GetFullWallet(const std::string& instrument_id,
+                     const std::string& address_id,
+                     const std::string& merchant_domain,
+                     const Cart& cart,
+                     const std::string& google_transaction_id,
+                     const std::string& encrypted_otp,
+                     const std::string& session_material,
+                     WalletClientObserver* observer);
+  void GetWalletItems(WalletClientObserver* observer);
+  void SendExtendedAutofillStatus(bool success,
+                                  const std::string& merchant_domain,
+                                  const std::string& reason,
+                                  const std::string& google_transaction_id,
+                                  WalletClientObserver* observer);
+  virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;
+ private:
+  friend class base::RefCountedThreadSafe<Core>;
+  // TODO(ahutter): Implement this.
+  std::string GetRiskParams() { return ""; }
+
+  enum RequestType {
+    NO_PENDING_REQUEST,
+    ACCEPT_LEGAL_DOCUMENTS,
+    ENCRYPT_OTP,
+    GET_FULL_WALLET,
+    GET_WALLET_ITEMS,
+    SEND_STATUS,
+  };
+
+  virtual ~Core() {}
+
+  void MakeWalletRequest(const GURL url,

[Ilya Sherman, 2012/12/14 04:56:43]

nit: Pass by const-reference.

[ahutter, 2012/12/15 01:06:31]

Done.

+                         const std::string& post_body,
+                         WalletClient::WalletClientObserver* observer,
+                         const std::string& content_type);
+
+  void HandleResponse(const net::URLFetcher* source,
+                      bool* should_retry_request);
+
+  void MaybeRetry(scoped_ptr<net::URLFetcher>* old_request,
+                  const net::URLFetcher* source,
+                  bool* should_retry_request);
+
+  scoped_refptr<net::URLRequestContextGetter> context_getter_;

[Evan Stade, 2012/12/07 19:57:44]

docs for all these

[ahutter, 2012/12/15 01:06:31]

Done.

+  WalletClient::WalletClientObserver* observer_;
+  scoped_ptr<net::URLFetcher> request_;
+  RequestType request_type_;
+  int num_retries_;
+  DISALLOW_COPY_AND_ASSIGN(Core);
+};
+
+void WalletClient::Core::AcceptLegalDocuments(
+    const std::vector<std::string>& document_ids,
+    const std::string& google_transaction_id,
+    WalletClient::WalletClientObserver* observer) {
+  DCHECK_EQ(request_type_, NO_PENDING_REQUEST);
+  request_type_ = ACCEPT_LEGAL_DOCUMENTS;
+  GURL url = GetAcceptLegalDocumentsUrl();
+  DictionaryValue request_dict;
+  request_dict.SetString("api_key", wallet::kApiKey);
+  request_dict.SetString("google_transaction_id", google_transaction_id);
+  ListValue* docs_list = new ListValue();
+  for (std::vector<std::string>::const_iterator it = document_ids.begin();
+       it != document_ids.end();
+       ++it) {
+    docs_list->AppendString(*it);
+  }
+  request_dict.Set("accepted_legal_document", docs_list);
+
+  std::string post_body;
+  base::JSONWriter::Write(&request_dict, &post_body);
+  MakeWalletRequest(url, post_body, observer, kJsonMimeType);
+}
+
+void WalletClient::Core::EncryptOtp(
+    const void* otp,
+    size_t length,
+    WalletClient::WalletClientObserver* observer) {
+  DCHECK_EQ(request_type_, NO_PENDING_REQUEST);
+  request_type_ = ENCRYPT_OTP;
+  GURL url = GetSecureUrl();
+  size_t num_bits = length * 8;
+  DCHECK_LT(num_bits, kMaxBits);
+  std::string post_body = StringPrintf(kEncryptOtpBodyFormat,
+                                       base::HexEncode(&num_bits, 1).c_str(),
+                                       base::HexEncode(otp, length).c_str());
+  MakeWalletRequest(url, post_body, observer, kApplicationMimeType);
+}
+
+void WalletClient::Core::GetFullWallet(
+    const std::string& instrument_id,
+    const std::string& address_id,
+    const std::string& merchant_domain,
+    const Cart& cart,
+    const std::string& google_transaction_id,
+    const std::string& encrypted_otp,
+    const std::string& session_material,
+    WalletClient::WalletClientObserver* observer) {
+  DCHECK_EQ(request_type_, NO_PENDING_REQUEST);
+  request_type_ = GET_FULL_WALLET;
+  GURL url = GetGetFullWalletUrl();
+  DictionaryValue request_dict;
+  request_dict.SetString("api_key", wallet::kApiKey);
+  request_dict.SetString("risk_params", GetRiskParams());
+  request_dict.SetString("selected_instrument_id", instrument_id);
+  request_dict.SetString("selected_address_id", address_id);
+  request_dict.SetString("merchant_domain", merchant_domain);
+  request_dict.SetString("google_transaction_id", google_transaction_id);
+  request_dict.Set("cart", cart.ToDictionary().release());
+  request_dict.SetString("encrypted_otp", encrypted_otp);
+  request_dict.SetString("session_material", session_material);
+  std::string post_body;
+  base::JSONWriter::Write(&request_dict, &post_body);
+  MakeWalletRequest(url, post_body, observer, kJsonMimeType);
+}
+
+void WalletClient::Core::GetWalletItems(
+    WalletClient::WalletClientObserver* observer) {

[Evan Stade, 2012/12/07 19:57:44]

more vertical whitespace in this fn, and pretty much every fn in this file

[ahutter, 2012/12/15 01:06:31]

Done.

+  DCHECK_EQ(request_type_, NO_PENDING_REQUEST);
+  request_type_ = GET_WALLET_ITEMS;
+  GURL url = GetGetWalletItemsUrl();
+  DictionaryValue request_dict;
+  request_dict.SetString("api_key", wallet::kApiKey);
+  request_dict.SetString("risk_params", GetRiskParams());
+  std::string post_body;
+  base::JSONWriter::Write(&request_dict, &post_body);
+  MakeWalletRequest(url, post_body, observer, kJsonMimeType);
+}
+
+void WalletClient::Core::SendExtendedAutofillStatus(
+    bool success,
+    const std::string& merchant_domain,
+    const std::string& reason,
+    const std::string& google_transaction_id,
+    WalletClient::WalletClientObserver* observer) {
+  DCHECK_EQ(request_type_, NO_PENDING_REQUEST);
+  request_type_ = SEND_STATUS;
+  GURL url = GetSendStatusUrl();
+  DictionaryValue request_dict;
+  request_dict.SetString("api_key", wallet::kApiKey);
+  request_dict.SetBoolean("success", success);
+  request_dict.SetString("hostname", merchant_domain);
+  if (!success) {
+    // TODO(ahutter): Probably want to do some checks on reason.
+    request_dict.SetString("reason", reason);
+  }
+  request_dict.SetString("google_transaction_id", google_transaction_id);
+  std::string post_body;
+  base::JSONWriter::Write(&request_dict, &post_body);
+  MakeWalletRequest(url, post_body, observer, kJsonMimeType);
+}
+
+void WalletClient::Core::MakeWalletRequest(
+    GURL url,
+    const std::string& post_body,
+    WalletClient::WalletClientObserver* observer,
+    const std::string& content_type) {
+  DCHECK(!request_.get()) << "Tried to fetch two things at once!";
+  DCHECK(observer);
+  num_retries_ = 0;
+  observer_ = observer;
+  request_.reset(net::URLFetcher::Create(
+      0, url, net::URLFetcher::POST, this));
+  request_->SetRequestContext(context_getter_);
+  DVLOG(1) << "Request body: " << post_body;
+  request_->SetUploadData(content_type, post_body);
+  request_->SetMaxRetries(3);
+  request_->Start();
+}
+
+
+void WalletClient::Core::OnURLFetchComplete(
+    const net::URLFetcher* source) {
+  bool should_retry;
+  HandleResponse(source, &should_retry);
+  if (should_retry) {
+    // Explicitly call ReceivedContentWasMalformed() to ensure the current
+    // request gets counted as a failure for calculation of the back-off
+    // period.  If it was already a failure by status code, this call will
+    // be ignored.
+    request_->ReceivedContentWasMalformed();
+    num_retries_++;
+    // Set context_getter_ again because
+    // URLFetcher::Core::RetryOrCompleteUrlFetch resets it to NULL.
+    request_->SetRequestContext(context_getter_);
+    request_->Start();
+  }
+}
+
+void WalletClient::Core::MaybeRetry(
+    scoped_ptr<net::URLFetcher>* old_request,
+    const net::URLFetcher* source,
+    bool* should_retry_request) {
+  if (source->GetMaxRetries() != -1 &&
+      num_retries_ >= source->GetMaxRetries()) {
+    request_type_ = NO_PENDING_REQUEST;
+    observer_->OnNetworkError(source->GetResponseCode());
+  } else {
+    request_ = old_request->Pass();
+    *should_retry_request = true;
+  }
+}
+
+void WalletClient::Core::HandleResponse(
+    const net::URLFetcher* source,
+    bool* should_retry_request) {
+  // Keep the URLFetcher object in case we need to reuse it.
+  scoped_ptr<net::URLFetcher> old_request = request_.Pass();
+  DCHECK_EQ(source, old_request.get());
+  *should_retry_request = false;
+
+  DVLOG(1) << "Got response from " << source->GetOriginalURL();
+
+  int response_code = source->GetResponseCode();
+  std::string data;
+  scoped_ptr<DictionaryValue> response_dict;
+  switch (response_code) {
+    // HTTP_BAD_REQUEST means the arguments are invalid. No point retrying.
+    case net::HTTP_BAD_REQUEST: {
+      request_type_ = NO_PENDING_REQUEST;
+      observer_->OnWalletError();
+      return;
+    }
+    // HTTP_OK holds a valid response and HTTP_INTERNAL_SERVER_ERROR holds an
+    // error code and message for the user.
+    case net::HTTP_OK:
+    case net::HTTP_INTERNAL_SERVER_ERROR: {
+      source->GetResponseAsString(&data);
+      DVLOG(1) << "Response body: " << data;
+      scoped_ptr<Value> message_value(base::JSONReader::Read(data));
+      if (message_value.get() &&
+          message_value->IsType(Value::TYPE_DICTIONARY)) {
+        response_dict.reset(
+            static_cast<DictionaryValue*>(message_value.release()));
+      }
+      if (response_code == net::HTTP_INTERNAL_SERVER_ERROR) {
+        request_type_ = NO_PENDING_REQUEST;
+        // TODO(ahutter): Do something with the response. See
+        // http://crbug.com/164410.
+        observer_->OnWalletError();
+        return;
+      }
+      break;
+    }
+    // Anything else is an error.
+    default: {
+      MaybeRetry(&old_request, source, should_retry_request);
+      return;
+    }
+  }
+
+  RequestType type = request_type_;
+  request_type_ = NO_PENDING_REQUEST;
+
+  switch (type) {
+    case ACCEPT_LEGAL_DOCUMENTS: {
+      observer_->OnAcceptLegalDocuments();
+      break;
+    }
+    case SEND_STATUS: {
+      observer_->OnSendExtendedAutofillStatus();
+      break;
+    }
+    case ENCRYPT_OTP: {
+      if (!data.empty()) {
+        std::vector<std::string> splits;
+        base::SplitString(data, '|', &splits);
+        if (splits.size() == 2)
+          observer_->OnEncryptOtp(splits[1], splits[0]);
+        else
+          observer_->OnNetworkError(response_code);
+      } else {
+        MaybeRetry(&old_request, source, should_retry_request);
+      }
+      break;
+    }
+    case GET_FULL_WALLET: {
+      if (response_dict.get()) {
+        scoped_ptr<FullWallet> full_wallet(
+            FullWallet::CreateFullWallet(*response_dict));
+        if (full_wallet.get())
+          observer_->OnGetFullWallet(full_wallet.release());
+        else
+          observer_->OnNetworkError(response_code);
+      } else {
+        MaybeRetry(&old_request, source, should_retry_request);
+      }
+      break;
+    }
+    case GET_WALLET_ITEMS: {
+      if (response_dict.get()) {
+        scoped_ptr<WalletItems> wallet_items(
+            WalletItems::CreateWalletItems(*response_dict));
+        if (wallet_items.get())
+          observer_->OnGetWalletItems(wallet_items.release());
+        else
+          observer_->OnNetworkError(response_code);
+      } else {
+        MaybeRetry(&old_request, source, should_retry_request);
+      }
+      break;
+    }
+    default: {
+      NOTREACHED();
+    }
+  }
+}
+
+WalletClient::WalletClient(
+    net::URLRequestContextGetter* context_getter) {
+  DCHECK(context_getter);
+  core_ = new Core(context_getter);
+}
+
+WalletClient::~WalletClient() {}
+
+void WalletClient::GetWalletItems(
+    WalletClient::WalletClientObserver* observer) {
+  core_->GetWalletItems(observer);
+}
+
+void WalletClient::AcceptLegalDocuments(
+    const std::vector<std::string>& document_ids,
+    const std::string& google_transaction_id,
+    WalletClient::WalletClientObserver* observer) {
+  core_->AcceptLegalDocuments(document_ids,
+                              google_transaction_id,
+                              observer);
+}
+
+void WalletClient::EncryptOtp(
+    const void* otp,
+    size_t length,
+    WalletClient::WalletClientObserver* observer) {
+  core_->EncryptOtp(otp, length, observer);
+}
+
+void WalletClient::GetFullWallet(
+    const std::string& instrument_id,
+    const std::string& address_id,
+    const std::string& merchant_domain,
+    const Cart& cart,
+    const std::string& google_transaction_id,
+    const std::string& encrypted_otp,
+    const std::string& session_material,
+    WalletClient::WalletClientObserver* observer) {
+  core_->GetFullWallet(instrument_id,
+                       address_id,
+                       merchant_domain,
+                       cart,
+                       google_transaction_id,
+                       encrypted_otp,
+                       session_material,
+                       observer);
+}
+
+void WalletClient::SendExtendedAutofillStatus(
+    bool success,
+    const std::string& merchant_domain,
+    const std::string& reason,
+    const std::string& google_transaction_id,
+    WalletClient::WalletClientObserver* observer) {
+  core_->SendExtendedAutofillStatus(success,
+                                    merchant_domain,
+                                    reason,
+                                    google_transaction_id,
+                                    observer);
+}
+
+}  // end wallet namespace
+