#include "kernel/devices/storage/controller.hpp"

#include "kapi/devices/device.hpp"

#include <kstd/memory>
#include <kstd/vector>

#include <algorithm>
#include <cstddef>

namespace kernel::devices::storage
{
  auto controller::set_ids(size_t major, size_t minors_per_dev) -> void
  {
    m_major = major;
    m_minors_per_device = minors_per_dev;
  }

  auto controller::major() const -> size_t
  {
    return m_major;
  }

  auto controller::device_by_minor(size_t minor) const -> kstd::shared_ptr<kapi::devices::device>
  {
    auto it = std::ranges::find_if(m_devices, [minor](auto const & device) { return device->minor() == minor; });

    if (it != m_devices.end())
    {
      return *it;
    }
    return nullptr;
  }

  auto controller::devices_count() const -> size_t
  {
    return m_devices.size();
  }

  auto controller::all_devices() const -> kstd::vector<kstd::shared_ptr<kapi::devices::device>> const &
  {
    return m_devices;
  }
}  // namespace kernel::devices::storage