//tic-tac-toe.c:

/*
 *      Copyright (C) Philipp 'ph3-der-loewe' Schafft - 2012-2013
 *
 *  This file is part of roard a part of RoarAudio,
 *  a cross-platform sound system for both, home and professional use.
 *  See README for details.
 *
 *  This file is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 3
 *  as published by the Free Software Foundation.
 *
 *  RoarAudio is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this software; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 51 Franklin Street, Fifth Floor,
 *  Boston, MA 02110-1301, USA.
 *
 */

#include <roaraudio.h>

#define _FREE     ' '
#define _PLAYER   'X'
#define _COMPUTER 'O'
#define _NOBODY   '#'

#define _DISPLAY \
/* Bring Telnet in right mode: */ \
/* IAC WILL ECHO IAC WILL SUPPRESS_GO_AHEAD IAC WONT LINEMODE */ \
/* 255  251    1 255  251                 3 255  252       34 */ \
"\377\373\001\377\373\003\377\374\042" \
"\e[2J\e[H" \
" Tic Tac Toe\r\n" \
" (running: %s, API: %s)\r\n" \
"\r\n" \
"Game board:     IDs:\r\n"        \
" %c | %c | %c      7 | 8 | 9\r\n"   \
"---+---+---     ---+---+---\r\n" \
" %c | %c | %c      4 | 5 | 6\r\n"   \
"---+---+---     ---+---+---\r\n" \
" %c | %c | %c      1 | 2 | 3\r\n"   \
"\r\n" \
"%s\r\n"
#define _DISPLAY_BUFFER_LEN 1024

#define _HELP \
"\e[2J\e[H" \
" Tic Tac Toe\r\n" \
" (running: roard " ROAR_VERSION_STRING ")\r\n" \
"\r\n" \
"IDs:            Alternative IDs:\r\n" \
" 7 | 8 | 9       a | s | d\r\n"   \
"---+---+---     ---+---+---\r\n"  \
" 4 | 5 | 6       f |SPC| h\r\n"   \
"---+---+---     ---+---+---\r\n"  \
" 1 | 2 | 3       j | k | l\r\n"   \
"\r\n" \
"Other keys:\r\n" \
" q g , .  - Quit\r\n" \
" n +      - New game\r\n" \
" H 0      - Help\r\n" \
" e        - Exit help / Redraw game\r\n" \
"\r\n" \
"%s\r\n"

#define MSG__YOUR_TURN       "It's your turn. -- press any ID or H for help, q to quit --"
#define MSG__CANNOT_PUT_HERE "You can not put your mark here. -- try again another position --"
#define MSG__WON_PLAYER      "You won. -- press any key to continue or H for help, q to quit --"
#define MSG__WON_COMPUTER    "I won. -- press any key to continue or H for help, q to quit --"
#define MSG__WON_NOBODY      "Nobody won. -- press any key to continue or H for help, q to quit --"
#define MSG__HELP            "-- press e to continue or q to quit --"

typedef char game_state[9];

struct test_result {
 int score_player, score_computer;
 int free;
};

struct stats {
 int free;
 struct test_result results[9];
};

static void new_game(game_state * state) {
 memset(state, _FREE, sizeof(game_state));
}

static int try_put(game_state * state, int id, char player) {
 if ( id < 0 || id > 8 )
  return -1;

 if ( (*state)[id] == _FREE ) {
  (*state)[id] = player;
  return 0;
 } else {
  return -1;
 }
};

static void analyze(struct stats * stats, game_state * state) {
 static const int textvec[3][3][2] = {
  {{0, 3}, {1, 3}, {2,  3}},
  {{0, 1}, {3, 1}, {6,  1}},
  {{0, 1}, {4, 0}, {8, -1}}
 };
 struct test_result * res;
 int i, j;
 int test;

 stats->free = 0;

 for (i = 0; i < 9; i++)
  if ( (*state)[i] == _FREE )
   stats->free++;

 for (test = 0; test < 3; test++) {
  for (i = 0; i < 3; i++) {
   res = &(stats->results[i+test*3]);
   res->score_player = 0;
   res->score_computer = 0;
   res->free = -1;
   for (j = 0; j < 3; j++) {
    switch ((*state)[textvec[test][j][0] + textvec[test][j][1]*i]) {
     case _PLAYER: res->score_player++; break;
     case _COMPUTER: res->score_computer++; break;
     case _FREE: res->free = textvec[test][j][0] + textvec[test][j][1]*i; break;
    }
   }
  }
 }
}

