"""
测试辅助函数
"""
import json
import time
import numpy as np
import sounddevice as sd
from typing import Optional


def play_audio(audio_data: np.ndarray, sample_rate: int = 24000):
    """播放音频"""
    if len(audio_data) == 0:
        print("   ⚠️  音频数据为空，跳过播放")
        return
    
    try:
        # 转换为 float32
        audio_float = audio_data.astype(np.float32) / 32768.0
        
        # 播放
        sd.play(audio_float, samplerate=sample_rate, blocking=True)
        print(f"   🔊 音频播放完成 ({len(audio_data)} samples, {len(audio_data)/sample_rate:.2f}s)")
    except Exception as e:
        print(f"   ❌ 音频播放失败: {e}")


def print_json(data: dict, indent: int = 2):
    """打印 JSON"""
    print(json.dumps(data, ensure_ascii=False, indent=indent))


def calculate_metrics(
    llm_ms: Optional[int],
    tts_ms: Optional[int],
    audio_length_s: float,
    total_time_s: float,
) -> dict:
    """计算性能指标"""
    metrics = {
        "llm_latency_ms": llm_ms,
        "tts_latency_ms": tts_ms,
        "audio_length_s": audio_length_s,
        "total_time_s": total_time_s,
    }
    
    if llm_ms:
        metrics["llm_latency_s"] = llm_ms / 1000
    
    if tts_ms:
        metrics["tts_latency_s"] = tts_ms / 1000
    
    if audio_length_s > 0 and total_time_s > 0:
        metrics["realtime_factor"] = audio_length_s / total_time_s
    
    return metrics


def print_metrics(metrics: dict):
    """打印性能指标"""
    print("\n   📊 性能指标:")
    if "llm_latency_ms" in metrics:
        print(f"      LLM 推理: {metrics['llm_latency_ms']}ms ({metrics.get('llm_latency_s', 0):.2f}s)")
    if "tts_latency_ms" in metrics:
        print(f"      TTS 合成: {metrics['tts_latency_ms']}ms ({metrics.get('tts_latency_s', 0):.2f}s)")
    print(f"      音频长度: {metrics['audio_length_s']:.2f}s")
    print(f"      总耗时: {metrics['total_time_s']:.2f}s")
    if "realtime_factor" in metrics:
        print(f"      实时率: {metrics['realtime_factor']:.2f}x")


class TestResult:
    """测试结果"""
    def __init__(self, test_name: str):
        self.test_name = test_name
        self.success = False
        self.errors = []
        self.details = {}
        self.start_time = time.time()
    
    def add_error(self, error: str):
        self.errors.append(error)
    
    def add_detail(self, key: str, value):
        self.details[key] = value
    
    def add_warning(self, warning: str):
        """添加警告信息"""
        self.details.setdefault("warnings", []).append(warning)
    
    def mark_success(self):
        self.success = True
    
    def finalize(self):
        self.elapsed = time.time() - self.start_time
        return self
    
    def print_report(self):
        print(f"\n{'='*60}")
        print(f"  测试: {self.test_name}")
        print(f"  耗时: {self.elapsed:.2f}s")
        print(f"{'='*60}")
        
        if self.success:
            print(f"  ✅ 通过")
        else:
            print(f"  ❌ 失败")
            for error in self.errors:
                print(f"      - {error}")
        
        if self.details:
            print(f"\n  详细信息:")
            for key, value in self.details.items():
                print(f"    {key}: {value}")
        
        print()
        return self.success