// Copyright 2018 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/bluetooth/bluetooth_service_data_map.h"

#include "third_party/blink/renderer/core/typed_arrays/dom_data_view.h"
#include "third_party/blink/renderer/modules/bluetooth/bluetooth_remote_gatt_utils.h"

namespace blink {

class BluetoothServiceDataMapIterationSource final
    : public PairSyncIterable<BluetoothServiceDataMap>::IterationSource {
 public:
  explicit BluetoothServiceDataMapIterationSource(
      const BluetoothServiceDataMap& map)
      : map_(map), iterator_(map_->Map().begin()) {}

  bool FetchNextItem(ScriptState* script_state,
                     String& map_key,
                     NotShared<DOMDataView>& map_value) override {
    if (iterator_ == map_->Map().end())
      return false;
    map_key = iterator_->key;
    map_value = NotShared<DOMDataView>(
        BluetoothRemoteGATTUtils::ConvertSpanToDataView(iterator_->value));
    ++iterator_;
    return true;
  }

  void Trace(Visitor* visitor) const override {
    visitor->Trace(map_);
    PairSyncIterable<BluetoothServiceDataMap>::IterationSource::Trace(visitor);
  }

 private:
  // Needs to be kept alive while we're iterating over it.
  const Member<const BluetoothServiceDataMap> map_;
  BluetoothServiceDataMap::MapType::const_iterator iterator_;
};

BluetoothServiceDataMap::BluetoothServiceDataMap(
    const BluetoothServiceDataMap::MapType& map)
    : parameter_map_(map) {}

BluetoothServiceDataMap::~BluetoothServiceDataMap() {}

PairSyncIterable<BluetoothServiceDataMap>::IterationSource*
BluetoothServiceDataMap::CreateIterationSource(ScriptState*) {
  return MakeGarbageCollected<BluetoothServiceDataMapIterationSource>(*this);
}

bool BluetoothServiceDataMap::GetMapEntry(ScriptState*,
                                          const String& key,
                                          NotShared<DOMDataView>& value) {
  auto it = parameter_map_.find(key);
  if (it == parameter_map_.end())
    return false;

  value = BluetoothRemoteGATTUtils::ConvertSpanToDataView(it->value);
  return true;
}

}  // namespace blink