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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// 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/device_notifications/device_system_tray_icon.h"
#include "chrome/browser/device_notifications/device_system_tray_icon_renderer.h"
#include "chrome/browser/profiles/profile.h"
#include "content/public/browser/browser_thread.h"
DeviceSystemTrayIcon::DeviceSystemTrayIcon(
std::unique_ptr<DeviceSystemTrayIconRenderer> icon_renderer)
: icon_renderer_(std::move(icon_renderer)) {}
DeviceSystemTrayIcon::~DeviceSystemTrayIcon() = default;
void DeviceSystemTrayIcon::StageProfile(Profile* profile) {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
auto it = profiles_.find(profile);
if (it != profiles_.end()) {
// If the |profile| is tracked, it must be unstaging.
CHECK(!it->second);
// Connection tracker's connection count is updated even the profile is
// just moved from unstaging to staging.
NotifyConnectionCountUpdated(profile);
it->second = true;
return;
}
profiles_[profile] = true;
ProfileAdded(profile);
}
void DeviceSystemTrayIcon::UnstageProfile(Profile* profile, bool immediate) {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
auto it = profiles_.find(profile);
// The |profile| must be tracked. However, it can be unstaging. For example,
// A profile is scheduled to be removed followed by profile destruction.
CHECK(it != profiles_.end());
if (immediate) {
profiles_.erase(it);
ProfileRemoved(profile);
return;
}
// For non-immediate case, the |profile| must be staging.
CHECK(it->second);
it->second = false;
// In order to avoid bouncing the system tray icon, schedule |profile| to be
// removed from the system tray icon later.
content::BrowserThread::GetTaskRunnerForThread(content::BrowserThread::UI)
->PostDelayedTask(
FROM_HERE,
// This class is supposedly safe as it is owned by
// g_browser_process. However, to avoid corner
// scenarios in tests, use weak ptr just to be safe.
base::BindOnce(&DeviceSystemTrayIcon::CleanUpProfile,
weak_factory_.GetWeakPtr(), profile->GetWeakPtr()),
kProfileUnstagingTime);
// Connection tracker's connection count is updated even in scheduled
// removal case.
NotifyConnectionCountUpdated(profile);
}
void DeviceSystemTrayIcon::ProfileAdded(Profile* profile) {
if (icon_renderer_) {
icon_renderer_->AddProfile(profile);
}
}
void DeviceSystemTrayIcon::ProfileRemoved(Profile* profile) {
if (icon_renderer_) {
icon_renderer_->RemoveProfile(profile);
}
}
void DeviceSystemTrayIcon::NotifyConnectionCountUpdated(Profile* profile) {
if (icon_renderer_) {
icon_renderer_->NotifyConnectionUpdated(profile);
}
}
void DeviceSystemTrayIcon::CleanUpProfile(base::WeakPtr<Profile> profile) {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
if (profile) {
auto it = profiles_.find(profile.get());
if (it != profiles_.end() && !it->second) {
profiles_.erase(it);
ProfileRemoved(profile.get());
}
return;
}
// If the |profile| is destroyed, |profiles_| shouldn't have an entry for
// |profile|. This is because DeviceConnectionTracker::CleanUp() is called on
// browser context (i.e. profile) shutdown and calls UnstageProfile() with
// immediate set to true so the entry will be removed from |profiles_|
// immediately.
}
|
