import os
from pathlib import Path

import yaml
from dotenv import load_dotenv

load_dotenv()

# ─── YAML Config Loader ────────────────────────────────────────────────────────

_CONFIG_PATH = Path(__file__).resolve().parent / "config.yaml"
_yaml: dict = {}
if _CONFIG_PATH.is_file():
    with open(_CONFIG_PATH, "r", encoding="utf-8") as f:
        _yaml = yaml.safe_load(f) or {}

def _yaml_get(*keys, default=None):
    """Navigate nested yaml dict, return default if any key missing."""
    d = _yaml
    for k in keys:
        if isinstance(d, dict) and k in d:
            d = d[k]
        else:
            return default
    return d if d is not None else default


# ─── Credential / Secret (hanya dari .env) ─────────────────────────────────────

llm_baseurl  = os.getenv("LLM_BASE_URL",  default="http://localhost:11434/v1")
llm_model    = os.getenv("LLM_MODEL",     default="granite4.1:8b")
llm_api_key  = os.getenv("LLM_API_KEY",   default="ollama")
llm_timeout  = int(os.getenv("LLM_TIMEOUT", default="600"))

XMPP_USERNAME = os.getenv("XMPP_USERNAME", default="")
XMPP_PASSWORD = os.getenv("XMPP_PASSWORD", default="")


# ─── Agent Config (YAML, bisa di-override dari .env) ────────────────────────────

AGENT_MAX_ITERATIONS  = int(os.getenv("AGENT_MAX_ITERATIONS",  default=_yaml_get("agent", "max_iterations",  default="30")))
AGENT_MAX_TOOL_OUTPUT = int(os.getenv("AGENT_MAX_TOOL_OUTPUT", default=_yaml_get("agent", "max_tool_output", default="40000")))


# ─── Persona / Mode (YAML, bisa di-override dari .env) ──────────────────────────

PERSONA_MODE      = os.getenv("PERSONA_MODE",      default=_yaml_get("persona", "mode",      default="programmer")).strip().lower()
PERSONA_NAME      = os.getenv("PERSONA_NAME",      default=_yaml_get("persona", "name",      default="OWL")).strip() or "OWL"
PERSONA_AGE       = os.getenv("PERSONA_AGE",       default=_yaml_get("persona", "age",       default="")).strip()
PERSONA_GENDER    = os.getenv("PERSONA_GENDER",    default=_yaml_get("persona", "gender",    default="")).strip()
PERSONA_TONE      = os.getenv("PERSONA_TONE",      default=_yaml_get("persona", "tone",      default="casual")).strip().lower() or "casual"
PERSONA_VERBOSITY = os.getenv("PERSONA_VERBOSITY", default=_yaml_get("persona", "verbosity", default="balanced")).strip().lower() or "balanced"
PERSONA_HUMOR     = os.getenv("PERSONA_HUMOR",     default=_yaml_get("persona", "humor",     default="light")).strip().lower() or "light"
PERSONA_LANGUAGE  = os.getenv("PERSONA_LANGUAGE",  default=_yaml_get("persona", "language",  default="id")).strip().lower() or "id"
PERSONA_MOOD      = os.getenv("PERSONA_MOOD",      default=_yaml_get("persona", "mood",      default="cheerful")).strip().lower() or "cheerful"
PERSONA_CATCHPHRASES = os.getenv("PERSONA_CATCHPHRASES", default=_yaml_get("persona", "catchphrases", default="")).strip()


# ─── Character & Skills (YAML, bisa di-override dari .env) ─────────────────────

AGENT_CHARACTER = os.getenv("AGENT_CHARACTER", default=_yaml_get("character", "preset", default="")).strip().lower()
AGENT_SKILLS    = os.getenv("AGENT_SKILLS",    default=_yaml_get("character", "skills", default="")).strip().lower()


# ─── XMPP (non-credential dari YAML, credential dari .env) ─────────────────────

XMPP_ENABLED = os.getenv("XMPP_ENABLED", default=str(_yaml_get("xmpp", "enabled", default="false"))).strip().lower() in ("true", "1", "yes")
XMPP_MUC_ROOMS = os.getenv("XMPP_MUC_ROOMS", default=_yaml_get("xmpp", "muc_rooms", default="")).strip()
XMPP_NICKNAME  = os.getenv("XMPP_NICKNAME",  default=_yaml_get("xmpp", "nickname",  default="")).strip()

XMPP_SELECTIVE_RESPONSE = os.getenv("XMPP_SELECTIVE_RESPONSE", default=str(_yaml_get("xmpp", "selective_response", default="true"))).strip().lower() in ("true", "1", "yes")


# ─── RAG (YAML) ─────────────────────────────────────────────────────────────────

RAG_PERSIST_DIR = os.getenv("RAG_PERSIST_DIR", default=_yaml_get("rag", "persist_dir", default="chroma_db"))


# ─── Humanize Delay (YAML) ─────────────────────────────────────────────────────

READ_DELAY_MIN = float(os.getenv("READ_DELAY_MIN", default=_yaml_get("delay", "read_min",     default="1.0")))
READ_DELAY_MAX = float(os.getenv("READ_DELAY_MAX", default=_yaml_get("delay", "read_max",     default="2.0")))
TYPING_SPEED   = float(os.getenv("TYPING_SPEED",   default=_yaml_get("delay", "typing_speed", default="15.0")))
TYPING_MAX     = float(os.getenv("TYPING_MAX",     default=_yaml_get("delay", "typing_max",   default="10.0")))


# ─── Character Preset Override ──────────────────────────────────────────────────
# Jika AGENT_CHARACTER di-set, baca character.md dari agent/characters/<preset>/
# dan override nilai persona yang relevan.

ENV_CHARACTERS_DIR = Path(__file__).resolve().parent / "agent" / "characters"
ENV_CHARACTER_CONFIG_PATH = ENV_CHARACTERS_DIR / AGENT_CHARACTER / "character.md" if AGENT_CHARACTER else None

if ENV_CHARACTER_CONFIG_PATH and ENV_CHARACTER_CONFIG_PATH.is_file():
    _character_env: dict[str, str] = {}
    for line in ENV_CHARACTER_CONFIG_PATH.read_text(encoding="utf-8").splitlines():
        line = line.strip()
        if not line or line.startswith("#"):
            continue
        if "=" in line:
            key, value = line.split("=", 1)
            _character_env[key.strip()] = value.strip()

    _character_overrides = {
        "PERSONA_MODE": PERSONA_MODE,
        "PERSONA_NAME": PERSONA_NAME,
        "PERSONA_AGE": PERSONA_AGE,
        "PERSONA_GENDER": PERSONA_GENDER,
        "PERSONA_TONE": PERSONA_TONE,
        "PERSONA_VERBOSITY": PERSONA_VERBOSITY,
        "PERSONA_HUMOR": PERSONA_HUMOR,
        "PERSONA_LANGUAGE": PERSONA_LANGUAGE,
        "PERSONA_MOOD": PERSONA_MOOD,
        "PERSONA_CATCHPHRASES": PERSONA_CATCHPHRASES,
    }
    for key, fallback in _character_overrides.items():
        value = _character_env.get(key, fallback).strip()
        if key in {"PERSONA_MODE", "PERSONA_TONE", "PERSONA_VERBOSITY", "PERSONA_HUMOR", "PERSONA_LANGUAGE", "PERSONA_MOOD"}:
            value = value.lower() or fallback
        if key == "PERSONA_NAME" and not value:
            value = fallback
        _character_overrides[key] = value

    for key, value in _character_overrides.items():
        globals()[key] = value
        os.environ[key] = value