[Media Router] Add MediaSink subtypes

chrome/browser/media/router/discovery/media_sink_internal.cc

Index: chrome/browser/media/router/discovery/media_sink_internal.cc
diff --git a/chrome/browser/media/router/discovery/media_sink_internal.cc b/chrome/browser/media/router/discovery/media_sink_internal.cc
new file mode 100644
index 0000000000000000000000000000000000000000..fe381f267f09683951e6190f6482059befb260d0
--- /dev/null
+++ b/chrome/browser/media/router/discovery/media_sink_internal.cc
@@ -0,0 +1,175 @@
+// Copyright 2017 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/media/router/discovery/media_sink_internal.h"
+
+#include "base/logging.h"
+#include "base/strings/string_util.h"
+
+namespace media_router {
+
+MediaSinkInternal::MediaSinkInternal(const MediaSink& sink,
+                                     const DialSinkExtraData& dial_data)
+    : sink_(sink), sink_type_(SinkType::DIAL) {
+  dial_data_.Init(dial_data);
+}
+
+MediaSinkInternal::MediaSinkInternal(const MediaSink& sink,
+                                     const CastSinkExtraData& cast_data)
+    : sink_(sink), sink_type_(SinkType::CAST) {
+  cast_data_.Init(cast_data);
+}
+
+MediaSinkInternal::MediaSinkInternal() : sink_type_(SinkType::GENERIC) {}
+
+MediaSinkInternal::MediaSinkInternal(const MediaSinkInternal& other) {
+  InternalCopyConstructFrom(other);
+}
+
+MediaSinkInternal::~MediaSinkInternal() {
+  InternalCleanup();
+}
+
+MediaSinkInternal& MediaSinkInternal::operator=(
+    const MediaSinkInternal& other) {
+  if (this != &other) {
+    if (sink_type_ == other.sink_type_) {
+      InternalCopyAssignFrom(other);
+    } else {
+      InternalCleanup();
+      InternalCopyConstructFrom(other);
+    }
+  }
+
+  return *this;
+}
+
+bool MediaSinkInternal::operator==(const MediaSinkInternal& other) const {
+  if (sink_type_ != other.sink_type_)
+    return false;
+
+  if (sink_ != other.sink_)
+    return false;
+
+  switch (sink_type_) {
+    case SinkType::DIAL:
+      return *dial_data_ == *(other.dial_data_);
+    case SinkType::CAST:
+      return *cast_data_ == *(other.cast_data_);
+    case SinkType::GENERIC:
+      return true;
+  }
+  NOTREACHED();
+  return false;
+}
+
+void MediaSinkInternal::set_sink(const MediaSink& sink) {
+  sink_ = sink;
+}
+
+void MediaSinkInternal::set_dial_data(const DialSinkExtraData& dial_data) {
+  DCHECK(sink_type_ != SinkType::CAST);
+  InternalCleanup();
+
+  sink_type_ = SinkType::DIAL;
+  dial_data_.Init(dial_data);
+}
+
+const DialSinkExtraData& MediaSinkInternal::dial_data() const {
+  DCHECK(is_dial_sink());
+  return *dial_data_;
+}
+
+void MediaSinkInternal::set_cast_data(const CastSinkExtraData& cast_data) {
+  DCHECK(sink_type_ != SinkType::DIAL);
+  InternalCleanup();
+
+  sink_type_ = SinkType::CAST;
+  cast_data_.Init(cast_data);
+}
+
+const CastSinkExtraData& MediaSinkInternal::cast_data() const {
+  DCHECK(is_cast_sink());
+  return *cast_data_;
+}
+
+// static
+bool MediaSinkInternal::IsValidSinkId(const std::string& sink_id) {
+  if (sink_id.empty() || !base::IsStringASCII(sink_id)) {
+    DLOG(WARNING) << "Invalid [sink_id]: " << sink_id;
+    return false;
+  }
+
+  return true;
+}
+
+void MediaSinkInternal::InternalCopyAssignFrom(const MediaSinkInternal& other) {
+  sink_ = other.sink_;
+  sink_type_ = other.sink_type_;
+
+  switch (sink_type_) {
+    case SinkType::DIAL:
+      *dial_data_ = *other.dial_data_;
+      return;
+    case SinkType::CAST:
+      *cast_data_ = *other.cast_data_;
+      return;
+    case SinkType::GENERIC:
+      return;
+  }
+  NOTREACHED();
+}
+
+void MediaSinkInternal::InternalCopyConstructFrom(
+    const MediaSinkInternal& other) {
+  sink_ = other.sink_;
+  sink_type_ = other.sink_type_;
+
+  switch (sink_type_) {
+    case SinkType::DIAL:
+      dial_data_.Init(*other.dial_data_);
+      return;
+    case SinkType::CAST:
+      cast_data_.Init(*other.cast_data_);
+      return;
+    case SinkType::GENERIC:
+      return;
+  }
+  NOTREACHED();
+}
+
+void MediaSinkInternal::InternalCleanup() {
+  switch (sink_type_) {
+    case SinkType::DIAL:
+      dial_data_.Destroy();
+      return;
+    case SinkType::CAST:
+      cast_data_.Destroy();
+      return;
+    case SinkType::GENERIC:
+      return;
+  }
+  NOTREACHED();
+}
+
+DialSinkExtraData::DialSinkExtraData() = default;
+DialSinkExtraData::DialSinkExtraData(const DialSinkExtraData& other) = default;
+DialSinkExtraData::~DialSinkExtraData() = default;
+
+bool DialSinkExtraData::operator==(const DialSinkExtraData& other) const {
+  return ip_address == other.ip_address && model_name == other.model_name &&
+         app_url == other.app_url;
+}
+
+CastSinkExtraData::CastSinkExtraData() = default;
+CastSinkExtraData::CastSinkExtraData(const CastSinkExtraData& other) = default;
+CastSinkExtraData::~CastSinkExtraData() = default;
+
+bool CastSinkExtraData::operator==(const CastSinkExtraData& other) const {
+  return ip_address == other.ip_address && model_name == other.model_name &&
+         capabilities == other.capabilities &&
+         cast_channel_id == other.cast_channel_id;
+}
+
+}  // namespace media_router