third_party/blink/renderer/modules/nfc/ndef_message.cc

// Copyright 2019 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 "third_party/blink/renderer/modules/nfc/ndef_message.h"

#include "services/device/public/mojom/nfc.mojom-blink.h"
#include "third_party/blink/renderer/bindings/modules/v8/string_or_array_buffer_or_ndef_message_init.h"
#include "third_party/blink/renderer/modules/nfc/ndef_message_init.h"
#include "third_party/blink/renderer/modules/nfc/ndef_record.h"
#include "third_party/blink/renderer/platform/bindings/exception_state.h"

namespace blink {

// static
NDEFMessage* NDEFMessage::Create(const NDEFMessageInit* init,
                                 ExceptionState& exception_state) {
  NDEFMessage* message = MakeGarbageCollected<NDEFMessage>();
  message->url_ = init->url();
  if (init->hasRecords()) {
    for (const NDEFRecordInit* record_init : init->records()) {
      NDEFRecord* record = NDEFRecord::Create(record_init, exception_state);
      if (exception_state.HadException())
        return nullptr;
      DCHECK(record);
      message->records_.push_back(record);
    }
  }
  return message;
}

// static
NDEFMessage* NDEFMessage::Create(const NDEFMessageSource& source,
                                 ExceptionState& exception_state) {
  if (source.IsString()) {
    NDEFMessage* message = MakeGarbageCollected<NDEFMessage>();
    message->records_.push_back(
        MakeGarbageCollected<NDEFRecord>(source.GetAsString()));
    return message;
  }

  if (source.IsArrayBuffer()) {
    NDEFMessage* message = MakeGarbageCollected<NDEFMessage>();
    message->records_.push_back(
        MakeGarbageCollected<NDEFRecord>(source.GetAsArrayBuffer()));
    return message;
  }

  if (source.IsNDEFMessageInit()) {
    return Create(source.GetAsNDEFMessageInit(), exception_state);
  }

  NOTREACHED();
  return nullptr;
}

NDEFMessage::NDEFMessage() = default;

NDEFMessage::NDEFMessage(const device::mojom::blink::NDEFMessage& message)
    : url_(message.url) {
  for (wtf_size_t i = 0; i < message.data.size(); ++i) {
    records_.push_back(MakeGarbageCollected<NDEFRecord>(*message.data[i]));
  }
}

const String& NDEFMessage::url() const {
  return url_;
}

const HeapVector<Member<NDEFRecord>>& NDEFMessage::records() const {
  return records_;
}

void NDEFMessage::Trace(blink::Visitor* visitor) {
  visitor->Trace(records_);
  ScriptWrappable::Trace(visitor);
}

}  // namespace blink