import os
import requests
import logging
from config import config, log, console
from rich.progress import (
    BarColumn,
    DownloadColumn,
    Progress,
    TextColumn,
    TimeRemainingColumn,
    TransferSpeedColumn,
)

progress = Progress(
    TextColumn("[bold blue]{task.fields[filename]}", justify="right"),
    BarColumn(bar_width=None),
    "[progress.percentage]{task.percentage:>3.1f}%",
    "•",
    DownloadColumn(),
    "•",
    TransferSpeedColumn(),
    "•",
    TimeRemainingColumn(),
    console=console
)


def check_if_tool_exists(tool: str) -> bool:
    if not os.path.exists(config["folders"]["tools"]):
        log.info(f"creating `tools` folder: {config['folders']['tools']}")
        os.mkdir(config["folders"]["tools"])

    if not os.path.exists(f"{config['folders']['tools']}/{tool}"):
        return False
    elif os.path.exists(f"{config['folders']['tools']}/{tool}") and os.path.isdir(
        f"{config['folders']['tools']}/{tool}"
    ):
        log.warning(f"`{config['folders']['tools']}/{tool}` is a folder")
        return True
    else:
        return True

requests_log = logging.getLogger("urllib3.connectionpool")
requests_log.setLevel(logging.WARNING)

def download_tool(url: str, tool: str):
    if not check_if_tool_exists(tool):
        progress.start()
        
        try:
            log.info(f"Requesting {url}")
            response = requests.get(url, stream=True)
            total = int(response.headers.get("content-length", None))
            task_id = progress.add_task("download", start=False, total=total, filename=tool)

            with open(f"{config['folders']['tools']}/{tool}", "wb") as file:
                progress.start_task(task_id)
                for bytes in response.iter_content(chunk_size=32768):
                    size = file.write(bytes)
                    progress.update(task_id, advance=size)

            log.info(f"`{tool}` downloaded")
        except Exception as e:
            log.error(f"error while downloading `{tool}`: {e}")

        progress.stop()

def check_and_download_all_tools():
    for tool in config["tools"]:
        download_tool(tool["url"], tool["tool"])