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
97
98
99
|
// Copyright 2016 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.
#include "chrome/browser/ui/toolbar/media_router_action_controller.h"
#include "chrome/browser/media/router/media_router_factory.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
#include "chrome/browser/ui/toolbar/toolbar_actions_model.h"
#include "chrome/common/pref_names.h"
MediaRouterActionController::MediaRouterActionController(Profile* profile)
: MediaRouterActionController(
profile,
media_router::MediaRouterFactory::GetApiForBrowserContext(profile),
ToolbarActionsModel::Get(profile),
ToolbarActionsModel::Get(profile)->component_migration_helper()) {
DCHECK(component_action_delegate_);
DCHECK(component_migration_helper_);
}
MediaRouterActionController::~MediaRouterActionController() {
DCHECK_EQ(dialog_count_, 0u);
}
void MediaRouterActionController::OnIssue(const media_router::Issue& issue) {
has_issue_ = true;
MaybeAddOrRemoveAction();
}
void MediaRouterActionController::OnIssuesCleared() {
has_issue_ = false;
MaybeAddOrRemoveAction();
}
void MediaRouterActionController::OnRoutesUpdated(
const std::vector<media_router::MediaRoute>& routes,
const std::vector<media_router::MediaRoute::Id>& joinable_route_ids) {
has_local_display_route_ =
std::find_if(routes.begin(), routes.end(),
[](const media_router::MediaRoute& route) {
return route.is_local() && route.for_display();
}) != routes.end();
MaybeAddOrRemoveAction();
}
void MediaRouterActionController::OnDialogShown() {
dialog_count_++;
MaybeAddOrRemoveAction();
}
void MediaRouterActionController::OnDialogHidden() {
DCHECK_GT(dialog_count_, 0u);
if (dialog_count_)
dialog_count_--;
MaybeAddOrRemoveAction();
}
MediaRouterActionController::MediaRouterActionController(
Profile* profile,
media_router::MediaRouter* router,
extensions::ComponentMigrationHelper::ComponentActionDelegate*
component_action_delegate,
extensions::ComponentMigrationHelper* component_migration_helper)
: media_router::IssuesObserver(router),
media_router::MediaRoutesObserver(router),
profile_(profile),
component_action_delegate_(component_action_delegate),
component_migration_helper_(component_migration_helper) {
DCHECK(profile_);
media_router::IssuesObserver::Init();
pref_change_registrar_.Init(profile->GetPrefs());
pref_change_registrar_.Add(
prefs::kToolbarMigratedComponentActionStatus,
base::Bind(&MediaRouterActionController::MaybeAddOrRemoveAction,
base::Unretained(this)));
}
void MediaRouterActionController::MaybeAddOrRemoveAction() {
if (ShouldEnableAction()) {
if (!component_action_delegate_->HasComponentAction(
ComponentToolbarActionsFactory::kMediaRouterActionId)) {
component_action_delegate_->AddComponentAction(
ComponentToolbarActionsFactory::kMediaRouterActionId);
}
} else if (component_action_delegate_->HasComponentAction(
ComponentToolbarActionsFactory::kMediaRouterActionId)) {
component_action_delegate_->RemoveComponentAction(
ComponentToolbarActionsFactory::kMediaRouterActionId);
}
}
bool MediaRouterActionController::ShouldEnableAction() const {
return has_local_display_route_ || has_issue_ || dialog_count_ ||
component_migration_helper_->GetComponentActionPref(
ComponentToolbarActionsFactory::kMediaRouterActionId);
}
|
