[Presentation API] 1-UA: send message between controller and receiver page

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 63 matching lines @@
   }
 }
 
 }  // 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 332 matching lines @@
   DCHECK(!connection.is_null());
   controller_->didCloseConnection(
       new PresentationConnectionClient(std::move(connection)),
       GetWebPresentationConnectionCloseReasonFromMojo(reason),
       blink::WebString::fromUTF8(message));
 }
 
 void PresentationDispatcher::OnSessionMessagesReceived(
     blink::mojom::PresentationSessionInfoPtr session_info,
     std::vector<blink::mojom::SessionMessagePtr> messages) {
-  if (!controller_)
+  auto observer =
+      (receiver_
+           ? static_cast<blink::WebPresentationMessageObserver*>(receiver_)
+           : static_cast<blink::WebPresentationMessageObserver*>(controller_));
+  if (!observer)
     return;
 
   for (size_t i = 0; i < messages.size(); ++i) {
     // Note: Passing batches of messages to the Blink layer would be more
     // efficient.
     std::unique_ptr<PresentationConnectionClient> session_client(
         new PresentationConnectionClient(session_info->url, session_info->id));
     switch (messages[i]->type) {
       case blink::mojom::PresentationMessageType::TEXT: {
         // TODO(mfoltz): Do we need to DCHECK(messages[i]->message)?
-        controller_->didReceiveSessionTextMessage(
+        observer->didReceiveSessionTextMessage(
             session_client.release(),
             blink::WebString::fromUTF8(messages[i]->message.value()));
         break;
       }
       case blink::mojom::PresentationMessageType::ARRAY_BUFFER:
       case blink::mojom::PresentationMessageType::BLOB: {
         // TODO(mfoltz): Do we need to DCHECK(messages[i]->data)?
-        controller_->didReceiveSessionBinaryMessage(
-            session_client.release(), &(messages[i]->data->front()),
-            messages[i]->data->size());
+        observer->didReceiveSessionBinaryMessage(session_client.release(),
+                                                 &(messages[i]->data->front()),
+                                                 messages[i]->data->size());
         break;
       }
       default: {
         NOTREACHED();
         break;
       }
     }
   }
 }
 
@@ skipping 73 matching lines @@
 PresentationDispatcher::AvailabilityStatus::AvailabilityStatus(
     const GURL& availability_url)
     : url(availability_url),
       last_known_availability(false),
       listening_state(ListeningState::INACTIVE) {}
 
 PresentationDispatcher::AvailabilityStatus::~AvailabilityStatus() {
 }
 
 }  // namespace content