matteoferla/DnD-battler

from .ability_die import AbilityDie
from typing import *
from ..log import log

class SkillRoll:

    log = log

    def __init__(self, ability_die: AbilityDie, modifier: int = 0, success_on_crit=True):
        """
        Meant for an ability check or a saving throw.
        This allows to add an extra modifier to the ability_die (in addition to ability bonus and proficiency)
        RAW in 5e: a crit on an ability roll is not a given success.

        :param ability_die:
        :param modifier:
        :param success_on_crit:
        """
        self.ability_die = ability_die
        self.success_on_crit = success_on_crit
        self.modifier = modifier

    def base_roll(self, advantage:Optional[int]=None) -> int:
        return self.ability_die.base_roll(advantage=advantage, success_on_crit=self.success_on_crit)

    def roll(self, advantage:Optional[int]=None) -> int:
        return self.base_roll(advantage=advantage) + self.bonuses

    @property
    def bonuses(self):
        return self.ability_die.bonus + self.ability_die.proficiency.bonus + self.modifier

    def __str__(self):
        return f'{self.ability_die}+{self.ability_die.proficiency.bonus}+{self.modifier}'