/*
 * Copyright (c) 2018 naehrwert
 * Copyright (c) 2018-2023 CTCaer
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <storage/sd.h>
#include <storage/sdmmc.h>
#include <storage/sdmmc_driver.h>
#include <gfx_utils.h>
#include <libs/fatfs/ff.h>
#include <mem/heap.h>

#ifndef BDK_SDMMC_UHS_DDR200_SUPPORT
#define SD_DEFAULT_SPEED SD_UHS_SDR104
#else
#define SD_DEFAULT_SPEED SD_UHS_DDR208
#endif

static bool sd_mounted = false;
static bool sd_init_done = false;
static bool insertion_event = false;
static u16  sd_errors[3] = { 0 }; // Init and Read/Write errors.
static u32  sd_mode = SD_DEFAULT_SPEED;


sdmmc_t sd_sdmmc;
sdmmc_storage_t sd_storage;
FATFS sd_fs;

void sd_error_count_increment(u8 type)
{
	switch (type)
	{
	case SD_ERROR_INIT_FAIL:
		sd_errors[0]++;
		break;
	case SD_ERROR_RW_FAIL:
		sd_errors[1]++;
		break;
	case SD_ERROR_RW_RETRY:
		sd_errors[2]++;
		break;
	}
}

u16 *sd_get_error_count()
{
	return sd_errors;
}

bool sd_get_card_removed()
{
	if (insertion_event && !sdmmc_get_sd_inserted())
		return true;

	return false;
}

bool sd_get_card_initialized()
{
	return sd_init_done;
}

bool sd_get_card_mounted()
{
	return sd_mounted;
}

u32 sd_get_mode()
{
	return sd_mode;
}

int sd_init_retry(bool power_cycle)
{
	u32 bus_width = SDMMC_BUS_WIDTH_4;
#ifndef BDK_SDMMC_UHS_DDR200_SUPPORT
	u32 type = SDHCI_TIMING_UHS_SDR104;
#else
	u32 type = SDHCI_TIMING_UHS_DDR200;
#endif

	// Power cycle SD card.
	if (power_cycle)
	{
		sd_mode--;
		sdmmc_storage_end(&sd_storage);
	}

	// Get init parameters.
	switch (sd_mode)
	{
	case SD_INIT_FAIL: // Reset to max.
		return 0;

	case SD_1BIT_HS25:
		bus_width = SDMMC_BUS_WIDTH_1;
		type = SDHCI_TIMING_SD_HS25;
		break;

	case SD_4BIT_HS25:
		type = SDHCI_TIMING_SD_HS25;
		break;

	case SD_UHS_SDR82:
		type = SDHCI_TIMING_UHS_SDR82;
		break;

	case SD_UHS_SDR104:
		type = SDHCI_TIMING_UHS_SDR104;
		break;

#ifdef BDK_SDMMC_UHS_DDR200_SUPPORT
	case SD_UHS_DDR208:
		type = SDHCI_TIMING_UHS_DDR200;
		break;
#endif

	default:
		sd_mode = SD_DEFAULT_SPEED;
		break;
	}

	int res = sdmmc_storage_init_sd(&sd_storage, &sd_sdmmc, bus_width, type);
	if (res)
	{
		sd_init_done    = true;
		insertion_event = true;
	}
	else
		sd_init_done = false;

	return res;
}

bool sd_initialize(bool power_cycle)
{
	if (power_cycle)
		sdmmc_storage_end(&sd_storage);

	int res = !sd_init_retry(false);

	while (true)
	{
		if (!res)
			return true;
		else if (!sdmmc_get_sd_inserted()) // SD Card is not inserted.
		{
			sd_mode = SD_DEFAULT_SPEED;
			break;
		}
		else
		{
			sd_errors[SD_ERROR_INIT_FAIL]++;

			if (sd_mode == SD_INIT_FAIL)
				break;
			else
				res = !sd_init_retry(true);
		}
	}

	sdmmc_storage_end(&sd_storage);

	return false;
}

bool sd_mount()
{
	if (sd_init_done && sd_mounted)
		return true;

	int res = 0;

	if (!sd_init_done)
		res = !sd_initialize(false);

	if (res)
	{
		gfx_con.mute = false;
		EPRINTF("Failed to init SD card.");
		if (!sdmmc_get_sd_inserted())
			EPRINTF("Make sure that it is inserted.");
		else
			EPRINTF("SD Card Reader is not properly seated!");
	}
	else
	{
		if (!sd_mounted)
			res = f_mount(&sd_fs, "0:", 1); // Volume 0 is SD.
		if (res == FR_OK)
		{
			sd_mounted = true;
			return true;
		}
		else
		{
			gfx_con.mute = false;
			EPRINTFARGS("Failed to mount SD card (FatFS Error %d).\nMake sure that a FAT partition exists..", res);
		}
	}

	return false;
}

static void _sd_deinit(bool deinit)
{
	if (deinit)
	{
		insertion_event = false;
		if (sd_mode == SD_INIT_FAIL)
			sd_mode = SD_DEFAULT_SPEED;
	}

	if (sd_init_done)
	{
		if (sd_mounted)
			f_mount(NULL, "0:", 1); // Volume 0 is SD.

		if (deinit)
		{
			sdmmc_storage_end(&sd_storage);
			sd_init_done = false;
		}
	}
	sd_mounted = false;
}

void sd_unmount() { _sd_deinit(false); }
void sd_end()     { _sd_deinit(true); }

bool sd_is_gpt()
{
	return sd_fs.part_type;
}

void *sd_file_read(const char *path, u32 *fsize)
{
	FIL fp;
	if (!sd_get_card_mounted())
		return NULL;

	if (f_open(&fp, path, FA_READ) != FR_OK)
		return NULL;

	u32 size = f_size(&fp);
	if (fsize)
		*fsize = size;

	void *buf = malloc(size);

	if (f_read(&fp, buf, size, NULL) != FR_OK)
	{
		free(buf);
		f_close(&fp);

		return NULL;
	}

	f_close(&fp);

	return buf;
}

int sd_save_to_file(void *buf, u32 size, const char *filename)
{
	FIL fp;
	u32 res = 0;
	if (!sd_get_card_mounted())
		return FR_DISK_ERR;

	res = f_open(&fp, filename, FA_CREATE_ALWAYS | FA_WRITE);
	if (res)
	{
		EPRINTFARGS("Error (%d) creating file\n%s.\n", res, filename);
		return res;
	}

	f_write(&fp, buf, size, NULL);
	f_close(&fp);

	return 0;
}