#ifndef _KERNEL_DEVICE_IDEDRV_H
#define _KERNEL_DEVICE_IDEDRV_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;
};

bool idedrv_init (struct device* device, void* arg, struct proc* proc, struct reschedule_ctx* rctx);

void idedrv_fini (struct device* device, struct proc* proc, struct reschedule_ctx* rctx);

int idedrv_read (struct device* device, struct proc* proc, struct reschedule_ctx* rctx,
                 uint64_t* lockflags, void* a1, void* a2, void* a3, void* a4);

int idedrv_write (struct device* device, struct proc* proc, struct reschedule_ctx* rctx,
                  uint64_t* lockflags, void* a1, void* a2, void* a3, void* a4);

int idedrv_get_device_type (struct device* device, struct proc* proc, struct reschedule_ctx* rctx,
                            uint64_t* lockflags, void* a1, void* a2, void* a3, void* a4);

int idedrv_get_sector_size (struct device* device, struct proc* proc, struct reschedule_ctx* rctx,
                            uint64_t* lockflags, void* a1, void* a2, void* a3, void* a4);

int idedrv_get_size (struct device* device, struct proc* proc, struct reschedule_ctx* rctx,
                     uint64_t* lockflags, void* a1, void* a2, void* a3, void* a4);

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

#endif // _KERNEL_DEVICE_IDEDRV_H