Download autocheckout whitelist and enable autocheckout for whitelisted sites only.

chrome/browser/autofill/autocheckout/whitelist_manager.cc

+// Copyright (c) 2013 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/autocheckout/whitelist_manager.h"
+
+#include "base/command_line.h"
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/string_split.h"
+#include "base/supports_user_data.h"
+#include "chrome/browser/autofill/autocheckout/whitelist_url.h"
+#include "chrome/common/chrome_switches.h"
+#include "content/public/browser/browser_context.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_request_context_getter.h"
+
+namespace {
+
+// Back off in seconds after each whitelist download is attempted.
+const int kDownloadIntervalSeconds = 86400;  // 1 day
+
+// The delay in seconds after startup before download whitelist. This helps
+// to reduce contention at startup time.
+const int kInitialDownloadDelaySeconds = 3;
+
+const char kWhiteListKeyName[] = "autocheckout_whitelist_manager";
+
+} //  namespace
+
+
+namespace autocheckout {
+
+// static
+WhitelistManager* WhitelistManager::GetForBrowserContext(
+    content::BrowserContext* context) {
+  DCHECK(context);
+  WhitelistManager* wm = static_cast<WhitelistManager*>(
+      context->GetUserData(kWhiteListKeyName));
+  if (!wm) {
+    wm = new WhitelistManager(context->GetRequestContext());
+    wm->ScheduleDownload(kInitialDownloadDelaySeconds);
+    context->SetUserData(kWhiteListKeyName, wm);
+  }
+  return wm;
+}
+
+WhitelistManager::WhitelistManager(
+    net::URLRequestContextGetter* context_getter)
+    : context_getter_(context_getter),
+      callback_pending_(false),
+      experimental_form_filling_enabled_(
+          CommandLine::ForCurrentProcess()->HasSwitch(
+              switches::kEnableExperimentalFormFilling)) {
+  DCHECK(context_getter);
+}
+
+bool WhitelistManager::ScheduleDownload(int interval_seconds) {
+  if (!experimental_form_filling_enabled_) {
+    // The feature is not enabled: do not do the request.
+    return false;
+  }
+  if (download_timer_.IsRunning() || callback_pending_) {
+    // A download activity is already scheduled or happening.
+    DVLOG(1) << "Autocheckout DownloadWhitelist scheduler is already running.";
+    return false;
+  }
+
+  download_timer_.Start(FROM_HERE,
+                        base::TimeDelta::FromSeconds(interval_seconds),
+                        this,
+                        &WhitelistManager::TriggerDownload);
+  DVLOG(1) << "Autocheckout DownloadWhitelist was scheduled for "
+           << interval_seconds << " seconds.";
+  return true;
+}
+
+void WhitelistManager::TriggerDownload() {
+  callback_pending_ = true;
+  DVLOG(1) << "Autocheckout DownloadWhitelist...";
+
+  request_.reset(net::URLFetcher::Create(
+      0, GetAutocheckoutWhitelistUrl(), net::URLFetcher::GET, this));
+  request_->SetRequestContext(context_getter_);
+  request_->Start();
+  return;
+}
+
+void WhitelistManager::OnURLFetchComplete(
+    const net::URLFetcher* source) {
+  DCHECK(callback_pending_);
+  callback_pending_ = false;
+  scoped_ptr<net::URLFetcher> old_request = request_.Pass();
+  DCHECK_EQ(source, old_request.get());
+
+  DVLOG(1) << "Autocheckout got response from " << source->GetOriginalURL()
+           << ". Response code: " << source->GetResponseCode();
+
+  if (source->GetResponseCode() != net::HTTP_OK)
+    return;
+
+  std::string data;
+  source->GetResponseAsString(&data);
+  DVLOG(1) << "Autocheckout whitelist response data: " << data;
+  BuildWhitelist(data);
+
+  ScheduleDownload(kDownloadIntervalSeconds);
+}
+
+bool WhitelistManager::IsAutocheckoutEnabled(const GURL& url) {
+  if (!experimental_form_filling_enabled_) {
+    // The feature is not enabled, return false.
+    return false;
+  }
+
+  if (url.is_empty())
+    return false;
+
+  for (std::vector<std::string>::iterator it = url_prefixes_.begin();
+      it != url_prefixes_.end(); ++it) {
+    // This is only for ~20 sites initially, liner search is sufficient.
+    // TODO(benquan): Look for optimization options when we support
+    // more sites.
+    if (url.spec().compare(0, it->size(), *it) == 0)
+      return true;
+  }
+  return false;
+}
+
+void WhitelistManager::BuildWhitelist(const std::string& data) {
+  // TODO(benquan): find a better way to parse csv data.
+  std::vector<std::string> new_url_prefixes;
+
+  std::stringstream dataStream(data);
+  std::string line;
+  while (std::getline(dataStream, line)) {
+    if (!line.empty()) {
+      std::vector<std::string> fields;
+      base::SplitString(line, ',', &fields);
+      // The whilist file is a simple CSV file, and the first column is the url
+      // prefix.
+      if (!fields[0].empty())
+        new_url_prefixes.push_back(fields[0]);
+    }
+  }
+  url_prefixes_ = new_url_prefixes;
+}
+
+}  // namespace autocheckout
+