1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/screen_ai/screen_ai_downloader_non_chromeos.h"
#include "base/no_destructor.h"
#include "base/task/thread_pool.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/component_updater/screen_ai_component_installer.h"
#include "chrome/browser/screen_ai/pref_names.h"
#include "components/prefs/pref_service.h"
#include "components/update_client/crx_update_item.h"
#include "content/public/browser/browser_thread.h"
#include "services/screen_ai/public/cpp/utilities.h"
namespace screen_ai {
// static
std::unique_ptr<ScreenAIInstallState> ScreenAIInstallState::Create() {
return std::make_unique<ScreenAIDownloaderNonChromeOS>();
}
// static
ScreenAIInstallState* ScreenAIInstallState::CreateForTesting() {
static base::NoDestructor<ScreenAIDownloaderNonChromeOS> install_state;
return install_state.get();
}
ScreenAIDownloaderNonChromeOS::ScreenAIDownloaderNonChromeOS() = default;
ScreenAIDownloaderNonChromeOS::~ScreenAIDownloaderNonChromeOS() = default;
void ScreenAIDownloaderNonChromeOS::DownloadComponentInternal() {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
component_updater::RegisterScreenAIComponent(
g_browser_process->component_updater());
if (!component_updater_observation_.IsObserving()) {
component_updater_observation_.Observe(
g_browser_process->component_updater());
}
}
void ScreenAIDownloaderNonChromeOS::SetLastUsageTime() {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
g_browser_process->local_state()->SetTime(
prefs::kScreenAILastUsedTimePrefName, base::Time::Now());
}
void ScreenAIDownloaderNonChromeOS::OnEvent(
const update_client::CrxUpdateItem& item) {
if (item.id !=
component_updater::ScreenAIComponentInstallerPolicy::GetOmahaId()) {
return;
}
if (item.state == update_client::ComponentState::kDownloading ||
item.state == update_client::ComponentState::kDownloadingDiff) {
SetDownloadProgress(static_cast<double>(item.downloaded_bytes) /
item.total_bytes);
}
}
} // namespace screen_ai
|
