tsmp-api/game.py

# Copyright (C) 2023  Evgenij Titarenko
#
#     This program is free software: you can redistribute it and/or modify
#     it under the terms of the GNU Affero General Public License as published
#     by the Free Software Foundation, either version 3 of the License, or
#     (at your option) any later version.
#
#     This program 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 Affero General Public License for more details.
#
#     You should have received a copy of the GNU Affero General Public License
#     along with this program.  If not, see <https://www.gnu.org/licenses/>.

from uuid import uuid4
from enum import Enum
from random import choice, randint

MAX_PLAYERS = 8
MIN_PLAYERS = 2
STREETS = []


class GameState(Enum):
    WAITING = 0
    IN_GAME = 1
    FINISHED = 2


def int_to_hex_str(x: int):
    hex_x = format(x, "x")
    return hex_x if len(hex_x) > 1 else f"0{hex_x}"


class Color:
    def __init__(self, red: int, green: int, blue: int):
        self.red = red
        self.green = green
        self.blue = blue
        hex_red = int_to_hex_str(red)
        hex_green = int_to_hex_str(green)
        hex_blue = int_to_hex_str(blue)
        self.hex = f'#{hex_red}{hex_green}{hex_blue}'

    def get_dict(self):
        return {
            "color": self.hex
        }


class DefaultColors(Enum):
    RED = Color(255, 0, 0)
    GREEN = Color(0, 255, 0)
    BLUE = Color(0, 0, 255)
    YELLOW = Color(255, 255, 0)
    CYAN = Color(0, 255, 255)
    MAGENTA = Color(255, 0, 255)


class User:
    def __init__(self, username: str, password: str):
        self.username = username
        self.password = password
        self.color = None
        self.initiative = None
        self.money = None
        self.token = str(uuid4())
        self.in_room = ""
        self.room_password = ""

    def get_dict(self):
        return {
            "username": self.username,
            "color": get_color(self.color).get_dict(),
            "initiative": self.initiative,
            "money": self.money,
        }


class StreetType(Enum):
    NORMAL = 0
    START = 1
    CHEST = 2
    TAX = 3
    TRAIN = 4
    CHANCE = 5
    PRISON = 6
    SERVICE = 7
    STOP = 8
    TO_PRISON = 9


def get_color(color: Color | DefaultColors) -> Color:
    if color is None:
        return Color(0, 0, 0)
    if color in list(DefaultColors):
        return color.value
    return color


class Street:
    def __init__(self, street_type: StreetType, name: str, color: Color, price: int, home_price: int,
                 hotel_price: int, mortgage: int, rent: tuple):
        self.street_type = street_type
        self.name = name
        self.color = color
        self.price = price
        self.home_price = home_price
        self.hotel_price = hotel_price
        self.mortgage = mortgage
        self.rent = rent
        self.owner = None

    def get_dict(self):
        return {
            "street_type": self.street_type.name,
            "name": self.name,
            "color": get_color(self.color).get_dict(),
            "price": self.price,
            "home_price": self.home_price,
            "hotel_price": self.hotel_price,
            "mortgage": self.mortgage,
            "rent": self.rent,
            "owner": self.owner
        }


def str_to_street_type(s: str):
    return StreetType[s.upper()]


def load_streets():
    with open("streets.txt") as f:
        for line in f:
            street_info = line.split(":")
            street_type = str_to_street_type(street_info[0])
            name = street_info[1]
            color = Color(*map(int, street_info[2].split(",")[:3]))
            price = int(street_info[3])
            home_price = int(street_info[4])
            hotel_price = int(street_info[5])
            mortgage = int(street_info[6])
            rent = tuple(map(int, street_info[7].split(",")))
            street = Street(street_type, name, color, price, home_price, hotel_price, mortgage, rent)
            STREETS.append(street)


def copy_streets():
    new_list = []
    for street in STREETS:
        new_list.append(Street(street.street_type, street.name, street.color, street.price, street.home_price,
                               street.hotel_price, street.mortgage, street.rent))
    return new_list


def choice_color_from_list(colors: list):
    return choice(colors)


def roll_dices(dice: int, n: int):
    def roll():
        return sum(randint(1, dice) for _ in range(n))
    return roll


roll_2d6 = roll_dices(6, 2)


def get_dict_user_or_none(user: User | None):
    if user:
        return user.get_dict()
    return user


class Room:
    def __init__(self, password: str = ""):
        self.room_id = str(uuid4())
        self.users = []
        self.state = GameState.WAITING
        self.password = password
        self.current_player = None
        self.streets = copy_streets()

    def get_dict(self):
        return {
            "room_id": self.room_id,
            "users": [user.get_dict()for user in self.users],
            "state": self.state.name,
            "current_player": self.current_player.get_dict() if self.current_player else self.current_player,
            "streets": [street.get_dict() for street in self.streets]
        }

    def check_color(self, color: Color):
        return color not in [user.color for user in self.users]

    def add_user(self, user: User):
        if len(self.users) == MAX_PLAYERS:
            return  # TODO
        available_colors = list(filter(self.check_color, list(DefaultColors)))
        user.color = choice_color_from_list(available_colors)
        self.users.append(user)

    def sort_players_by_initiative(self):
        self.users.sort(key=lambda user: user.initiative, reverse=True)

    def roll_initiative(self):
        places = list(range(MAX_PLAYERS))
        for user in self.users:
            place = choice(places)
            user.initiative = place
            places.remove(place)

    def start_game(self):
        if self.state == GameState.IN_GAME or len(self.users) < MIN_PLAYERS:
            return  # TODO
        self.roll_initiative()
        self.sort_players_by_initiative()
        self.current_player = self.users[0]
        self.state = GameState.IN_GAME

    def remove_user(self, user: User):
        if user in self.users:
            self.users.remove(user)