content/common/url_loader_factory_bundle.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 "content/common/url_loader_factory_bundle.h"

#include <map>
#include <string>

#include "base/macros.h"
#include "services/network/public/interfaces/url_loader_factory.mojom.h"
#include "url/gurl.h"

class GURL;

namespace content {

URLLoaderFactoryBundleInfo::URLLoaderFactoryBundleInfo(
    URLLoaderFactoryBundleInfo&&) = default;

URLLoaderFactoryBundleInfo::URLLoaderFactoryBundleInfo(
    network::mojom::URLLoaderFactoryPtrInfo default_factory_info,
    std::map<std::string, network::mojom::URLLoaderFactoryPtrInfo>
        factories_info)
    : default_factory_info(std::move(default_factory_info)),
      factories_info(std::move(factories_info)) {}

URLLoaderFactoryBundleInfo::~URLLoaderFactoryBundleInfo() = default;

URLLoaderFactoryBundle::URLLoaderFactoryBundle() = default;

URLLoaderFactoryBundle::URLLoaderFactoryBundle(URLLoaderFactoryBundle&&) =
    default;

URLLoaderFactoryBundle::URLLoaderFactoryBundle(
    URLLoaderFactoryBundleInfo info) {
  default_factory_.Bind(std::move(info.default_factory_info));
  for (auto& factory_info : info.factories_info)
    factories_[factory_info.first].Bind(std::move(factory_info.second));
}

URLLoaderFactoryBundle::~URLLoaderFactoryBundle() = default;

URLLoaderFactoryBundle& URLLoaderFactoryBundle::operator=(
    URLLoaderFactoryBundle&&) = default;

void URLLoaderFactoryBundle::SetDefaultFactory(
    network::mojom::URLLoaderFactoryPtr factory) {
  default_factory_ = std::move(factory);
}

void URLLoaderFactoryBundle::RegisterFactory(
    const base::StringPiece& scheme,
    network::mojom::URLLoaderFactoryPtr factory) {
  DCHECK(factory.is_bound());
  auto result = factories_.emplace(std::string(scheme), std::move(factory));
  DCHECK(result.second);
}

network::mojom::URLLoaderFactory* URLLoaderFactoryBundle::GetFactoryForRequest(
    const GURL& url) {
  auto it = factories_.find(url.scheme());
  if (it == factories_.end()) {
    DCHECK(default_factory_.is_bound());
    return default_factory_.get();
  }
  return it->second.get();
}

URLLoaderFactoryBundleInfo URLLoaderFactoryBundle::PassInfo() {
  std::map<std::string, network::mojom::URLLoaderFactoryPtrInfo> factories_info;
  for (auto& factory : factories_)
    factories_info.emplace(factory.first, factory.second.PassInterface());
  DCHECK(default_factory_.is_bound());
  return URLLoaderFactoryBundleInfo(default_factory_.PassInterface(),
                                    std::move(factories_info));
}

URLLoaderFactoryBundle URLLoaderFactoryBundle::Clone() {
  DCHECK(default_factory_.is_bound());
  network::mojom::URLLoaderFactoryPtr cloned_default_factory;
  default_factory_->Clone(mojo::MakeRequest(&cloned_default_factory));

  URLLoaderFactoryBundle new_bundle;
  new_bundle.SetDefaultFactory(std::move(cloned_default_factory));
  for (auto& factory : factories_) {
    network::mojom::URLLoaderFactoryPtr cloned_factory;
    factory.second->Clone(mojo::MakeRequest(&cloned_factory));
    new_bundle.RegisterFactory(factory.first, std::move(cloned_factory));
  }

  return new_bundle;
}

}  // namespace content