"""
通知工具 - 支持桌面通知、声音提示等
"""

import platform
import subprocess
import logging
from typing import Optional


def send_notification(title: str, message: str, urgency: str = "normal") -> bool:
    """
    发送桌面通知
    
    Args:
        title: 通知标题
        message: 通知内容
        urgency: 通知紧急程度 (low, normal, high)
    
    Returns:
        是否成功发送
    """
    system = platform.system()
    
    try:
        if system == "Windows":
            # Windows 使用 PowerShell 发送通知
            ps_script = f"""
            [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType=WindowsRuntime] > $null
            $template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastText02)
            $template.SelectSingleNode("//text[@id='1']").AppendChild($template.CreateTextNode('{title}')) > $null
            $template.SelectSingleNode("//text[@id='2']").AppendChild($template.CreateTextNode('{message}')) > $null
            $toast = [Windows.UI.Notifications.ToastNotification]::new($template)
            [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier("Screen2Feishu").Show($toast)
            """
            
            # 使用 PowerShell 执行
            subprocess.run(
                ["powershell", "-Command", ps_script],
                capture_output=True,
                timeout=5
            )
            return True
            
        elif system == "Darwin":  # macOS
            # macOS 使用 osascript 发送通知
            script = f'display notification "{message}" with title "{title}"'
            subprocess.run(
                ["osascript", "-e", script],
                capture_output=True,
                timeout=5
            )
            return True
            
        elif system == "Linux":
            # Linux 使用 notify-send
            try:
                subprocess.run(
                    ["notify-send", "-u", urgency, title, message],
                    capture_output=True,
                    timeout=5
                )
                return True
            except FileNotFoundError:
                # 如果 notify-send 不可用，尝试使用 zenity
                try:
                    subprocess.run(
                        ["zenity", "--info", "--title", title, "--text", message],
                        capture_output=True,
                        timeout=5
                    )
                    return True
                except FileNotFoundError:
                    return False
        else:
            return False
            
    except Exception as e:
        logging.error(f"发送通知失败: {str(e)}")
        return False


def send_notification_with_icon(title: str, message: str, icon_path: Optional[str] = None) -> bool:
    """
    发送带图标的桌面通知
    
    Args:
        title: 通知标题
        message: 通知内容
        icon_path: 图标文件路径
    
    Returns:
        是否成功发送
    """
    system = platform.system()
    
    try:
        if system == "Linux" and icon_path:
            # Linux 支持图标
            subprocess.run(
                ["notify-send", "-u", "normal", "-i", icon_path, title, message],
                capture_output=True,
                timeout=5
            )
            return True
        else:
            # 其他系统使用普通通知
            return send_notification(title, message)
            
    except Exception as e:
        logging.error(f"发送带图标通知失败: {str(e)}")
        return False


