Source code for VegansDeluxe.rebuild.States.DamageThreshold

from VegansDeluxe.core import RegisterState, RegisterEvent
from VegansDeluxe.core import StateContext, EventContext
from VegansDeluxe.core import HPLossGameEvent
from VegansDeluxe.core import Session
from VegansDeluxe.core import State


class DamageThreshold(State):
    id = 'damage-threshold'

    def __init__(self):
        super().__init__()
        self.threshold = 6


@RegisterState(DamageThreshold)
async def register(root_context: StateContext[DamageThreshold]):
    session: Session = root_context.session
    state = root_context.state

    @RegisterEvent(session.id, event=HPLossGameEvent)
    async def func(context: EventContext[HPLossGameEvent]):
        if not context.event.damage:
            return
        context.event.hp_loss = (context.event.damage // max(state.threshold, 1)) + 1