/*
 * Copyright (c) 2006-2022, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2016-09-20     Bernard      the first version
 */

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

#include <rtthread.h>
#include <rtdevice.h>

#include "partition.h"

// #define PARTITION_DEBUG

#ifdef PARTITION_DEBUG
#define DEBUG_TRACE         rt_kprintf("[part] "); rt_kprintf
#else
#define DEBUG_TRACE(...)
#endif

struct partition_device
{
    struct rt_device parent;            /* parent block device */

    int sector_size;
    int block_size;

    struct rt_device *block_device;
    const struct rt_partition *partition;
};

/* RT-Thread device interface */
static rt_err_t partition_init(rt_device_t dev)
{
    return RT_EOK;
}

static rt_err_t partition_open(rt_device_t dev, rt_uint16_t oflag)
{
    return RT_EOK;
}

static rt_err_t partition_close(rt_device_t dev)
{
    return RT_EOK;
}

#if RTTHREAD_VERSION >= 30000
static rt_err_t partition_control(rt_device_t dev, int cmd, void *args)
#else
static rt_err_t partition_control(rt_device_t dev, rt_uint8_t cmd, void *args)
#endif
{
    struct partition_device *part;

    part = (struct partition_device*) dev;
    RT_ASSERT(part != RT_NULL);

    if (cmd == RT_DEVICE_CTRL_BLK_GETGEOME)
    {
        struct rt_device_blk_geometry *geometry;

        geometry = (struct rt_device_blk_geometry *)args;
        if (geometry == RT_NULL) return -RT_ERROR;

        geometry->bytes_per_sector = part->sector_size;
        geometry->sector_count = part->partition->size / part->sector_size;
        geometry->block_size = part->block_size;
    }

    return RT_EOK;
}

static rt_ssize_t partition_read(rt_device_t dev,
                                   rt_off_t pos,
                                   void* buffer,
                                   rt_size_t size)
{
    struct partition_device *part;

    part = (struct partition_device*) dev;
    RT_ASSERT(part != RT_NULL);

    DEBUG_TRACE("read block: %d, size=%d\n", pos + part->partition->offset/part->sector_size, size);

    return rt_device_read(part->block_device, pos + part->partition->offset/part->sector_size, buffer, size);
}

static rt_ssize_t partition_write(rt_device_t dev,
                                    rt_off_t pos,
                                    const void* buffer,
                                    rt_size_t size)
{
    struct partition_device *part;

    part = (struct partition_device*) dev;
    RT_ASSERT(part != RT_NULL);

    if (part->partition->flags & PARTITION_WRITEABLE)
    {
        DEBUG_TRACE("write block: %d, size=%d\n", pos + part->partition->offset/part->sector_size, size);

        return rt_device_write(part->block_device, pos + part->partition->offset/part->sector_size,
                buffer, size);
    }

    DEBUG_TRACE("read-only partition!\n");
    return 0;
}

#ifdef RT_USING_DEVICE_OPS
const static struct rt_device_ops _partition_ops =
{
    .init = partition_init,
    .open = partition_open,
    .close = partition_close,
    .read = partition_read,
    .write = partition_write,
    .control = partition_control,
};
#endif /* RT_USING_DEVICE_OPS */

int rt_partition_init(const char* flash_device, const struct rt_partition* parts, rt_size_t num)
{
    struct rt_device *device;
    struct partition_device *part_dev;
    const struct rt_partition* part;
    struct rt_device_blk_geometry geometry;

    device = (struct rt_device *)rt_device_find(flash_device);
    if(device == RT_NULL)
    {
        DEBUG_TRACE("block device %s not found!\r\n", flash_device);
        return -RT_ENOSYS;
    }

    /* open the block device */
    rt_device_open(device, RT_DEVICE_OFLAG_RDWR);
    /* get block device geometry */
    rt_device_control(device, RT_DEVICE_CTRL_BLK_GETGEOME, &geometry);

    /* create each partition device */
    while(num --)
    {
        part = parts ++;
        part_dev = (struct partition_device*)rt_malloc (sizeof(struct partition_device));
        if (part_dev)
        {
            part_dev->block_device = device;
            part_dev->partition = part;
            part_dev->sector_size = geometry.bytes_per_sector;
            part_dev->block_size  = geometry.block_size;

            /* register device */
            part_dev->parent.type    = RT_Device_Class_Block;
#ifndef RT_USING_DEVICE_OPS
            part_dev->parent.init    = partition_init;
            part_dev->parent.open    = partition_open;
            part_dev->parent.close   = partition_close;
            part_dev->parent.read    = partition_read;
            part_dev->parent.write   = partition_write;
            part_dev->parent.control = partition_control;
#else
            part_dev->parent.ops     = &_partition_ops;
#endif /* RT_USING_DEVICE_OPS */
            /* no private */
            part_dev->parent.user_data = RT_NULL;

            DEBUG_TRACE("add partition: %s\n", part->name);
            rt_device_register(RT_DEVICE(part_dev), part->name,
                               RT_DEVICE_FLAG_RDWR | RT_DEVICE_FLAG_STANDALONE);
        }
    }

    return RT_EOK;
}