1 files changed, 101 insertions, 0 deletions
diff --git a/chromium/third_party/nearby/src/connections/payload.cc b/chromium/third_party/nearby/src/connections/payload.cc
new file mode 100644
index 00000000000..4979d0549ff
--- /dev/null
+++ b/chromium/third_party/nearby/src/connections/payload.cc
@@ -0,0 +1,101 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "connections/payload.h"
+
+namespace location {
+namespace nearby {
+namespace connections {
+
+// Payload is default-constructible, and moveable, but not copyable container
+// that holds at most one instance of one of:
+// ByteArray, InputStream, or InputFile.
+Payload::Payload(Payload&& other) noexcept = default;
+Payload::~Payload() = default;
+Payload& Payload::operator=(Payload&& other) noexcept = default;
+
+// Default (invalid) payload.
+Payload::Payload() : content_(absl::monostate()) {}
+
+// Constructors for outgoing payloads.
+Payload::Payload(ByteArray&& bytes) : content_(std::move(bytes)) {}
+
+Payload::Payload(const ByteArray& bytes) : content_(bytes) {}
+
+Payload::Payload(InputFile file)
+    : content_(std::move(file)),
+      id_(std::hash<std::string>()(file.GetFilePath())) {}
+
+// TODO(jfcarroll): Convert std::function to function pointer
+Payload::Payload(std::function<InputStream&()> stream)
+    : content_(std::move(stream)) {}
+
+// Constructors for incoming payloads.
+Payload::Payload(Id id, ByteArray&& bytes)
+    : content_(std::move(bytes)), id_(id) {}
+
+Payload::Payload(Id id, const ByteArray& bytes) : content_(bytes), id_(id) {}
+
+Payload::Payload(Id id, InputFile file) : content_(std::move(file)), id_(id) {}
+
+// TODO(jfcarroll): Convert std::function to function pointer
+Payload::Payload(Id id, std::function<InputStream&()> stream)
+    : content_(std::move(stream)), id_(id) {}
+
+// Returns ByteArray payload, if it has been defined, or empty ByteArray.
+const ByteArray& Payload::AsBytes() const& {
+  static const ByteArray empty;  // NOLINT: function-level static is OK.
+  auto* result = absl::get_if<ByteArray>(&content_);
+  return result ? *result : empty;
+}
+ByteArray&& Payload::AsBytes() && {
+  auto* result = absl::get_if<ByteArray>(&content_);
+  return result ? std::move(*result) : std::move(ByteArray());
+}
+// Returns InputStream* payload, if it has been defined, or nullptr.
+InputStream* Payload::AsStream() {
+  auto* result = absl::get_if<std::function<InputStream&()>>(&content_);
+  return result ? &(*result)() : nullptr;
+}
+// Returns InputFile* payload, if it has been defined, or nullptr.
+InputFile* Payload::AsFile() { return absl::get_if<InputFile>(&content_); }
+
+// Returns Payload unique ID.
+Payload::Id Payload::GetId() const { return id_; }
+
+// Returns Payload type.
+Payload::Type Payload::GetType() const { return type_; }
+
+// Sets the payload offset in bytes
+void Payload::SetOffset(size_t offset) {
+  CHECK(type_ == Type::kFile || type_ == Type::kStream);
+  InputFile* file = AsFile();
+  if (file != nullptr) {
+    CHECK(file->GetTotalSize() > 0 && offset < (size_t)file->GetTotalSize());
+  }
+  offset_ = offset;
+}
+
+size_t Payload::GetOffset() { return offset_; }
+
+// Generate Payload Id; to be passed to outgoing file constructor.
+Payload::Id Payload::GenerateId() { return Prng().NextInt64(); }
+
+Payload::Type Payload::FindType() const {
+  return static_cast<Type>(content_.index());
+}
+
+}  // namespace connections
+}  // namespace nearby
+}  // namespace location