from telegram.ext import CommandHandler, run_async
from bot.helper import download_tools, gdriveTools, listeners
from bot import LOGGER, dispatcher
from bot.helper import fs_utils
from bot import download_dict, status_reply_dict, DOWNLOAD_STATUS_UPDATE_INTERVAL
from bot.helper.message_utils import *
from time import sleep
from bot.helper.bot_utils import get_readable_message
LOGGER.info('mirror.py')


class MirrorListener(listeners.MirrorListeners):
    def __init__(self, context, update, reply_message):
        super().__init__(context, update, reply_message)

    def onDownloadStarted(self, link):
        LOGGER.info("Adding link: " + link)

    def onDownloadProgress(self, progress_status_list: list, index: int):
        msg = get_readable_message(progress_status_list)
        LOGGER.info("Editing message")
        editMessage(msg, self.context, self.reply_message)

    def onDownloadComplete(self, progress_status_list, index: int):
        msg = get_readable_message(progress_status_list)
        LOGGER.info("Download completed")
        editMessage(msg, self.context, self.reply_message)
        gdrive = gdriveTools.GoogleDriveHelper(self)
        gdrive.upload(progress_status_list[index].name())

    def onDownloadError(self, error, progress_status_list: list, index: int):
        LOGGER.error(error)
        editMessage(error, self.context, self.reply_message)
        del download_dict[self.update.update_id]
        fs_utils.clean_download(progress_status_list[index].path())

    def onUploadStarted(self, progress_status_list: list, index: int):
        msg = get_readable_message(progress_status_list)
        editMessage(msg, self.context, self.reply_message)

    def onUploadComplete(self, link: str, progress_status_list: list, index: int):
        msg = '<a href="{}">{}</a>'.format(link, progress_status_list[index].name())
        deleteMessage(self.context, self.reply_message)
        sendMessage(msg, self.context, self.update)
        del download_dict[self.update.update_id]
        fs_utils.clean_download(progress_status_list[index].path())

    def onUploadError(self, error: str, progress_status: list, index: int):
        LOGGER.error(error)
        editMessage(error, self.context, self.reply_message)
        del download_dict[self.update.update_id]
        fs_utils.clean_download(progress_status[index].path())


@run_async
def mirror(update, context):
    message = update.message.text
    link = message.replace('/mirror', '')[1:]
    reply_msg = sendMessage('Starting Download', context, update)
    listener = MirrorListener(context, update, reply_msg)
    aria = download_tools.DownloadHelper(listener)
    aria.add_download(link)


@run_async
def mirror_status(update, context):
    try:
        deleteMessage(context, status_reply_dict[update.effective_chat])
        del status_reply_dict[update.effective_chat]
    except KeyError:
        pass

    while True:
        message = get_readable_message()
        if len(message) == 0:
            message = "No active downloads"
            try:
                deleteMessage(context, status_reply_dict[update.effective_chat])
                del status_reply_dict[update.effective_chat]
            except KeyError:
                pass
            break
        try:
            editMessage(message, context, status_reply_dict[update.effective_chat])
        except KeyError:
            status_reply_dict[update.effective_chat] = sendMessage(message, context, update)
        sleep(DOWNLOAD_STATUS_UPDATE_INTERVAL)


mirror_handler = CommandHandler('mirror', mirror)
mirror_status_handler = CommandHandler('status', mirror_status)
dispatcher.add_handler(mirror_handler)
dispatcher.add_handler(mirror_status_handler)