[Presentation API] Mojo typemap for content::PresentationConnectionMessage

content/common/presentation/presentation_struct_traits.h

 // 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.
 
 #ifndef CONTENT_COMMON_PRESENTATION_PRESENTATION_STRUCT_TRAITS_H_
 #define CONTENT_COMMON_PRESENTATION_PRESENTATION_STRUCT_TRAITS_H_
 
 #include <string>
+#include <vector>
 
 #include "base/macros.h"
+#include "base/optional.h"
 #include "base/strings/string_util.h"
+#include "content/public/common/presentation_connection_message.h"
 #include "content/public/common/presentation_session.h"
 #include "mojo/common/common_custom_types_struct_traits.h"
 #include "third_party/WebKit/public/platform/modules/presentation/presentation.mojom.h"
 #include "url/mojo/url.mojom.h"
 
 namespace mojo {
 
 template <>
 struct EnumTraits<blink::mojom::PresentationErrorType,
                   content::PresentationErrorType> {
@@ skipping 168 matching lines @@
       return false;
 
     if (!base::IsStringUTF8(out->message) ||
         out->message.length() > content::PresentationError::kMaxMessageLength)
       return false;
 
     return true;
   }
 };
 
+template <>
+struct UnionTraits<blink::mojom::PresentationConnectionMessageDataView,
+                   content::PresentationConnectionMessage> {
+  static blink::mojom::PresentationConnectionMessageDataView::Tag GetTag(
+      const content::PresentationConnectionMessage& message) {
+    return message.is_binary()
+               ? blink::mojom::PresentationConnectionMessageDataView::Tag::DATA
+               : blink::mojom::PresentationConnectionMessageDataView::Tag::
+                     MESSAGE;
+  }
+
+  static const std::string& message(
+      const content::PresentationConnectionMessage& message) {
+    DCHECK(!message.is_binary());
+    return message.message.value();
+  }
+
+  static const std::vector<uint8_t>& data(
+      const content::PresentationConnectionMessage& message) {
+    DCHECK(message.is_binary());
+    return message.data.value();
+  }
+
+  static bool Read(blink::mojom::PresentationConnectionMessageDataView data,
+                   content::PresentationConnectionMessage* out) {

[dcheng, 2017/02/25 07:17:27]

Please out-of-line this method.

+    if (data.is_message()) {
+      out->message = std::string();

[dcheng, 2017/02/25 07:17:27]

This is unnecessary.

+      if (!data.ReadMessage(&(out->message)))
+        return false;
+
+      if (out->message->length() >
+          content::kMaxPresentationConnectionMessageSize)
+        return false;
+    } else {
+      out->data = std::vector<uint8_t>();

[dcheng, 2017/02/25 07:17:27]

Nit: please omit this.

+      if (!data.ReadData(&(out->data)))
+        return false;
+
+      if (out->data->size() > content::kMaxPresentationConnectionMessageSize)
+        return false;
+    }
+
+    return true;
+  }
+};
+
 }  // namespace mojo
 
 #endif  // CONTENT_COMMON_PRESENTATION_PRESENTATION_STRUCT_TRAITS_H_