#ifndef _KERNEL_DEVICE_IDEDRV_H
#define _KERNEL_DEVICE_IDEDRV_H

#include <device/def_device_op.h>
#include <libk/list.h>
#include <libk/std.h>
#include <proc/proc.h>
#include <proc/reschedule.h>
#include <proc/suspension_q.h>

#define IDE_PROBE_AVAIL (1 << 0)
#define IDE_PROBE_LBA48 (1 << 1)

struct device;

struct idedrv_init {
  bool lba48;
  size_t sector_count;
  size_t sector_size;
  uint16_t io, ctrl;
  uint8_t devno;
  uint8_t irq;
  uint16_t bmbase;
  bool irqs_support;
  bool bm_support;
};

struct idedrv_request {
  uint16_t* buffer;
  size_t sector_done_count;
  size_t sector_count;
  int type;
  atomic_int done;
};

struct ide_prd_entry {
  uint32_t phys_addr;
  uint16_t size;
  uint16_t rsvd_eot;
} PACKED;

struct idedrv {
  struct device* device;
  bool lba48;
  size_t sector_count;
  size_t sector_size;
  uint16_t io, ctrl;
  uint8_t devno;
  uint8_t irq;
  struct idedrv_request* current_req;
  bool irqs_support;
  uint16_t bmbase;
  bool bm_support;
  struct ide_prd_entry* prdt;
  uintptr_t prdt_phys;
  size_t prdt_entry_count;
  uintptr_t bounce_buffer_phys;
  void* bounce_buffer;
};

struct ide_probe {
  uint16_t flags;
  size_t sector_count;
  size_t sector_size;
  uint16_t io, ctrl;
  uint8_t devno;
  uint8_t irq;
  uint16_t bmbase;
  bool irqs_support;
  bool bm_support;
};

DEFINE_DEVICE_INIT(idedrv_init);

DEFINE_DEVICE_FINI(idedrv_fini);

DEFINE_DEVICE_OP(idedrv_read);

DEFINE_DEVICE_OP(idedrv_write);

DEFINE_DEVICE_OP(idedrv_get_device_type);

DEFINE_DEVICE_OP(idedrv_get_sector_size);

DEFINE_DEVICE_OP(idedrv_get_size);

DEFINE_DEVICE_OP(idedrv_partition_rescan);

void ide_probe(uint16_t io, uint16_t ctrl, uint8_t devno, struct ide_probe* probe);

#endif // _KERNEL_DEVICE_IDEDRV_H