from server.Files import save_file
from datetime import datetime
import os

class GameStorage:
  save_folder_dict = {
    "standart": "01_Standart",
    "duel": "02_Duels",
    "constrictor": "04_Constrictor",
  }

  def __init__(self, snake:str, path:str, no_store_turns:int=10):
    self.snake_type = snake
    self.folder = path
    self.winner_snake_names = None
    self.turns = {}
    self.not_store_file_when_you_are_the_winner_and_turns_are = no_store_turns

  def start_new_game(self, game_info:dict, game_board:dict, snake:dict):
    self.init_game_board = game_board
    self.game_info = game_info
    
    self.my_snake_id = snake['id']
    self.moves = []
    self.now_date = datetime.now()
    self.game_url = self._get_game_url(True if game_info["ruleset"]["version"] == "cli" else False)

  def add_moves(self, turn:int, game_board:dict, my_move:str):
    self.turns[turn] = {
      "move": my_move,
      "game_board": game_board
    }

  def add_end_state(self, game_board:dict, snake_history_state:list[dict], final_turns:int):
    self.final_turns = final_turns
    self.snake_history = snake_history_state
    self._set_winner_snake_name(game_board['snakes'])

  def _get_game_url(self, local_game:bool):
    if local_game:
      return None
    return f"https://play.battlesnake.com/game/{self.game_info['id']}"

  def _set_winner_snake_name(self, snakes:list[dict]):
    if self.my_snake_id in [ x["id"] for x in snakes]:
      self.winner_snake_names = "me"
    else:
      self.winner_snake_names = [ x["name"] for x in snakes]
      if len(self.winner_snake_names) == 0:
        self.winner_snake_names = None

  def _get_type_of_gameboard(self):
    if len(self.init_game_board["snakes"]) == 2:
      return {"name": "duel", "is_ladder": True if self.game_info["source"] == "ladder" else False}
    elif self.game_info["ruleset"]["name"] == "constrictor":
      return {"name": "constrictor", "is_ladder": True if self.game_info["source"] == "ladder" else False}

    return {"name": "standart", "is_ladder": True if self.game_info["source"] == "ladder" else False}

  def _get_correct_folder_for_save_file(self, file_name:str, game_type:str, leader_board:bool, winner:bool):
    storage_folder = self.folder
    if leader_board:
      storage_folder = os.path.join(storage_folder, "00_Leaderboards")

    storage_folder = os.path.join(storage_folder, self.save_folder_dict[game_type])
    storage_folder = os.path.join(storage_folder, self.now_date.strftime('%Y'), self.now_date.strftime('%m_%B'), self.now_date.strftime('%d'))

    if winner:
      storage_folder = os.path.join(storage_folder, "Winner")
    else:
      storage_folder = os.path.join(storage_folder, "Lost")

    return os.path.join(storage_folder, file_name)

  def save(self, file_name:str, callback=None, **kwargs):
    if self.winner_snake_names == "me" and self.final_turns <= self.not_store_file_when_you_are_the_winner_and_turns_are:
      return None

    game_type = self._get_type_of_gameboard()
    save_file_path = self._get_correct_folder_for_save_file(file_name, game_type["name"], game_type["is_ladder"], True if self.winner_snake_names == "me" else False)

    save_file(save_file_path, {
      "winner": self.winner_snake_names,
      "game": {
        "url": self.game_url,
        "id": self.game_info["id"],
        "final_turns": self.final_turns,
        "map": self.game_info["map"],
        "type": game_type,
        "ruleset": self.game_info["ruleset"],
      },
      "moves": self.turns,
      "snake": {
        "type": self.snake_type,
        "calculations": self.snake_history,
      },
    }, callback=callback, **kwargs)

  def __str__(self):
    return f"<{self.__class__.__name__}> Snake: {self.snake_type}, Folder: {self.folder}, Winner: {self.winner_snake_names}, Old Moves: {self.moves}"