#pragma once

#include "DiskDevice.h"
#include "InodeIdentifier.h"
#include "InodeMetadata.h"
#include "Limits.h"
#include "UnixTypes.h"
#include <AK/ByteBuffer.h>
#include <AK/HashMap.h>
#include <AK/OwnPtr.h>
#include <AK/Retainable.h>
#include <AK/RetainPtr.h>
#include <AK/String.h>
#include <AK/Function.h>
#include <AK/kstdio.h>

static const dword mepoch = 476763780;

class CoreInode;
class FileDescriptor;

class FS : public Retainable<FS> {
public:
    static void initializeGlobals();
    virtual ~FS();

    dword id() const { return m_id; }
    static FS* fromID(dword);

    virtual bool initialize() = 0;
    virtual const char* class_name() const = 0;
    virtual InodeIdentifier rootInode() const = 0;
    virtual bool writeInode(InodeIdentifier, const ByteBuffer&) = 0;
    virtual InodeMetadata inodeMetadata(InodeIdentifier) const = 0;

    virtual Unix::ssize_t read_inode_bytes(InodeIdentifier, Unix::off_t offset, Unix::size_t count, byte* buffer, FileDescriptor*) const = 0;

    struct DirectoryEntry {
        DirectoryEntry(const char* name, InodeIdentifier, byte fileType);
        DirectoryEntry(const char* name, Unix::size_t name_length, InodeIdentifier, byte fileType);
        char name[256];
        Unix::size_t name_length { 0 };
        InodeIdentifier inode;
        byte fileType { 0 };
    };

    virtual bool set_mtime(InodeIdentifier, dword timestamp) = 0;
    virtual InodeIdentifier create_inode(InodeIdentifier parentInode, const String& name, Unix::mode_t, unsigned size) = 0;
    virtual InodeIdentifier create_directory(InodeIdentifier parentInode, const String& name, Unix::mode_t) = 0;

    virtual InodeIdentifier find_parent_of_inode(InodeIdentifier) const = 0;

    virtual RetainPtr<CoreInode> get_inode(InodeIdentifier) const = 0;

    ByteBuffer readEntireInode(InodeIdentifier, FileDescriptor* = nullptr) const;

protected:
    FS();

private:
    dword m_id { 0 };
};

class CoreInode : public Retainable<CoreInode> {
    friend class VFS;
public:
    virtual ~CoreInode();

    FS& fs() { return m_fs; }
    const FS& fs() const { return m_fs; }
    unsigned fsid() const;
    unsigned index() const { return m_index; }

    size_t size() const { return metadata().size; }
    bool is_symlink() const { return metadata().isSymbolicLink(); }
    bool is_directory() const { return metadata().isDirectory(); }

    InodeIdentifier identifier() const { return { fsid(), index() }; }
    const InodeMetadata& metadata() const { if (!m_metadata.isValid()) { populate_metadata(); } return m_metadata; }

    ByteBuffer read_entire(FileDescriptor* = nullptr);

    virtual Unix::ssize_t read_bytes(Unix::off_t, Unix::size_t, byte* buffer, FileDescriptor*) = 0;
    virtual bool traverse_as_directory(Function<bool(const FS::DirectoryEntry&)>) = 0;
    virtual InodeIdentifier lookup(const String& name) = 0;
    virtual String reverse_lookup(InodeIdentifier) = 0;

protected:
    CoreInode(FS& fs, unsigned index)
        : m_fs(fs)
        , m_index(index)
    {
    }

    virtual void populate_metadata() const = 0;

    mutable InodeMetadata m_metadata;
private:
    FS& m_fs;
    unsigned m_index { 0 };
};

inline FS* InodeIdentifier::fileSystem()
{
    return FS::fromID(m_fileSystemID);
}

inline const FS* InodeIdentifier::fileSystem() const
{
    return FS::fromID(m_fileSystemID);
}

inline InodeMetadata InodeIdentifier::metadata() const
{
    if (!isValid())
        return InodeMetadata();
    return fileSystem()->inodeMetadata(*this);
}

inline bool InodeIdentifier::isRootInode() const
{
    return (*this) == fileSystem()->rootInode();
}

inline unsigned CoreInode::fsid() const
{
    return m_fs.id();
}

namespace AK {

template<>
struct Traits<InodeIdentifier> {
    // FIXME: This is a shitty hash.
    static unsigned hash(const InodeIdentifier& inode) { return Traits<unsigned>::hash(inode.fsid()) + Traits<unsigned>::hash(inode.index()); }
    static void dump(const InodeIdentifier& inode) { kprintf("%02u:%08u", inode.fsid(), inode.index()); }
};

}