from dataclasses import dataclass, field

from components import (
    Health,
    Satiety,
    Location,
    LocationType,
    Name,
    Hydration,
    Hygiene,
    Tags,
)
from ecs import World


def create_pet(world: World, name: str):
    return world.spawn(
        Name(name),
        Health(100),
        Satiety(80),
        Hydration(80),
        Hygiene(80),
        Tags(),
    )


def move_pet_to(world: World, pet_id, container_entity=None) -> bool:
    component_map = world.get_components(pet_id, pop=True)
    if component_map is None:
        return False
    component_map[Location].id = container_entity
    component_map[Location].type = LocationType.PADDOCK
    return True