static char check_won(game_state * state) {
 struct stats stats;
 int i;

 analyze(&stats, state);

 for (i = 0; i < 9; i++) {
  if ( stats.results[i].score_player == 3 )
   return _PLAYER;
  if ( stats.results[i].score_computer == 3 )
   return _COMPUTER;
 }

 if ( stats.free == 0)
  return _NOBODY;

 return 0;
}

static void auto_move(game_state * state, int player_move) {
 static const int corner[4] = {0, 2, 6, 8};
 static const int side[4]   = {1, 3, 5, 7};
 struct stats stats;
 struct test_result * res;
 int i;

 analyze(&stats, state);

 // step 0: try to win.
 for (i = 0; i < 9; i++) {
  res = &(stats.results[i]);
  if ( res->score_computer == 2 && res->free != -1 ) {
   if ( try_put(state, res->free, _COMPUTER) == 0 )
    return;
  }
 }

 // step 1: block winning
 for (i = 0; i < 9; i++) {
  res = &(stats.results[i]);
  if ( res->score_player == 2 && res->free != -1 ) {
   if ( try_put(state, res->free, _COMPUTER) == 0 )
    return;
  }
 }

 // step 2: fork
 // TODO: implement

 // step 3: block forking
 // TODO: implement

 // step 4: play center
 if ( try_put(state, 4, _COMPUTER) == 0 )
  return;

 // step 5: "Opposite corner"
 for (i = 0; i < 4; i++)
  if ( player_move == corner[i] )
   if ( try_put(state, 8 - player_move, _COMPUTER) == 0 )
    return;

 // step 6: Empty corner
 for (i = 0; i < 4; i++)
  if ( try_put(state, corner[i], _COMPUTER) == 0 )
   return;

 // step 7: Empty side
 for (i = 0; i < 4; i++)
  if ( try_put(state, side[i], _COMPUTER) == 0 )
   return;
}

static void draw_help(struct roar_buffer ** obuffer) {
 struct roar_buffer * buf;
 ssize_t len = 0;
 void * data;

 if ( roar_buffer_new_data(&buf, _DISPLAY_BUFFER_LEN, &data) == -1 )
  return;

 snprintf(data, _DISPLAY_BUFFER_LEN, _HELP, MSG__HELP);

 len = roar_mm_strlen(data);

 if ( roar_buffer_set_len(buf, len) == -1 ) {
  roar_buffer_free(buf);
 }

 _LIBROAR_IGNORE_RET(roar_buffer_moveintoqueue(obuffer, &buf));
}

static void draw_game(game_state * state, const char * info, struct roar_buffer ** obuffer, struct roar_dl_librarypara * pluginpara) {
 struct roar_buffer * buf;
 ssize_t len = 0;
 void * data;

 if ( roar_buffer_new_data(&buf, _DISPLAY_BUFFER_LEN, &data) == -1 )
  return;

 snprintf(data, _DISPLAY_BUFFER_LEN, _DISPLAY,
          pluginpara == NULL || pluginpara->appname    == NULL ? "<unknown>" : pluginpara->appname,
          pluginpara == NULL || pluginpara->abiversion == NULL ? "<unknown>" : pluginpara->abiversion,
          (*state)[0], (*state)[1], (*state)[2],
          (*state)[3], (*state)[4], (*state)[5],
          (*state)[6], (*state)[7], (*state)[8],
          info);

 len = roar_mm_strlen(data);

 if ( roar_buffer_set_len(buf, len) == -1 ) {
  roar_buffer_free(buf);
 }

 _LIBROAR_IGNORE_RET(roar_buffer_moveintoqueue(obuffer, &buf));
}

static int _set_proto(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
 game_state * state;

 (void)client, (void)vio, (void)obuffer, (void)protopara, (void)protoparalen;

 ROAR_DBG("_set_proto(*) = ?");

 state = roar_mm_malloc(sizeof(game_state));
 if ( state == NULL )
  return -1;

 new_game(state);

 *userdata = state;

 draw_game(state, MSG__YOUR_TURN, obuffer, pluginpara);

 ROAR_DBG("_set_proto(client=%i, vio=%p, obuffer=%p, userdata=%p{%p}, protopara=%p, protoparalen=?, pluginpara=%p) = 0", client, vio, obuffer, userdata, *userdata, protopara, pluginpara);
 return 0;
}

