#include <device_info.h>
#include <devices.h>
#include <malloc.h>
#include <mprintf.h>
#include <process.h>
#include <status.h>
#include <str_status.h>
#include <string.h>
#include <system.h>
#include <volume_info.h>

static void usb_poll(void) {
  struct device_info* infos = malloc(sizeof(struct device_info) * 1024);
  memset(infos, 0, sizeof(struct device_info) * 1024);

  int device_count = get_device_info(infos, 1024);
  struct device_info info;
  bool found = false;

  for (int dev = 0; dev < device_count; dev++) {
    info = infos[dev];

    if (info.type == DEVICE_TYPE_USB_CTRL) {
      found = true;
      break;
    }
  }

  if (found) {
    mprintf("Polling device %s\n", info.key);

    for (;;) {
      device_do(info.key, XUSBCTRL_POLL_DRIVER, NULL, NULL, NULL, NULL);

      for (volatile int i = 0; i < 1000; i++) {
        sched();
#if defined(__x86_64__)
        __asm__ volatile("pause" ::: "memory");
#endif
      }
    }
  }

  free(infos);
}

static void usb_eject(const char* dev_key) {
  struct volume_info* volume_infos = malloc(sizeof(*volume_infos) * 100);
  memset(volume_infos, 0, sizeof(*volume_infos) * 100);

  size_t count = get_volume_info(volume_infos, 100);

  for (size_t i = 0; i < count; i++) {
    struct volume_info* info = &volume_infos[i];

    if (strcmp(info->device_key, dev_key) == 0) {
      int ret = volume_delete(info->volume_name);
      mprintf("Deleted volume %s: %s\n", info->volume_name, str_status[ret < 0 ? -ret : ret]);
      break;
    }
  }

  free(volume_infos);
}

void app_main(void) {
  char commandbuf[32];
  memset(commandbuf, 0, sizeof(commandbuf));
  char devnamebuf[32];
  memset(devnamebuf, 0, sizeof(devnamebuf));

  if (env_get(process_get_pgid(), "C", (void*)commandbuf, sizeof(commandbuf)) != ST_OK) {
    mprintf("ERROR C=???. No command provided\n");
    return;
  }

  if (strcmp(commandbuf, "poll") == 0) {
    usb_poll();
  } else if (strcmp(commandbuf, "eject") == 0) {
    if (env_get(process_get_pgid(), "dev", (void*)devnamebuf, sizeof(devnamebuf)) != ST_OK) {
      mprintf("ERROR No device provided\n");
      return;
    }

    usb_eject(devnamebuf);
  } else {
    mprintf("ERROR unknown command %s\n", commandbuf);
  }
}