from collections import OrderedDict

from voice_dialogue.core.constants import (
    voice_state_manager, session_manager, dropped_audio_cache,
    user_still_speaking_event, chat_history_cache, is_debug_mode
)
from voice_dialogue.models.voice_task import VoiceTask
from voice_dialogue.utils.logger import logger


class TaskStatusMixin:
    """提供语音任务状态检查和中断处理的通用功能"""

    def is_task_interrupted(self, voice_task: VoiceTask) -> bool:
        """检查语音任务是否被其他任务中断"""
        if not voice_state_manager.interrupt_task_id:
            return False

        if voice_task.id != voice_state_manager.interrupt_task_id:
            logger.info(f"任务<{voice_task.id}> 被任务<{voice_state_manager.interrupt_task_id}> 中断")
            return True
        
        return False

    def is_task_valid(self, voice_task: VoiceTask) -> bool:
        """检查语音任务是否有效（会话匹配、未被丢弃等）"""
        if self.is_task_interrupted(voice_task):
            return False
        if voice_task.session_id != session_manager.current_id:
            logger.info(f"任务<{voice_task.id}> 会话不匹配: {voice_task.session_id} != {session_manager.current_id}")
            return False
        if voice_task.answer_id in dropped_audio_cache:
            logger.info(f"任务<{voice_task.id}> 被丢弃: {voice_task.answer_id}")
            return False
        return True

    def handle_user_speaking_interruption(self, voice_task: VoiceTask) -> bool:
        """处理用户继续说话导致的中断"""
        if user_still_speaking_event.is_set():
            logger.info(f'用户仍在说话，丢弃任务 {voice_task.id}')
            voice_state_manager.drop_audio_task(voice_task.id)
            dropped_audio_cache[voice_task.answer_id] = voice_task.answer_id
            user_still_speaking_event.clear()
            return True
        return False


class HistoryMixin:
    """提供更新聊天历史记录的功能"""

    def update_chat_history(self, voice_task: VoiceTask) -> None:
        """更新会话的聊天历史"""
        chat_history = chat_history_cache.get(voice_task.session_id, OrderedDict())
        task_answer_id = voice_task.answer_id

        user_question_key = f'{task_answer_id}:human'
        if user_question_key not in chat_history:
            chat_history[user_question_key] = voice_task.transcribed_text

        ai_answer_key = f'{task_answer_id}:ai'
        cached_ai_answer = chat_history.get(ai_answer_key, [])
        cached_ai_answer.append(voice_task.answer_sentence)
        chat_history[ai_answer_key] = cached_ai_answer

        chat_history_cache[voice_task.session_id] = chat_history


class PerformanceLogMixin:
    """提供记录任务性能日志的功能"""

    def log_task_performance(self, voice_task: VoiceTask, task_name: str = "任务"):
        """记录ASR, LLM, TTS各阶段耗时和音频长度"""
        if not is_debug_mode():
            return

        try:
            from voice_dialogue.utils.audio_utils import calculate_audio_duration

            asr_duration = getattr(voice_task, 'whisper_end_time', 0) - getattr(voice_task, 'whisper_start_time', 0)
            llm_duration = getattr(voice_task, 'llm_end_time', 0) - getattr(voice_task, 'llm_start_time', 0)
            tts_duration = getattr(voice_task, 'tts_end_time', 0) - getattr(voice_task, 'tts_start_time', 0)

            audio_duration = 0
            if hasattr(voice_task, 'tts_generated_sentence_audio') and voice_task.tts_generated_sentence_audio:
                audio_data, sample_rate = voice_task.tts_generated_sentence_audio
                audio_duration = calculate_audio_duration(audio_data, sample_rate)

            logger.info(
                f"\n"
                f"┌───────────────────────── 任务信息  ───────────────────────┐\n"
                f"│ 任务ID: {voice_task.id}\n"
                f"├───────────────────────── 性能统计 ────────────────────────┤\n"
                f"│ ASR 耗时: {asr_duration:.2f}s\n"
                f"│ LLM 耗时: {llm_duration:.2f}s\n"
                f"│ TTS 耗时: {tts_duration:.2f}s\n"
                f"│ 音频长度: {audio_duration:.2f}s\n"
                f"├───────────────────────── 生成内容 ────────────────────────┤\n"
                f"│-> {voice_task.answer_sentence}\n"
                f"└──────────────────────────────────────────────────────────┘"
            )
        except Exception as e:
            logger.error(f"记录任务性能信息时出错: {e}")

    def log_task_user_question(self, voice_task: VoiceTask):
        if not is_debug_mode():
            return

        from voice_dialogue.config.paths import PROJECT_ROOT
        output_path = PROJECT_ROOT / "output"
        if not output_path.exists():
            output_path.mkdir(parents=True, exist_ok=True)

        import soundfile as sf
        output_filename = output_path / (voice_task.id + ".wav")
        sf.write(output_filename.as_posix(), voice_task.user_voice, 16000, subtype="PCM_16")