#include <Adafruit_NeoPixel.h>
#include <WiFi.h>
#include <WebServer.h>
#include <ArduinoJson.h>
#include "config.h"
#include "index.h"
#include "fonts.h"
#include "lowLevel.h"
#include "upload_page.h"
#include "prototypes.h"

#ifdef __AVR__
 #include <avr/power.h> // Required for 16 MHz Adafruit Trinket
#endif

Adafruit_NeoPixel pixels(NUMPIXELS, PIN, NEO_GRB + NEO_KHZ800);

TextNode text_nodes[MAX_TEXT_NODES_COUNT];
MultiColorTextNode multi_color_text_node[MAX_TEXT_NODES_COUNT];
Cursor cursor;
unsigned char ever_created_text_nodes = 0;
unsigned char ever_created_multi_color_text_nodes = 0;

Image saved_images[MAX_IMAGES_SAVED] = {
	{},{}
};



short getTextNodeY2(TextNode *node)
{
  return node->pos_y + node->characterSize.height - 1;
}

short getTextNodeX2(TextNode *node)
{
  if (node->character_count == 0) return node->pos_x;
  return node->pos_x + (node->characterSize.width * node->character_count) + (node->character_count - 1) - 1;
}

void addNewTextNode
  (
    char text[TEXT_MAX_LENGTH + 1], uint32_t color, bool handle_pos_via_cursor, short pos_x, short pos_y,
    unsigned char scroll_slowness, bool is_scrolling, bool is_small, short disappear_time, bool is_repeating
  )
{
  unsigned char text_length = strlen(text);
  if (text_length == 0){return;}
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (text_nodes[i].disappear_time == 0)
    {
      strncpy(text_nodes[i].content, text, TEXT_MAX_LENGTH);
      text_nodes[i].color = color;

      if (handle_pos_via_cursor)
      {
		    text_nodes[i].pos_x = cursor.x;
        text_nodes[i].pos_y = cursor.y;
      }
      else
      {
        text_nodes[i].pos_x = pos_x;
        text_nodes[i].pos_y = pos_y;
      }

      text_nodes[i].characterSize.height = is_small ? SMALL_TEXT_HEIGHT : MEDIUM_TEXT_HEIGHT;
      text_nodes[i].characterSize.width = is_small ? SMALL_TEXT_WIDTH : MEDIUM_TEXT_WIDTH;
      text_nodes[i].character_count = text_length;
      text_nodes[i].scroll_slowness = scroll_slowness;
      text_nodes[i].is_scrolling = is_scrolling;
      text_nodes[i].is_repeating = is_repeating;
      text_nodes[i].disappear_time = disappear_time;
      text_nodes[i].id = ever_created_text_nodes;
      ever_created_text_nodes++;

		  if (handle_pos_via_cursor)
      {
        cursor.x += (text_nodes[i].characterSize.width + 1) * text_length;
      }
      break;
    }
  }
}

void modifyTextNodeById(unsigned char id, uint32_t new_color, char new_text[TEXT_MAX_LENGTH + 1], unsigned char new_slowness)
{
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (text_nodes[i].id == id && text_nodes[i].disappear_time != 0)
    {
      text_nodes[i].color = new_color;
      strncpy(text_nodes[i].content, new_text, TEXT_MAX_LENGTH);
      text_nodes[i].character_count = strlen(new_text);
      text_nodes[i].scroll_slowness = new_slowness;
      break;
    }
  }
}

