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

chrome/browser/ui/webui/media_router/media_router_webui_message_handler.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_WEBUI_MESSAGE_HANDLER_H_
 #define CHROME_BROWSER_UI_WEBUI_MEDIA_ROUTER_MEDIA_ROUTER_WEBUI_MESSAGE_HANDLER_H_
 
 #include <vector>
 
 #include "base/macros.h"
@@ skipping 21 matching lines @@
 
 // The handler for Javascript messages related to the media router dialog.
 class MediaRouterWebUIMessageHandler : public content::WebUIMessageHandler {
  public:
   explicit MediaRouterWebUIMessageHandler(MediaRouterUI* media_router_ui);
   ~MediaRouterWebUIMessageHandler() override;
 
   // Methods to update the status displayed by the dialog.
   void UpdateSinks(const std::vector<MediaSinkWithCastModes>& sinks);
   void UpdateRoutes(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);
   void UpdateCastModes(const CastModeSet& cast_modes,
                        const std::string& source_host);
   void OnCreateRouteResponseReceived(const MediaSink::Id& sink_id,
                                      const MediaRoute* route);
   void ReturnSearchResult(const std::string& sink_id);
 
   // Does not take ownership of |issue|. Note that |issue| can be nullptr, when
   // there are no more issues.
   void UpdateIssue(const Issue* issue);
 
@@ skipping 38 matching lines @@
   // |args| contains additional parameter that varies based on |type|.
   // Returns |true| if the action was successfully performed.
   bool ActOnIssueType(const IssueAction::Type& type,
                       const base::DictionaryValue* args);
 
   // May update the first run flow related properties in the WebUI. This is
   // called after the initial data is received to avoid unnecessary work when
   // initializing the WebUI.
   void MaybeUpdateFirstRunFlowData();
 
+  // Returns the current cast mode for the route with ID |route_id| or -1 if the
+  // route has no current cast mode.
+  int CurrentCastModeForRouteId(
+      const MediaRoute::Id& route_id,
+      const std::unordered_map<MediaRoute::Id, MediaCastMode>&
+          current_cast_modes) const;
+
   // Converts |routes| and |joinable_route_ids| into base::ListValue that can be
   // passed to WebUI.
   std::unique_ptr<base::ListValue> RoutesToValue(
       const std::vector<MediaRoute>& routes,
-      const std::vector<MediaRoute::Id>& joinable_route_ids) const;
+      const std::vector<MediaRoute::Id>& joinable_route_ids,
+      const std::unordered_map<MediaRoute::Id, MediaCastMode>&
+          current_cast_modes) const;
 
   // Retrieve the account info for email and domain of signed in users. This is
   // used when updating sinks to determine if identity should be displayed.
   // Marked virtual for tests.
   virtual AccountInfo GetAccountInfo();
 
   // |true| if the associated Profile is off the record.
   bool off_the_record_;
 
   // Keeps track of whether a command to close the dialog has been issued.
   bool dialog_closing_;
 
   MediaRouterUI* media_router_ui_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaRouterWebUIMessageHandler);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_UI_WEBUI_MEDIA_ROUTER_MEDIA_ROUTER_WEBUI_MESSAGE_HANDLER_H_