#include <kapi/devices/bus.hpp>

#include <kapi/devices.hpp>
#include <kapi/devices/bus_protocol.hpp>
#include <kapi/devices/driver_registry.hpp>
#include <kapi/system.hpp>

#include <kstd/memory.hpp>
#include <kstd/string.hpp>
#include <kstd/vector.hpp>

#include <span>
#include <utility>

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

  bus::bus(kstd::string const & name, bus_protocol & protocol)
      : device{name}
      , m_protocol{&protocol}
  {}

  auto bus::add_child(kstd::shared_ptr<device> child) -> void
  {
    child->set_parent(kstd::static_pointer_cast<bus>(shared_from_this()));

    if (!kapi::devices::device_registry::get().add(child))
    {
      kapi::system::panic("[OS:DEV] Failed to register child device {}", child->name());
    }

    auto & attached = m_devices.emplace_back(std::move(child));
    attached->set_state(state::present);

    driver_registry::get().device_attached(attached);
  }

  [[nodiscard]] auto bus::children() const -> std::span<kstd::shared_ptr<device> const>
  {
    return {m_devices.data(), m_devices.size()};
  }

  auto bus::query_facet(kapi::capabilities::facet_id facet) -> void *
  {
    if (facet == bus::id)
    {
      return this;
    }
    else if (facet == bus_protocol::id)
    {
      return m_protocol;
    }

    return device::query_facet(facet);
  }

}  // namespace kapi::devices