#pragma once

#include <BAN/Function.h>
#include <BAN/Iteration.h>
#include <BAN/WeakPtr.h>

#include <kernel/FS/FAT/Definitions.h>
#include <kernel/FS/Inode.h>

namespace Kernel
{

	class FATFS;

	class FATInode final : public Inode, public BAN::Weakable<FATInode>
	{
	public:
		virtual ino_t ino() const override { return m_ino; };
		virtual Mode mode() const override { return Mode { ((m_entry.attr & FAT::FileAttr::DIRECTORY) ? Mode::IFDIR : Mode::IFREG) | 0777 }; }
		virtual nlink_t nlink() const override { return 1; }
		virtual uid_t uid() const override { return 0; }
		virtual gid_t gid() const override { return 0; }
		virtual off_t size() const override { return m_entry.file_size; }
		virtual timespec atime() const override;
		virtual timespec mtime() const override;
		virtual timespec ctime() const override;
		virtual blksize_t blksize() const override;
		virtual blkcnt_t blocks() const override { return m_block_count; }
		virtual dev_t dev() const override { return 0; }
		virtual dev_t rdev() const override { return 0; }

		virtual const FileSystem* filesystem() const override;

		const FAT::DirectoryEntry& entry() const { return m_entry; }

	protected:
		virtual BAN::ErrorOr<BAN::RefPtr<Inode>> find_inode_impl(BAN::StringView) override;
		virtual BAN::ErrorOr<size_t> list_next_inodes_impl(off_t, struct dirent*, size_t) override;
		//virtual BAN::ErrorOr<void> create_file_impl(BAN::StringView, mode_t, uid_t, gid_t) override;
		//virtual BAN::ErrorOr<void> create_directory_impl(BAN::StringView, mode_t, uid_t, gid_t) override;
		//virtual BAN::ErrorOr<void> unlink_impl(BAN::StringView) override;

		virtual BAN::ErrorOr<size_t> read_impl(off_t, BAN::ByteSpan) override;
		//virtual BAN::ErrorOr<size_t> write_impl(off_t, BAN::ConstByteSpan) override;
		//virtual BAN::ErrorOr<void> truncate_impl(size_t) override;
		//virtual BAN::ErrorOr<void> chmod_impl(mode_t) override;
		virtual BAN::ErrorOr<void> fsync_impl() override { return {}; }

		virtual bool can_read_impl() const override { return true; }
		virtual bool can_write_impl() const override { return true; }
		virtual bool has_error_impl() const override { return false; }

	private:
		FATInode(FATFS& fs, const FAT::DirectoryEntry& entry, ino_t ino, uint32_t block_count)
			: m_fs(fs)
			, m_entry(entry)
			, m_ino(ino)
			, m_block_count(block_count)
		{ }
		~FATInode()	{}

		BAN::ErrorOr<void> for_each_directory_entry(BAN::ConstByteSpan, BAN::Function<BAN::Iteration(const FAT::DirectoryEntry&)>);
		BAN::ErrorOr<void> for_each_directory_entry(BAN::ConstByteSpan, BAN::Function<BAN::Iteration(const FAT::DirectoryEntry&, BAN::String, uint32_t)>);

	private:
		FATFS& m_fs;
		FAT::DirectoryEntry m_entry;
		const ino_t m_ino;
		uint32_t m_block_count;

		friend class Ext2FS;
		friend class BAN::RefPtr<FATInode>;
	};

}