from bot import aria2


def get_download(gid):
    return aria2.get_download(gid)


class DownloadStatus:
    STATUS_UPLOADING = "Uploading"
    STATUS_DOWNLOADING = "Downloading"
    STATUS_WAITING = "Queued"
    STATUS_FAILED = "Failed. Cleaning download"

    def __init__(self, gid):
        self.__gid = gid
        self.__download = get_download(gid)

    def __update(self):
        self.__download = get_download(self.__gid)

    def progress(self):
        self.__update()
        return self.__download.progress_string()

    def speed(self):
        self.__update()
        return self.__download.download_speed_string()

    def name(self):
        return self.__download.name

    def size(self):
        return self.__download.total_length_string()

    def eta(self):
        self.__update()
        return self.__download.eta_string()

    def status(self):
        self.__update()
        status = None
        if self.__download.is_waiting:
            status = DownloadStatus.STATUS_WAITING
        elif self.__download.is_complete:
            # If download exists and is complete the it must be uploading
            # otherwise the gid would have been removed from the download_list
            status = DownloadStatus.STATUS_UPLOADING
        elif self.__download.is_active:
            status = DownloadStatus.STATUS_DOWNLOADING
        return status

    def download(self):
        self.__update()
        return self.__download