Add Media Router Mojo impl code.

chrome/browser/media/router/media_router_mojo_impl.cc

+// Copyright 2015 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/media_router_mojo_impl.h"
+
+#include "base/bind.h"
+#include "base/guid.h"
+#include "base/logging.h"
+#include "base/observer_list.h"
+#include "chrome/browser/media/router/media_router_impl.h"
+#include "chrome/browser/media/router/media_router_impl_factory.h"
+#include "chrome/browser/media/router/media_router_mojo_impl_factory.h"
+#include "chrome/browser/media/router/media_router_type_converters.h"
+#include "chrome/browser/media/router/media_routes_observer.h"
+#include "chrome/browser/media/router/media_sinks_observer.h"
+#include "extensions/browser/process_manager.h"
+
+#define DVLOG_WITH_INSTANCE(level) \
+  DVLOG(level) << "MR #" << instance_id_ << ": "
+
+#define DVLOG_WITH_INSTANCE(level) DLOG(level) << "MR #" << instance_id_ << ": "
+
+namespace media_router {
+namespace {
+
+// Converts the callback result of calling Mojo CreateRoute() into a local
+// callback.
+void CreateRouteFinished(const MediaSinkId& sink_id,
+                         const MediaRouteResponseCallback& callback,
+                         interfaces::MediaRoutePtr media_route,
+                         const mojo::String& error_text) {
+  if (media_route.is_null()) {
+    // An error occurred.
+    DCHECK(!error_text.is_null());
+    callback.Run(nullptr, !error_text.get().empty() ? error_text.get()
+                                                    : "Unknown error.");
+    return;
+  }
+  callback.Run(make_scoped_ptr(new MediaRoute(media_route.To<MediaRoute>())),
+               "");
+}
+
+void EventPageWakeComplete(bool success) {
+  if (!success)
+    LOG(ERROR) << "An error encountered while waking the event page.";
+}
+
+}  // namespace
+
+MediaRouterMojoImpl::MediaRouterMojoImpl()
+    : event_page_tracker_(nullptr), instance_id_(base::GenerateGUID()) {
+}
+
+MediaRouterMojoImpl::MediaRouterMojoImpl(
+    const std::string& mrpm_extension_id,
+    extensions::EventPageTracker* event_page_tracker_for_test)
+    : MediaRouterMojoImpl() {
+  DCHECK(!mrpm_extension_id.empty());
+  DCHECK(event_page_tracker_for_test);
+  mrpm_extension_id_ = mrpm_extension_id;
+  event_page_tracker_ = event_page_tracker_for_test;
+}
+
+MediaRouterMojoImpl::~MediaRouterMojoImpl() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+}
+
+// static
+void MediaRouterMojoImpl::BindToRequest(
+    const std::string& extension_id,
+    content::BrowserContext* context,
+    mojo::InterfaceRequest<interfaces::MediaRouterObserver> request) {
+  MediaRouterMojoImpl* impl =
+      MediaRouterMojoImplFactory::GetApiForBrowserContext(context);
+  DCHECK(impl);
+  impl->Bind(request.Pass());
+  impl->MonitorExtension(extension_id, context);
+}
+
+void MediaRouterMojoImpl::Bind(
+    mojo::InterfaceRequest<interfaces::MediaRouterObserver> request) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  binding_.reset(
+      new mojo::Binding<interfaces::MediaRouterObserver>(this, request.Pass()));
+  binding_->set_error_handler(this);
+}
+
+void MediaRouterMojoImpl::OnConnectionError() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  mojo_media_router_.reset();
+  binding_.reset();
+}
+
+void MediaRouterMojoImpl::ProvideMediaRouter(
+    interfaces::MediaRouterPtr media_router_ptr,
+    const interfaces::MediaRouterObserver::ProvideMediaRouterCallback&
+        callback) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  mojo_media_router_ = media_router_ptr.Pass();
+  mojo_media_router_.set_error_handler(this);
+  callback.Run(instance_id_);
+  ExecutePendingRequests();

