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

#ifndef EXTENSIONS_BROWSER_API_WEB_REQUEST_WEB_REQUEST_INFO_H_
#define EXTENSIONS_BROWSER_API_WEB_REQUEST_WEB_REQUEST_INFO_H_

#include <stdint.h>

#include <memory>
#include <optional>
#include <string>
#include <vector>

#include "base/containers/flat_map.h"
#include "base/memory/scoped_refptr.h"
#include "base/values.h"
#include "content/public/browser/child_process_id.h"
#include "content/public/browser/global_routing_id.h"
#include "extensions/browser/api/declarative_net_request/request_action.h"
#include "extensions/browser/api/web_request/web_request_resource_type.h"
#include "extensions/browser/extension_api_frame_id_map.h"
#include "extensions/buildflags/buildflags.h"
#include "ipc/constants.mojom.h"
#include "net/http/http_request_headers.h"
#include "net/http/http_response_headers.h"
#include "net/ssl/ssl_info.h"
#include "services/network/public/cpp/resource_request.h"
#include "services/network/public/mojom/url_response_head.mojom-forward.h"
#include "url/gurl.h"
#include "url/origin.h"

static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));

namespace extensions {

class ExtensionNavigationUIData;

// Helper struct to initialize WebRequestInfo.
struct WebRequestInfoInitParams {
  WebRequestInfoInitParams();

  // Initializes a WebRequestInfoInitParams from information provided over a
  // URLLoaderFactory interface.
  WebRequestInfoInitParams(
      uint64_t request_id,
      int render_process_id,
      int frame_routing_id,
      std::unique_ptr<ExtensionNavigationUIData> navigation_ui_data,
      const network::ResourceRequest& request,
      bool is_download,
      bool is_async,
      bool is_service_worker_script,
      std::optional<int64_t> navigation_id);

  WebRequestInfoInitParams(const WebRequestInfoInitParams&) = delete;
  WebRequestInfoInitParams(WebRequestInfoInitParams&& other);

  WebRequestInfoInitParams& operator=(const WebRequestInfoInitParams&) = delete;
  WebRequestInfoInitParams& operator=(WebRequestInfoInitParams&& other);

  ~WebRequestInfoInitParams();

  uint64_t id = 0;
  GURL url;
  int render_process_id = -1;
  int frame_routing_id = IPC::mojom::kRoutingIdNone;
  std::string method;
  bool is_navigation_request = false;
  std::optional<url::Origin> initiator;
  WebRequestResourceType web_request_type = WebRequestResourceType::OTHER;
  bool is_async = false;
  net::HttpRequestHeaders extra_request_headers;
  std::optional<base::Value::Dict> request_body_data;
  bool is_web_view = false;
  int web_view_instance_id = -1;
  int web_view_rules_registry_id = -1;
  content::ChildProcessId web_view_embedder_process_id;
  ExtensionApiFrameIdMap::FrameData frame_data;
  bool is_service_worker_script = false;
  std::optional<int64_t> navigation_id;
  content::GlobalRenderFrameHostId parent_routing_id;

 private:
  void InitializeWebViewAndFrameData(
      const ExtensionNavigationUIData* navigation_ui_data);
};

// A URL request representation used by WebRequest API internals. This structure
// carries information about an in-progress request.
struct WebRequestInfo {
  explicit WebRequestInfo(WebRequestInfoInitParams params);

  WebRequestInfo(const WebRequestInfo&) = delete;
  WebRequestInfo& operator=(const WebRequestInfo&) = delete;

  ~WebRequestInfo();

  // Fill in response data for this request.
  void AddResponseInfoFromResourceResponse(
      const network::mojom::URLResponseHead& response);

  void AddSslInfo(const std::optional<net::SSLInfo>& ssl_info);

  // Erases all actions in `dnr_actions` that are associated with the given
  // `extension_id`.
  void EraseDNRActionsForExtension(const ExtensionId& extension_id);

  // Erases all actions in `dnr_actions` that have a priority less than that of
  // the max priority allow rule for that action's extension in
  // `allow_rule_max_priority`. This prevents lower priority actions matched in
  // a different request stage to modify a request if they should've been
  // bypassed by higher priority allow rules.
  void EraseOutprioritizedDNRActions();

