Implement WebBluetooth getDescriptor[s]

third_party/WebKit/Source/modules/bluetooth/BluetoothAttributeInstanceMap.cpp

 // Copyright 2016 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 "modules/bluetooth/BluetoothAttributeInstanceMap.h"
 
 #include "modules/bluetooth/BluetoothDevice.h"
 #include "modules/bluetooth/BluetoothRemoteGATTService.h"
 #include "public/platform/modules/bluetooth/WebBluetoothRemoteGATTCharacteristicInit.h"
+#include "public/platform/modules/bluetooth/WebBluetoothRemoteGATTDescriptorInit.h"
+
 #include "public/platform/modules/bluetooth/WebBluetoothRemoteGATTService.h"
 #include <memory>
 #include <utility>
 
 namespace blink {
 
 BluetoothAttributeInstanceMap::BluetoothAttributeInstanceMap(
     BluetoothDevice* device)
     : m_device(device) {}
 
@@ skipping 36 matching lines @@
   }
 
   return characteristic;
 }
 
 bool BluetoothAttributeInstanceMap::containsCharacteristic(
     const String& characteristicInstanceId) {
   return m_characteristicIdToObject.contains(characteristicInstanceId);
 }
 
+BluetoothRemoteGATTDescriptor*
+BluetoothAttributeInstanceMap::getOrCreateBluetoothRemoteGATTDescriptor(
+    std::unique_ptr<WebBluetoothRemoteGATTDescriptorInit> webDescriptor,
+    BluetoothRemoteGATTCharacteristic* characteristic) {
+  String descriptorInstanceId = webDescriptor->descriptorInstanceID;
+
+  BluetoothRemoteGATTDescriptor* descriptor =
+      m_descriptorIdToObject.get(descriptorInstanceId);
+
+  if (!descriptor) {
+    descriptor = BluetoothRemoteGATTDescriptor::create(std::move(webDescriptor),
+                                                       characteristic);
+    m_descriptorIdToObject.add(descriptorInstanceId, descriptor);
+  }
+
+  return descriptor;
+}
+
+bool BluetoothAttributeInstanceMap::containsDescriptor(
+    const String& descriptorInstanceId) {
+  return m_descriptorIdToObject.contains(descriptorInstanceId);
+}
+
 void BluetoothAttributeInstanceMap::Clear() {
   m_serviceIdToObject.clear();
   m_characteristicIdToObject.clear();
+  m_descriptorIdToObject.clear();
 }
 
 DEFINE_TRACE(BluetoothAttributeInstanceMap) {
   visitor->trace(m_device);
   visitor->trace(m_serviceIdToObject);
   visitor->trace(m_characteristicIdToObject);
+  visitor->trace(m_descriptorIdToObject);
 }
 
 }  // namespace blink