"""
System Info Module
System monitoring and space information.
"""
import os
import json
import platform
import psutil

from .config import STORAGE_DIR, HF_TOKEN
from .utils import format_size


def get_system_info() -> str:
    """Get detailed system information"""
    try:
        mem = psutil.virtual_memory()
        disk = psutil.disk_usage('/')
        cpu = psutil.cpu_percent(interval=0.1)

        return json.dumps({
            "system": {
                "platform": platform.system(),
                "python": platform.python_version(),
            },
            "cpu": {
                "cores": psutil.cpu_count(),
                "usage": f"{cpu}%",
            },
            "memory": {
                "total": format_size(mem.total),
                "used": format_size(mem.used),
                "percent": f"{mem.percent}%",
            },
            "disk": {
                "total": format_size(disk.total),
                "free": format_size(disk.free),
                "percent": f"{disk.percent}%",
            }
        }, indent=2)
    except Exception as e:
        return json.dumps({"error": str(e)})


def get_system_dashboard() -> str:
    """Get ASCII dashboard with bars"""
    try:
        mem = psutil.virtual_memory()
        disk = psutil.disk_usage('/')
        cpu = psutil.cpu_percent(interval=0.1)

        def bar(pct):
            filled = int(pct / 5)
            return '█' * filled + '░' * (20 - filled)

        return f"""
╔════════════════════════════════════════════════════════╗
║              📊 SYSTEM DASHBOARD                       ║
╠════════════════════════════════════════════════════════╣
║  💾 RAM:  [{bar(mem.percent)}] {mem.percent:5.1f}%     ║
║          {format_size(mem.used):>10} / {format_size(mem.total):<10}       ║
╠════════════════════════════════════════════════════════╣
║  💿 DISK: [{bar(disk.percent)}] {disk.percent:5.1f}%     ║
║          {format_size(disk.used):>10} / {format_size(disk.total):<10}       ║
╠════════════════════════════════════════════════════════╣
║  🔥 CPU:  [{bar(cpu)}] {cpu:5.1f}%     ║
║          {psutil.cpu_count()} cores                              ║
╚════════════════════════════════════════════════════════╝
""".strip()
    except Exception as e:
        return f"Error: {e}"


def get_space_info() -> str:
    """Get HuggingFace space information"""
    return json.dumps({
        "space_id": os.getenv("SPACE_ID", "local"),
        "author": os.getenv("SPACE_AUTHOR_NAME", "unknown"),
        "hardware": os.getenv("SPACE_HARDWARE", "cpu-basic"),
        "hf_token": "set" if HF_TOKEN else "not_set",
        "storage_path": str(STORAGE_DIR),
    }, indent=2)


def get_storage_stats() -> str:
    """Get storage statistics"""
    try:
        files = list(STORAGE_DIR.rglob('*'))
        total = sum(f.stat().st_size for f in files if f.is_file())
        return json.dumps({
            "files": len([f for f in files if f.is_file()]),
            "folders": len([f for f in files if f.is_dir()]),
            "total_size": format_size(total),
        }, indent=2)
    except Exception as e:
        return json.dumps({"error": str(e)})