[imcheng (use chromium acct), 2015/05/14 21:30:13]

Consider inlining this since there's only 1 caller.

[Kevin M, 2015/05/14 22:46:35]

I added an "inline" keyword. I think that organizing it under its own function
adds clarity to the code structure. Do you still think I need to inline the code
itself?

+}
+
+void MediaRouterMojoImpl::OnMessage(const mojo::String& route_id,
+                                    const mojo::String& message) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  NOTIMPLEMENTED();
+}
+
+void MediaRouterMojoImpl::OnIssue(const interfaces::IssuePtr issue) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  NOTIMPLEMENTED();
+}
+
+void MediaRouterMojoImpl::OnSinksReceived(
+    const mojo::String& media_source,
+    mojo::Array<interfaces::MediaSinkPtr> sinks) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DVLOG_WITH_INSTANCE(1) << "OnSinksReceived";
+  std::vector<MediaSink> sinks_converted;
+  sinks_converted.reserve(sinks.size());
+
+  for (size_t i = 0; i < sinks.size(); ++i) {
+    sinks_converted.push_back(sinks[i].To<MediaSink>());
+  }
+
+  auto it = sinks_observers_.find(media_source);
+  if (it == sinks_observers_.end()) {
+    DVLOG_WITH_INSTANCE(1)
+        << "Received sink list without any active observers: " << media_source;
+  } else {
+    FOR_EACH_OBSERVER(MediaSinksObserver, *it->second,
+                      OnSinksReceived(sinks_converted));
+  }
+}
+
+void MediaRouterMojoImpl::OnRoutesUpdated(
+    mojo::Array<interfaces::MediaRoutePtr> routes) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DVLOG_WITH_INSTANCE(1) << "OnRoutesUpdated";
+
+  std::vector<MediaRoute> routes_converted;
+  routes_converted.reserve(routes.size());
+
+  for (size_t i = 0; i < routes.size(); ++i) {
+    routes_converted.push_back(routes[i].To<MediaRoute>());
+  }
+
+  FOR_EACH_OBSERVER(MediaRoutesObserver, routes_observers_,
+                    OnRoutesUpdated(routes_converted));
+}
+
+// ----------------------------------------------------------------------------
+// MediaRouter methods.
+
+void MediaRouterMojoImpl::CreateRoute(
+    const MediaSourceId& source_id,
+    const MediaSinkId& sink_id,
+    const MediaRouteResponseCallback& callback) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoCreateRoute,
+                        base::Unretained(this), source_id, sink_id, callback));
+}
+
+void MediaRouterMojoImpl::CloseRoute(const MediaRouteId& route_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoCloseRoute,
+                        base::Unretained(this), route_id));
+}
+
+void MediaRouterMojoImpl::PostMessage(const MediaRouteId& route_id,
+                                      const std::string& message) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoPostMessage,
+                        base::Unretained(this), route_id, message));
+}
+
+void MediaRouterMojoImpl::ClearIssue(const Issue::IssueId& issue_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoClearIssue,
+                        base::Unretained(this), issue_id));
+}
+
+bool MediaRouterMojoImpl::RegisterMediaSinksObserver(
+    MediaSinksObserver* observer) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  // Create an observer list for the media source and add |observer|
+  // to it. Fail if |observer| is already registered.
+  const std::string& source_id = observer->source().id();
+  linked_ptr<ObserverList<MediaSinksObserver>> observer_list =
+      sinks_observers_[source_id];
+  if (!observer_list.get()) {
+    observer_list = make_linked_ptr(new ObserverList<MediaSinksObserver>);
+    sinks_observers_[source_id] = observer_list;
+  } else {
+    if (observer_list->HasObserver(observer)) {
+      DLOG_WITH_INSTANCE(FATAL)
+          << "Redundant RegisterMediaSinksObserver call detected.";
+      return true;

[imcheng (use chromium acct), 2015/05/14 21:30:13]

return false instead?

[Kevin M, 2015/05/14 22:46:35]

Done

+    }
+  }
+
+  // Turn on sink observation if this is the first time we've started observing
+  // sinks for the media source.
+  if (!observer_list->might_have_observers()) {

[imcheng (use chromium acct), 2015/05/14 21:30:13]

Because we don't have caching anymore, we should always be calling
DoStartObservingMediaSinks whenever a sink is added. Otherwise sinks other than
the first will not receive results.

[Kevin M, 2015/05/14 22:46:35]

Done.

+    RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoStartObservingMediaSinks,
+                          base::Unretained(this), source_id));
+  }
+  observer_list->AddObserver(observer);
+
+  return true;
+}
+
+void MediaRouterMojoImpl::UnregisterMediaSinksObserver(
+    MediaSinksObserver* observer) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  const std::string& source_id = observer->source().id();
+  const auto observer_list_it = sinks_observers_.find(source_id);
+  if (observer_list_it == sinks_observers_.end() ||
+      !observer_list_it->second->HasObserver(observer)) {
+    return;
+  }
+  observer_list_it->second->RemoveObserver(observer);
+
+  // If we are removing the final observer for the source, then stop
+  // observing sinks for it.
+  if (!observer_list_it->second->might_have_observers()) {
+    RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoStopObservingMediaSinks,
+                          base::Unretained(this), source_id));
+    sinks_observers_.erase(observer_list_it);
+  }
+}
+
+bool MediaRouterMojoImpl::RegisterMediaRoutesObserver(
+    MediaRoutesObserver* observer) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (routes_observers_.HasObserver(observer)) {
+    DLOG(FATAL) << "Redundant RegisterMediaRoutesObserver() call detected.";
+    return true;
+  }
+
+  // TODO(kmarshall): add result caching and max limits.

