#include <stdio.h>
#include <string.h>
#include "splash.h"
#include "oled.h"
#include "ff.h"
#include "buttons.h"

void show_message(char *text, uint8_t wait)
{
  char* second_line;

  oled_draw_rectangle(0, oled_get_line() + OLED_HEIGHT, OLED_WIDTH - 1, oled_get_line() + OLED_HEIGHT * 2 - 1, 1, 0);

  second_line = strstr(text, "\n");
  if (!second_line)
  {
    oled_draw_text(&SPLASH_REGULAR_FONT, text, OLED_WIDTH / 2 - oled_get_text_length(&SPLASH_REGULAR_FONT, text) / 2,
      oled_get_line() + OLED_HEIGHT + OLED_HEIGHT / 2 - SPLASH_REGULAR_FONT.char_height / 2, 0, 0);
  } else {
    // two-lined text
    second_line++;
    // we need to make a copy of text because text can point to non-writable memory
    char first_line[second_line - text];
    memcpy(first_line, text, second_line - text - 1);
    first_line[second_line - text - 1] = 0;
    oled_draw_text(&SPLASH_REGULAR_FONT, first_line, OLED_WIDTH / 2 - oled_get_text_length(&SPLASH_REGULAR_FONT, first_line) / 2,
      oled_get_line() + OLED_HEIGHT + OLED_HEIGHT / 2 - SPLASH_REGULAR_FONT.char_height, 0, 0);
    oled_draw_text(&SPLASH_REGULAR_FONT, second_line, OLED_WIDTH / 2 - oled_get_text_length(&SPLASH_REGULAR_FONT, second_line) / 2,
      oled_get_line() + OLED_HEIGHT + OLED_HEIGHT / 2, 0, 0);
  }

  oled_update_invisible();
  oled_switch_to_invisible();

  if (wait)
  {
    while (button_up_holding() || button_down_holding() || button_left_holding() || button_right_holding())
      HAL_Delay(1);
    while (!button_up_newpress() && !button_down_newpress() && !button_left_newpress() && !button_right_newpress())
      HAL_Delay(1);
  }
}

void show_updating_screen()
{
  oled_draw_rectangle(0, oled_get_line() + OLED_HEIGHT, OLED_WIDTH - 1, oled_get_line() + OLED_HEIGHT * 2 - 1, 1, 0);
  oled_draw_text(&SPLASH_REGULAR_FONT, "Updating", 10, oled_get_line() + OLED_HEIGHT + OLED_HEIGHT / 2 - SPLASH_REGULAR_FONT.char_height / 2, 0, 0);
  oled_draw_image(&IMAGE_COGS, OLED_WIDTH - IMAGE_COGS.width - 10, oled_get_line() + OLED_HEIGHT + OLED_HEIGHT / 2 - IMAGE_COGS.height / 2, 0, 0);
  oled_update_invisible();
  oled_switch_to_invisible();
}

void show_error_screen(char *text, uint8_t fatal)
{
  if (fatal)
    __disable_irq();

  // we need to re-init i2c in case of interrupt
  // because it can be used and we need it to
  // output error message
  HAL_I2C_DeInit(&hi2c1);
  HAL_I2C_Init(&hi2c1);

  oled_draw_rectangle(0, oled_get_line() + OLED_HEIGHT, OLED_WIDTH - 1, oled_get_line() + OLED_HEIGHT * 2 - 1, 1, 0);
  oled_draw_text(&SPLASH_ERROR_TITLE_FONT, "Error :(", 4, oled_get_line() + OLED_HEIGHT, 0, 0);
  oled_draw_text(&SPLASH_ERROR_FONT, text, 4, oled_get_line() + OLED_HEIGHT + 20, 0, 0);
  oled_update_invisible();
  oled_switch_to_invisible();

  if (fatal)
  {
    while (1);
  } else {
    while (button_up_holding() || button_down_holding() || button_left_holding() || button_right_holding())
      HAL_Delay(1);
    while (!button_up_newpress() && !button_down_newpress() && !button_left_newpress() && !button_right_newpress())
      HAL_Delay(1);
  }
}

void show_error_screen_fr(FRESULT fr, uint8_t fatal)
{
  char *text;
  switch ((int)fr)
  {
    case FR_OK: return;        /* (0) Succeeded */
    case FR_DISK_ERR: text = "SD card error"; break;      /* (1) A hard error occurred in the low level disk I/O layer */
    case FR_INT_ERR: text = "Internal FAT error"; break;       /* (2) Assertion failed */
    case FR_NOT_READY: text = "SD card not ready"; break;     /* (3) The physical drive cannot work */
    case FR_NO_FILE: text = "File not found"; break;       /* (4) Could not find the file */
    case FR_NO_PATH: text = "Path not found"; break;       /* (5) Could not find the path */
    case FR_INVALID_NAME: text = "Invalid path name"; break;    /* (6) The path name format is invalid */
    case FR_DENIED: text = "Access denied"; break;        /* (7) Access denied due to prohibited access or directory full */
    case FR_EXIST: text = "Already exists"; break;       /* (8) Access denied due to prohibited access */
    case FR_INVALID_OBJECT: text = "Invalid object"; break;    /* (9) The file/directory object is invalid */
    case FR_WRITE_PROTECTED: text = "Write protected"; break;   /* (10) The physical drive is write protected */
    case FR_INVALID_DRIVE: text = "Invalid drive"; break;   /* (11) The logical drive number is invalid */
    case FR_NOT_ENABLED: text = "Not enabled"; break;     /* (12) The volume has no work area */
    case FR_NO_FILESYSTEM: text = "No filesystem"; break;   /* (13) There is no valid FAT volume */
    case FR_MKFS_ABORTED: text = "MKFS aborted"; break;    /* (14) The f_mkfs() aborted due to any problem */
    case FR_TIMEOUT: text = "SD card timeout"; break;       /* (15) Could not get a grant to access the volume within defined period */
    case FR_LOCKED: text = "Locked"; break;        /* (16) The operation is rejected according to the file sharing policy */
    case FR_NOT_ENOUGH_CORE: text = "Not enough core"; break;   /* (17) LFN working buffer could not be allocated */
    case FR_TOO_MANY_OPEN_FILES: text = "Too many open files"; break; /* (18) Number of open files > _FS_LOCK */
    case FR_INVALID_PARAMETER: text = "Invalid parameter"; break;  /* (19) Given parameter is invalid */
    default: text = "Unknown error";
  }
  show_error_screen(text, fatal);
}

void show_error_screen_sd(SD_RESULT r, uint8_t fatal)
{
  char text[32];
  sprintf(text, "SD card error %d", r);
  show_error_screen(text, fatal);
}