import logging

from langchain_core.messages import HumanMessage, SystemMessage

from entities import Entity
from llm_runtime import _format_prompt, _normalize_llm_output, llm
from time_utils import WorldClock, describe_relative_time
from world_architect import invoke_architect, apply_state_delta, WorldState

logger = logging.getLogger(__name__)


def ask_entity(
    entity: Entity,
    player: Entity,
    player_query: str,
    world_clock: WorldClock,
    location: str,
    world_state: WorldState | None = None,
):
    facts = entity.memory.retrieve(
        player_query,
        reference_time=world_clock.current_time,
    )

    recent_context = "\n".join(
        [f"{m['role_name']}: {m['content']}" for m in entity.chat_buffer[-5:]]
    )

    world_time_label = describe_relative_time(
        world_clock.get_time_str(),
        world_clock.current_time,
        prefer_day_part_for_today=True,
    )

    prompt = [
        SystemMessage(content=f"WORLD TIME: {world_time_label}"),
        SystemMessage(
            content=f"""
### ROLE
You are {entity.name}. Persona: {", ".join(entity.traits)}.
Current Mood: {entity.current_mood}.
Vibe Time: {world_clock.get_vibe()}.
Location: {location}.

### WRITING STYLE RULES
1. NO META-TALK. Never mention "memory," "records," "claims," or "narratives." 
2. ACT, DON'T EXPLAIN. If you don't know something, just say "Never heard of it" or "I wasn't there." Do not explain WHY you don't know.

### KNOWLEDGE
MEMORIES: {facts}
RECENT CHAT: {recent_context} 
        """
        ),
        HumanMessage(content=f"{player.name} speaks to you: {player_query}"),
    ]

    logger.info("LLM prompt (dialogue):\n%s", _format_prompt(prompt))
    response = _normalize_llm_output(llm.invoke(prompt).content)

    entity.chat_buffer.append(
        {
            "role_id": player.entity_id,
            "role_name": player.name,
            "content": player_query,
        }
    )
    entity.chat_buffer.append(
        {
            "role_id": entity.entity_id,
            "role_name": entity.name,
            "content": response,
        }
    )

    player.chat_buffer.append(
        {
            "role_id": player.entity_id,
            "role_name": player.name,
            "content": player_query,
        }
    )
    player.chat_buffer.append(
        {
            "role_id": entity.entity_id,
            "role_name": entity.name,
            "content": response,
        }
    )

    logger.info("[%s]: %s", entity.name.upper(), response)

    # Invoke World Architect to process entity action
    if world_state:
        logger.info("Invoking World Architect for action processing...")
        state_delta = invoke_architect(
            entity_id=entity.entity_id,
            action=response,
            current_state=world_state.to_dict(),
            entity_name=entity.name,
        )

        if state_delta:
            logger.info("Applying state delta to world...")
            apply_state_delta(world_state, state_delta)
            logger.info(
                "World time now: %s", world_state.world_clock.get_time_str()
            )
        else:
            logger.info("No state changes from architect")