content/browser/web_package/bundled_exchanges_handle.h

// Copyright 2019 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.

#ifndef CONTENT_BROWSER_WEB_PACKAGE_BUNDLED_EXCHANGES_HANDLE_H_
#define CONTENT_BROWSER_WEB_PACKAGE_BUNDLED_EXCHANGES_HANDLE_H_

#include <memory>
#include <string>

#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "base/optional.h"
#include "content/browser/web_package/bundled_exchanges_source.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/data_decoder/public/mojom/bundled_exchanges_parser.mojom.h"
#include "services/network/public/mojom/url_loader_factory.mojom.h"
#include "url/gurl.h"

namespace content {

class BundledExchangesReader;
class BundledExchangesURLLoaderFactory;
class NavigationLoaderInterceptor;

// A class to provide interfaces to communicate with a BundledExchanges for
// loading. Running on the UI thread.
// TODO(crbug.com/966753): Add browser tests once core parts are submitted and
// basic sequence can pass.
class BundledExchangesHandle final {
 public:
  BundledExchangesHandle();
  explicit BundledExchangesHandle(const BundledExchangesSource& source);
  ~BundledExchangesHandle();

  // Creates a NavigationLoaderInterceptor instance to handle the request for
  // a BundledExchanges, to redirect to the entry URL of the BundledExchanges,
  // and to load the main exchange from the BundledExchanges.
  std::unique_ptr<NavigationLoaderInterceptor> CreateInterceptor();

  // Creates a URLLoaderFactory to load resources from the BundledExchanges.
  void CreateURLLoaderFactory(
      mojo::PendingReceiver<network::mojom::URLLoaderFactory> receiver,
      mojo::Remote<network::mojom::URLLoaderFactory> fallback_factory);

  // Checks if a valid BundledExchanges is attached, opened, and ready for use.
  bool IsReadyForLoading();

 private:
  class PrimaryURLRedirectLoader;

  void CreatePrimaryURLLoader(const network::ResourceRequest& resource_request,
                              network::mojom::URLLoaderRequest request,
                              network::mojom::URLLoaderClientPtr client);
  void MayRedirectPrimaryURLLoader();
  void OnMetadataReady(data_decoder::mojom::BundleMetadataParseErrorPtr error);

  const BundledExchangesSource source_;

  base::WeakPtr<PrimaryURLRedirectLoader> redirect_loader_;
  std::unique_ptr<BundledExchangesReader> reader_;
  std::unique_ptr<BundledExchangesURLLoaderFactory> url_loader_factory_;
  GURL primary_url_;
  data_decoder::mojom::BundleMetadataParseErrorPtr metadata_error_;
  bool is_redirected_ = false;

  base::WeakPtrFactory<BundledExchangesHandle> weak_factory_{this};

  DISALLOW_COPY_AND_ASSIGN(BundledExchangesHandle);
};

}  // namespace content

#endif  // CONTENT_BROWSER_WEB_PACKAGE_BUNDLED_EXCHANGES_HANDLE_H_