// Copyright 2013 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_WEBSTORE_DATA_FETCHER_H_
#define EXTENSIONS_BROWSER_WEBSTORE_DATA_FETCHER_H_

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

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "extensions/browser/cws_item_service.pb.h"
#include "extensions/buildflags/buildflags.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
#include "services/network/public/cpp/resource_request.h"
#include "services/network/public/cpp/simple_url_loader.h"
#include "url/gurl.h"

static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));

namespace network::mojom {
class URLLoaderFactory;
class URLResponseHead;
}  // namespace network::mojom

namespace extensions {

class WebstoreDataFetcherDelegate;

// WebstoreDataFetcher fetches web store data and parses it into a
// DictionaryValue.
class WebstoreDataFetcher {
 public:
  WebstoreDataFetcher(WebstoreDataFetcherDelegate* delegate,
                      const GURL& referrer_url,
                      const std::string& webstore_item_id);

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

  ~WebstoreDataFetcher();

  static void SetLogResponseCodeForTesting(bool enabled);

  // Sets a mock response that is returned by the item snippets API when
  // FetchItemSnippet is called. `mock_response` is owned by the test that calls
  // this method.
  static void SetMockItemSnippetReponseForTesting(
      FetchItemSnippetResponse* mock_response);

  // Starts the request to fetch web store data using the item snippets API.
  void Start(network::mojom::URLLoaderFactory* url_loader_factory);

  void set_max_auto_retries(int max_retries) {
    max_auto_retries_ = max_retries;
  }

 private:
  // Initializes `simple_url_loader_` for the given `request` and `annotation`.
  void InitializeSimpleLoaderForRequest(
      std::unique_ptr<network::ResourceRequest> request,
      const net::NetworkTrafficAnnotationTag& annotation);

  void OnResponseStarted(const GURL& final_url,
                         const network::mojom::URLResponseHead& response_head);

  // Called when a response is received from the item snippet API.
  void OnFetchItemSnippetResponseReceived(
      std::optional<std::string> response_body);

  raw_ptr<WebstoreDataFetcherDelegate> delegate_;
  GURL referrer_url_;
  std::string id_;
  std::string post_data_;

  // For fetching webstore JSON data.
  std::unique_ptr<network::SimpleURLLoader> simple_url_loader_;

  // Maximum auto retry times on server 5xx error or ERR_NETWORK_CHANGED.
  // Default is 0 which means to use the URLFetcher default behavior.
  int max_auto_retries_ = 0;

  base::WeakPtrFactory<WebstoreDataFetcher> weak_ptr_factory_{this};
};

}  // namespace extensions

#endif  // EXTENSIONS_BROWSER_WEBSTORE_DATA_FETCHER_H_