| // Copyright 2017 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 "device/u2f/u2f_discovery.h" |
| |
| #include <utility> |
| |
| #include "device/u2f/u2f_device.h" |
| |
| namespace device { |
| |
| U2fDiscovery::U2fDiscovery() = default; |
| |
| U2fDiscovery::~U2fDiscovery() = default; |
| |
| void U2fDiscovery::AddObserver(Observer* observer) { |
| observers_.AddObserver(observer); |
| } |
| |
| void U2fDiscovery::RemoveObserver(Observer* observer) { |
| observers_.RemoveObserver(observer); |
| } |
| |
| void U2fDiscovery::NotifyDiscoveryStarted(bool success) { |
| for (auto& observer : observers_) |
| observer.DiscoveryStarted(this, success); |
| } |
| |
| void U2fDiscovery::NotifyDiscoveryStopped(bool success) { |
| for (auto& observer : observers_) |
| observer.DiscoveryStopped(this, success); |
| } |
| |
| void U2fDiscovery::NotifyDeviceAdded(U2fDevice* device) { |
| for (auto& observer : observers_) |
| observer.DeviceAdded(this, device); |
| } |
| |
| void U2fDiscovery::NotifyDeviceRemoved(U2fDevice* device) { |
| for (auto& observer : observers_) |
| observer.DeviceRemoved(this, device); |
| } |
| |
| std::vector<U2fDevice*> U2fDiscovery::GetDevices() { |
| std::vector<U2fDevice*> devices; |
| devices.reserve(devices_.size()); |
| for (const auto& device : devices_) |
| devices.push_back(device.second.get()); |
| return devices; |
| } |
| |
| std::vector<const U2fDevice*> U2fDiscovery::GetDevices() const { |
| std::vector<const U2fDevice*> devices; |
| devices.reserve(devices_.size()); |
| for (const auto& device : devices_) |
| devices.push_back(device.second.get()); |
| return devices; |
| } |
| |
| U2fDevice* U2fDiscovery::GetDevice(base::StringPiece device_id) { |
| return const_cast<U2fDevice*>( |
| static_cast<const U2fDiscovery*>(this)->GetDevice(device_id)); |
| } |
| |
| const U2fDevice* U2fDiscovery::GetDevice(base::StringPiece device_id) const { |
| auto found = devices_.find(device_id); |
| return found != devices_.end() ? found->second.get() : nullptr; |
| } |
| |
| bool U2fDiscovery::AddDevice(std::unique_ptr<U2fDevice> device) { |
| std::string device_id = device->GetId(); |
| const auto result = devices_.emplace(std::move(device_id), std::move(device)); |
| if (result.second) |
| NotifyDeviceAdded(result.first->second.get()); |
| return result.second; |
| } |
| |
| bool U2fDiscovery::RemoveDevice(base::StringPiece device_id) { |
| auto found = devices_.find(device_id); |
| if (found == devices_.end()) |
| return false; |
| |
| auto device = std::move(found->second); |
| devices_.erase(found); |
| NotifyDeviceRemoved(device.get()); |
| return true; |
| } |
| |
| } // namespace device |