/*
 * Plants-OS VDiskDriver
 * Copyright by min0911.
 */

#include "../include/vdisk.h"
#include "../include/printf.h"
#include "../include/fifo.h"
#include "../include/task.h"

int getReadyDisk();

vdisk vdisk_ctl[26];

int init_vdisk() {
    for (int i = 0; i < 26; i++) {
        vdisk_ctl[i].flag = 0; // 设置为未使用
    }
}

int register_vdisk(vdisk vd) {
    for (int i = 0; i < 26; i++) {
        if (!vdisk_ctl[i].flag) {
            vdisk_ctl[i] = vd; // 找到了！
            return i + ('A');  // 注册成功，返回drive
        }
    }
    printk("[vdisk]not found\n");
    return 0; // 注册失败
}
int logout_vdisk(char drive) {
    int indx = drive - ('A');
    if (indx > 26) {
        return 0; // 失败
    }
    if (vdisk_ctl[indx].flag) {
        vdisk_ctl[indx].flag = 0; // 设置为没有
        return 1;                 // 成功
    } else {
        return 0; // 失败
    }
}
int rw_vdisk(char drive, unsigned int lba, unsigned char *buffer,
             unsigned int number, int read) {
    int indx = drive - ('A');
    if (indx > 26) {
        return 0; // 失败
    }
    if (vdisk_ctl[indx].flag) {
        if (read) {
            vdisk_ctl[indx].Read(drive, buffer, number, lba);
        } else {
            vdisk_ctl[indx].Write(drive, buffer, number, lba);
        }
        return 1; // 成功
    } else {
        return 0; // 失败
    }
}
bool have_vdisk(char drive) {
    int indx = drive - 'A';
    // printk("drive=%c\n",drive);
    if (indx > 26) {
        return 0; // 失败
    }
    if (vdisk_ctl[indx].flag) {
        return 1; // 成功
    } else {
        return 0; // 失败
    }
}
// 基于vdisk的通用读写

static unsigned char *drive_name[16] = {NULL, NULL, NULL, NULL, NULL, NULL,
                                        NULL, NULL, NULL, NULL, NULL, NULL,
                                        NULL, NULL, NULL, NULL};
static struct FIFO8 drive_fifo[16];
static unsigned char drive_buf[16][256];
bool SetDrive(unsigned char *name) {
    for (int i = 0; i != 16; i++) {
        if (drive_name[i] == NULL) {
            drive_name[i] = name;
            fifo8_init(&drive_fifo[i], 256, drive_buf[i]);
            return true;
        }
    }
    return false;
}
unsigned int GetDriveCode(unsigned char *name) {
    for (int i = 0; i != 16; i++) {
        if (strcmp((char *)drive_name[i], (char *)name) == 0) {
            return i;
        }
    }
    return 16;
}

bool DriveSemaphoreTake(unsigned int drive_code) {
    return true;
}
void DriveSemaphoreGive(unsigned int drive_code) {
    return;
}

void Disk_Read(unsigned int lba, unsigned int number, void *buffer,
               char drive) {
    if (have_vdisk(drive)) {
        if (DriveSemaphoreTake(GetDriveCode((unsigned char *)"DISK_DRIVE"))) {
            for (int i = 0; i < number; i += SECTORS_ONCE) {
                int sectors = ((number - i) >= SECTORS_ONCE) ? SECTORS_ONCE : (number - i);
                rw_vdisk(drive, lba + i, buffer + i * 512, sectors, 1);
            }
            DriveSemaphoreGive(GetDriveCode((unsigned char *)"DISK_DRIVE"));
        }
    }
}
unsigned int disk_Size(char drive) {
    unsigned char drive1 = drive;
    if (have_vdisk(drive1)) {
        int indx = drive1 - 'A';
        return vdisk_ctl[indx].size;
    } else {
        return 0;
    }

    return 0;
}
bool DiskReady(char drive) { return have_vdisk(drive); }
int getReadyDisk() { return 0; }
void Disk_Write(unsigned int lba, unsigned int number, void *buffer,
                char drive) {
//  printk("%d\n",lba);
    if (have_vdisk(drive)) {
        if (DriveSemaphoreTake(GetDriveCode((unsigned char *)"DISK_DRIVE"))) {
            // printk("*buffer(%d %d) = %02x\n",lba,number,*(unsigned char *)buffer);
            for (int i = 0; i < number; i += SECTORS_ONCE) {
                int sectors = ((number - i) >= SECTORS_ONCE) ? SECTORS_ONCE : (number - i);
                rw_vdisk(drive, lba + i, buffer + i * 512, sectors, 0);
            }
            DriveSemaphoreGive(GetDriveCode((unsigned char *)"DISK_DRIVE"));
        }
    }
}