1 // Copyright 2017 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/offline_pages/prefetch/prefetch_service_factory.h"
6
7 #include <memory>
8 #include <string>
9 #include <utility>
10
11 #include "base/callback_helpers.h"
12 #include "base/files/file_path.h"
13 #include "base/memory/singleton.h"
14 #include "base/task/task_traits.h"
15 #include "base/task/thread_pool.h"
16 #include "base/threading/sequenced_task_runner_handle.h"
17 #include "chrome/browser/chrome_content_browser_client.h"
18 #include "chrome/browser/download/download_service_factory.h"
19 #include "chrome/browser/image_fetcher/image_fetcher_service_factory.h"
20 #include "chrome/browser/net/system_network_context_manager.h"
21 #include "chrome/browser/offline_pages/offline_page_model_factory.h"
22 #include "chrome/browser/offline_pages/prefetch/gcm_token.h"
23 #include "chrome/browser/offline_pages/prefetch/offline_metrics_collector_impl.h"
24 #include "chrome/browser/offline_pages/prefetch/prefetch_background_task_handler_impl.h"
25 #include "chrome/browser/profiles/profile.h"
26 #include "chrome/browser/profiles/profile_key.h"
27 #include "chrome/browser/transition_manager/full_browser_transition_manager.h"
28 #include "chrome/common/channel_info.h"
29 #include "chrome/common/chrome_constants.h"
30 #include "components/feed/feed_feature_list.h"
31 #include "components/image_fetcher/core/cached_image_fetcher.h"
32 #include "components/image_fetcher/core/image_fetcher_impl.h"
33 #include "components/image_fetcher/core/image_fetcher_service.h"
34 #include "components/keyed_service/core/simple_dependency_manager.h"
35 #include "components/offline_pages/core/offline_page_feature.h"
36 #include "components/offline_pages/core/prefetch/prefetch_dispatcher_impl.h"
37 #include "components/offline_pages/core/prefetch/prefetch_downloader_impl.h"
38 #include "components/offline_pages/core/prefetch/prefetch_gcm_app_handler.h"
39 #include "components/offline_pages/core/prefetch/prefetch_importer_impl.h"
40 #include "components/offline_pages/core/prefetch/prefetch_network_request_factory_impl.h"
41 #include "components/offline_pages/core/prefetch/prefetch_service_impl.h"
42 #include "components/offline_pages/core/prefetch/store/prefetch_store.h"
43 #include "content/public/browser/browser_task_traits.h"
44 #include "content/public/browser/browser_thread.h"
45 #include "services/network/public/cpp/shared_url_loader_factory.h"
46
47 namespace offline_pages {
48
49 namespace {
50
GetImageFetcher(ProfileKey * key,image_fetcher::ImageFetcherConfig config)51 image_fetcher::ImageFetcher* GetImageFetcher(
52 ProfileKey* key,
53 image_fetcher::ImageFetcherConfig config) {
54 image_fetcher::ImageFetcherService* image_fetcher_service =
55 ImageFetcherServiceFactory::GetForKey(key);
56 DCHECK(image_fetcher_service);
57 return image_fetcher_service->GetImageFetcher(config);
58 }
59
SwitchToFullBrowserImageFetcher(PrefetchServiceImpl * prefetch_service,ProfileKey * key)60 void SwitchToFullBrowserImageFetcher(PrefetchServiceImpl* prefetch_service,
61 ProfileKey* key) {
62 // We don't need to switch the image_fetcher if it isn't created.
63 if (!prefetch_service->GetImageFetcher())
64 return;
65
66 prefetch_service->ReplaceImageFetcher(
67 GetImageFetcher(key, image_fetcher::ImageFetcherConfig::kDiskCacheOnly));
68 }
69
OnProfileCreated(PrefetchServiceImpl * prefetch_service,Profile * profile)70 void OnProfileCreated(PrefetchServiceImpl* prefetch_service, Profile* profile) {
71 if (IsPrefetchingOfflinePagesEnabled()) {
72 // Trigger an update of the cached GCM token. This needs to be post tasked
73 // because otherwise leads to circular dependency between
74 // PrefetchServiceFactory and GCMProfileServiceFactory. See
75 // https://crbug.com/944952
76 // Update is not a priority so make sure it happens after the critical
77 // startup path.
78 content::GetUIThreadTaskRunner({base::TaskPriority::BEST_EFFORT})
79 ->PostTask(FROM_HERE,
80 base::BindOnce(
81 &GetGCMToken, profile, kPrefetchingOfflinePagesAppId,
82 base::BindOnce(&PrefetchServiceImpl::GCMTokenReceived,
83 prefetch_service->GetWeakPtr())));
84 }
85
86 SwitchToFullBrowserImageFetcher(prefetch_service, profile->GetProfileKey());
87 }
88
89 } // namespace
90
PrefetchServiceFactory()91 PrefetchServiceFactory::PrefetchServiceFactory()
92 : SimpleKeyedServiceFactory("OfflinePagePrefetchService",
93 SimpleDependencyManager::GetInstance()) {
94 DependsOn(DownloadServiceFactory::GetInstance());
95 DependsOn(OfflinePageModelFactory::GetInstance());
96 DependsOn(ImageFetcherServiceFactory::GetInstance());
97 }
98
99 // static
GetInstance()100 PrefetchServiceFactory* PrefetchServiceFactory::GetInstance() {
101 return base::Singleton<PrefetchServiceFactory>::get();
102 }
103
104 // static
GetForKey(SimpleFactoryKey * key)105 PrefetchService* PrefetchServiceFactory::GetForKey(SimpleFactoryKey* key) {
106 return static_cast<PrefetchService*>(
107 GetInstance()->GetServiceForKey(key, true));
108 }
109
BuildServiceInstanceFor(SimpleFactoryKey * key) const110 std::unique_ptr<KeyedService> PrefetchServiceFactory::BuildServiceInstanceFor(
111 SimpleFactoryKey* key) const {
112 ProfileKey* profile_key = ProfileKey::FromSimpleFactoryKey(key);
113
114 OfflinePageModel* offline_page_model =
115 OfflinePageModelFactory::GetForKey(profile_key);
116 DCHECK(offline_page_model);
117
118 auto offline_metrics_collector =
119 std::make_unique<OfflineMetricsCollectorImpl>(profile_key->GetPrefs());
120
121 auto prefetch_dispatcher =
122 std::make_unique<PrefetchDispatcherImpl>(profile_key->GetPrefs());
123
124 auto* system_network_context_manager =
125 SystemNetworkContextManager::GetInstance();
126 scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory;
127 if (system_network_context_manager) {
128 url_loader_factory =
129 system_network_context_manager->GetSharedURLLoaderFactory();
130 } else {
131 // In unit_tests, NetworkService might not be available and
132 // |system_network_context_manager| would be null.
133 url_loader_factory = nullptr;
134 }
135
136 auto prefetch_network_request_factory =
137 std::make_unique<PrefetchNetworkRequestFactoryImpl>(
138 url_loader_factory, chrome::GetChannel(), GetUserAgent(),
139 profile_key->GetPrefs());
140
141 scoped_refptr<base::SequencedTaskRunner> background_task_runner =
142 base::ThreadPool::CreateSequencedTaskRunner({base::MayBlock()});
143 base::FilePath store_path =
144 profile_key->GetPath().Append(chrome::kOfflinePagePrefetchStoreDirname);
145 auto prefetch_store =
146 std::make_unique<PrefetchStore>(background_task_runner, store_path);
147
148 image_fetcher::ImageFetcher* image_fetcher = GetImageFetcher(
149 profile_key, image_fetcher::ImageFetcherConfig::kReducedMode);
150
151 auto prefetch_downloader = std::make_unique<PrefetchDownloaderImpl>(
152 DownloadServiceFactory::GetForKey(profile_key), chrome::GetChannel(),
153 profile_key->GetPrefs());
154
155 auto prefetch_importer = std::make_unique<PrefetchImporterImpl>(
156 prefetch_dispatcher.get(), offline_page_model, background_task_runner);
157
158 auto prefetch_background_task_handler =
159 std::make_unique<PrefetchBackgroundTaskHandlerImpl>(
160 profile_key->GetPrefs());
161
162 auto service = std::make_unique<PrefetchServiceImpl>(
163 std::move(offline_metrics_collector), std::move(prefetch_dispatcher),
164 std::move(prefetch_network_request_factory), offline_page_model,
165 std::move(prefetch_store), std::move(prefetch_downloader),
166 std::move(prefetch_importer), std::make_unique<PrefetchGCMAppHandler>(),
167 std::move(prefetch_background_task_handler), image_fetcher,
168 profile_key->GetPrefs());
169
170 auto callback = base::BindOnce(&OnProfileCreated, service.get());
171 FullBrowserTransitionManager::Get()->RegisterCallbackOnProfileCreation(
172 profile_key, std::move(callback));
173
174 return service;
175 }
176
177 } // namespace offline_pages
178