#include <kapi/devices/device.hpp>

#include <kapi/devices.hpp>
#include <kapi/devices/bus.hpp>
#include <kapi/devices/resource.hpp>

#include <kstd/memory.hpp>
#include <kstd/mutex.hpp>
#include <kstd/result.hpp>
#include <kstd/string.hpp>
#include <kstd/system_error.hpp>
#include <kstd/vector.hpp>

#include <algorithm>
#include <cstddef>
#include <ranges>

namespace kapi::devices
{
  device::device(kstd::string const & name)
      : m_name(name)
  {}

  auto device::facet(kapi::capabilities::facet_id facet) noexcept -> void *
  {
    return query_facet(facet);
  }

  auto device::facet(kapi::capabilities::facet_id facet) const noexcept -> void const *
  {
    return const_cast<device *>(this)->query_facet(facet);
  }

  auto device::has_facet(kapi::capabilities::facet_id facet) const noexcept -> bool
  {
    return const_cast<device *>(this)->query_facet(facet) != nullptr;
  }

  auto device::name() const -> kstd::string const &
  {
    return m_name;
  }

  auto device::parent() const -> kstd::shared_ptr<bus>
  {
    auto guard = kstd::lock_guard{m_lock};
    return m_parent.lock();
  }

  auto device::state() const noexcept -> enum state
  {
    auto guard = kstd::lock_guard{m_lock};
    return m_state;
  }

  auto device::set_state(enum state state) -> void
  {
    auto guard = kstd::lock_guard{m_lock};
    m_state = state;
  }

  auto device::bound_driver() const noexcept -> driver *
  {
    auto guard = kstd::lock_guard{m_lock};
    return m_driver.lock().get();
  }

  auto device::bind_driver(kstd::weak_ptr<struct driver> driver) -> void
  {
    auto guard = kstd::lock_guard{m_lock};
    m_driver = driver;
  }

  auto device::driver_data() const -> kstd::shared_ptr<void>
  {
    auto guard = kstd::lock_guard{m_lock};
    return m_driver_data;
  }

  auto device::set_driver_data(kstd::shared_ptr<void> data) -> void
  {
    auto guard = kstd::lock_guard{m_lock};
    m_driver_data = data;
  }

  auto device::resources() const -> kstd::vector<resource>
  {
    auto guard = kstd::lock_guard{m_lock};
    return m_resources;
  }

  auto device::request_resource(resource_type type, std::size_t index) const -> kstd::result<resource>
  {
    auto guard = kstd::lock_guard{m_lock};

    auto numbered_resources = std::views::enumerate(m_resources);
    auto found = std::ranges::find_if(numbered_resources, [&](auto entry) {
      auto const & [number, resource] = entry;
      return resource.type == type && static_cast<std::size_t>(number) == index;
    });

    if (found != numbered_resources.end())
    {
      return kstd::success(*found.base());
    }

    return kstd::failure(make_error_code(kstd::errc::invalid_argument));
  }

  auto device::set_resources(kstd::vector<resource> resources) -> void
  {
    auto guard = kstd::lock_guard{m_lock};
    m_resources = resources;
  }

  auto device::query_facet(kapi::capabilities::facet_id) -> void *
  {
    return nullptr;
  }

  auto device::set_parent(kstd::weak_ptr<bus> parent) -> void
  {
    auto guard = kstd::lock_guard{m_lock};
    m_parent = parent;
  }

}  // namespace kapi::devices