from ...ext_utils.status_utils import MirrorStatus
from ....modules.stats import commands


class RcloneStatus:
    def __init__(self, listener, obj, gid, status):
        self._obj = obj
        self._gid = gid
        self._status = status
        self.listener = listener
        self.tool = "rclone"
        self.engine = f"Rclone v{commands["rclone"]}"

    def gid(self):
        return self._gid

    def progress(self):
        return self._obj.percentage

    def speed(self):
        return self._obj.speed

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

    def size(self):
        return self._obj.size

    def eta(self):
        return self._obj.eta

    def status(self):
        if self._status == "dl":
            return MirrorStatus.STATUS_DOWNLOAD
        elif self._status == "up":
            return MirrorStatus.STATUS_UPLOAD
        else:
            return MirrorStatus.STATUS_CLONE

    def processed_bytes(self):
        return self._obj.transferred_size

    def task(self):
        return self._obj