CharacterService_ren.py

from __future__ import annotations
from typing import Optional

from renpy import store
import renpy.exports as renpy

from game.characters.Relationship_ren import Relationship
from game.characters.Moods_ren import Moods
from game.characters.NonPlayableCharacter_ren import NonPlayableCharacter
from game.characters.PlayableCharacters_ren import PlayableCharacter

"""renpy
init python:
"""


class CharacterService:
    @staticmethod
    def get_relationship(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> Relationship:
        if target is None:
            target = store.mc

        if not hasattr(character, "relationships"):
            character.relationships = {}

        return character.relationships.setdefault(target.name, Relationship.FRIEND)

    @staticmethod
    def has_relationship(
        character: NonPlayableCharacter,
        relationship: Relationship,
        target: Optional[PlayableCharacter] = None,
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.get_relationship(character, target) == relationship

    @staticmethod
    def set_relationship(
        character: NonPlayableCharacter,
        relationship: Relationship,
        target: Optional[PlayableCharacter] = None,
    ) -> None:
        if target is None:
            target = store.mc

        if not hasattr(character, "relationships"):
            character.relationships = {}

        if not hasattr(target, "relationships"):
            target.relationships = {}

        if (
            character.relationships.setdefault(target.name, Relationship.FRIEND)
            == relationship
        ):
            return

        character.relationships[target.name] = relationship
        target.relationships[character.name] = relationship

    @staticmethod
    def get_mood(character: NonPlayableCharacter) -> Moods:
        return character.mood

    @staticmethod
    def has_mood(character: NonPlayableCharacter, mood: Moods) -> bool:
        return mood == character.mood or character.mood & mood == mood

    @staticmethod
    def set_mood(character: NonPlayableCharacter, mood: Moods) -> None:
        if mood == character.mood:
            return

        character.mood = mood

    @staticmethod
    def reset_mood(character: NonPlayableCharacter) -> None:
        character.mood = Moods.NORMAL

    @staticmethod
    def add_mood(character: NonPlayableCharacter, mood: Moods) -> None:
        if mood == character.mood:
            return

        if character.mood == Moods.NORMAL:
            character.mood = mood
            return

        character.mood = character.mood | mood

    @staticmethod
    def remove_mood(character: NonPlayableCharacter, mood: Moods) -> None:
        character.mood = character.mood & ~mood

    @staticmethod
    def get_profile_pictures(character_name: str) -> list[str]:
        directory: str = f"characters/images/{character_name.lower()}"

        try:
            return [file for file in renpy.list_files() if file.startswith(directory)]
        except OSError:
            return [
                file
                for file in renpy.list_files()
                if file.startswith("characters/images/chloe")
            ]

    @staticmethod
    def is_girlfriend(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(
            character, Relationship.GIRLFRIEND, target
        )

    @staticmethod
    def is_fwb(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(character, Relationship.FWB, target)

    @staticmethod
    def is_dating(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(character, Relationship.DATING, target)

    @staticmethod
    def is_kissed(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(character, Relationship.KISSED, target)

    @staticmethod
    def is_friend(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(character, Relationship.FRIEND, target)

    @staticmethod
    def is_ex(
        character: NonPlayableCharacter, target: Optional[PlayableCharacter] = None
    ) -> bool:
        if target is None:
            target = store.mc

        return CharacterService.has_relationship(character, Relationship.EX, target)

    @staticmethod
    def is_mad(character: NonPlayableCharacter) -> bool:
        return CharacterService.has_mood(character, Moods.MAD)

    @staticmethod
    def is_threatened(character: NonPlayableCharacter) -> bool:
        return CharacterService.has_mood(character, Moods.THREATENED)