#include "fsctl/fugue/dirent.h"
#include "fsctl/fugue/cluster.h"
#include "fsctl/fugue/logger.h"
#include "fsctl/fugue/bits/fat.h"

//---
// Internals
//---

/* _fugue_dirent_checksum() : Fugue checksum handling */
static uint8_t _fugue_dirent_checksum(uintptr_t directory)
{
    uint8_t *block;
    uint8_t checksum;
    int a;
    int b;

    checksum = 0;
    block = (void *)directory;
    for (int i = 0 ; i < 11 ; ++i)
    {
        checksum = checksum & 0xff;
        a = checksum / 2;
        b = checksum * 128;
        checksum = a | b;
        checksum = checksum + block[i];
    }
    checksum = checksum & 0xff;
    return checksum;
}

//---
// Public
//---

/* fugue_dirent_init() : init dirent object */
int fugue_dirent_init(fugue_dirent_t *dirent, fugue_fs_t *fs)
{
    if (fs == NULL || dirent == NULL)
        return -1;
    memset(dirent, 0x00, sizeof(fugue_dirent_t));
    memcpy(&dirent->_private.fs, &fs, sizeof(fugue_fs_t));
    dirent->current_dir_addr   = (uintptr_t)fs->_private.root;
    dirent->cluster_addr_start = (uintptr_t)fs->_private.root;
    dirent->cluster_addr_end   = (uintptr_t)fs->_private.root;
    dirent->cluster_addr_end  += fs->props.sector_root_nb * 512;
    return 0;
}

/* fugue_dirent_dir_fetch() : fetch the current dir blob and walk to next */
void *fugue_dirent_dir_fetch(fugue_dirent_t *dirent)
{
    void *dir;

    dir = (void*)dirent->current_dir_addr;
    if (dir == NULL)
        return NULL;

    dirent->current_dir_addr += 32;
    if (dirent->current_dir_addr < dirent->cluster_addr_end)
        return dir;

    dirent->cluster_addr_start = fugue_cluster_find_next(
        &dirent->_private.fs,
        dirent->cluster_idx
    );
    dirent->cluster_addr_end = dirent->cluster_addr_start;
    if (dirent->cluster_addr_start != 0x00000000)
        dirent->cluster_addr_end += dirent->_private.fs.props.cluster_size;
    return dir;
}

/* fugue_dirent_name_fetch() : fetch fragment */
int fugue_dirent_name_fetch(
    fugue_dirent_t *dirent,
    fugue_file_t *file,
    void *dir
) {
    fugue_logger_t *log;
    struct fugue_fat_dir_name *lfn;

    lfn = (void *)dir;
    log = dirent->_private.fs._private.logger;

    //---
    // check directory block validity
    //---

    if (lfn->DIR_Attr != FUGUE_DIR_ATTR_LFN)
        return -1;
    if (lfn->DIR_Type != 0x00)
        return -2;
    if (lfn->DIR_FstClus != 0x0000)
        return -3;
    //TODO : check Secquence validity
    if (lfn->checksum != _fugue_dirent_checksum(dirent->current_dir_addr)) {
        fugue_logger_write(log, "[WARN] checksum error at %p", lfn);
        return -5;
    }

    //---
    // Dump name fragment
    //---

    //TODO: dump
    (void)file;
    fugue_logger_write(log, "[ERR] name dump error at %p", lfn);
    return -1;
}