[Media Router] Assign each route a current cast mode if possible

chrome/browser/ui/webui/media_router/media_router_ui.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_UI_WEBUI_MEDIA_ROUTER_MEDIA_ROUTER_UI_H_
 #define CHROME_BROWSER_UI_WEBUI_MEDIA_ROUTER_MEDIA_ROUTER_UI_H_
 
 #include <memory>
 #include <set>
 #include <string>
@@ skipping 119 matching lines @@
   std::string GetTruncatedPresentationRequestSourceName() const;
   bool HasPendingRouteRequest() const {
     return current_route_request_id_ != -1;
   }
   const std::vector<MediaSinkWithCastModes>& sinks() const { return sinks_; }
   const std::vector<MediaRoute>& routes() const { return routes_; }
   const std::vector<MediaRoute::Id>& joinable_route_ids() const {
     return joinable_route_ids_;
   }
   const std::set<MediaCastMode>& cast_modes() const { return cast_modes_; }
+  const std::unordered_map<MediaRoute::Id, MediaCastMode>& current_cast_modes()
+      const {
+    return current_cast_modes_;
+  }
   const content::WebContents* initiator() const { return initiator_; }
 
   // Marked virtual for tests.
   virtual const std::string& GetRouteProviderExtensionId() const;
 
   // Called to track UI metrics.
   void SetUIInitializationTimer(const base::Time& start_time);
   void OnUIInitiallyLoaded();
   void OnUIInitialDataReceived();
 
   void UpdateMaxDialogHeight(int height);
 
   void InitForTest(MediaRouter* router,
                    content::WebContents* initiator,
                    MediaRouterWebUIMessageHandler* handler);
 
  private:
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest, SortedSinks);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
                            UIMediaRoutesObserverFiltersNonDisplayRoutes);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
       UIMediaRoutesObserverFiltersNonDisplayJoinableRoutes);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
+      UIMediaRoutesObserverAssignsCurrentCastModes);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
+      UIMediaRoutesObserverSkipsUnavailableCastModes);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
+      UIMediaRoutesObserverFiltersNonDisplayJoinableRoutes);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest, GetExtensionNameExtensionPresent);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
                            GetExtensionNameEmptyWhenNotInstalled);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
                            GetExtensionNameEmptyWhenNotExtensionURL);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest,
                            RouteCreationTimeoutForPresentation);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterUITest, RouteRequestFromIncognito);
 
   class UIIssuesObserver;
 
   class UIMediaRoutesObserver : public MediaRoutesObserver {
    public:
-    using RoutesUpdatedCallback =
-        base::Callback<void(const std::vector<MediaRoute>&,
-            const std::vector<MediaRoute::Id>&)>;
-    UIMediaRoutesObserver(MediaRouter* router, const MediaSource::Id& source_id,
-                          const RoutesUpdatedCallback& callback);
+    using RoutesUpdatedCallback = base::Callback<void(
+        const std::vector<MediaRoute>&,
+        const std::vector<MediaRoute::Id>&,
+        const std::unordered_map<MediaRoute::Id, MediaCastMode>&)>;
+    using SourceCastModeMapCallback =
+        base::Callback<std::unordered_map<MediaSource::Id, MediaCastMode>()>;
+    UIMediaRoutesObserver(
+        MediaRouter* router,
+        const MediaSource::Id& source_id,
+        const SourceCastModeMapCallback& source_cast_mode_callback,
+        const RoutesUpdatedCallback& callback);
     ~UIMediaRoutesObserver() override;
 
     // MediaRoutesObserver
     void OnRoutesUpdated(
         const std::vector<MediaRoute>& routes,
         const std::vector<MediaRoute::Id>& joinable_route_ids) override;
 
    private:
-    // Callback to the owning MediaRouterUI instance.
+    // Callbacks to the owning MediaRouterUI instance.
+    SourceCastModeMapCallback source_cast_mode_callback_;
     RoutesUpdatedCallback callback_;
 
     DISALLOW_COPY_AND_ASSIGN(UIMediaRoutesObserver);
   };
 
   static std::string GetExtensionName(const GURL& url,
                                       extensions::ExtensionRegistry* registry);
 
   // QueryResultManager::Observer
   void OnResultsUpdated(
       const std::vector<MediaSinkWithCastModes>& sinks) override;
 
   // Called by |issues_observer_| when the top issue has changed.
   // If the UI is already initialized, notifies |handler_| to update the UI.
   // Ignored if the UI is not yet initialized.
   void SetIssue(const Issue* issue);
 
+  // Called by |routes_observer_| to get an updated map from media sources to
+  // their current cast mode.
+  std::unordered_map<MediaSource::Id, MediaCastMode> GetAvailableSourceMap();
+
   // Called by |routes_observer_| when the set of active routes has changed.
   void OnRoutesUpdated(const std::vector<MediaRoute>& routes,
-      const std::vector<MediaRoute::Id>& joinable_route_ids);
+      const std::vector<MediaRoute::Id>& joinable_route_ids,
+      const std::unordered_map<MediaRoute::Id, MediaCastMode>&
+          current_cast_modes);
 
   // Callback passed to MediaRouter to receive response to route creation
   // requests.
   void OnRouteResponseReceived(
       int route_request_id,
       const MediaSink::Id& sink_id,
       MediaCastMode cast_mode,
       const base::string16& presentation_request_source_name,
       const RouteRequestResult& result);
 
@@ skipping 57 matching lines @@
   int route_request_counter_;
 
   // Used for locale-aware sorting of sinks by name. Set during |InitCommon()|
   // using the current locale. Set to null
   std::unique_ptr<icu::Collator> collator_;
 
   std::vector<MediaSinkWithCastModes> sinks_;
   std::vector<MediaRoute> routes_;
   std::vector<MediaRoute::Id> joinable_route_ids_;
   CastModeSet cast_modes_;
+  std::unordered_map<MediaRoute::Id, MediaCastMode> current_cast_modes_;
 
   std::unique_ptr<QueryResultManager> query_result_manager_;
 
   // If set, then the result of the next presentation route request will
   // be handled by this object.
   std::unique_ptr<CreatePresentationConnectionRequest> create_session_request_;
 
   // Set to the presentation request corresponding to the presentation cast
   // mode, if supported. Otherwise set to nullptr.
   std::unique_ptr<PresentationRequest> presentation_request_;
@@ skipping 19 matching lines @@
   // NOTE: Weak pointers must be invalidated before all other member variables.
   // Therefore |weak_factory_| must be placed at the end.
   base::WeakPtrFactory<MediaRouterUI> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaRouterUI);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_UI_WEBUI_MEDIA_ROUTER_MEDIA_ROUTER_UI_H_