//
// Created by Stepan Usatiuk on 07.06.2023.
//

#include "../includes/RepoFS.h"

#define FUSE_USE_VERSION 26

#include <errno.h>
#include <fcntl.h>
#include <fuse.h>
#include <iostream>
#include <stdio.h>
#include <string.h>

#include "Serialize.h"
#include "objects/Chunk.h"

RepoFS::RepoFS(Repository *repon, Object::idType archiveId, std::string path) : archive(Serialize::deserialize<Archive>(repon->getObject(archiveId))), path(std::move(path)) {
    RepoFS::repo = repon;
    for (auto const &f: archive.files) {
        auto file = Serialize::deserialize<File>(repo->getObject(f));
        auto path = std::filesystem::u8path(file.name);
        DirEntry *entry = &root;
        for (auto const &subp: path) {
            entry = &entry->children[subp];
        }
        entry->file.emplace(file);
    }
    //    thread = std::thread(&RepoFS::workerFn, this);
}

RepoFS::~RepoFS() {
    //    stop = true;
    //    thread.join();
}

static const char *hello_str = "Hello World!\n";
static const char *hello_path = "/hello";

DirEntry *getf(std::string path) {
    auto p = std::filesystem::relative(std::filesystem::u8path(path), "/");
    DirEntry *entry = &RepoFS::root;
    if (p != ".")
        for (auto const &subp: p) {
            entry = &entry->children.at(subp);
        }
    return entry;
}

static int rfsGetattr(const char *path, struct stat *stbuf) {
    int res = 0;

    memset(stbuf, 0, sizeof(struct stat));
    if (strcmp(path, "/") == 0) {
        stbuf->st_mode = S_IFDIR | 0755;
        stbuf->st_nlink = 2;
        return res;
    }
    DirEntry *e;
    try {
        e = getf(path);
        if (e->file->fileType == File::Type::Directory) {
            stbuf->st_mode = S_IFDIR | 0755;
            stbuf->st_nlink = 1;
        } else if (e->file->fileType == File::Type::Normal) {
            stbuf->st_mode = S_IFREG | 0444;
            stbuf->st_nlink = 1;
            stbuf->st_size = e->file->bytes;
        }
    } catch (...) { return -ENOENT; }

    return res;
}

static int rfsReaddir(const char *path, void *buf, fuse_fill_dir_t filler,
                      off_t offset, struct fuse_file_info *fi) {
    (void) offset;
    (void) fi;

    DirEntry *entry = &RepoFS::root;
    if (std::string(path) != "/")
        try {
            entry = getf(path);
        } catch (...) { return -ENOENT; }

    filler(buf, ".", NULL, 0);
    filler(buf, "..", NULL, 0);

    for (auto const &e: entry->children) {
        auto pstr = std::filesystem::u8path(e.second.file->name).filename().u8string();
        std::cout << pstr << std::endl;
        filler(buf, pstr.c_str(), NULL, 0);
    }

    return 0;
}

static int rfsOpen(const char *path, struct fuse_file_info *fi) {
    DirEntry *entry = &RepoFS::root;
    if (std::string(path) != "/")
        try {
            entry = getf(path);
        } catch (...) { return -ENOENT; }

    if ((fi->flags & 3) != O_RDONLY)
        return -EACCES;

    return 0;
}

static int rfsRead(const char *path, char *buf, size_t size, off_t offset,
                   struct fuse_file_info *fi) {
    size_t len;
    (void) fi;
    DirEntry *entry = &RepoFS::root;
    if (std::string(path) != "/")
        try {
            entry = getf(path);
        } catch (...) { return -ENOENT; }

    std::vector<char> data;

    for (auto const &id: entry->file->chunks) {
        auto ch = Serialize::deserialize<Chunk>(RepoFS::repo->getObject(id));
        data.insert(data.end(), ch.data.begin(), ch.data.end());
    }

    len = data.size();
    if (offset < len) {
        if (offset + size > len)
            size = len - offset;
        memcpy(buf, data.data() + offset, size);
    } else
        size = 0;

    return size;
}

static struct fuse_operations rfsOps = {
        .getattr = rfsGetattr,
        .readdir = rfsReaddir,
        .open = rfsOpen,
        .read = rfsRead,
};

void RepoFS::workerFn() {
    int argc = 3;
    char *argv[] = {"", "-d", const_cast<char *>(path.c_str())};
    std::cout << static_cast<int>(fuse_main(argc, argv, &rfsOps, nullptr));
    //    while (!stop) {
    //    }
}