#pragma once

#include <glacier/container/array_view.h>
#include <mammoth/util/memory_region.h>

#include "xhci/xhci.h"

class TrbRing {
 public:
  TrbRing();

  uint64_t PhysicalAddress() { return phys_address_; }
  XhciTrb GetTrbFromPhysical(uint64_t address);

 protected:
  uint64_t phys_address_;
  mmth::OwnedMemoryRegion page_;
  glcr::ArrayView<XhciTrb> trb_list_;
};

class TrbRingWriter : public TrbRing {
 public:
  uint64_t EnqueueTrb(const XhciTrb& trb);

 private:
  uint64_t enqueue_ptr_ = 0;
};

class TrbRingReader : public TrbRing {
 public:
  bool HasNext();
  XhciTrb Read();

  uint64_t DequeuePtr() { return phys_address_ + dequeue_ptr_; }

 private:
  uint64_t dequeue_ptr_ = 0;
  uint8_t cycle_bit_ = 1;
};