"""
Toon格式轉換工具

將JSON數據轉換為toon格式以減少token消耗
"""
from toon_format import encode, decode, estimate_savings, compare_formats
import json
from typing import Union, Dict, List


def convert_json_to_toon(json_data: Union[str, dict, list]) -> str:
    """
    將JSON數據轉換為toon格式
    
    Args:
        json_data: JSON字符串或Python字典/列表
        
    Returns:
        toon格式的字符串
    """
    if isinstance(json_data, str):
        try:
            data = json.loads(json_data)
        except json.JSONDecodeError:
            # 如果無法解析，直接返回原始字符串
            return json_data
    else:
        data = json_data
    
    try:
        return encode(data)
    except Exception as e:
        print(f"警告：toon轉換失敗：{e}，返回原始數據")
        return json.dumps(data, ensure_ascii=False) if not isinstance(json_data, str) else json_data


def convert_toon_to_json(toon_data: str) -> dict:
    """
    將toon數據轉換回JSON/Python字典
    
    Args:
        toon_data: toon格式的字符串
        
    Returns:
        Python字典
    """
    try:
        return decode(toon_data)
    except Exception as e:
        print(f"警告：toon解碼失敗：{e}")
        # 嘗試作為JSON解析
        try:
            return json.loads(toon_data)
        except:
            return {"error": "無法解析數據", "原始數據": toon_data}


def show_toon_savings(data: Union[dict, list]) -> Dict[str, float]:
    """
    顯示使用toon格式的token節省情況
    
    Args:
        data: Python字典或列表
        
    Returns:
        包含節省百分比和token數的字典
    """
    try:
        result = estimate_savings(data)
        print(f"Token節省: {result['savings_percent']:.1f}%")
        print(f"JSON tokens: {result['json_tokens']}")
        print(f"Toon tokens: {result['toon_tokens']}")
        return result
    except Exception as e:
        print(f"警告：無法計算節省：{e}")
        return {"savings_percent": 0, "json_tokens": 0, "toon_tokens": 0}


def compare_format_display(data: Union[dict, list]) -> None:
    """
    顯示JSON與toon格式的直觀比較
    
    Args:
        data: Python字典或列表
    """
    try:
        print(compare_formats(data))
    except Exception as e:
        print(f"警告：無法顯示格式比較：{e}")