blob: 5383086a519b2aa37324ab4335a5df99ef229d1e [file] [log] [blame]
// Copyright 2016 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.
#include <map>
#include <memory>
#include <string>
#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "components/data_use_measurement/core/data_use_user_data.h"
#include "net/url_request/url_fetcher_delegate.h"
#include "net/url_request/url_request.h"
#include "url/gurl.h"
namespace net {
class URLFetcher;
class URLRequestContextGetter;
} // namespace net
namespace image_fetcher {
class ImageDataFetcher : public net::URLFetcherDelegate {
using ImageDataFetcherCallback =
base::Callback<void(const std::string& image_data)>;
using DataUseServiceName = data_use_measurement::DataUseUserData::ServiceName;
explicit ImageDataFetcher(
net::URLRequestContextGetter* url_request_context_getter);
~ImageDataFetcher() override;
// Sets a service name against which to track data usage.
void SetDataUseServiceName(DataUseServiceName data_use_service_name);
// Fetches the raw image bytes from the given |image_url| and calls the given
// |callback|. The callback is run even if fetching the URL fails. In case
// of an error an empty string is passed to the callback.
void FetchImageData(const GURL& image_url,
const ImageDataFetcherCallback& callback);
// Like above, but lets the caller set a referrer.
void FetchImageData(const GURL& image_url,
const ImageDataFetcherCallback& callback,
const std::string& referrer,
net::URLRequest::ReferrerPolicy referrer_policy);
struct ImageDataFetcherRequest;
// Method inherited from URLFetcherDelegate
void OnURLFetchComplete(const net::URLFetcher* source) override;
// All active image url requests.
std::map<const net::URLFetcher*, std::unique_ptr<ImageDataFetcherRequest>>
scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
DataUseServiceName data_use_service_name_;
// The next ID to use for a newly created URLFetcher. Each URLFetcher gets an
// id when it is created. The |url_fetcher_id_| is incremented by one for each
// newly created URLFetcher. The URLFetcher ID can be used during testing to
// get individual URLFetchers and modify their state. Outside of tests this ID
// is not used.
int next_url_fetcher_id_;
} // namespace image_fetcher