Source code for VegansDeluxe.rebuild.Skills.ShieldGen

from VegansDeluxe.core import AttachedAction, StateContext, RegisterState
from VegansDeluxe.core.Actions.StateAction import DecisiveStateAction
from VegansDeluxe.core import EventContext
from VegansDeluxe.core import At
from VegansDeluxe.core import Entity
from VegansDeluxe.core import PostDamageGameEvent

from VegansDeluxe.core import Session
from VegansDeluxe.core.Skills.Skill import Skill
from VegansDeluxe.core import Allies
from VegansDeluxe.core.Translator.LocalizedString import ls


class ShieldGen(Skill):
    id = 'shield-gen'
    name = ls("skill_shield_gen_name")
    description = ls("skill_shield_gen_description")

    def __init__(self):
        super().__init__()
        self.cooldown_turn = 0


@RegisterState(ShieldGen)
def register(root_context: StateContext[ShieldGen]):
    session: Session = root_context.session
    source = root_context.entity


@AttachedAction(ShieldGen)
class ShieldGenAction(DecisiveStateAction):
    id = 'shield-gen'
    name = ls("skill_shield_gen_action_name")
    target_type = Allies()
    priority = -2

    def __init__(self, session: Session, source: Entity, skill: ShieldGen):
        super().__init__(session, source, skill)
        self.state = skill

    @property
    def hidden(self) -> bool:
        return self.session.turn < self.state.cooldown_turn

    def func(self, source, target):
        self.state.cooldown_turn = self.session.turn + 5
        if target == source:
            self.session.say(ls("skill_shield_gen_action_text").format(source.name))
        else:
            self.session.say(ls("skill_shield_gen_action_targeted").format(source.name, target.name))

        @At(self.session.id, turn=self.session.turn, event=PostDamageGameEvent)
        def shield_block(context: EventContext[PostDamageGameEvent]):
            if context.event.target != target:
                return
            if not context.event.damage:
                return
            context.event.damage = 0