components/custom_handlers/protocol_handler_throttle.h

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_CUSTOM_HANDLERS_PROTOCOL_HANDLER_THROTTLE_H_
#define COMPONENTS_CUSTOM_HANDLERS_PROTOCOL_HANDLER_THROTTLE_H_

#include "base/memory/ref_counted.h"
#include "url/gurl.h"

#include "third_party/blink/public/common/loader/url_loader_throttle.h"

namespace custom_handlers {

class ProtocolHandlerRegistry;

class ProtocolHandlerThrottle : public blink::URLLoaderThrottle {
 public:
  explicit ProtocolHandlerThrottle(ProtocolHandlerRegistry&);
  ~ProtocolHandlerThrottle() override;

  void WillStartRequest(network::ResourceRequest* request,
                        bool* defer) override;

  void WillRedirectRequest(
      net::RedirectInfo* redirect_info,
      const network::mojom::URLResponseHead& response_head,
      bool* defer,
      std::vector<std::string>* to_be_removed_headers,
      net::HttpRequestHeaders* modified_headers,
      net::HttpRequestHeaders* modified_cors_exempt_headers) override;

 private:
  // If the @url has a custom protocol handler, the argument will return the
  // translated url.
  void TranslateUrl(GURL& url);
  // The ProtocolHandlerRegistry instance is a KeyedService which ownership is
  // managed by the BrowserContext. BrowserContext can be destroyed before this
  // throttle.
  base::WeakPtr<ProtocolHandlerRegistry> protocol_handler_registry_;
};

}  // namespace custom_handlers

#endif  // COMPONENTS_CUSTOM_HANDLERS_PROTOCOL_HANDLER_THROTTLE_H_