/*-----------------------------------------------------------------------*/
/* Low level disk I/O module SKELETON for FatFs     (C)ChaN, 2019        */
/*-----------------------------------------------------------------------*/
/* If a working storage control module is available, it should be        */
/* attached to the FatFs via a glue function rather than modifying it.   */
/* This is an example of glue functions to attach various exsisting      */
/* storage control modules to the FatFs module with a defined API.       */
/*-----------------------------------------------------------------------*/

#include "ff.h"			/* Obtains integer types */
#include "diskio.h"		/* Declarations of disk functions */
#include "debug.h"
#include "bsp_sd_spi.h"

/*-----------------------------------------------------------------------*/
/* Get Drive Status                                                      */
/*-----------------------------------------------------------------------*/

DSTATUS disk_status (
	BYTE pdrv		/* Physical drive nmuber to identify the drive */
)
{
	return 0;
}



/*-----------------------------------------------------------------------*/
/* Inidialize a Drive                                                    */
/*-----------------------------------------------------------------------*/

DSTATUS disk_initialize (
	BYTE pdrv				/* Physical drive nmuber to identify the drive */
)
{
	if(bsp_sd_init() != bsp_sd_error_none) {
	    return STA_NOINIT;
	}

	bsp_card_info_t card_info;
	bsp_sd_spi_get_card_info(&card_info);

    LOG_I("sd card successfully initialized, capacity: %dMB", (uint32_t)card_info.sector_count >> 11);

	return 0;
}



/*-----------------------------------------------------------------------*/
/* Read Sector(s)                                                        */
/*-----------------------------------------------------------------------*/

DRESULT disk_read (
	BYTE pdrv,		/* Physical drive nmuber to identify the drive */
	BYTE *buff,		/* Data buffer to store read data */
	LBA_t sector,	/* Start sector in LBA */
	UINT count		/* Number of sectors to read */
)
{
    bsp_sd_error_t sd_error;

	if (count == 1) {
	    sd_error = bsp_sd_read_block(buff, sector);
	} else {
	    sd_error = bsp_sd_read_multi_blocks(buff, sector, count);
	}

	if(sd_error == bsp_sd_error_none) {
		return RES_OK;
	}

    LOG_E("error while reading disk, error code=%d", sd_error);
	return RES_PARERR;
}



/*-----------------------------------------------------------------------*/
/* Write Sector(s)                                                       */
/*-----------------------------------------------------------------------*/

#if FF_FS_READONLY == 0

DRESULT disk_write (
    BYTE pdrv,          /* Physical drive nmuber to identify the drive */
    const BYTE *buff,   /* Data to be written */
    LBA_t sector,       /* Start sector in LBA */
    UINT count          /* Number of sectors to write */
)
{
    bsp_sd_error_t sd_result;

    if (count > 1) {
      sd_result = bsp_sd_write_multi_blocks(buff, sector, count);
    } else {
      sd_result = bsp_sd_write_block(buff, sector);
    }

    if(sd_result == bsp_sd_error_none) {
        return RES_OK;
    }

    LOG_E("error while writing disk, error code=%d", sd_error);
    return RES_ERROR;
}

#endif



/*-----------------------------------------------------------------------*/
/* Miscellaneous Functions                                               */
/*-----------------------------------------------------------------------*/

DRESULT disk_ioctl (
	BYTE pdrv,		/* Physical drive nmuber (0..) */
	BYTE cmd,		/* Control code */
	void *buff		/* Buffer to send/receive control data */
)
{
	switch(cmd) {
	case CTRL_SYNC:
		return RES_OK;
    case GET_SECTOR_SIZE:
    case GET_BLOCK_SIZE:
        *(DWORD *)buff = BSP_SD_BLOCK_SIZE;
        return RES_OK;
    case GET_SECTOR_COUNT:
        bsp_card_info_t card_info;
        bsp_sd_spi_get_card_info(&card_info);
        *(DWORD *)buff = card_info.sector_count;
        return RES_OK;
	default:
		return RES_PARERR;
	}
}