// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "third_party/blink/renderer/modules/push_messaging/push_message_data.h"

#include <memory>

#include "base/containers/span.h"
#include "third_party/blink/renderer/bindings/core/v8/v8_binding_for_core.h"
#include "third_party/blink/renderer/bindings/core/v8/v8_union_arraybuffer_arraybufferview_usvstring.h"
#include "third_party/blink/renderer/core/fileapi/blob.h"
#include "third_party/blink/renderer/core/typed_arrays/dom_array_buffer.h"
#include "third_party/blink/renderer/core/typed_arrays/dom_array_piece.h"
#include "third_party/blink/renderer/platform/bindings/exception_state.h"
#include "third_party/blink/renderer/platform/bindings/script_state.h"
#include "third_party/blink/renderer/platform/blob/blob_data.h"
#include "third_party/blink/renderer/platform/heap/garbage_collected.h"
#include "third_party/blink/renderer/platform/wtf/text/text_encoding.h"
#include "v8/include/v8.h"

namespace blink {

PushMessageData* PushMessageData::Create(const String& message_string) {
  // The standard supports both an empty but valid message and a null message.
  // In case the message is explicitly null, return a null pointer which will
  // be set in the PushEvent.
  if (message_string.IsNull())
    return nullptr;
  return PushMessageData::Create(
      MakeGarbageCollected<V8UnionArrayBufferOrArrayBufferViewOrUSVString>(
          message_string));
}

PushMessageData* PushMessageData::Create(
    const V8UnionArrayBufferOrArrayBufferViewOrUSVString* message_data) {
  if (!message_data)
    return nullptr;
  switch (message_data->GetContentType()) {
    case V8UnionArrayBufferOrArrayBufferViewOrUSVString::ContentType::
        kArrayBuffer: {
      const DOMArrayBuffer* buffer = message_data->GetAsArrayBuffer();
      return MakeGarbageCollected<PushMessageData>(buffer->ByteSpan());
    }
    case V8UnionArrayBufferOrArrayBufferViewOrUSVString::ContentType::
        kArrayBufferView: {
      const DOMArrayBufferView* buffer_view =
          message_data->GetAsArrayBufferView().Get();
      return MakeGarbageCollected<PushMessageData>(buffer_view->ByteSpan());
    }
    case V8UnionArrayBufferOrArrayBufferViewOrUSVString::ContentType::
        kUSVString: {
      std::string encoded_string = Utf8Encoding().Encode(
          message_data->GetAsUSVString(), UnencodableHandling::kNone);
      return MakeGarbageCollected<PushMessageData>(
          base::as_byte_span(encoded_string));
    }
  }
  NOTREACHED();
}

PushMessageData::PushMessageData(base::span<const uint8_t> data) {
  data_.append_range(data);
}

PushMessageData::~PushMessageData() = default;

DOMArrayBuffer* PushMessageData::arrayBuffer() const {
  return DOMArrayBuffer::Create(data_);
}

Blob* PushMessageData::blob() const {
  auto blob_data = std::make_unique<BlobData>();
  blob_data->AppendBytes(data_);

  // Note that the content type of the Blob object is deliberately not being
  // provided, following the specification.

  const uint64_t byte_length = blob_data->length();
  return MakeGarbageCollected<Blob>(
      BlobDataHandle::Create(std::move(blob_data), byte_length));
}

NotShared<DOMUint8Array> PushMessageData::bytes() const {
  return NotShared(DOMUint8Array::Create(data_));
}

ScriptValue PushMessageData::json(ScriptState* script_state) const {
  return ScriptValue(script_state->GetIsolate(),
                     FromJSONString(script_state, text()));
}

String PushMessageData::text() const {
  return Utf8Encoding().Decode(data_);
}

}  // namespace blink