[imcheng (use chromium acct), 2015/05/14 21:30:13]

ditto on caching

[Kevin M, 2015/05/14 22:46:35]

Done.

+  if (!routes_observers_.might_have_observers()) {
+    RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoStartObservingMediaRoutes,
+                          base::Unretained(this)));
+  }
+  routes_observers_.AddObserver(observer);
+  return true;
+}
+
+void MediaRouterMojoImpl::UnregisterMediaRoutesObserver(
+    MediaRoutesObserver* observer) {
+  if (!routes_observers_.HasObserver(observer)) {

[imcheng (use chromium acct), 2015/05/14 21:30:13]

curly brace not needed for a single line

[Kevin M, 2015/05/14 22:46:35]

Done.

+    return;
+  }
+  routes_observers_.RemoveObserver(observer);
+  if (!routes_observers_.might_have_observers()) {
+    RunOrDefer(base::Bind(&MediaRouterMojoImpl::DoStopObservingMediaRoutes,
+                          base::Unretained(this)));
+  }
+}
+
+void MediaRouterMojoImpl::AddIssuesObserver(IssuesObserver* observer) {
+  NOTIMPLEMENTED();
+}
+
+void MediaRouterMojoImpl::RemoveIssuesObserver(IssuesObserver* observer) {
+  NOTIMPLEMENTED();
+}
+
+void MediaRouterMojoImpl::DoCreateRoute(
+    const MediaSourceId& source_id,
+    const MediaSinkId& sink_id,
+    const MediaRouteResponseCallback& callback) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "CreateRoute " << source_id << "=>" << sink_id;
+  mojo_media_router_->CreateRoute(
+      source_id, sink_id, base::Bind(&CreateRouteFinished, sink_id, callback));
+}
+
+void MediaRouterMojoImpl::DoCloseRoute(const MediaRouteId& route_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "CloseRoute " << route_id;
+  mojo_media_router_->CloseRoute(route_id);
+}
+
+void MediaRouterMojoImpl::DoPostMessage(const MediaRouteId& route_id,
+                                        const std::string& message) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "PostMessage " << route_id;
+  mojo_media_router_->PostMessage(route_id, message);
+}
+
+void MediaRouterMojoImpl::DoClearIssue(const Issue::IssueId& issue_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "ClearIssue " << issue_id;
+  mojo_media_router_->ClearIssue(issue_id);
+}
+
+void MediaRouterMojoImpl::DoStartObservingMediaSinks(
+    const std::string& source_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "StartObservingMediaSinks: " << source_id;
+  mojo_media_router_->StartObservingMediaSinks(source_id);
+}
+
+void MediaRouterMojoImpl::DoStopObservingMediaSinks(
+    const std::string& source_id) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "StopObservingMediaSinks: " << source_id;
+  mojo_media_router_->StopObservingMediaSinks(source_id);
+}
+
+void MediaRouterMojoImpl::DoStartObservingMediaRoutes() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "StartObservingMediaRoutes";
+  mojo_media_router_->StartObservingMediaRoutes();
+}
+
+void MediaRouterMojoImpl::DoStopObservingMediaRoutes() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DVLOG_WITH_INSTANCE(1) << "StopObservingMediaRoutes";
+  mojo_media_router_->StopObservingMediaRoutes();
+}
+
+void MediaRouterMojoImpl::EnqueueTask(const base::Closure& closure) {
+  DVLOG_WITH_INSTANCE(2) << "EnqueueTask (size=" << pending_requests_.size()
+                         << ")";
+  pending_requests_.push_back(closure);
+}
+
+void MediaRouterMojoImpl::MonitorExtension(const std::string& extension_id,
+                                           content::BrowserContext* context) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(context);
+  mrpm_extension_id_ = extension_id;
+  event_page_tracker_ = extensions::ProcessManager::Get(context);
+}
+
+void MediaRouterMojoImpl::RunOrDefer(const base::Closure& request_cb) {
+  DCHECK(event_page_tracker_);
+  if (event_page_tracker_->IsEventPageSuspended(mrpm_extension_id_)) {
+    DVLOG_WITH_INSTANCE(1) << "Waking event page.";
+    pending_requests_.push_back(request_cb);
+    if (!event_page_tracker_->WakeEventPage(
+            mrpm_extension_id_, base::Bind(&EventPageWakeComplete))) {
+      LOG(ERROR) << "An error encountered while waking the event page.";
+    }
+    mojo_media_router_.reset();
+  } else if (!mojo_media_router_) {
+    DVLOG_WITH_INSTANCE(1) << "Extension is awake, awaiting MRPM connection.";
+    pending_requests_.push_back(request_cb);
+  } else {
+    request_cb.Run();

[imcheng (use chromium acct), 2015/05/14 21:30:13]

I wonder if we can add a DCHECK(mojo_media_router_); here and remove it from all
of the Do* functions. Also I don't think we need
DCHECK(thread_checker_.CalledOnValidThread()); in the Do* functions as the
callers already have that DCHECK.

[Kevin M, 2015/05/14 22:46:35]

No need to have a DCHECK on mojo_media_router_ because the conditional already
enforces that condition.

Removed thread checker & mojo_media_router_ checks from Do* fns.

+  }
+}
+
+void MediaRouterMojoImpl::ExecutePendingRequests() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(mojo_media_router_);
+  DCHECK(event_page_tracker_);
+  if (event_page_tracker_->IsEventPageSuspended(mrpm_extension_id_)) {

[imcheng (use chromium acct), 2015/05/14 21:30:13]

In what conditions will that happen? Should it be DCHECK instead?

[Kevin M, 2015/05/14 22:46:35]

If the extension shuts down immediately after ProvideMediaRouter is called. Not
likely to occur but it's worth having a check, anyways. It's not costly to
verify, so I don't think we need to demote it to a debug-only check.

+    DVLOG_WITH_INSTANCE(1)
+        << "ExecutePendingRequests was called while extension is suspended.";
+    return;
+  }
+
+  for (const auto& next_request : pending_requests_) {

[imcheng (use chromium acct), 2015/05/14 21:30:13]

curly braces not needed

[Kevin M, 2015/05/14 22:46:35]

Done.

+    next_request.Run();
+  }
+  pending_requests_.clear();
+}
+
+}  // namespace media_router