#include "raylib.h"
#include <math.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>

#include "game.h"
#include "ui.h"

const Color tile_colors[] = {
    LIGHTGRAY, // 0 - Empty
    BEIGE,     // 2
    GOLD,      // 4
    ORANGE,    // 8
    ORANGE,    // 16
    RED,       // 32
    MAROON,    // 64
    YELLOW,    // 128
    GOLD,      // 256
    ORANGE,    // 512
    RED,       // 1024
    MAROON,    // 2048
    PINK,      // 4096
    MAGENTA,   // 8192
};

static GameState_t game_state;

int main() {
  srand(time(NULL));

  reset_board(&game_state);
  FILE* game_file = fopen("save.dat", "rb");
  if (game_file != NULL) {
    fseek(game_file, 0, SEEK_SET);
    fread(&game_state, sizeof(GameState_t), 1, game_file);
    fclose(game_file);
  }

  InitWindow(WINDOW_WIDTH, WINDOW_HEIGHT, "2048");
  SetTargetFPS(30); // So my laptop doesn't burn

  while (!WindowShouldClose()) {
    ClearBackground(RAYWHITE);
    BeginDrawing();

    for (int y = 0; y < BOARD_HEIGHT; y++) {
      for (int x = 0; x < BOARD_WIDTH; x++) {
        int tile_value = game_state.board[y][x];
        int pos_x = BOARD_MARGIN + x * (TILE_SIZE + TILE_SPACING);
        int pos_y = BOARD_MARGIN + y * (TILE_SIZE + TILE_SPACING);
        int color_index = (tile_value == 0) ? 0 : (int)log2(tile_value);

        if (color_index >= sizeof(tile_colors) / sizeof(tile_colors[0])) {
          color_index = sizeof(tile_colors) / sizeof(tile_colors[0]) - 1; // Cap to the last color
        }

        DrawRectangle(pos_x, pos_y, TILE_SIZE, TILE_SIZE, tile_colors[color_index]);
        if (tile_value != 0) {
          const char *text = TextFormat("%d", tile_value);
          DrawText(text, pos_x + TILE_SIZE / 2 - MeasureText(text, 20) / 2, pos_y + TILE_SIZE / 2 - 10, 20, BLACK);
        }
      }
    }

    DrawText(TextFormat("Score: %d", game_state.score), WINDOW_HEIGHT, BOARD_MARGIN, 20, BLACK);
    DrawText(TextFormat("Moves used: %d", game_state.moves_made), WINDOW_HEIGHT, BOARD_MARGIN + 20, 20, BLACK);

    if (game_state.status == PLAYING) {
      DrawText("Drag with the mouse to move tiles", 400 - MeasureText("Drag with the mouse to move tiles", 20) / 2,
               WINDOW_HEIGHT - 40, 20, GRAY);

      // Reset button
      DrawRectangle(500, 455, 275, 24, RED);
      DrawText("Reset", 500 + 275 / 2 - MeasureText("Reset", 20) / 2, 455 + 4, 20, WHITE);
    } else {
      Color overlay = Fade(BLACK, 0.7f);
      DrawRectangle(0, 0, WINDOW_WIDTH, 700, overlay);

      if (game_state.status == WON) {
        DrawText("You win!", 400 - MeasureText("You win!", 40) / 2, 250, 40, GREEN);
      } else if (game_state.status == LOST) {
        DrawText("Game over!", 400 - MeasureText("Game over!", 40) / 2, 250, 40, RED);
      }

      // Draw the scores under the win/loss message
      DrawText(TextFormat("Score: %d", game_state.score),
               400 - MeasureText(TextFormat("Score: %d", game_state.score), 20) / 2, 290, 20, WHITE);
      DrawText(TextFormat("Moves used: %d", game_state.moves_made),
               400 - MeasureText(TextFormat("Moves used: %d", game_state.moves_made), 20) / 2, 310, 20, WHITE);

      DrawRectangle(400 - 100, 450, 200, 40, RED);
      DrawText("Try again", 400 - MeasureText("Try again", 20) / 2, 455 + 4, 20, WHITE);
    }

    EndDrawing();
    handle_mouse_input(&game_state);
  }
  CloseWindow();
  return 0;
}