import os
from typing import Optional
from win10toast import ToastNotifier
from windows_toasts import ToastButton, ToastActivatedEventArgs, Toast, InteractableWindowsToaster, ToastDisplayImage
import subprocess
from abc import ABC, abstractmethod

import utils

class INotifier(ABC):
    @abstractmethod
    def notify(self, title, msg):
        pass

    @abstractmethod
    def show_received_file(self, folder: str, filename: str, ori_filename: str) -> None:
        """
        显示收到的文件
        图片文件会显示图片内容，非图片文件只显示文件名
        :param folder: 文件的保存目录
        :param filename: 保存的文件
        :param ori_filename: 原始文件名
        :return: None
        """
        pass

    @abstractmethod
    def show_received_files(self, folder: str, ori_filename_list: list):
        """
        显示收到的批量文件
        :param folder: 文件的保存目录
        :param ori_filename_list: 原始文件列表
        :return: None
        """
        pass

    @abstractmethod
    def show_future_files(self, folder: Optional[str], filename_list: list, to_mobile: bool):
        """
        显示接下来将要收到哪些文件
        :param folder: 文件的保存目录
        :param filename_list: 文件列表
        :param to_mobile: 是否是发送到iOS
        :return:
        """
        pass

"""
不带按钮和图片显示的通知，适用于旧版系统
"""
class BasicNotifier(INotifier):
    def __init__(self):
        self.notifier = ToastNotifier()

    def notify(self, title, msg):
        self.notifier.show_toast(title, msg, threaded=True)

    def show_received_file(self, folder: str, filename: str, ori_filename: str) -> None:
        self.notify('收到文件:', ori_filename)

    def show_received_files(self, folder: str, ori_filename_list: list):
        num_files = len(ori_filename_list)
        if num_files == 0:
            raise ValueError('文件数量不能为0')
        msg = ", ".join(ori_filename_list)
        self.notify(f"收到 {num_files} 个文件:", msg)

    def show_future_files(self, folder: str, filename_list: list, to_mobile: bool):
        num_files = len(filename_list)
        if num_files == 0:
            raise ValueError('文件数量不能为0')
        msg = ", ".join(filename_list)
        action_str = '发送' if to_mobile else '接收'
        self.notify(f"开始{action_str} {num_files} 个文件:", msg)

"""
可交互式通知
"""
class Notifier(INotifier):
    def __init__(self):
        self.toaster = InteractableWindowsToaster("", 'Microsoft.Windows.Explorer')

    @staticmethod
    def _button_callback(args: ToastActivatedEventArgs):
        if '=' not in args.arguments:
            return
        action, arg = args.arguments.split('=')
        if action == 'select':
            subprocess.Popen(f'explorer /select,{arg}')
        elif action == 'open':
            subprocess.Popen(f'explorer {arg}')

    def notify(self, title, msg):
        self.clear_toasts()
        toast = Toast([title, msg])
        self.toaster.show_toast(toast)

    def clear_toasts(self):
        self.toaster.clear_scheduled_toasts()
        self.toaster.clear_toasts()

    def show_received_file(self, folder, filename, ori_filename):
        self.clear_toasts()
        toast = Toast([f"收到文件: {ori_filename}"])
        file_path = os.path.join(folder, filename)
        if utils.is_image_file(file_path):
            toast.AddImage(ToastDisplayImage.fromPath(file_path))
        toast.AddAction(ToastButton("打开文件夹", arguments=f'select={file_path}'))
        toast.AddAction(ToastButton("关闭", arguments='ignore='))
        toast.on_activated = self._button_callback
        self.toaster.show_toast(toast)

    def show_received_files(self, folder, ori_filename_list):
        self.clear_toasts()
        num_files = len(ori_filename_list)
        if num_files == 0:
            raise ValueError('文件数量不能为0')
        content = [f"收到 {num_files} 个文件:", ', '.join(ori_filename_list)]
        toast = Toast(content)
        toast.AddAction(ToastButton("打开文件夹", arguments=f'open={folder}'))
        toast.AddAction(ToastButton("关闭", arguments='ignore='))
        toast.on_activated = self._button_callback
        self.toaster.show_toast(toast)

    def show_future_files(self, folder: str, filename_list: list, to_mobile: bool):
        self.clear_toasts()
        num_files = len(filename_list)
        if num_files == 0:
            raise ValueError('文件数量不能为0')
        action_str = '发送' if to_mobile else '接收'
        toast = Toast([f"开始{action_str} {num_files} 个文件:", ', '.join(filename_list)])
        if not to_mobile:
            toast.AddAction(ToastButton("打开文件夹", arguments=f'open={folder}'))
            toast.AddAction(ToastButton("关闭", arguments='ignore='))
            toast.on_activated = self._button_callback
        self.toaster.show_toast(toast)

def create_notifier(basic: bool = True):
    return BasicNotifier() if basic else Notifier()