# Generation of data for a campaign of Ticket to Ride Legacy: Legends of the West

# Imports
import os
import random
import jsonpickle
from src.MamaOConnel import MamaOConnel

class Circus:
  # There are:
  # 4 wagons of each colors (until 2 players' stop)
  # 2 wagons of each colors (until 3 players' stop)
  # 2 wagons of each colors (until 4 players' stop)
  # 2 wagons of each colors (until 5 players' stop)
  # NOTE: In the original distribution, the same color never appears twice in a row
  #   We chose to remove this constraint, for the sake of simplicity

  def __init__(self, playersNb):
    wagonsPerColors = 2 * playersNb
    colors = "blue", "green", "black", "red", "yellow"
    remaining = {"blue": wagonsPerColors, "green": wagonsPerColors, "black": wagonsPerColors, "red": wagonsPerColors,\
      "yellow": wagonsPerColors}
    self.wagons = []
    for n in range(0, 5 * wagonsPerColors):
      color = colors[random.randint(0, 4)]
      while remaining[color] == 0:
        color = colors[random.randint(0, 4)]
      self.wagons.append(color)
      remaining[color] = remaining[color] - 1
    self.state = 0 # 0 means Circus is not used in game yet, 1 means next wagon in list is #1, and so on

  def enable(self):
    if self.state > 0:
      print("Circus is already in game.")
      return False
    self.state = 1

  def getNextColor(self):
    return self.wagons[self.state - 1]

  def takeWagon(self):
    if self.state == 0:
      print("Circus is not or no longer in game.")
      return False
    print("Player obtained a " + self.wagons[self.state - 1] + " circus sticker!")
    self.state = self.state + 1
    if self.state > len(self.wagons):
      print("Circus stickers are now depleted.")
      self.state = 0 # effectively disable circus

class TreasureCard:

  def __init__(self):
    self.treasures = [36, 33, 29, 27, 24]
    self.state = -1 # disabled

  def enable(self):
    self.state = 0 # 0 means 0 treasures have been found

  def getFoundNumber(self):
    if self.state == -1:
      print("Treasure hunt has not started yet or is completed.")
      return False
    return self.state

  def takeTreasure(self):
    if self.state == -1:
      print("Treasure hunt has not started yet or is completed.")
      return False
    print("Congrats! You obtained $" + str(self.treasures[self.state]))
    self.state = self.state + 1
    if self.state == 5:
      print("All treasures have been found.")
      self.state = -1 # effectively disable treasure card


class ConcessionCard:
  # Each card contains:
  # 6 cities to connect to the player's concession. There is no need to implement those.
  # 7 gold nuggets, as rewards when the player succefully connects a city to their concession. The player can choose
  # which nugget to earn, and then discovers the associated reward (between 11 and 17 coins).
  # A ConcessionCard is then composed of 7 nuggets, with the rewards randomly distributed.

  def __init__(self):
    availableValues = [11, 12, 13, 14, 15, 16, 17]
    nuggetsNb = len(availableValues) - 1
    self.nuggets = []
    for n in range(0, 7):
      choice = random.randint(0, nuggetsNb - n)
      self.nuggets.append(availableValues[choice])
      availableValues.pop(choice)
    self.taken = set()

  def getRemainingPepites(self):
    return {1, 2, 3, 4, 5, 6, 7} - self.taken

  def takeReward(self, choice):
    if not choice.isdigit():
      print("Please choose a number.")
      return False
    choice = int(choice)
    if choice not in range(1, 8):
      print("This is not a valid number")
      return False
    if choice in self.taken:
      print(str(choice) + " has already been obtained.")
      return False
    self.taken.add(choice)
    print("Congrats! You obtained $" + str(self.nuggets[choice - 1]))

class Player:

  def __init__(self, color):
    self.color = color
    self.concession = ConcessionCard()

