[Media Router] Custom Controls 2 - add MediaRouter::GetRouteController()

chrome/browser/media/router/media_router.h

 // 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.
 
 #ifndef CHROME_BROWSER_MEDIA_ROUTER_MEDIA_ROUTER_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_MEDIA_ROUTER_H_
 
 #include <stdint.h>
 
 #include <string>
@@ skipping 15 matching lines @@
 class WebContents;
 }
 
 namespace url {
 class Origin;
 }  // namespace url
 
 namespace media_router {
 
 class IssuesObserver;
+class MediaRouteController;
 class MediaRoutesObserver;
 class MediaSinksObserver;
 class PresentationConnectionStateObserver;
 class RouteRequestResult;
 
 // Type of callback used in |CreateRoute()|, |JoinRoute()|, and
 // |ConnectRouteByRouteId()|. Callback is invoked when the route request either
 // succeeded or failed.
 using MediaRouteResponseCallback =
     base::Callback<void(const RouteRequestResult& result)>;
@@ skipping 143 matching lines @@
       const content::PresentationConnectionStateChangedCallback& callback) = 0;
 
   // Called when the incognito profile for this instance is being shut down.
   // This will terminate all incognito media routes.
   virtual void OnIncognitoProfileShutdown() = 0;
 
   // Returns the media routes that currently exist. To get notified whenever
   // there is a change to the media routes, subclass MediaRoutesObserver.
   virtual std::vector<MediaRoute> GetCurrentRoutes() const = 0;
 
+  // Returns a controller for sending media commands to a route. Returns a
+  // nullptr if no MediaRoute exists for the given |route_id|.
+  virtual scoped_refptr<MediaRouteController> GetRouteController(
+      const MediaRoute::Id& route_id) = 0;
+
  private:
   friend class IssuesObserver;
   friend class MediaSinksObserver;
+  friend class MediaRouteController;
   friend class MediaRoutesObserver;
   friend class PresentationConnectionStateObserver;
   friend class RouteMessageObserver;
 
   // The following functions are called by friend Observer classes above.
 
   // Registers |observer| with this MediaRouter. |observer| specifies a media
   // source and will receive updates with media sinks that are compatible with
   // that source. The initial update may happen synchronously.
   // NOTE: This class does not assume ownership of |observer|. Callers must
@@ skipping 35 matching lines @@
   // route and will receive messages from the MediaSink connected to the
   // route. Note that MediaRouter does not own |observer|. |observer| should be
   // unregistered before it is destroyed. Registering the same observer more
   // than once will result in undefined behavior.
   virtual void RegisterRouteMessageObserver(RouteMessageObserver* observer) = 0;
 
   // Unregisters a previously registered RouteMessagesObserver. |observer| will
   // stop receiving further updates.
   virtual void UnregisterRouteMessageObserver(
       RouteMessageObserver* observer) = 0;
+
+  // Removes the MediaRouteController for |route_id| from the list of
+  // controllers held by |this|. Called by MediaRouteController when it is
+  // invalidated.
+  virtual void DetachRouteController(const MediaRoute::Id& route_id,
+                                     MediaRouteController* controller) = 0;
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_MEDIA_ROUTER_H_