#include <kapi/devices.hpp>

#include <kernel/bus/root.hpp>

#include <kapi/system.hpp>
#include <kapi/test_support/devices.hpp>

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

#include <string_view>

namespace kapi::devices
{
  namespace
  {
    auto constinit root_bus = kstd::shared_ptr<kernel::bus::root>{};
    auto constinit device_tree = kstd::flat_map<kstd::string, kstd::weak_ptr<device>>{};
  }  // namespace

  auto init() -> void
  {
    if (root_bus)
    {
      kapi::system::panic("[OS:DEV] The device subsystem has already been initialized");
    }

    device_registry::init();
    facet_registry::init();

    root_bus = kstd::make_shared<kernel::bus::root>();
    device_registry::get().add(root_bus);
  }

  auto get_root_bus() -> kstd::shared_ptr<bus>
  {
    if (!root_bus)
    {
      kapi::system::panic("[OS:DEV] Root bus not initialized!");
    }
    return root_bus;
  }

  auto remove_device(device & device) -> bool
  {
    auto parent = device.parent();
    if (!parent)
    {
      return false;
    }

    return parent->remove_child(device);
  }

}  // namespace kapi::devices

namespace kapi::test_support::devices
{
  auto deinit() -> void
  {
    kapi::devices::root_bus.reset();

    deinit_driver_registry();
    deinit_facet_registry();
    deinit_device_registry();
  }
}  // namespace kapi::test_support::devices