// 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 "content/browser/webauth/virtual_discovery.h"

#include <memory>
#include <string_view>
#include <utility>

#include "base/check.h"
#include "base/functional/bind.h"
#include "base/location.h"
#include "base/memory/weak_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "device/fido/fido_device.h"
#include "device/fido/fido_device_discovery.h"
#include "device/fido/public/fido_transport_protocol.h"
#include "device/fido/virtual_fido_device_authenticator.h"

namespace content {

VirtualFidoDiscovery::VirtualFidoDiscovery(
    ::device::FidoTransportProtocol transport)
    : FidoDeviceDiscovery(transport) {}

VirtualFidoDiscovery::~VirtualFidoDiscovery() = default;

void VirtualFidoDiscovery::AddVirtualDevice(
    std::unique_ptr<device::VirtualFidoDevice> device) {
  // The real implementation would never notify the client's observer about
  // devices before the client calls Start(), mimic the same behavior.
  if (is_start_requested()) {
    auto authenticator =
        std::make_unique<device::VirtualFidoDeviceAuthenticator>(
            std::move(device));
    FidoDeviceDiscovery::AddAuthenticator(std::move(authenticator));
  } else {
    devices_pending_discovery_start_.push_back(std::move(device));
  }
}

bool VirtualFidoDiscovery::RemoveVirtualDevice(std::string_view device_id) {
  DCHECK(is_start_requested());
  return ::device::FidoDeviceDiscovery::RemoveDevice(device_id);
}

void VirtualFidoDiscovery::StartInternal() {
  for (auto& device : devices_pending_discovery_start_) {
    auto authenticator =
        std::make_unique<device::VirtualFidoDeviceAuthenticator>(
            std::move(device));
    FidoDeviceDiscovery::AddAuthenticator(std::move(authenticator));
  }
  devices_pending_discovery_start_.clear();

  base::SingleThreadTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE,
      base::BindOnce(&VirtualFidoDiscovery::NotifyDiscoveryStarted,
                     weak_ptr_factory_.GetWeakPtr(), /*success=*/true));
}

}  // namespace content