#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>
#include <sync/spin_lock.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;
  bool irqs_support;
};

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

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;
};

struct ide_probe {
  uint16_t flags;
  size_t sector_count;
  size_t sector_size;
  uint16_t io, ctrl;
  uint8_t devno;
  uint8_t irq;
  bool irqs_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);

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

#endif // _KERNEL_DEVICE_IDEDRV_H