#include <stdio.h>
#include <stdlib.h>
#include <curses.h>
#include "board.h"

void draw_board(BoardUnit* board, WINDOW* win) {
  wmove(win, 0, 0);
  for (int row = 0; row < BOARD_ROW_UNITS; row++) {
    for (int i = 0; i < BOARD_ROW_UNITS; i++) {
      int index = BOARD_ROW_UNITS * row + i;
      if (board[index].feature == EMPTY && is_center_index(index)) waddch(win, '*');
      else waddch(win, board[index].feature | COLOR_PAIR(board[index].structure_group % 6));
    }
  }
  wrefresh(win);
}

int main() {
  /* initialize curses */
  initscr();
  cbreak();
  use_default_colors();

  /* colors */
  start_color();
  for (int i = 1; i < 6; i++) init_pair(i, COLOR_BLACK + i, COLOR_BLACK);

  /* create board window */
  WINDOW* board_box = newwin(BOARD_ROW_UNITS + 2, BOARD_ROW_UNITS + 2, 0, 0);
  WINDOW* board_win = derwin(board_box, BOARD_ROW_UNITS, BOARD_ROW_UNITS, 1, 1);
  box(board_box, 0, 0);
  mvwaddstr(board_box, 0, 1, "Board");
  wrefresh(board_box);

  /* create messages window */
  WINDOW* messages_box = newwin(BOARD_ROW_UNITS + 2, 60, 0, BOARD_ROW_UNITS + 3);
  WINDOW* messages_win = derwin(messages_box, BOARD_ROW_UNITS, 60 - 2, 1, 1);
  box(messages_box, 0, 0);
  mvwaddstr(messages_box, 0, 1, "Log");
  wrefresh(messages_box);


  BoardUnit board[BOARD_UNITS];
  initialize_board(board);

  Tile tile = { "FRCR", 'R', 0 };
  place_tile(tile, translate_coordinate(BOARD_WIDTH * BOARD_WIDTH / 2), board, 1);

  Tile tileset[5] = {
    { "RRRR", 'R', 0 },
    { "FCCC", 'C', 0 },
    { "FFFC", 'C', 0 },
    { "CCRR", '*', 0 },
    { "RFRF", 'R', 0 }
  };

  /* main loop */
  char input_key;
  while (1) {
    /* board */
    refresh_structure_groups(board);
    draw_board(board, board_win);
    if (wgetch(board_win) == 'q') {
      endwin();
      return 0;
    }

    /* tile placement */
    tile = tileset[rand() % 5];
    int position = 0;
    BoardUnit board_preview[BOARD_UNITS];

    while (1) {
      for (int i = 0; i < BOARD_UNITS; i++) {
        board_preview[i].feature = board[i].feature;
        board_preview[i].structure_group = board[i].structure_group;
      }
      place_tile(tile, translate_coordinate(position), board_preview, 1);

      int is_allowed = is_allowed_placement(tile, translate_coordinate(position), board);

      if (is_allowed) wattron(board_win, COLOR_PAIR(2));
      draw_board(board_preview, board_win);
      wattroff(board_win, COLOR_PAIR(2));

      input_key = wgetch(board_win);
      if (input_key == 10 && is_allowed) break; /* enter key */
      else if (input_key == 'l') position += 1;
      else if (input_key == 'h') position -= 1;
      else if (input_key == 'j') position += BOARD_WIDTH;
      else if (input_key == 'k') position -= BOARD_WIDTH;
      else if (input_key == 'r') rotate_tile(&tile, 3);
    }

    int result = place_tile(tile, translate_coordinate(position), board, 0);
    if (result) wprintw(messages_win, "Placed tile %s (%c) at position %i\n", tile.edges, tile.center, position);
    else wprintw(messages_win, "Could not place tile %s (%c) at position %i\n", tile.edges, tile.center, position);
    wrefresh(messages_win);
  }

  endwin();
  return 0;
}