def play_sound(sound_type: str = "success") -> bool:
    """
    播放系统声音
    
    Args:
        sound_type: 声音类型 (success, error, warning, info)
    
    Returns:
        是否成功播放
    """
    system = platform.system()
    
    try:
        if system == "Windows":
            # Windows 使用 PowerShell 播放系统声音
            sound_map = {
                "success": "SystemAsterisk",
                "error": "SystemHand",
                "warning": "SystemExclamation",
                "info": "SystemDefault"
            }
            
            sound = sound_map.get(sound_type, "SystemDefault")
            ps_script = f'[System.Media.SystemSounds]::{sound}.Play()'
            
            subprocess.run(
                ["powershell", "-Command", ps_script],
                capture_output=True,
                timeout=5
            )
            return True
            
        elif system == "Darwin":  # macOS
            # macOS 使用 afplay 播放系统声音
            sound_map = {
                "success": "/System/Library/Sounds/Purr.aiff",
                "error": "/System/Library/Sounds/Basso.aiff",
                "warning": "/System/Library/Sounds/Funk.aiff",
                "info": "/System/Library/Sounds/Glass.aiff"
            }
            
            sound_file = sound_map.get(sound_type, "/System/Library/Sounds/Glass.aiff")
            
            # 异步播放声音
            subprocess.Popen(
                ["afplay", sound_file],
                stdout=subprocess.DEVNULL,
                stderr=subprocess.DEVNULL
            )
            return True
            
        elif system == "Linux":
            # Linux 使用 paplay 或 aplay 播放声音
            # 这里使用简单的 beep 声音
            try:
                # 尝试使用 beep
                subprocess.run(
                    ["beep"],
                    capture_output=True,
                    timeout=5
                )
                return True
            except FileNotFoundError:
                # 如果 beep 不可用，尝试使用 paplay
                try:
                    # 创建临时的声音文件
                    import tempfile
                    import wave
                    import struct
                    
                    # 创建简单的正弦波声音
                    with tempfile.NamedTemporaryFile(suffix='.wav', delete=False) as f:
                        # 写入 WAV 文件头
                        sample_rate = 44100
                        duration = 0.2
                        num_samples = int(sample_rate * duration)
                        
                        wav_file = wave.open(f.name, 'w')
                        wav_file.setnchannels(1)
                        wav_file.setsampwidth(2)
                        wav_file.setframerate(sample_rate)
                        
                        # 生成正弦波
                        for i in range(num_samples):
                            value = int(32767 * 0.3 * (1 if i < num_samples // 2 else 0))
                            wav_file.writeframes(struct.pack('<h', value))
                        
                        wav_file.close()
                        
                        # 播放声音
                        subprocess.run(
                            ["paplay", f.name],
                            capture_output=True,
                            timeout=5
                        )
                        
                        # 删除临时文件
                        import os
                        os.unlink(f.name)
                        
                    return True
                except Exception:
                    return False
        else:
            return False
            
    except Exception as e:
        logging.error(f"播放声音失败: {str(e)}")
        return False


def notify_task_processed(success: bool, file_name: str, count: int = 1) -> bool:
    """
    通知任务处理结果
    
    Args:
        success: 是否成功
        file_name: 文件名
        count: 处理数量
    
    Returns:
        是否成功发送通知
    """
    if success:
        title = "  任务处理成功"
        message = f"已成功处理 {count} 个文件: {file_name}"
        sound_type = "success"
    else:
        title = "  任务处理失败"
        message = f"处理文件失败: {file_name}"
        sound_type = "error"
    
    # 发送桌面通知
    notification_sent = send_notification(title, message)
    
    # 播放声音提示
    sound_played = play_sound(sound_type)
    
    return notification_sent or sound_played


def notify_batch_result(results: dict) -> bool:
    """
    通知批量处理结果
    
    Args:
        results: 处理结果字典 {文件名: 是否成功}
    
    Returns:
        是否成功发送通知
    """
    success_count = sum(1 for v in results.values() if v)
    total_count = len(results)
    
    if success_count == total_count:
        title = "  批量处理完成"
        message = f"所有 {total_count} 个文件处理成功"
        sound_type = "success"
    elif success_count == 0:
        title = "  批量处理失败"
        message = f"所有 {total_count} 个文件处理失败"
        sound_type = "error"
    else:
        title = "⚠️ 批量处理部分成功"
        message = f"{success_count}/{total_count} 个文件处理成功"
        sound_type = "warning"
    
    # 发送桌面通知
    notification_sent = send_notification(title, message)
    
    # 播放声音提示
    sound_played = play_sound(sound_type)
    
    return notification_sent or sound_played


if __name__ == "__main__":
    # 测试通知功能
    import logging
    logging.basicConfig(level=logging.INFO)
    
    print("测试桌面通知...")
    success = send_notification("测试通知", "这是一个测试通知消息")
    print(f"通知发送: {'成功' if success else '失败'}")
    
    print("测试声音播放...")
    success = play_sound("success")
    print(f"声音播放: {'成功' if success else '失败'}")
    
    print("测试任务处理通知...")
    success = notify_task_processed(True, "test_image.png")
    print(f"任务通知: {'成功' if success else '失败'}")