  // Returns if the provided `allow_action` which was already matched against
  // the request, should be recorded as a match, relative to the already matched
  // actions in `dnr_actions`.
  // This method currently makes two assumptions:
  //  - `allow_action` outprioritizes all actions in `dnr_actions` after a prior
  //    call to EraseOutprioritizedDNRActions().
  //  - this is called in onHeadersReceived
  bool ShouldRecordMatchedAllowRuleInOnHeadersReceived(
      const declarative_net_request::RequestAction& allow_action) const;

  // A unique identifier for this request.
  const uint64_t id;

  // The URL of the request.
  const GURL url;

  // The ID of the render process which initiated the request, or -1 of not
  // applicable (i.e. if initiated by the browser).
  const int render_process_id;

  // The frame routing ID of the frame which initiated this request, or
  // IPC::mojom::kRoutingIdNone if the request was not initiated by a frame.
  const int frame_routing_id = IPC::mojom::kRoutingIdNone;

  // The HTTP method used for the request, if applicable.
  const std::string method;

  // Indicates whether the request is for a browser-side navigation.
  const bool is_navigation_request;

  // The origin of the context which initiated the request. May be null for
  // browser-initiated requests such as navigations.
  const std::optional<url::Origin> initiator;

  // Extension API frame data corresponding to details of the frame which
  // initiate this request.
  ExtensionApiFrameIdMap::FrameData frame_data;

  // The resource type being requested.
  const WebRequestResourceType web_request_type = WebRequestResourceType::OTHER;

  // Indicates if this request is asynchronous.
  const bool is_async;

  const net::HttpRequestHeaders extra_request_headers;

  // HTTP response code for this request if applicable. -1 if not.
  int response_code = -1;

  // All response headers for this request, if set.
  scoped_refptr<net::HttpResponseHeaders> response_headers;

  // The stringified IP address of the host which provided the response to this
  // request, if applicable and available.
  std::string response_ip;

  // Indicates whether or not the request response (if applicable) came from
  // cache.
  bool response_from_cache = false;

  // A dictionary of request body data matching the format expected by
  // WebRequest API consumers. This may have a "formData" key and/or a "raw"
  // key. See WebRequest API documentation for more details.
  std::optional<base::Value::Dict> request_body_data;

  // Indicates whether this request was initiated by a <webview> instance.
  const bool is_web_view;

  // If `is_web_view` is true, the instance ID, rules registry ID, and embedder
  // process ID pertain to the webview instance.
  const int web_view_instance_id;
  const int web_view_rules_registry_id;
  const content::ChildProcessId web_view_embedder_process_id;

  // The Declarative Net Request (DNR) actions associated with this request that
  // are matched during the onBeforeRequest stage. Mutable since this is lazily
  // computed. Cached to avoid redundant computations. Valid when not null. In
  // case no actions are taken, populated with an empty vector.
  mutable std::optional<std::vector<declarative_net_request::RequestAction>>
      dnr_actions;

  // A map from an extension ID to the highest priority matching allow or
  // allowAllRequests rule for this request. This is set from the same field as
  // what's in declarative_net_request::RequestParams so it can be used between
  // different request stages where DNR rules will be matched.
  mutable base::flat_map<ExtensionId,
                         std::optional<declarative_net_request::RequestAction>>
      max_priority_allow_action;

  const bool is_service_worker_script;

  // Valid if this request corresponds to a navigation.
  const std::optional<int64_t> navigation_id;

  // ID of the RenderFrameHost corresponding to the parent frame. Only valid for
  // document subresource and sub-frame requests.
  // TODO(karandeepb, mcnee): For subresources, having "parent" in the name is
  // misleading. This should be renamed to indicate that this is the initiator.
  const content::GlobalRenderFrameHostId parent_routing_id;

  // For SecurityInfo object.
  std::optional<net::SSLInfo> ssl_info;
};

}  // namespace extensions

#endif  // EXTENSIONS_BROWSER_API_WEB_REQUEST_WEB_REQUEST_INFO_H_