status_effect.py

from __future__ import annotations
from copy import deepcopy

from typing import TYPE_CHECKING

import color
from components.ai import ConfusedEnemy

if TYPE_CHECKING:
    from entity import Actor


class StatusEffect:
    def __init__(self, name: str, duration: int):
        self.name = name
        self.duration = duration

    def apply(self, entity: Actor, target: Actor):
        raise NotImplementedError()

    def remove(self, entity: Actor):
        raise NotImplementedError()


class Grappled(StatusEffect):
    def __init__(self, duration=1):
        super().__init__("grappled", duration)

    def apply(self, entity: Actor, target: Actor):
        engine = entity.parent.engine
        # check if the entity is the player
        if target is engine.player and not target.status.grappled:
            engine.message_log.add_message("You are being grappled!", color.yellow)
        # Make sure to use deepcopy of the status effect to avoid changing the original
        target.status.set_active_status_effect(deepcopy(self))

    def remove(self, entity: Actor):
        pass


class Confused(StatusEffect):
    def __init__(self, duration=1):
        super().__init__("confused", duration)

    def apply(self, entity: Actor, target: Actor):
        engine = entity.parent.engine

        # Avoid stacking confusion
        if target.status.confused:
            return

        if target is engine.player and not target.status.confused:
            engine.message_log.add_message("You are feeling confused!", color.yellow)
        target.ai = ConfusedEnemy(target)
        target.status.set_active_status_effect(deepcopy(self))

    def remove(self, entity: Actor):
        engine = entity.parent.engine
        if entity == engine.player:
            engine.message_log.add_message(
                "You are regaining your senses!", color.white
            )
        entity.restore_ai()


class BloodDrain(StatusEffect):
    def __init__(self, heal_amount: int, duration=0):
        self.heal_amount = heal_amount
        super().__init__("blood drain", duration)

    def apply(self, entity: Actor, target: Actor):
        engine = entity.parent.engine
        entity.fighter.heal(self.heal_amount)
        if target is engine.player:
            engine.message_log.add_message(
                f"{entity.name} is draining your blood!", color.yellow
            )

    def remove(self, entity: Actor):
        pass