// 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 "content/public/common/url_loader_throttle.h"
#include "net/http/http_request_headers.h"

namespace chromecast {

class CastURLLoaderThrottle : public content::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)> callback) = 0;

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

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

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

  void ResumeRequest(int error, net::HttpRequestHeaders 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_