from typing import Dict, Any
from datetime import datetime
import uuid


async def harnessed_get_agent_config(context: Dict[str, Any]) -> Dict[str, Any]:
    """
    Get agent configuration for current user

    Args:
        context: Request context containing user information

    Returns:
        Configuration data for the current user
    """
    from .core import HermesAgent
    agent = HermesAgent()
    user_id = agent._get_current_user_id(context)

    # Query database for user configuration
    sql = """
        SELECT * FROM harnessed_agent_config
        WHERE user_id = %(user_id)s
        ORDER BY created_at DESC
        LIMIT 1
    """
    try:
        rows = await agent.db.sql(sql, {"user_id": user_id})
        if rows:
            config = rows[0]
            # Convert string booleans to actual booleans for UI
            config["auto_cleanup_enabled"] = config["auto_cleanup_enabled"] == "1"
            return {"success": True, "config": config}
        else:
            # Return default configuration
            default_config = {
                "id": f"default_{user_id}",
                "user_id": user_id,
                "work_dir": "./hermes_work",
                "skills_path": "~/.hermes/skills",
                "max_memory_tokens": 2000,
                "default_priority": 50,
                "high_priority_threshold": 70,
                "low_priority_threshold": 30,
                "auto_cleanup_enabled": True,
                "min_retention_days": 30,
                "created_at": datetime.now().isoformat(),
                "updated_at": datetime.now().isoformat()
            }
            return {"success": True, "config": default_config}
    except Exception as e:
        return {"success": False, "error": str(e)}


async def harnessed_save_agent_config(context: Dict[str, Any], config_data: Dict[str, Any]) -> Dict[str, Any]:
    """
    Save agent configuration for current user

    Args:
        context: Request context containing user information
        config_data: Configuration data to save

    Returns:
        Success status
    """
    from .core import HermesAgent
    agent = HermesAgent()
    user_id = agent._get_current_user_id(context)

    # Convert boolean to string for database storage
    config_data["auto_cleanup_enabled"] = "1" if config_data.get("auto_cleanup_enabled") else "0"
    config_data["user_id"] = user_id
    config_data["updated_at"] = datetime.now().isoformat()

    # Check if config already exists
    sql_check = "SELECT id FROM harnessed_agent_config WHERE user_id = %(user_id)s"
    try:
        rows = await agent.db.sql(sql_check, {"user_id": user_id})
        if rows:
            # Update existing config
            config_data["id"] = rows[0]["id"]
            config_data["created_at"] = rows[0]["created_at"]  # Keep original created_at
            sql_update = """
                UPDATE harnessed_agent_config SET
                    work_dir = %(work_dir)s,
                    skills_path = %(skills_path)s,
                    max_memory_tokens = %(max_memory_tokens)s,
                    default_priority = %(default_priority)s,
                    high_priority_threshold = %(high_priority_threshold)s,
                    low_priority_threshold = %(low_priority_threshold)s,
                    auto_cleanup_enabled = %(auto_cleanup_enabled)s,
                    min_retention_days = %(min_retention_days)s,
                    updated_at = %(updated_at)s
                WHERE id = %(id)s
            """
            await agent.db.sql(sql_update, config_data)
        else:
            # Create new config
            config_data["id"] = str(uuid.uuid4()).replace("-", "")[:32]
            config_data["created_at"] = config_data["updated_at"]
            sql_insert = """
                INSERT INTO harnessed_agent_config (
                    id, user_id, work_dir, skills_path, max_memory_tokens,
                    default_priority, high_priority_threshold, low_priority_threshold,
                    auto_cleanup_enabled, min_retention_days, created_at, updated_at
                ) VALUES (
                    %(id)s, %(user_id)s, %(work_dir)s, %(skills_path)s, %(max_memory_tokens)s,
                    %(default_priority)s, %(high_priority_threshold)s, %(low_priority_threshold)s,
                    %(auto_cleanup_enabled)s, %(min_retention_days)s, %(created_at)s, %(updated_at)s
                )
            """
            await agent.db.sql(sql_insert, config_data)

        return {"success": True}
    except Exception as e:
        return {"success": False, "error": str(e)}