static char map_input(char c) {
 switch (c) {
  case ',':
  case '.':
  case 'g':
    return 'q';
   break;
  case 'a': return '0'; break;
  case 's': return '1'; break;
  case 'd': return '2'; break;
  case 'f': return '3'; break;
  case ' ': return '4'; break;
  case 'h': return '5'; break;
  case 'j': return '6'; break;
  case 'k': return '7'; break;
  case 'l': return '8'; break;
  case '+': return 'n'; break;
  case '0': return 'H'; break;
 }

 if ( c > '9' )
  return c;
 if ( c >= '7' )
  return c - 7;
 if ( c >= '4' )
  return c - 1;
 if ( c >= '1' )
  return c + 5;
 return c;
}

static int _handle(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
 char buf[32];
 ssize_t len;
 ssize_t i;
 const char * msg = MSG__YOUR_TURN;
 char won;
 int player_move = -1;
 char c;

 (void)client, (void)obuffer, (void)protopara, (void)protoparalen;

 ROAR_DBG("_handle(client=%i, vio=%p, obuffer=%p, userdata=%p{%p}, protopara=%p, protoparalen=?, pluginpara=%p) = ?", client, vio, obuffer, userdata, *userdata, protopara, pluginpara);

 len = roar_vio_read(vio, buf, sizeof(buf));

 if ( len < 1 ) {
  return -1;
 }

 if ( check_won(*userdata) ) {
  for (i = 0; i < len; i++) {
   if ( map_input(buf[i]) == 'q' ) {
    return -1;
   } else {
    new_game(*userdata);
    draw_game(*userdata, MSG__YOUR_TURN, obuffer, pluginpara);
    return 0;
   }
  }
 }

 for (i = 0; i < len; i++) {
  c = map_input(buf[i]);
  switch (c) {
   case 'q':
     return -1;
    break;
   case 'n':
     new_game(*userdata);
     draw_game(*userdata, MSG__YOUR_TURN, obuffer, pluginpara);
     return 0;
    break;
   case 'H':
     draw_help(obuffer);
     return 0;
    break;
   case 'e':
     draw_game(*userdata, MSG__YOUR_TURN, obuffer, pluginpara);
     return 0;
    break;
   case '0': case '1': case '2': case '3': case '4':
   case '5': case '6': case '7': case '8':
     if ( try_put(*userdata, c - '0', _PLAYER) == -1 ) {
      msg = MSG__CANNOT_PUT_HERE;
      draw_game(*userdata, msg, obuffer, pluginpara);
      return 0;
     } else {
      msg = MSG__YOUR_TURN;
      i = len; // end the loop.
      player_move = c - '0';
     }
    break;
  }
 }

 won = check_won(*userdata);

 if ( !won && player_move != -1 ) {
  auto_move(*userdata, player_move);
  won = check_won(*userdata);
 }

 switch (won) {
  case _PLAYER: msg = MSG__WON_PLAYER; break;
  case _COMPUTER: msg = MSG__WON_COMPUTER; break;
  case _NOBODY: msg = MSG__WON_NOBODY; break;
 }

 draw_game(*userdata, msg, obuffer, pluginpara);

 ROAR_DBG("check_client(client=%i, vio=%p) = 0", client, vio);
 return 0;
}


static const struct roar_dl_proto proto = {
 .proto = ROAR_PROTO_GAME,
 .description = "Simple Tic-Tac-Toe game",
 .flags = ROAR_DL_PROTO_FLAGS_NONE,
 .set_proto = _set_proto,
 .unset_proto = NULL,
 .handle = _handle,
 .flush = NULL,
 .flushed = NULL,
 .status = NULL
};

static int __reg_proto(struct roar_dl_librarypara * para, struct roar_dl_libraryinst * lib) {
 (void)para, (void)lib;
 ROAR_DL_PLUGIN_REG_FN(ROAR_DL_PROTO_SUBTYPE, proto, ROAR_DL_PROTO_VERSION);
 return 0;
}

ROAR_DL_PLUGIN_START(protocol_tic_tac_toe) {
 ROAR_DL_PLUGIN_META_PRODUCT_NIV("protocol-tic-tac-toe", ROAR_VID_ROARAUDIO, ROAR_VNAME_ROARAUDIO);
 ROAR_DL_PLUGIN_META_VERSION(ROAR_VERSION_STRING);
 ROAR_DL_PLUGIN_META_LICENSE_TAG(GPLv3_0);
 ROAR_DL_PLUGIN_META_CONTACT_FLNE("Philipp", "Schafft", "ph3-der-loewe", "lion@lion.leolix.org");
 ROAR_DL_PLUGIN_META_DESC("Implementation of a the well known game Tic-Tac-Toe");

 ROAR_DL_PLUGIN_REG(ROAR_DL_FN_PROTO, __reg_proto);
} ROAR_DL_PLUGIN_END

//ll