/*
 * Copyright (C) 2022-2024, Xiaohua Semiconductor Co., Ltd.
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2022-04-28     CDT          first version
 */

#include <board.h>
#include <drv_spi.h>
#include <rtdevice.h>
#include <rthw.h>
#include <finsh.h>
#include <dfs_fs.h>
#include <fal.h>

#ifdef BSP_USING_SPI_FLASH

#include "spi_flash.h"
#ifdef RT_USING_SFUD
    #include "spi_flash_sfud.h"
#endif

#if defined(HC32F4A0) || defined(HC32F448)
    #define SPI_BUS_NAME                "spi1"
    #define SPI_FLASH_DEVICE_NAME       "spi10"
    #define SPI_FLASH_CHIP              "w25q64"
    #define SPI_FLASH_SS_PORT           GPIO_PORT_C
    #define SPI_FLASH_SS_PIN            GPIO_PIN_07
#elif defined(HC32F460)
    #define SPI_BUS_NAME                "spi3"
    #define SPI_FLASH_DEVICE_NAME       "spi30"
    #define SPI_FLASH_CHIP              "w25q64"
    #define SPI_FLASH_SS_PORT           GPIO_PORT_C
    #define SPI_FLASH_SS_PIN            GPIO_PIN_07
#elif defined(HC32F472)
    #define SPI_BUS_NAME                "spi1"
    #define SPI_FLASH_DEVICE_NAME       "spi10"
    #define SPI_FLASH_CHIP              "w25q64"
    #define SPI_FLASH_SS_PORT           GPIO_PORT_B
    #define SPI_FLASH_SS_PIN            GPIO_PIN_12
#endif
#define SPI_FLASH_CMD_ENABLE_RESET      0x66
#define SPI_FLASH_CMD_RESET_DEVICE      0x99

/* Partition Name */
#define FS_PARTITION_NAME              "filesystem"


#ifdef RT_USING_SFUD
static void rt_hw_spi_flash_reset(char *spi_dev_name)
{
    struct rt_spi_device *spi_dev_w25;
    rt_uint8_t w25_en_reset = SPI_FLASH_CMD_ENABLE_RESET;
    rt_uint8_t w25_reset_dev = SPI_FLASH_CMD_RESET_DEVICE;

    spi_dev_w25 = (struct rt_spi_device *)rt_device_find(spi_dev_name);
    if (!spi_dev_w25)
    {
        rt_kprintf("Can't find %s device!\n", spi_dev_name);
    }
    else
    {
        rt_spi_send(spi_dev_w25, &w25_en_reset, 1U);
        rt_spi_send(spi_dev_w25, &w25_reset_dev, 1U);
        DDL_DelayMS(1U);
        rt_kprintf("Reset ext flash!\n");
    }
}

static int rt_hw_spi_flash_with_sfud_init(void)
{
    rt_hw_spi_device_attach(SPI_BUS_NAME, SPI_FLASH_DEVICE_NAME, SPI_FLASH_SS_PORT, SPI_FLASH_SS_PIN);

    if (RT_NULL == rt_sfud_flash_probe(SPI_FLASH_CHIP, SPI_FLASH_DEVICE_NAME))
    {
        rt_hw_spi_flash_reset(SPI_FLASH_DEVICE_NAME);
        if (RT_NULL == rt_sfud_flash_probe(SPI_FLASH_CHIP, SPI_FLASH_DEVICE_NAME))
        {
            return -RT_ERROR;
        }
    }

    return RT_EOK;
}
INIT_COMPONENT_EXPORT(rt_hw_spi_flash_with_sfud_init);

static int rt_hw_fs_init(void)
{
    struct rt_device *mtd_dev = RT_NULL;

    /* 初始化 fal */
    fal_init();
    /* 生成 mtd 设备 */
    mtd_dev = fal_mtd_nor_device_create(FS_PARTITION_NAME);
    if (!mtd_dev)
    {
        LOG_E("Can't create a mtd device on '%s' partition.", FS_PARTITION_NAME);
        return -RT_ERROR;
    }
    else
    {
        /* 挂载 littlefs */
        if (RT_EOK == dfs_mount(FS_PARTITION_NAME, "/", "lfs", 0, 0))
        {
            LOG_I("Filesystem initialized!");
            return RT_EOK;
        }
        else
        {
            /* 格式化文件系统 */
            if (RT_EOK == dfs_mkfs("lfs", FS_PARTITION_NAME))
            {
                /* 挂载 littlefs */
                if (RT_EOK == dfs_mount(FS_PARTITION_NAME, "/", "lfs", 0, 0))
                {
                    LOG_I("Filesystem initialized!");
                    return RT_EOK;
                }
                else
                {
                    LOG_E("Failed to initialize filesystem!");
                    return -RT_ERROR;
                }
            }
            else
            {
                LOG_E("Failed to Format fs!");
                return -RT_ERROR;
            }
        }
    }
}
INIT_APP_EXPORT(rt_hw_fs_init);

#endif /* RT_USING_SFUD */

#endif /* BSP_USING_SPI_FLASH */