cast_url_loader_throttle.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 CHROMECAST_COMMON_CAST_URL_LOADER_THROTTLE_H_
#define CHROMECAST_COMMON_CAST_URL_LOADER_THROTTLE_H_

#include <string>

#include "base/callback_forward.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "net/http/http_request_headers.h"
#include "third_party/blink/public/common/loader/url_loader_throttle.h"

namespace chromecast {

class CastURLLoaderThrottle : public blink::URLLoaderThrottle {
 public:
  // An interface for CastURLLoaderThrottle to modify the resource request,
  // possibly also defer the request (by returning net::IO_PENDING) in some
  // scenarios where blocking operations are needed.
  class Delegate {
   public:
    virtual int WillStartResourceRequest(
        network::ResourceRequest* request,
        const std::string& session_id,
        base::OnceCallback<void(int,
                                net::HttpRequestHeaders,
                                net::HttpRequestHeaders)> callback) = 0;

   protected:
    virtual ~Delegate() = default;
  };

  CastURLLoaderThrottle(Delegate* delegate, const std::string& session_id);
  ~CastURLLoaderThrottle() override;

 private:
  // blink::URLLoaderThrottle implementation:
  void DetachFromCurrentSequence() override;
  void WillStartRequest(network::ResourceRequest* request,
                        bool* defer) override;
  bool makes_unsafe_redirect() override;

  void ResumeRequest(int error,
                     net::HttpRequestHeaders headers,
                     net::HttpRequestHeaders cors_exempt_headers);

  bool deferred_ = false;
  Delegate* const settings_delegate_;
  const std::string session_id_;

  base::WeakPtr<CastURLLoaderThrottle> weak_this_;
  base::WeakPtrFactory<CastURLLoaderThrottle> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(CastURLLoaderThrottle);
};

}  // namespace chromecast

#endif  // CHROMECAST_COMMON_CAST_URL_LOADER_THROTTLE_H_