[Presentation API] (browser side) Implement reconnect() for 1-UA mode

chrome/browser/media/router/offscreen_presentation_manager.h

 // Copyright 2016 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_OFFSCREEN_PRESENTATION_MANAGER_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_OFFSCREEN_PRESENTATION_MANAGER_H_
 
 #include <map>
 #include <memory>
 #include <string>
 #include <unordered_map>
 
 #include "base/macros.h"
+#include "base/optional.h"
 #include "base/threading/thread_checker.h"
+#include "chrome/browser/media/router/media_route.h"
 #include "chrome/browser/media/router/render_frame_host_id.h"
 #include "components/keyed_service/core/keyed_service.h"
 #include "content/public/browser/presentation_service_delegate.h"
 
 class GURL;
 
 namespace media_router {
 // Manages all offscreen presentations started in the associated Profile and
 // facilitates communication between the controllers and the receiver of an
 // offscreen presentation.
@@ skipping 78 matching lines @@
   // Creates a new presentation if no presentation with |presentation_id|
   // exists.
   // |controller_connection_ptr|, |receiver_connection_request|: Not owned by
   // this class. Ownership is transferred to presentation receiver via
   // |receiver_callback| passed below.
   virtual void RegisterOffscreenPresentationController(
       const std::string& presentation_id,
       const GURL& presentation_url,
       const RenderFrameHostId& render_frame_id,
       content::PresentationConnectionPtr controller_connection_ptr,
-      content::PresentationConnectionRequest receiver_connection_request);
+      content::PresentationConnectionRequest receiver_connection_request,
+      const MediaRoute& route);
 
   // Unregisters controller PresentationConnectionPtr to presentation with
   // |presentation_id|, |render_frame_id|. It does nothing if there is no
   // controller that matches the provided arguments. It removes presentation
   // that matches the arguments if the presentation has no |receiver_callback|
   // and any other pending controller.
   virtual void UnregisterOffscreenPresentationController(
       const std::string& presentation_id,
       const RenderFrameHostId& render_frame_id);
 
   // Registers |receiver_callback| to presentation with |presentation_id| and
   // |presentation_url|.
   virtual void OnOffscreenPresentationReceiverCreated(
       const std::string& presentation_id,
       const GURL& presentation_url,
       const content::ReceiverConnectionAvailableCallback& receiver_callback);
 
   // Unregisters ReceiverConnectionAvailableCallback associated with
   // |presentation_id|.
   virtual void OnOffscreenPresentationReceiverTerminated(
       const std::string& presentation_id);
 
+  // Returns true if this class has an offscreen presentation with
+  // |presentation_id|.
+  virtual bool IsOffscreenPresentation(const std::string& presentation_id);
+
+  // Returns nullptr if |presentation_id| is not associated with an offscreen
+  // presentation.
+  virtual const MediaRoute* GetRoute(const std::string& presentation_id);
+
  private:
   // Represents an offscreen presentation registered with
   // OffscreenPresentationManager. Contains callback to the receiver to inform
   // it of new connections established from a controller. Contains set of
   // controllers registered to OffscreenPresentationManager before corresponding
   // receiver.
   class OffscreenPresentation {
    public:
     OffscreenPresentation(const std::string& presentation_id,
                           const GURL& presentation_url);
     ~OffscreenPresentation();
 
     // Register controller with |render_frame_id|. If |receiver_callback_| has
     // been set, invoke |receiver_callback_| with |controller_connection_ptr|
     // and |receiver_connection_request| as parameter, else creates a
     // ControllerConnection object with |controller_connection_ptr| and
     // |receiver_connection_request|, and store it in |pending_controllers_|
     // map.
     void RegisterController(
         const RenderFrameHostId& render_frame_id,
         content::PresentationConnectionPtr controller_connection_ptr,
-        content::PresentationConnectionRequest receiver_connection_request);
+        content::PresentationConnectionRequest receiver_connection_request,
+        const MediaRoute& route);
 
     // Unregister controller with |render_frame_id|. Do nothing if there is no
     // pending controller with |render_frame_id|.
     void UnregisterController(const RenderFrameHostId& render_frame_id);
 
     // Register |receiver_callback| to current offscreen_presentation object.
     // For each controller in |pending_controllers_| map, invoke
     // |receiver_callback| with controller as parameter. Clear
     // |pending_controllers_| map afterwards.
     void RegisterReceiver(
         const content::ReceiverConnectionAvailableCallback& receiver_callback);
 
    private:
     friend class OffscreenPresentationManagerTest;
     friend class OffscreenPresentationManager;
 
     // Returns false if receiver_callback_ is null and there are no pending
     // controllers.
     bool IsValid() const;
 
     const std::string presentation_id_;
     const GURL presentation_url_;
+    base::Optional<MediaRoute> route_;
 
     // Callback to invoke whenever a receiver connection is available.
     content::ReceiverConnectionAvailableCallback receiver_callback_;
 
     // Stores controller information.
     // |controller_connection_ptr|: Mojo::InterfacePtr to
     // blink::PresentationConnection object in controlling frame;
     // |receiver_connection_request|: Mojo::InterfaceRequest to be bind to
     // blink::PresentationConnection object in receiver frame.
     struct ControllerConnection {
@@ skipping 40 matching lines @@
   OffscreenPresentationMap offscreen_presentations_;
 
   base::ThreadChecker thread_checker_;
 
   DISALLOW_COPY_AND_ASSIGN(OffscreenPresentationManager);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_OFFSCREEN_PRESENTATION_MANAGER_H_