// Copyright (c) 2017 GitHub, Inc.
// Use of this source code is governed by the MIT license that can be
// found in the LICENSE file.

#include "atom/browser/io_thread.h"

#include <utility>

#include "components/net_log/chrome_net_log.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/network_service_instance.h"
#include "net/cert/caching_cert_verifier.h"
#include "net/cert/cert_verifier.h"
#include "net/cert/cert_verify_proc.h"
#include "net/cert/multi_threaded_cert_verifier.h"
#include "net/proxy_resolution/proxy_resolution_service.h"
#include "net/url_request/url_request_context.h"
#include "services/network/network_service.h"
#include "services/network/public/cpp/features.h"
#include "services/network/url_request_context_builder_mojo.h"

using content::BrowserThread;

IOThread::IOThread(net_log::ChromeNetLog* net_log,
                   SystemNetworkContextManager* system_network_context_manager)
    : net_log_(net_log) {
  BrowserThread::SetIOThreadDelegate(this);

  system_network_context_manager->SetUp(
      &network_context_request_, &network_context_params_,
      &http_auth_static_params_, &http_auth_dynamic_params_);
}

IOThread::~IOThread() {
  BrowserThread::SetIOThreadDelegate(nullptr);
}

void IOThread::Init() {
  if (!base::FeatureList::IsEnabled(network::features::kNetworkService)) {
    std::unique_ptr<network::URLRequestContextBuilderMojo> builder =
        std::make_unique<network::URLRequestContextBuilderMojo>();

    auto cert_verifier = std::make_unique<net::CachingCertVerifier>(
        std::make_unique<net::MultiThreadedCertVerifier>(
            net::CertVerifyProc::CreateDefault()));
    builder->SetCertVerifier(std::move(cert_verifier));

    // Create the network service, so that shared host resolver
    // gets created which is required to set the auth preferences below.
    network::NetworkService* network_service = content::GetNetworkServiceImpl();
    network_service->SetUpHttpAuth(std::move(http_auth_static_params_));
    network_service->ConfigureHttpAuthPrefs(
        std::move(http_auth_dynamic_params_));

    system_network_context_ =
        network_service
            ->CreateNetworkContextWithBuilder(
                std::move(network_context_request_),
                std::move(network_context_params_), std::move(builder),
                &system_request_context_)
            .release();
  }
}

void IOThread::CleanUp() {
  if (!base::FeatureList::IsEnabled(network::features::kNetworkService))
    system_request_context_->proxy_resolution_service()->OnShutdown();

  if (net_log_)
    net_log_->ShutDownBeforeTaskScheduler();
}