[Presentation API] Mojo typemap for content::PresentationConnectionMessage

content/renderer/presentation/presentation_connection_proxy.cc

 // Copyright 2017 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_connection_proxy.h"
 
 #include "base/logging.h"
 #include "content/public/common/presentation_session.h"
 #include "content/renderer/presentation/presentation_dispatcher.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/modules/presentation/WebPresentationConnection.h"
 #include "third_party/WebKit/public/platform/modules/presentation/WebPresentationController.h"
 #include "third_party/WebKit/public/platform/modules/presentation/WebPresentationSessionInfo.h"
 
 namespace content {
 
 PresentationConnectionProxy::PresentationConnectionProxy(
     blink::WebPresentationConnection* source_connection)
     : binding_(this),
       target_connection_ptr_(nullptr),
       source_connection_(source_connection) {
   DCHECK(source_connection_);
 }
 
 PresentationConnectionProxy::~PresentationConnectionProxy() = default;
 
 void PresentationConnectionProxy::SendConnectionMessage(
-    blink::mojom::ConnectionMessagePtr connection_message,
+    PresentationConnectionMessage message,
     const OnMessageCallback& callback) const {
   DCHECK(target_connection_ptr_);
-  target_connection_ptr_->OnMessage(std::move(connection_message), callback);
+  target_connection_ptr_->OnMessage(std::move(message), callback);
 }
 
 void PresentationConnectionProxy::OnMessage(
-    blink::mojom::ConnectionMessagePtr message,
+    PresentationConnectionMessage message,
     const OnMessageCallback& callback) {
   DCHECK(!callback.is_null());
 
-  switch (message->type) {
-    case blink::mojom::PresentationMessageType::TEXT: {
-      DCHECK(message->message);
-      source_connection_->didReceiveTextMessage(
-          blink::WebString::fromUTF8(message->message.value()));
-      break;
-    }
-    case blink::mojom::PresentationMessageType::BINARY: {
-      DCHECK(message->data);
-      source_connection_->didReceiveBinaryMessage(&(message->data->front()),
-                                                  message->data->size());
-      break;
-    }
-    default: {
-      callback.Run(false);
-      NOTREACHED();
-      return;
-    }
+  if (message.is_binary()) {
+    source_connection_->didReceiveBinaryMessage(&(message.data->front()),
+                                                message.data->size());
+  } else {
+    source_connection_->didReceiveTextMessage(
+        blink::WebString::fromUTF8(*(message.message)));
   }
 
   callback.Run(true);
 }
 
 // TODO(crbug.com/588874): Ensure legal PresentationConnection state transitions
 // in a single place.
 void PresentationConnectionProxy::DidChangeState(
     content::PresentationConnectionState state) {
   if (state == content::PRESENTATION_CONNECTION_STATE_CONNECTED) {
@@ skipping 52 matching lines @@
     blink::mojom::PresentationConnectionPtr controller_connection_ptr) {
   DCHECK(!target_connection_ptr_);
   target_connection_ptr_ = std::move(controller_connection_ptr);
   target_connection_ptr_->DidChangeState(
       content::PRESENTATION_CONNECTION_STATE_CONNECTED);
 
   DidChangeState(content::PRESENTATION_CONNECTION_STATE_CONNECTED);
 }
 
 }  // namespace content