// Copyright (c) 2008 The Chromium Embedded Framework Authors.
// Portions copyright (c) 2006-2008 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 "browser_request_context.h"
#include "browser_resource_loader_bridge.h"
#include "build/build_config.h"

#include "base/file_path.h"
#include "net/base/cert_verifier.h"
#include "net/base/cookie_monster.h"
#include "net/base/host_resolver.h"
#include "net/base/ssl_config_service.h"
#include "net/base/static_cookie_policy.h"
#include "net/ftp/ftp_network_layer.h"
#include "net/http/http_auth_handler_factory.h"
#include "net/proxy/proxy_config_service.h"
#include "net/proxy/proxy_config_service_fixed.h"
#include "net/proxy/proxy_service.h"
#include "webkit/blob/blob_storage_controller.h"
#include "webkit/glue/webkit_glue.h"

#if defined(OS_WIN)

#include <winhttp.h>
#pragma comment(lib, "winhttp.lib")

namespace {

// ProxyConfigService implementation that does nothing.
class ProxyConfigServiceNull : public net::ProxyConfigService {
public:
  ProxyConfigServiceNull() {}
  virtual void AddObserver(Observer* observer) {}
  virtual void RemoveObserver(Observer* observer) {}
  virtual bool GetLatestProxyConfig(net::ProxyConfig* config) { return true; }
  virtual void OnLazyPoll() {}
};

} // namespace

#endif // defined(OS_WIN)

BrowserRequestContext::BrowserRequestContext() {
  Init(FilePath(), net::HttpCache::NORMAL, false);
}

BrowserRequestContext::BrowserRequestContext(
    const FilePath& cache_path,
    net::HttpCache::Mode cache_mode,
    bool no_proxy) {
  Init(cache_path, cache_mode, no_proxy);
}

void BrowserRequestContext::Init(
    const FilePath& cache_path,
    net::HttpCache::Mode cache_mode,
    bool no_proxy) {
  cookie_store_ = new net::CookieMonster(NULL, NULL);
  cookie_policy_ = new net::StaticCookiePolicy();

  // hard-code A-L and A-C for test shells
  accept_language_ = "en-us,en";
  accept_charset_ = "iso-8859-1,*,utf-8";

#if defined(OS_WIN)
  // Using the system proxy resolver on Windows when "Automatically detect
  // settings" (auto-detection) is checked under LAN Settings can hurt resource
  // loading performance because the call to WinHttpGetProxyForUrl in
  // proxy_resolver_winhttp.cc will block the IO thread.  This is especially
  // true for Windows 7 where auto-detection is checked by default. To avoid
  // slow resource loading on Windows we only use the system proxy resolver if
  // auto-detection is unchecked.
  WINHTTP_CURRENT_USER_IE_PROXY_CONFIG ie_config = {0};
  if (WinHttpGetIEProxyConfigForCurrentUser(&ie_config)) {
    if (ie_config.fAutoDetect == TRUE) {
      proxy_service_ = net::ProxyService::CreateWithoutProxyResolver(
          new ProxyConfigServiceNull(), NULL);
    }

    if (ie_config.lpszAutoConfigUrl)
      GlobalFree(ie_config.lpszAutoConfigUrl);
    if (ie_config.lpszProxy)
      GlobalFree(ie_config.lpszProxy);
    if (ie_config.lpszProxyBypass)
      GlobalFree(ie_config.lpszProxyBypass);
  }
#endif // defined(OS_WIN)
  
  if (!proxy_service_.get()) {
    // Use the system proxy resolver.
    scoped_ptr<net::ProxyConfigService> proxy_config_service(
        net::ProxyService::CreateSystemProxyConfigService(
            MessageLoop::current(), NULL));
    proxy_service_ = net::ProxyService::CreateUsingSystemProxyResolver(
        proxy_config_service.release(), 0, NULL);
  }

  host_resolver_ =
      net::CreateSystemHostResolver(net::HostResolver::kDefaultParallelism,
                                    NULL, NULL);
  cert_verifier_ = new net::CertVerifier;
  ssl_config_service_ = net::SSLConfigService::CreateSystemSSLConfigService();

  http_auth_handler_factory_ =
      net::HttpAuthHandlerFactory::CreateDefault(host_resolver_);

  net::HttpCache::DefaultBackend* backend = new net::HttpCache::DefaultBackend(
      cache_path.empty() ? net::MEMORY_CACHE : net::DISK_CACHE,
      cache_path, 0, BrowserResourceLoaderBridge::GetCacheThread());

  net::HttpCache* cache =
      new net::HttpCache(host_resolver_, cert_verifier_, NULL, NULL,
                         proxy_service_, ssl_config_service_,
                         http_auth_handler_factory_, NULL, NULL, backend);

  cache->set_mode(cache_mode);
  http_transaction_factory_ = cache;

  ftp_transaction_factory_ = new net::FtpNetworkLayer(host_resolver_);

  blob_storage_controller_.reset(new webkit_blob::BlobStorageController());
}

BrowserRequestContext::~BrowserRequestContext() {
  delete ftp_transaction_factory_;
  delete http_transaction_factory_;
  delete http_auth_handler_factory_;
  delete static_cast<net::StaticCookiePolicy*>(cookie_policy_);
  delete cert_verifier_;
  delete host_resolver_;
}

void BrowserRequestContext::SetAcceptAllCookies(bool accept_all_cookies) {
  net::StaticCookiePolicy::Type policy_type = accept_all_cookies ?
      net::StaticCookiePolicy::ALLOW_ALL_COOKIES :
      net::StaticCookiePolicy::BLOCK_THIRD_PARTY_COOKIES;
  static_cast<net::StaticCookiePolicy*>(cookie_policy())->set_type(policy_type);
}

const std::string& BrowserRequestContext::GetUserAgent(
    const GURL& url) const {
  return webkit_glue::GetUserAgent(url);
}