| // 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 CHROME_BROWSER_DOWNLOAD_DEFERRED_CLIENT_WRAPPER_H_ |
| #define CHROME_BROWSER_DOWNLOAD_DEFERRED_CLIENT_WRAPPER_H_ |
| |
| #include <string> |
| #include <vector> |
| |
| #include "base/callback.h" |
| #include "base/files/file_path.h" |
| #include "base/memory/weak_ptr.h" |
| #include "build/build_config.h" |
| #include "components/download/public/background_service/client.h" |
| #include "components/download/public/background_service/clients.h" |
| #include "net/http/http_response_headers.h" |
| #include "url/gurl.h" |
| |
| class SimpleFactoryKey; |
| class Profile; |
| |
| namespace download { |
| |
| using ClientFactory = base::OnceCallback<std::unique_ptr<Client>(Profile*)>; |
| |
| class DeferredClientWrapper : public Client { |
| public: |
| DeferredClientWrapper(DownloadClient client_id, |
| ClientFactory factory, |
| SimpleFactoryKey* key); |
| ~DeferredClientWrapper() override; |
| |
| // Client implementation. |
| void OnServiceInitialized( |
| bool state_lost, |
| const std::vector<DownloadMetaData>& downloads) override; |
| void OnServiceUnavailable() override; |
| void OnDownloadStarted( |
| const std::string& guid, |
| const std::vector<GURL>& url_chain, |
| const scoped_refptr<const net::HttpResponseHeaders>& headers) override; |
| void OnDownloadUpdated(const std::string& guid, |
| uint64_t bytes_uploaded, |
| uint64_t bytes_downloaded) override; |
| void OnDownloadFailed(const std::string& guid, |
| const download::CompletionInfo& info, |
| FailureReason reason) override; |
| void OnDownloadSucceeded(const std::string& guid, |
| const CompletionInfo& completion_info) override; |
| bool CanServiceRemoveDownloadedFile(const std::string& guid, |
| bool force_delete) override; |
| void GetUploadData(const std::string& guid, |
| GetUploadDataCallback callback) override; |
| |
| private: |
| // Forwarding functions |
| void ForwardOnServiceInitialized( |
| bool state_lost, |
| const std::vector<DownloadMetaData>& downloads); |
| void ForwardOnServiceUnavailable(); |
| void ForwardOnDownloadStarted( |
| const std::string& guid, |
| const std::vector<GURL>& url_chain, |
| const scoped_refptr<const net::HttpResponseHeaders>& headers); |
| void ForwardOnDownloadUpdated(const std::string& guid, |
| uint64_t bytes_uploaded, |
| uint64_t bytes_downloaded); |
| void ForwardOnDownloadFailed(const std::string& guid, |
| const download::CompletionInfo& info, |
| FailureReason reason); |
| void ForwardOnDownloadSucceeded(const std::string& guid, |
| const CompletionInfo& completion_info); |
| void ForwardCanServiceRemoveDownloadedFile(const std::string& guid, |
| bool force_delete); |
| void ForwardGetUploadData(const std::string& guid, |
| GetUploadDataCallback callback); |
| |
| void RunDeferredClosures(bool force_inflate); |
| void DoRunDeferredClosures(); |
| void InflateClient(Profile* profile); |
| |
| std::unique_ptr<download::Client> wrapped_client_; |
| std::vector<base::OnceClosure> deferred_closures_; |
| ClientFactory client_factory_; |
| SimpleFactoryKey* key_; |
| |
| #if defined(OS_ANDROID) |
| // This is needed to record UMA for when DownloadClientWrapper requests a full |
| // browser start while the browser is running in reduced mode. Reduced mode is |
| // only on Android so it is ifdefed out on other platforms to prevent the |
| // compiler from complaining that it is unused. |
| DownloadClient client_id_; |
| bool full_browser_requested_; |
| void LaunchFullBrowser(); |
| #endif |
| |
| base::WeakPtrFactory<DeferredClientWrapper> weak_ptr_factory_; |
| }; |
| |
| } // namespace download |
| |
| #endif // CHROME_BROWSER_DOWNLOAD_DEFERRED_CLIENT_WRAPPER_H_ |