#ifndef __ZVFS_SPDK_ENGINE_H__
#define __ZVFS_SPDK_ENGINE_H__

#include "common/uthash.h"
#include "proto/ipc_proto.h"
#include <stdint.h>
#include <sys/types.h>
#include <stdatomic.h>
#include <pthread.h>
#include <spdk/blob.h>


// blob_handle 结构体：底层 blob 信息，不含文件级 size（上层维护）
typedef struct zvfs_blob_handle {
    spdk_blob_id         blob_id;
    struct spdk_blob    *blob;         
    atomic_uint          ref_count;
    uint64_t             current_offset;
    uint32_t             status_flags;
    pthread_mutex_t      state_mu;
} zvfs_blob_handle_t;

struct zvfs_io_thread {
    struct spdk_thread     *thread;
    struct spdk_io_channel *channel;   // 每个 io 线程独占一个 channel
    pthread_t               tid;
    bool                    ready;
};

typedef uint64_t    zvfs_handle_id_t;

struct zvfs_blob_cache_entry {
    zvfs_handle_id_t              handle_id;   // key != blob_id
    struct zvfs_blob_handle *handle;
    UT_hash_handle           hh;
};

typedef struct zvfs_spdk_io_engine {
    struct spdk_bs_dev      *bs_dev;
    struct spdk_blob_store  *bs;


    /* 线程池：thread_pool[0] 固定为 md 线程，其余为 io 线程 */
    struct zvfs_io_thread   *thread_pool;       // 线程池
    int                      thread_count;      // 总线程数 (= CPU 核心数)
    int                      io_thread_count;   // 线程数量

    struct zvfs_blob_cache_entry *handle_cache;   // handle_id -> handle 映射
    pthread_mutex_t               cache_mu;

    uint64_t                 io_unit_size;
    uint64_t                 cluster_size;


    /**
     * 全局 DMA buf pool。
     * 所有 IO 请求（read / write）从这里借用 buf，完成后归还。
     * buf 大小固定为 ZVFS_DMA_BUF_SIZE（1MB），对齐到 io_unit_size。
     * 在 io_engine_init 完成、io_unit_size 确定后创建。
     */
    struct dma_buf_pool     *dma_pool;
} zvfs_spdk_io_engine_t;

/* ------------------------------------------------------------------ */
/* handle cache 操作（实现在 spdk_engine.c）                           */
/* ------------------------------------------------------------------ */
int engine_cache_insert(struct zvfs_blob_handle *handle, zvfs_handle_id_t *out_id);
struct zvfs_blob_handle *engine_cache_lookup(zvfs_handle_id_t handle_id);
void engine_cache_remove(zvfs_handle_id_t handle_id);

/* ------------------------------------------------------------------ */
/* 引擎公开接口                                                         */
/* ------------------------------------------------------------------ */
int io_engine_init(const char *bdev_name, const char *json_file, int thread_num);
int blob_create(struct zvfs_req *req);
int blob_open(struct zvfs_req *req);
int blob_write(struct zvfs_req *req);
int blob_read(struct zvfs_req *req);
int blob_resize(struct zvfs_req *req);
int blob_sync_md(struct zvfs_req *req);
int blob_close(struct zvfs_req *req);
int blob_delete(struct zvfs_req *req);
int blob_seek(struct zvfs_req *req);
int blob_get_pos(struct zvfs_req *req);
int blob_get_flags(struct zvfs_req *req);
int blob_set_flags(struct zvfs_req *req);
int blobstore_reset(struct zvfs_req *req);
bool io_engine_reset_in_progress(void);

#endif // __ZVFS_IO_ENGINE_H__