hazelmud/commands/encounter_cmdset.py

from commands.command import Command
from evennia import create_script
from evennia import CmdSet

class EncounterCommand(Command):
    energy_cost = 1
    allow_self_target = False

    def handler(self):
        return self.caller.ndb.encounter_handler

    def can_perform(self):
        # Returns whether the caller is having their turn and if they have enough energy
        # to make this action
        return self.handler().can_act(self)

    def has_target(self):
        # Targeting code:
        # Looks for the suggested target
        # If none is found but there is only one other character in the encounter, 
        # default to that other character
        # Else ask player for clarification
        if self.target:
            self.target = self.caller.search(self.target)
            if not self.target:
                default_target = self.handler().has_default_target(self.caller)
                if default_target:
                    self.caller.msg("Defaulting to other character in encounter")
                    self.target = default_target
                else:
                    self.caller.msg(f"Who are you trying to {self.key}?")
                    self.target = None
        else:
            default_target = self.handler().has_default_target(self.caller)
            if default_target:
                self.target = default_target
            else:
                self.caller.msg(f"Who are you trying to {self.key}?")
                self.target = None
        if self.caller == self.target and not self.allow_self_target:
            self.msg(f"You aren't supposed to {self.key} yourself!")
        return self.target
            

# Bite (Regular)
class CmdBite(EncounterCommand):
    "Bite down on your target with your teeth"
    key = "bite"
    help_category = "Violence"

    def parse(self):
        self.args = self.args.strip()
        target, *adverb = self.args.split(":")
        self.target = target.strip()
        self.adverb = ' ' + adverb[0].strip() if adverb else ''

    def func(self):
        if super().has_target() and super().can_perform():
            self.caller.msg(f"You bite {self.target.key}{self.adverb}")
            self.target.msg(f"{self.caller.key} bites you{self.adverb}")
            # damage calculations, chance to miss, etc

# Kiss (Regular)
class CmdKiss(EncounterCommand):
    "Kiss the target with your lips"
    key = "kiss"
    help_category = "Sex"

    def parse(self):
        self.args = self.args.strip()
        target, *adverb = self.args.split(':')
        self.target = target.strip()
        self.adverb = ' ' + adverb[0].strip() if adverb else ''

    def func(self):
        if super().has_target() and super().can_perform():
            self.caller.msg(f"You kiss {self.target}{self.adverb}")
            self.target.msg(f"{self.caller} kisses you{self.adverb}")

class SetSpeciesHuman(CmdSet):
    def at_cmdset_creation(self):
        self.add(CmdBite)
        self.add(CmdKiss)

# Special encounter commands

class CmdPass(EncounterCommand):
    "Pass the turn over to the next character"
    energy_cost = 0
    key = "pass"
    aliases = ["next", "finish", "done"]
    help_category = "Encounter"

    def func(self):
        if super().can_perform():
            self.caller.msg("You end your turn")
            super().handler().next_turn()

class CmdRP(EncounterCommand):
    "Free-form text input (for describing your actions in RP)"
    energy_cost = 0
    key = "rp"
    aliases = [">"]
    help_category = "Encounter"

    def func(self):
        if super().can_perform():
            super().handler().msg_all(self.args.strip())

class CmdFlee(EncounterCommand):
    "Exits the current encounter"
    energy_cost = 0
    key = "flee"
    aliases = ["escape", "run", "run away", "leave"]
    help_category = "Encounter"

    def func(self):
        super().handler().msg_all(f"{self.caller.key} left the encounter")
        super().handler().remove_character(self.caller)

class SetEncounterSpecial(CmdSet):
    def at_cmdset_creation(self):
        self.add(CmdPass)
        self.add(CmdRP)
        self.add(CmdFlee)    

# Encounter-related character commands

class CmdEngage(Command):
    """
    Initiates an encounter with the selected target
    """
    key = "engage"
    aliases = ["encounter"]
    def func(self):
        if not self.args:
            self.caller.msg("Usage: engage <target>")
            return
        target = self.caller.search(self.args)
        if not target:
            return
        if target.ndb.encounter_handler:
            target.ndb.encounter_handler.add_character(self.caller)
            target.ndb.encounter_handler.msg_all(f"{self.caller} joins the encounter")
        else:
            handler = create_script("encounter_handler.EncounterHandler")
            handler.add_character(self.caller)
            handler.add_character(target)
            target.msg(f"{self.caller} has engaged you in an encounter!")
            target.msg("You may leave using the flee command (or aliases)")
            self.caller.msg(f"You have entered an encounter with {target}. Be polite!")
            self.caller.msg("It is now your turn:")