from random import randint
from enum import Enum

from graphics.engine import Screen
from graphics.layers import Sprite

class StatsSet:
    #HP, ATK, DEF, INITIATIVE, XPCOEF
    def __init__(self, hp, atkP, defP, initiative, xpcoef):
        self.hp = hp
        self.atkP = atkP
        self.defP = defP
        self.initiative = initiative
        self.xpcoef = xpcoef


class ClassType(Enum):
    GUERRIER = StatsSet(20, 10, 15, 5, 8)
    MAGICIEN = StatsSet(15, 15, 6, 5, 7)
    VOLEUR = StatsSet(17, 22, 9, 7, 9)
    ELFE = StatsSet(12, 30, 1, 5, 10)


class Personnage:
    def __init__(self, nom, class_type, place):
        self.nom = nom
        self.class_name = class_type.name
        self.class_type = class_type.value

        self.__hp = self.class_type.hp
        self.stats = StatsSet(0, 0, 0, 0, 0)
        self.__xp = 1

        Sprite(3, self.class_name, place)
        #self.inventaire =
    
    def jet_attaque(self):
        damage = randint(1, 20)
        self.change_exp(self.class_type.xpcoef * self.__xp)
        return damage + self.class_type.initiative + self.stats.initiative

    def jet_defense(self):
        damage = randint(1, 20)
        self.change_exp(self.class_type.xpcoef * self.__xp)
        return damage + self.class_type.initiative + self.stats.initiative
    
    def get_hp(self):
        return self.__hp
    def change_hp(self, nb_hp):
        if self.__hp - nb_hp < 0:
            self.__hp = 0
        else:
            self.__hp -= nb_hp

    def get_xp(self):
        return self.__xp
    def change_exp(self, nb_exp):
        if nb_exp > 0:
            self.__xp += nb_exp
        else:
            raise ValueError("nb_exp attends un nombre positif")

    def affiche_caracteristiques(self):
        return (f"Nom: {self.nom}", 
                f"Type de classe: {self.class_name}",
                f"Vie: {self.__hp}",
                f"Expérience: {self.__xp}",
                "Stats (classe + inventaire):",
                f"- HP: {self.class_type.hp} + {self.stats.hp}",
                f"- ATK%: {self.class_type.atkP} + {self.stats.atkP}",
                f"- DEF%: {self.class_type.defP} + {self.stats.defP}",
                f"- INITIATIVE: {self.class_type.initiative} + {self.stats.initiative}",
                f"- XPCOEF: {self.class_type.xpcoef} + {self.stats.xpcoef}"
        )

    def reduced_stats(self):
        return(f"Ennemi: {self.nom}", 
            f"Type: {self.class_name}", 
            f"Vie: {self.__hp}",
            )