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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
// Copyright 2018 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/media/router/mojo/media_sink_service_status.h"
#include "base/json/json_string_value_serializer.h"
#include "base/strings/strcat.h"
#include "components/media_router/browser/log_util.h"
namespace media_router {
namespace {
constexpr size_t kMaxAvailableSinksSize = 100;
// Helper function to convert `value` to JSON string.
std::string ToJSONString(const base::Value::Dict& value) {
std::string json;
JSONStringValueSerializer serializer(&json);
serializer.set_pretty_print(true);
return serializer.Serialize(value) ? json : "";
}
// Helper function to convert `sink_internal` to JSON format represented by
// base::Value::Dict.
base::Value::Dict ToValue(const MediaSinkInternal& sink_internal) {
base::Value::Dict dict;
const MediaSink& sink = sink_internal.sink();
dict.Set("id", log_util::TruncateId(sink.id()));
dict.Set("name", sink.name());
dict.Set("icon_type", static_cast<int>(sink.icon_type()));
if (sink_internal.is_dial_sink()) {
DialSinkExtraData extra_data = sink_internal.dial_data();
dict.Set("ip_address", extra_data.ip_address.ToString());
dict.Set("model_name", extra_data.model_name);
dict.Set("app_url", extra_data.app_url.spec());
}
if (sink_internal.is_cast_sink()) {
CastSinkExtraData extra_data = sink_internal.cast_data();
dict.Set("ip_endpoint", extra_data.ip_endpoint.ToString());
dict.Set("model_name", extra_data.model_name);
dict.Set("capabilities",
base::checked_cast<int>(extra_data.capabilities.ToEnumBitmask()));
dict.Set("channel_id", extra_data.cast_channel_id);
dict.Set("discovered_by_dial",
extra_data.discovery_type == CastDiscoveryType::kDial);
}
return dict;
}
// Helper function to convert `sinks` to JSON format represented by
// base::Value::Dict. The key is the source id and the value is a list of
// sinks represented by base::Value::Dict.
base::Value::Dict ConvertDiscoveredSinksToValues(
const base::flat_map<std::string, std::vector<MediaSinkInternal>>& sinks) {
base::Value::Dict dict;
for (const auto& sinks_it : sinks) {
base::Value::List list;
for (const auto& inner_sink : sinks_it.second) {
list.Append(ToValue(inner_sink));
}
dict.Set(sinks_it.first, std::move(list));
}
return dict;
}
// Helper function to convert `available_sinks` to a dictionary of availability
// strings in JSON format represented by base::Value::Dict. The key is the
// source id and the value is a list of sink ids.
base::Value::Dict ConvertAvailableSinksToValues(
const base::LRUCache<std::string, std::vector<MediaSinkInternal>>&
available_sinks) {
base::Value::Dict dict;
for (const auto& sinks_it : available_sinks) {
base::Value::List list;
for (const auto& inner_sink : sinks_it.second) {
std::string sink_id = inner_sink.sink().id();
list.Append(log_util::TruncateId(sink_id));
}
dict.Set(sinks_it.first, std::move(list));
}
return dict;
}
} // namespace
MediaSinkServiceStatus::MediaSinkServiceStatus()
: available_sinks_(kMaxAvailableSinksSize) {}
MediaSinkServiceStatus::~MediaSinkServiceStatus() = default;
void MediaSinkServiceStatus::UpdateDiscoveredSinks(
const std::string& provider_name,
const std::vector<MediaSinkInternal>& discovered_sinks) {
discovered_sinks_[provider_name] = discovered_sinks;
}
void MediaSinkServiceStatus::UpdateAvailableSinks(
mojom::MediaRouteProviderId provider_id,
const std::string& media_source,
const std::vector<MediaSinkInternal>& available_sinks) {
// TODO(takumif): It'd be safer and more efficient to make use
// pair<MediaRouteProviderId, string> than a serialized "id:name" string.
std::string key =
base::StrCat({ProviderIdToString(provider_id), ":", media_source});
available_sinks_.Put(key, available_sinks);
}
base::Value::Dict MediaSinkServiceStatus::GetStatusAsValue() const {
base::Value::Dict status_dict;
status_dict.Set("discovered_sinks",
ConvertDiscoveredSinksToValues(discovered_sinks_));
status_dict.Set("available_sinks",
ConvertAvailableSinksToValues(available_sinks_));
return status_dict;
}
std::string MediaSinkServiceStatus::GetStatusAsJSONString() const {
return ToJSONString(GetStatusAsValue());
}
} // namespace media_router
|
