[Presentation API] (5th) (1-UA) integrate controller and receiver side for 1-UA messaging

content/renderer/presentation/presentation_dispatcher.cc

 // 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.
 
 #include "content/renderer/presentation/presentation_dispatcher.h"
 
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 58 matching lines @@
     case blink::mojom::PresentationConnectionCloseReason::CLOSED:
       return blink::WebPresentationConnectionCloseReason::Closed;
     case blink::mojom::PresentationConnectionCloseReason::WENT_AWAY:
       return blink::WebPresentationConnectionCloseReason::WentAway;
     default:
       NOTREACHED();
       return blink::WebPresentationConnectionCloseReason::Error;
   }
 }
 
+// TODO(zhaobin): remove this function and use typemap instead after
+// crbug.com/632623 lands.
+blink::mojom::SessionMessagePtr ToMojoTextMessage(
+    const blink::WebString& message) {
+  blink::mojom::SessionMessagePtr session_message =
+      blink::mojom::SessionMessage::New();
+  session_message->type = blink::mojom::PresentationMessageType::TEXT;
+  session_message->message = message.utf8();
+  return session_message;
+}
+
+// TODO(zhaobin): remove this function and use typemap instead after
+// crbug.com/632623 lands.
+blink::mojom::SessionMessagePtr ToMojoBinaryMessage(
+    blink::mojom::PresentationMessageType type,
+    const uint8_t* data,
+    size_t length) {
+  blink::mojom::SessionMessagePtr session_message =
+      blink::mojom::SessionMessage::New();
+  session_message->type = type;
+  session_message->data = std::vector<uint8_t>(data, data + length);
+  return session_message;
+}
+
 }  // namespace
 
 namespace content {
 
 PresentationDispatcher::PresentationDispatcher(RenderFrame* render_frame)
     : RenderFrameObserver(render_frame),
       controller_(nullptr),
-      binding_(this) {
-}
+      receiver_(nullptr),
+      binding_(this) {}
 
 PresentationDispatcher::~PresentationDispatcher() {
   // Controller should be destroyed before the dispatcher when frame is
   // destroyed.
   DCHECK(!controller_);
 }
 
 void PresentationDispatcher::setController(
     blink::WebPresentationController* controller) {
   // There shouldn't be any swapping from one non-null controller to another.
@@ skipping 102 matching lines @@
   // Start processing request if only one in the queue.
   if (message_request_queue_.size() == 1)
     DoSendMessage(message_request_queue_.front().get(), connection_proxy);
 }
 
 void PresentationDispatcher::DoSendMessage(
     SendMessageRequest* request,
     const blink::WebPresentationConnectionProxy* connection_proxy) {
   ConnectToPresentationServiceIfNeeded();
 
-  presentation_service_->SendSessionMessage(
-      std::move(request->session_info), std::move(request->message),
-      base::Bind(&PresentationDispatcher::HandleSendMessageRequests,
-                 base::Unretained(this), nullptr));
+  static_cast<const PresentationConnectionProxy*>(connection_proxy)
+      ->SendSessionMessage(std::move(request->session_info),
+                           std::move(request->message));
 }
 
 void PresentationDispatcher::HandleSendMessageRequests(
     const blink::WebPresentationConnectionProxy* connection_proxy,
     bool success) {
   // In normal cases, message_request_queue_ should not be empty at this point
   // of time, but when DidCommitProvisionalLoad() is invoked before receiving
   // the callback for previous send mojo call, queue would have been emptied.
   if (message_request_queue_.empty())
     return;
@@ skipping 154 matching lines @@
   }
   status->availability_callbacks.Clear();
   UpdateListeningState(status);
 }
 
 void PresentationDispatcher::OnDefaultSessionStarted(
     blink::mojom::PresentationSessionInfoPtr session_info) {
   if (!controller_)
     return;
 
-  if (!session_info.is_null()) {
-    presentation_service_->ListenForSessionMessages(session_info.Clone());
-    controller_->didStartDefaultSession(
-        new PresentationConnectionClient(std::move(session_info)));
-  }
+  if (session_info.is_null())
+    return;
+
+  presentation_service_->ListenForSessionMessages(session_info.Clone());
+  controller_->didStartDefaultSession(new PresentationConnectionClient(
+      session_info->url, session_info->id,
+      CreateAndSetControllerConnectionProxy(session_info.Clone())));
 }
 
 void PresentationDispatcher::OnSessionCreated(
     blink::WebPresentationConnectionClientCallbacks* callback,
     blink::mojom::PresentationSessionInfoPtr session_info,
     blink::mojom::PresentationErrorPtr error) {
   DCHECK(callback);
   if (!error.is_null()) {
     DCHECK(session_info.is_null());
     callback->onError(blink::WebPresentationError(
         GetWebPresentationErrorTypeFromMojo(error->error_type),
         blink::WebString::fromUTF8(error->message)));
     return;
   }
 
   DCHECK(!session_info.is_null());
   presentation_service_->ListenForSessionMessages(session_info.Clone());
-  callback->onSuccess(
-      base::MakeUnique<PresentationConnectionClient>(std::move(session_info)));
+
+  callback->onSuccess(base::MakeUnique<PresentationConnectionClient>(
+      session_info->url, session_info->id,
+      CreateAndSetControllerConnectionProxy(session_info.Clone())));
 }
 
 void PresentationDispatcher::OnReceiverConnectionAvailable(
-    blink::mojom::PresentationSessionInfoPtr session_info) {
-  if (receiver_) {
-    receiver_->onReceiverConnectionAvailable(
-        new PresentationConnectionClient(std::move(session_info)));
-  }
+    blink::mojom::PresentationSessionInfoPtr session_info,
+    blink::mojom::PresentationConnectionPtr controller) {
+  DCHECK(receiver_);
+
+  std::unique_ptr<PresentationConnectionProxy> receiver_connection_proxy =
+      base::MakeUnique<PresentationConnectionProxy>();
+
+  receiver_connection_proxy->RegisterOnMessageCallback(
+      base::Bind(&PresentationDispatcher::HandleSendMessageRequests,
+                 base::Unretained(this), receiver_connection_proxy.get()));
+
+  PresentationConnectionProxy* receiver_connection_ptr =
+      receiver_connection_proxy.get();
+  // Bind receiver_connection_proxy with PresentationConnection
+  // in receiver page.
+  receiver_->onReceiverConnectionAvailable(
+      new PresentationConnectionClient(session_info->url, session_info->id,
+                                       std::move(receiver_connection_proxy)));
+
+  controller->SetTargetConnection(receiver_connection_ptr->Bind());
+  receiver_connection_ptr->SetTargetConnection(std::move(controller));
 }
 
 void PresentationDispatcher::OnConnectionStateChanged(
     blink::mojom::PresentationSessionInfoPtr connection,
     blink::mojom::PresentationConnectionState state) {
   if (!controller_)
     return;
 
   DCHECK(!connection.is_null());
   controller_->didChangeSessionState(
@@ skipping 69 matching lines @@
   ConnectToPresentationServiceIfNeeded();
   if (should_listen) {
     status->listening_state = ListeningState::WAITING;
     presentation_service_->ListenForScreenAvailability(status->url);
   } else {
     status->listening_state = ListeningState::INACTIVE;
     presentation_service_->StopListeningForScreenAvailability(status->url);
   }
 }
 
+std::unique_ptr<PresentationConnectionProxy>
+PresentationDispatcher::CreateAndSetControllerConnectionProxy(
+    blink::mojom::PresentationSessionInfoPtr session_info) {
+  std::unique_ptr<PresentationConnectionProxy> controller_connection_proxy =
+      base::MakeUnique<PresentationConnectionProxy>();
+  controller_connection_proxy->RegisterOnMessageCallback(
+      base::Bind(&PresentationDispatcher::HandleSendMessageRequests,
+                 base::Unretained(this), controller_connection_proxy.get()));
+
+  // Pass controller_connection_proxy to PSImpl and register
+  // it with OffscreenPresentationManager.
+  presentation_service_->SetPresentationConnection(
+      session_info.Clone(), controller_connection_proxy->Bind());
+
+  return controller_connection_proxy;
+}
+
 PresentationDispatcher::SendMessageRequest::SendMessageRequest(
     blink::mojom::PresentationSessionInfoPtr session_info,
     blink::mojom::SessionMessagePtr message)
     : session_info(std::move(session_info)), message(std::move(message)) {}
 
 PresentationDispatcher::SendMessageRequest::~SendMessageRequest() {}
 
 // static
 PresentationDispatcher::SendMessageRequest*
 PresentationDispatcher::CreateSendTextMessageRequest(
     const blink::WebURL& presentationUrl,
     const blink::WebString& presentationId,
     const blink::WebString& message) {
   blink::mojom::PresentationSessionInfoPtr session_info =
       blink::mojom::PresentationSessionInfo::New();
   session_info->url = presentationUrl;
   session_info->id = presentationId.utf8();
 
-  blink::mojom::SessionMessagePtr session_message =
-      blink::mojom::SessionMessage::New();
-  session_message->type = blink::mojom::PresentationMessageType::TEXT;
-  session_message->message = message.utf8();
   return new SendMessageRequest(std::move(session_info),
-                                std::move(session_message));
+                                ToMojoTextMessage(message));
 }
 
 // static
 PresentationDispatcher::SendMessageRequest*
 PresentationDispatcher::CreateSendBinaryMessageRequest(
     const blink::WebURL& presentationUrl,
     const blink::WebString& presentationId,
     blink::mojom::PresentationMessageType type,
     const uint8_t* data,
     size_t length) {
   blink::mojom::PresentationSessionInfoPtr session_info =
       blink::mojom::PresentationSessionInfo::New();
   session_info->url = presentationUrl;
   session_info->id = presentationId.utf8();
 
-  blink::mojom::SessionMessagePtr session_message =
-      blink::mojom::SessionMessage::New();
-  session_message->type = type;
-  session_message->data = std::vector<uint8_t>(data, data + length);
   return new SendMessageRequest(std::move(session_info),
-                                std::move(session_message));
+                                ToMojoBinaryMessage(type, data, length));
 }
 
 PresentationDispatcher::AvailabilityStatus::AvailabilityStatus(
     const GURL& availability_url)
     : url(availability_url),
       last_known_availability(false),
       listening_state(ListeningState::INACTIVE) {}
 
 PresentationDispatcher::AvailabilityStatus::~AvailabilityStatus() {
 }
 
 }  // namespace content