[Media Router] Custom Controls 3 - add plumbing through MRUI and MRWebUIMessageHandler

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 <unordered_map>
 #include <vector>
 
 #include "base/macros.h"
 #include "chrome/browser/ui/webui/media_router/media_cast_mode.h"
 #include "chrome/browser/ui/webui/media_router/media_sink_with_cast_modes.h"
 #include "chrome/common/media_router/issue.h"
+#include "chrome/common/media_router/media_status.h"
 #include "components/signin/core/browser/account_info.h"
 #include "content/public/browser/web_ui_message_handler.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace base {
 class DictionaryValue;
 class ListValue;
 }  // namespace base
 
 namespace content {
@@ skipping 24 matching lines @@
                                      const MediaRoute* route);
   void ReturnSearchResult(const std::string& sink_id);
 
   void UpdateIssue(const Issue& issue);
   void ClearIssue();
 
   // Updates the maximum dialog height to allow the WebUI properly scale when
   // the browser window changes.
   void UpdateMaxDialogHeight(int height);
 
+  // Notifies the WebUI with an updated MediaStatus. Overridden in tests.
+  virtual void UpdateMediaRouteStatus(const MediaStatus& status);
+
+  // Notifies the WebUI that the controller for the selected route has been
+  // invalidated.
+  void OnRouteControllerInvalidated();
+
   void SetWebUIForTest(content::WebUI* webui);
   void set_incognito_for_test(bool incognito) { incognito_ = incognito; }
 
  private:
   FRIEND_TEST_ALL_PREFIXES(MediaRouterWebUIMessageHandlerTest,
                            RecordCastModeSelection);
   FRIEND_TEST_ALL_PREFIXES(MediaRouterWebUIMessageHandlerTest,
                            RetrieveCastModeSelection);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterWebUIMessageHandlerTest,
+                           OnRouteDetailsOpenedAndClosed);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterWebUIMessageHandlerTest,
+                           OnMediaCommandsReceived);
+  FRIEND_TEST_ALL_PREFIXES(MediaRouterWebUIMessageHandlerTest,
+                           OnInvalidMediaCommandsReceived);
 
   // WebUIMessageHandler implementation.
   void RegisterMessages() override;
 
   // Handlers for JavaScript messages.
   // See media_router_ui_interface.js for documentation on parameters.
   void OnRequestInitialData(const base::ListValue* args);
   void OnCreateRoute(const base::ListValue* args);
   void OnAcknowledgeFirstRunFlow(const base::ListValue* args);
   void OnActOnIssue(const base::ListValue* args);
   void OnCloseRoute(const base::ListValue* args);
   void OnJoinRoute(const base::ListValue* args);
   void OnCloseDialog(const base::ListValue* args);
   void OnReportBlur(const base::ListValue* args);
   void OnReportClickedSinkIndex(const base::ListValue* args);
   void OnReportFilter(const base::ListValue* args);
   void OnReportInitialAction(const base::ListValue* args);
   void OnReportInitialState(const base::ListValue* args);
   void OnReportNavigateToView(const base::ListValue* args);
   void OnReportRouteCreation(const base::ListValue* args);
   void OnReportRouteCreationOutcome(const base::ListValue* args);
   void OnReportSelectedCastMode(const base::ListValue* args);
   void OnReportSinkCount(const base::ListValue* args);
   void OnReportTimeToClickSink(const base::ListValue* args);
   void OnReportTimeToInitialActionClose(const base::ListValue* args);
+  void OnMediaControllerAvailable(const base::ListValue* args);
+  void OnMediaControllerClosed(const base::ListValue* args);
   void OnSearchSinksAndCreateRoute(const base::ListValue* args);
   void OnInitialDataReceived(const base::ListValue* args);
 
+  // Handlers for JavaScript messages to control the media.
+  void OnPlayCurrentMedia(const base::ListValue* args);
+  void OnPauseCurrentMedia(const base::ListValue* args);
+  void OnSeekCurrentMedia(const base::ListValue* args);
+  void OnSetCurrentMediaMute(const base::ListValue* args);
+  void OnSetCurrentMediaVolume(const base::ListValue* args);
+
   // Performs an action for an Issue of |type|.
   // |args| contains additional parameter that varies based on |type|.
   // Returns |true| if the action was successfully performed.
   bool ActOnIssueType(IssueInfo::Action 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();
@@ skipping 17 matching lines @@
   // used when updating sinks to determine if identity should be displayed.
   // Marked virtual for tests.
   virtual AccountInfo GetAccountInfo();
 
   // |true| if the associated Profile is incognito.
   bool incognito_;
 
   // Keeps track of whether a command to close the dialog has been issued.
   bool dialog_closing_;
 
+  // The media status currently shown in the UI.
+  base::Optional<MediaStatus> current_media_status_;
+
   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_