"""
Utility functions for Face Mask extension.

Provides helper functions for server status, cache info, etc.
"""

import os
import urllib.request
import urllib.error
import json
import tempfile
from typing import Dict, Tuple, Optional


def get_server_status() -> Dict:
    """
    Get server status and GPU information.

    Returns:
        dict: {
            'running': bool,
            'gpu_available': bool,
            'gpu_device': str or None,
            'gpu_count': int,
            'rocm_version': str or None,
        }
    """
    result = {
        'running': False,
        'gpu_available': False,
        'gpu_device': None,
        'gpu_count': 0,
        'rocm_version': None,
    }

    try:
        with urllib.request.urlopen("http://127.0.0.1:8181/status", timeout=1) as response:
            data = json.loads(response.read().decode('utf-8'))
            result['running'] = data.get('status') == 'running'
            result['gpu_available'] = data.get('gpu_available', False)
            result['gpu_device'] = data.get('gpu_device')
            result['gpu_count'] = data.get('gpu_count', 0)
            result['rocm_version'] = data.get('rocm_version')
    except (urllib.error.URLError, ConnectionRefusedError, TimeoutError):
        result['running'] = False

    return result


def get_cache_info(strip_name: Optional[str] = None) -> Tuple[str, int, int]:
    """
    Get cache directory information.

    Args:
        strip_name: If provided, get info for specific strip. Otherwise, get info for all cache.

    Returns:
        Tuple of (cache_path, total_size_bytes, file_count)
    """
    import bpy

    blend_file = bpy.data.filepath

    if strip_name:
        # Get cache for specific strip
        if blend_file:
            project_dir = os.path.dirname(blend_file)
            cache_path = os.path.join(project_dir, ".mask_cache", strip_name)
        else:
            cache_path = os.path.join(tempfile.gettempdir(), "blender_mask_cache", strip_name)
    else:
        # Get cache root
        if blend_file:
            project_dir = os.path.dirname(blend_file)
            cache_path = os.path.join(project_dir, ".mask_cache")
        else:
            cache_path = os.path.join(tempfile.gettempdir(), "blender_mask_cache")

    # Calculate size and count
    total_size = 0
    file_count = 0

    if os.path.exists(cache_path):
        for root, dirs, files in os.walk(cache_path):
            for file in files:
                if file.endswith('.png'):  # Only count mask images
                    file_path = os.path.join(root, file)
                    try:
                        total_size += os.path.getsize(file_path)
                        file_count += 1
                    except OSError:
                        pass

    return cache_path, total_size, file_count


def format_size(size_bytes: int) -> str:
    """
    Format bytes to human-readable size.

    Args:
        size_bytes: Size in bytes

    Returns:
        Formatted string (e.g., "1.5 MB")
    """
    for unit in ['B', 'KB', 'MB', 'GB']:
        if size_bytes < 1024.0:
            return f"{size_bytes:.1f} {unit}"
        size_bytes /= 1024.0
    return f"{size_bytes:.1f} TB"