class Game:

  # initGame() is used to generate all data for a campaign and initialize status variables
  def initGame(self):
    self.years = 1865, 1868, 1871, 1874, 1877, 1880, 1883, 1886, 1889, 1892, 1895, 1898

    playersNb = input("Enter the number of players (2-5): ")
    while not playersNb.isdigit() or int(playersNb) not in range(2, 6):
      playersNb = input("OUT OF RANGE! Enter the number of players (2-5): ")
    playersNb = int(playersNb)

    # Game status
    self.yearId = 0
    self.concession = False

    # General data to generate
    self.circus = Circus(playersNb)
    self.treasure = TreasureCard()
    self.mamaoc = MamaOConnel()
    while not self.mamaoc.solver():
      self.mamaoc = MamaOConnel()

    # Players
    self.players = []
    for n in range(0, playersNb):
      color = input("Enter player " + str(n + 1) + "'s color: ")
      self.players.append(Player(color))

  def getPlayerByColor(self, color):
    for player in self.players:
      if player.color.lower() == color.lower():
        return player
    print("ERROR: player '" + color + "' is not defined")
    return False

  def enableConcession(self):
    self.concession = True

  def disableConcession(self):
    self.concession = False

  # if number = 0 print Mama O'Connel's location, else print hint #number
  def printMamaHint(self, number):
    if number not in range(0, 4):
      return False
    os.system('clear')
    input("When you are ready, press Enter.")
    print("")
    match number:
      case 0:
        print("Mama O'Connel is at: " + self.mamaoc.location.name)
      case 1:
        print("Mama O'Connel is " + str(self.mamaoc.hint1[1]) + " cities from " + self.mamaoc.hint1[0].name)
      case 2:
        print("Mama O'Connel is " + str(self.mamaoc.hint2[1]) + " cities from " + self.mamaoc.hint2[0].name)
      case 3:
        print("Mama O'Connel is " + str(self.mamaoc.hint3[1]) + " cities from " + self.mamaoc.hint3[0].name)
    print("")
    input("When you are done, press Enter.")
    os.system('clear')

  def printStatus(self):
    print("")
    print("---------------------------------------------------------------------")
    print("")
    print("Ticket to Ride Legacy: Legends of the West")
    print("")
    print("Campaign - Year: " + str(self.years[self.yearId]))
    print("")
    print("Players:")
    for player in self.players:
      print("  " + player.color)
      if self.concession:
        print("     Concession card remaining nuggets: " + str(player.concession.getRemainingPepites()))
    print("")
    if self.circus.state > 0:
      print("Circus next sticker color: " + self.circus.getNextColor())
    if self.treasure.state > -1:
      print(f"Treasures found: {self.treasure.getFoundNumber()}")

  def saveData(self, path):
    savefile = open('./' + path, 'w+')
    savefile.write(jsonpickle.encode(self))
    savefile.close()

def loadData(path):
  with open('./' + path, 'r') as savefile:
    content = savefile.read()
  data = jsonpickle.decode(content)
  return data


print("")
print("Ticket to Ride Legacy: Legends of the West")
print("")
choice = input("What do you want to do (new/load)? ")
while choice not in ("new", "load"):
  choice = input("UNDEFINED! What do you want to do (new/load)? ")
if choice == "new":
  myGame = Game()
  myGame.initGame()
elif choice == "load":
  path = input("Path to save file (defaults to savefile.json)? ")
  if path == "":
    path = "savefile.json"
  myGame = loadData(path)

while True:
  myGame.printStatus()
  command = input("==> ")
  while command not in ('load', 'save', 'next year', 'exit', 'enable circus', 'take circus', 'enable concession',\
    'take concession', 'disable concession', 'enable treasure', 'take treasure', 'mama hint 1', 'mama hint 2',\
    'mama hint 3', 'mama location'):
    print("")
    print("Available commands:")
    print("  Main: 'load', 'save', 'next year', 'exit'")
    print("  Circus: 'enable circus', 'take circus'")
    print("  Treasure: 'enable treasure', 'take treasure'")
    print("  Concessions: 'enable concession', 'take concession', 'disable concession'")
    print("  Mama O'Connel tracking: 'mama hint 1', 'mama hint 2', 'mama hint 3', 'mama location'")
    command = input("==> ")
  print("")
  match command:
    case 'load':
      choice = input("WARNING! If you load, you will loose all unsaved changes. Are you sure? (type YES if you are) ")
      if choice == "YES":
        path = input("Path to save file (defaults to savefile.json)? ")
        if path == "":
          path = "savefile.json"
        myGame = loadData(path)
    case 'save':
      path = input("Path to save file (defaults to savefile.json)? ")
      if path == "":
        path = "savefile.json"
      myGame.saveData(path)
      print("Game saved!")
    case 'next year':
      myGame.yearId = myGame.yearId + 1
    case 'exit':
      exit()
    case 'enable circus':
      myGame.circus.enable()
    case 'take circus':
      myGame.circus.takeWagon()
    case 'enable treasure':
      myGame.treasure.enable()
    case 'take treasure':
      myGame.treasure.takeTreasure()
    case 'enable concession':
      myGame.enableConcession()
    case 'take concession':
      choice = input("Player? ==> ")
      player = myGame.getPlayerByColor(choice)
      if player == False:
        print("Not a valid player.")
      else:
        choice = input("Which nugget? ==> ")
        print("")
        player.concession.takeReward(choice)
    case 'disable concession':
      myGame.disableConcession()
    case 'mama hint 1':
      myGame.printMamaHint(1)
    case 'mama hint 2':
      myGame.printMamaHint(2)
    case 'mama hint 3':
      myGame.printMamaHint(3)
    case 'mama location':
      myGame.printMamaHint(0)