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
|
// Copyright 2020 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/apps/app_service/app_notifications.h"
#include "base/containers/contains.h"
namespace apps {
AppNotifications::AppNotifications() = default;
AppNotifications::~AppNotifications() = default;
void AppNotifications::AddNotification(const std::string& app_id,
const std::string& notification_id) {
app_id_to_notification_ids_[app_id].insert(notification_id);
notification_id_to_app_ids_[notification_id].insert(app_id);
}
void AppNotifications::RemoveNotification(const std::string& notification_id) {
auto it = notification_id_to_app_ids_.find(notification_id);
CHECK(it != notification_id_to_app_ids_.end());
for (const auto& app_id : it->second) {
auto app_id_it = app_id_to_notification_ids_.find(app_id);
app_id_it->second.erase(notification_id);
if (app_id_it->second.empty()) {
app_id_to_notification_ids_.erase(app_id_it);
}
}
notification_id_to_app_ids_.erase(it);
}
void AppNotifications::RemoveNotificationsForApp(const std::string& app_id) {
auto it = app_id_to_notification_ids_.find(app_id);
if (it == app_id_to_notification_ids_.end()) {
return;
}
for (const auto& notification_id : it->second) {
auto notification_id_it = notification_id_to_app_ids_.find(notification_id);
notification_id_it->second.erase(app_id);
if (notification_id_it->second.empty()) {
notification_id_to_app_ids_.erase(notification_id_it);
}
}
app_id_to_notification_ids_.erase(it);
}
bool AppNotifications::HasNotification(const std::string& app_id) {
return base::Contains(app_id_to_notification_ids_, app_id);
}
std::set<std::string> AppNotifications::GetAppIdsForNotification(
const std::string& notification_id) {
auto it = notification_id_to_app_ids_.find(notification_id);
if (it == notification_id_to_app_ids_.end()) {
return {};
}
return it->second;
}
AppPtr AppNotifications::CreateAppWithHasBadgeStatus(
AppType app_type,
const std::string& app_id) {
auto app = std::make_unique<App>(app_type, app_id);
app->has_badge = HasNotification(app_id);
return app;
}
} // namespace apps
|
