#include <dos.h>
#include <conio.h>
#include <i86.h>

#include "mouse_io.h"

struct MouseStatus *_status;

/**
 * Ensure this is called after setting the desired video mode.
 */
int activateMouse(struct MouseStatus *status) {
  union REGS regs;
  int mouseActivated;

  regs.w.ax = MOUSE_DRIVER_RESET;
  int386(MOUSE_DRIVER_INTERRUPT, &regs, &regs);

  mouseActivated = regs.w.ax;

  if (mouseActivated) {
    status->isActive = regs.w.ax;
    status->buttonCount = regs.w.bx;

    // set horiz and vert range
    regs.w.ax = 0x07;
    regs.w.cx = 0;
    regs.w.dx = VGA_DISPLAY_WIDTH - 1;
    int386(MOUSE_DRIVER_INTERRUPT, &regs, &regs);

    regs.w.ax = 0x08;
    regs.w.cx = 0;
    regs.w.dx = VGA_DISPLAY_HEIGHT - 1;
    int386(MOUSE_DRIVER_INTERRUPT, &regs, &regs);
  }

  _status = status;

  return mouseActivated;
}

void readMouse() {
  union REGS regs;
  int buttonStatus;

  regs.w.ax = MOUSE_DRIVER_READ_STATE;
  int386(MOUSE_DRIVER_INTERRUPT, &regs, &regs);

  buttonStatus = regs.w.bx;
  _status->xPosition = regs.w.cx;
  _status->yPosition = regs.w.dx;

  _status->leftButtonDown = buttonStatus & 1;
  _status->rightButtonDown = (buttonStatus >> 1) & 1;
}