#include "FreeRTOS.h"
#include "task.h"
#include "app_tasks.h"

#include "core/core.h"
#include "core/settings.h"
#include "hal.h"
#include "pwr.h"
#include "usb/usb.h"

#define VBAT_MIN 3200
#define VBAT_MAX 4100
#define VBAT_USB 4600

static void pwr_graceful_shutdown() {
  while(hal_flash_busy()) {
    ;
  }

  settings_commit();
}

void pwr_reboot() {
  pwr_graceful_shutdown();
  hal_reboot();
}

void pwr_shutdown() {
  pwr_graceful_shutdown();
  hal_gpio_set(GPIO_PWR_KILL, GPIO_SET);
}

void pwr_usb_plugged(bool from_isr) {
  if (g_settings.enable_usb && g_core.ready) {
    hal_usb_start();
  }

  if (from_isr) {
    BaseType_t xHigherPriorityTaskWoken = pdFALSE;
    vTaskNotifyGiveIndexedFromISR(APP_TASK(usb), USB_NOTIFICATION_IDX, &xHigherPriorityTaskWoken);
    portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
  } else {
    xTaskNotifyGiveIndexed(APP_TASK(usb), USB_NOTIFICATION_IDX);
  }
}

void pwr_usb_unplugged() {
  hal_usb_stop();
}

void pwr_smartcard_inserted() {
#ifndef TEST_APP
  pwr_reboot();
#endif
}

void pwr_smartcard_removed() {
#ifndef TEST_APP
  pwr_shutdown();
#endif
}

void pwr_inactivity_timer_elapsed() {
  pwr_shutdown();
}

uint8_t pwr_battery_level() {
  uint32_t vbat;
  hal_adc_read(ADC_VBAT, &vbat);

  if (vbat > VBAT_USB) {
    return PWR_BATTERY_CHARGING;
  } else if (vbat > VBAT_MAX) {
    vbat = VBAT_MAX;
  } else if (vbat < VBAT_MIN) {
    vbat = VBAT_MIN;
  }

  return ((vbat - VBAT_MIN) * 100) / (VBAT_MAX - VBAT_MIN);
}