import json
import os
import logging
from pathlib import Path

logger = logging.getLogger("friday.protocols")

PROTOCOLS_PATH = Path(__file__).parent.parent.parent / "data" / "learned_protocols.json"

def get_protocols() -> dict:
    """Load learned protocols from disk."""
    if not PROTOCOLS_PATH.exists():
        # Initialize with defaults
        os.makedirs(PROTOCOLS_PATH.parent, exist_ok=True)
        default = {
            "user_preferences": [],
            "interaction_style": "default",
            "learned_facts": {},
            "capabilities": {},
            "forbidden_topics": ["personal credentials", "passwords"]
        }
        save_protocols(default)
        return default
    
    try:
        with open(PROTOCOLS_PATH, "r") as f:
            return json.load(f)
    except Exception as e:
        logger.error(f"Failed to load protocols: {e}")
        return {}

def save_protocols(protocols: dict):
    """Save learned protocols to disk."""
    try:
        os.makedirs(PROTOCOLS_PATH.parent, exist_ok=True)
        with open(PROTOCOLS_PATH, "w") as f:
            json.dump(protocols, f, indent=4)
        logger.info("Learned protocols updated.")
    except Exception as e:
        logger.error(f"Failed to save protocols: {e}")

def update_protocol(category: str, detail: str):
    """Update a specific protocol category."""
    protocols = get_protocols()
    if category not in protocols:
        protocols[category] = []
    
    if isinstance(protocols[category], list):
        if detail not in protocols[category]:
            protocols[category].append(detail)
    elif isinstance(protocols[category], dict):
        # detail would be expected as "key:value"
        if ":" in detail:
            k, v = detail.split(":", 1)
            protocols[category][k.strip()] = v.strip()
    
    save_protocols(protocols)

def get_protocol_prompt() -> str:
    """Generate a prompt snippet from learned protocols."""
    protocols = get_protocols()
    if not protocols:
        return ""
    
    prompt = "\n[LEARNED ADAPTIVE PROTOCOLS]\n"
    if protocols.get("user_preferences"):
        prompt += "- Preferences: " + ", ".join(protocols["user_preferences"]) + "\n"
    if protocols.get("interaction_style") != "default":
        prompt += f"- Style Override: {protocols['interaction_style']}\n"
    for k, v in protocols.get("learned_facts", {}).items():
        prompt += f"- Known Fact: {k} is {v}\n"
    
    return prompt + "\n"