void scrollAllScrollableTexts(bool split_scroll_mode = false)
{
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (text_nodes[i].disappear_time == 0) {continue;}

    if (text_nodes[i].is_scrolling)
    {
        if (text_nodes[i].scroll_slowness > text_nodes[i].scroll_progress)
        {
          text_nodes[i].scroll_progress++;
        }
        else
        {
          text_nodes[i].scroll_progress = 0;

          short x1 = text_nodes[i].pos_x;
          short x2 = getTextNodeX2(&text_nodes[i]);
          if (split_scroll_mode || text_nodes[i].pos_y < 7)
          {
            if (x2 < 0)
            {
              if (text_nodes[i].is_repeating)
              {
                text_nodes[i].pos_x = DISPLAY_MAX_X;
              }
              else
              {
                text_nodes[i].disappear_time = 1;
              }
              continue;
            }
            text_nodes[i].pos_x--;
          }
          else
          {
            if (x1 > DISPLAY_MAX_X)
            {
              if (text_nodes[i].is_repeating)
              {
                text_nodes[i].pos_x = -getTextNodeX2(&text_nodes[i]) + text_nodes[i].pos_x;
              }
              else
              {
                text_nodes[i].disappear_time = 1;
              }
              continue;
            }
            text_nodes[i].pos_x++;
          }
        }
    }

    cursor.x = text_nodes[i].pos_x;
    cursor.y = text_nodes[i].pos_y;

    for (unsigned char j = 0; j < text_nodes[i].character_count; j++)
    {
      char ch = text_nodes[i].content[j];
      if (ch < '!' || ch > '~')
      {
        ch = ' ';
      }
      drawCharacter(font7x5[ch - ' '], text_nodes[i].characterSize.height, text_nodes[i].characterSize.width, text_nodes[i].color, &cursor);
    }
  }
}

void addNewMultiColor
(
  char text[TEXT_MAX_LENGTH + 1], RGBWithIndex colors[4], unsigned char color_count, bool handle_pos_via_cursor, short pos_x, short pos_y,
  unsigned char scroll_slowness, bool is_scrolling, bool is_small, short disappear_time, bool is_repeating
)
{
  unsigned char text_length = strlen(text);
  if (text_length == 0){return;}
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (multi_color_text_node[i].disappear_time == 0)
    {
      strncpy(multi_color_text_node[i].content, text, TEXT_MAX_LENGTH);
      multi_color_text_node[i].color_count = color_count;
      for(int j = 0; j < color_count; j++)
      {
        multi_color_text_node[i].colors[j] = colors[j];
      }

      if (handle_pos_via_cursor)
      {
		    multi_color_text_node[i].pos_x = cursor.x;
        multi_color_text_node[i].pos_y = cursor.y;
      }
      else
      {
        multi_color_text_node[i].pos_x = pos_x;
        multi_color_text_node[i].pos_y = pos_y;
      }

      multi_color_text_node[i].characterSize.height = is_small ? SMALL_TEXT_HEIGHT : MEDIUM_TEXT_HEIGHT;
      multi_color_text_node[i].characterSize.width = is_small ? SMALL_TEXT_WIDTH : MEDIUM_TEXT_WIDTH;
      multi_color_text_node[i].character_count = text_length;
      multi_color_text_node[i].scroll_slowness = scroll_slowness;
      multi_color_text_node[i].is_scrolling = is_scrolling;
      multi_color_text_node[i].is_repeating = is_repeating;
      multi_color_text_node[i].disappear_time = disappear_time;
      multi_color_text_node[i].id = ever_created_multi_color_text_nodes;
      ever_created_multi_color_text_nodes++;

		  if (handle_pos_via_cursor)
      {
        cursor.x += (multi_color_text_node[i].characterSize.width + 1) * text_length;
      }
      break;
    }
  }
}

short getMultiColorTextNodeX2(MultiColorTextNode *node)
{
  if (node->character_count == 0) return node->pos_x;
  return node->pos_x + (node->characterSize.width * node->character_count) + (node->character_count - 1) - 1;
}

