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

chrome/browser/media/router/media_router_base.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_BASE_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_MEDIA_ROUTER_BASE_H_
 
 #include <set>
 #include <unordered_map>
 #include <vector>
@@ skipping 16 matching lines @@
       const MediaRoute::Id& route_id,
       const content::PresentationConnectionStateChangedCallback& callback)
       override;
 
   // Called when the incognito profile for this instance is being shut down.
   // This will terminate all incognito media routes.
   void OnIncognitoProfileShutdown() override;
 
   std::vector<MediaRoute> GetCurrentRoutes() const override;
 
+  scoped_refptr<MediaRouteController> GetRouteController(
+      const MediaRoute::Id& route_id) override;
+
  protected:
   FRIEND_TEST_ALL_PREFIXES(MediaRouterMojoImplTest,
                            PresentationConnectionStateChangedCallback);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterMojoImplTest,
                            PresentationConnectionStateChangedCallbackRemoved);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterBaseTest, CreatePresentationIds);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterBaseTest, NotifyCallbacks);
 
   MediaRouterBase();
 
@@ skipping 31 matching lines @@
   void Initialize();
 
   // Called when a PresentationConnectionStateChangedCallback associated with
   // |route_id| is removed from |presentation_connection_state_callbacks_|.
   void OnPresentationConnectionStateCallbackRemoved(
       const MediaRoute::Id& route_id);
 
   // KeyedService
   void Shutdown() override;
 
+  // MediaRouter
+  void DetachRouteController(const MediaRoute::Id& route_id,
+                             MediaRouteController* controller) override;
+
   std::unique_ptr<InternalMediaRoutesObserver> internal_routes_observer_;
   bool initialized_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaRouterBase);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_MEDIA_ROUTER_BASE_H_