#include "kernel/filesystem/devfs/filesystem.hpp"

#include "kapi/devices/device.hpp"
#include "kernel/devices/storage/management.hpp"
#include "kernel/filesystem/devfs/inode.hpp"
#include "kernel/filesystem/device_inode.hpp"
#include "kernel/filesystem/inode.hpp"

#include <kstd/memory>

#include <algorithm>
#include <string_view>

namespace kernel::filesystem::devfs
{
  auto filesystem::mount(kstd::shared_ptr<kapi::devices::device> const &) -> int
  {
    m_root_inode = kstd::make_shared<inode>();
    build_device_inode_table();

    return 0;
  }

  auto filesystem::lookup(kstd::shared_ptr<kernel::filesystem::inode> const & parent, std::string_view name)
      -> kstd::shared_ptr<kernel::filesystem::inode>
  {
    if (!parent || !parent->is_directory())
    {
      return nullptr;
    }

    if (parent.get() != m_root_inode.get())
    {
      return nullptr;
    }

    auto it = std::ranges::find_if(m_inodes, [&](auto const & dev_node) {
      auto device_inode_ptr = static_cast<device_inode *>(dev_node.get());
      if (!device_inode_ptr)
      {
        return false;
      }
      return device_inode_ptr->device()->name() == name;
    });
    return (it != m_inodes.end()) ? *it : nullptr;
  }

  auto filesystem::build_device_inode_table() -> void
  {
    m_inodes.clear();

    auto storage_mgmt = devices::storage::management::get();
    std::ranges::for_each(storage_mgmt.all_controllers(), [&](auto const & controller) {
      std::ranges::for_each(controller->all_devices(),
                            [&](auto const & device) { m_inodes.push_back(kstd::make_shared<device_inode>(device)); });
    });
  }
}  // namespace kernel::filesystem::devfs