void scrollAllMultiColorTexts(bool split_scroll_mode = false)
{
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (multi_color_text_node[i].disappear_time == 0) {continue;}

    if (multi_color_text_node[i].is_scrolling)
    {
        if (multi_color_text_node[i].scroll_slowness > multi_color_text_node[i].scroll_progress)
        {
          multi_color_text_node[i].scroll_progress++;
        }
        else
        {
          multi_color_text_node[i].scroll_progress = 0;

          short x1 = multi_color_text_node[i].pos_x;
          short x2 = getMultiColorTextNodeX2(&multi_color_text_node[i]);
          if (split_scroll_mode || multi_color_text_node[i].pos_y < 7)
          {
            if (x2 < 0)
            {
              if (multi_color_text_node[i].is_repeating)
              {
                multi_color_text_node[i].pos_x = DISPLAY_MAX_X;
              }
              else
              {
                multi_color_text_node[i].disappear_time = 1;
              }
              continue;
            }
            multi_color_text_node[i].pos_x--;
          }
          else
          {
            if (x1 > DISPLAY_MAX_X)
            {
              if (multi_color_text_node[i].is_repeating)
              {
                multi_color_text_node[i].pos_x = -getMultiColorTextNodeX2(&multi_color_text_node[i]) + multi_color_text_node[i].pos_x;
              }
              else
              {
                multi_color_text_node[i].disappear_time = 1;
              }
              continue;
            }
            multi_color_text_node[i].pos_x++;
          }
        }
    }

    cursor.x = multi_color_text_node[i].pos_x;
    cursor.y = multi_color_text_node[i].pos_y;

    uint32_t current_color = pixels.Color(multi_color_text_node[i].colors[0].r, multi_color_text_node[i].colors[0].g, multi_color_text_node[i].colors[0].b);
    unsigned char color_index = 0;

    for (unsigned char j = 0; j < multi_color_text_node[i].character_count; j++)
    {
      if (color_index < multi_color_text_node[i].color_count - 1 && multi_color_text_node[i].colors[color_index+1].start_index == j)
      {
        color_index++;
        current_color = pixels.Color(multi_color_text_node[i].colors[color_index].r, multi_color_text_node[i].colors[color_index].g, multi_color_text_node[i].colors[color_index].b);
      }
      char ch = multi_color_text_node[i].content[j];
      if (ch < '!' || ch > '~')
      {
        ch = ' ';
      }
      drawCharacter(font7x5[ch - ' '], multi_color_text_node[i].characterSize.height, multi_color_text_node[i].characterSize.width, current_color, &cursor);
    }
  }
}

void drawImageFromMemoryByIndex(unsigned char image_index, short pos_x, short pos_y, unsigned char brightness)
{
	Image* img = &saved_images[image_index];

  auto dimBy = [brightness](unsigned char color)
  {
    return color * brightness / 100;
  };

	for (unsigned char y = 0; y < img->height; y++)
  {
    for (unsigned char x = 0; x < img->width; x++)
    {
      setPixel(x + pos_x, y + pos_y, pixels.Color(dimBy(img->pixels[y][x].r), dimBy(img->pixels[y][x].g), dimBy(img->pixels[y][x].b)));
    }
  }
}

void handleDisappearTimers()
{
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (text_nodes[i].disappear_time > 0)
    {
      text_nodes[i].disappear_time--;
      if (text_nodes[i].disappear_time == 0)
      {
        fillPixels(text_nodes[i].pos_x, text_nodes[i].pos_y, getTextNodeX2(&text_nodes[i]), getTextNodeY2(&text_nodes[i]), 0);
      }
    }
  }
}

void handleMultiColorDisappearTimers()
{
  for (unsigned char i = 0; i < MAX_TEXT_NODES_COUNT; i++)
  {
    if (multi_color_text_node[i].disappear_time > 0)
    {
      multi_color_text_node[i].disappear_time--;
      if (multi_color_text_node[i].disappear_time == 0)
      {
        fillPixels(multi_color_text_node[i].pos_x, multi_color_text_node[i].pos_y, getMultiColorTextNodeX2(&multi_color_text_node[i]), multi_color_text_node[i].pos_y + multi_color_text_node[i].characterSize.height - 1, 0);
      }
    }
  }
}

void setup()
{
  Serial.begin(115200);
  Serial.println("----------------------------------------------------");

  pixels.begin();
  pixels.clear();

  addNewTextNode("NET", 0xFF050505, false, 0, 0, 1, true, true, 100);
  addNewTextNode("AWAIT", 0xFF050505, false, 0, 9, 1, true, true, 100);

  pixels.show();
  start_server();
}

void loop()
{
  pixels.clear();
  handle_server();
  handleDisappearTimers();
  handleMultiColorDisappearTimers();
  scrollAllScrollableTexts();
  scrollAllMultiColorTexts();
  
  pixels.show();
}