import datetime
from os.path import exists
from os import remove
from typing import Optional

from httpx import get, stream


class Artifact:
    def __init__(self, data: dict):
        self.id: int = data.get("id", 0)
        self.name: str = "Grasscutter.zip"
        self.url: str = data.get("archive_download_url", "")
        size: int = data.get("archive_size", 0)
        self.size: str = f"{size / 1024 / 1024:.2f} MB"
        self.expired: bool = data.get("expired", False)
        created_at: str = data.get("created_at", "")
        # 2022-04-27T06:31:46Z
        try:
            self.created_at: datetime.datetime = datetime.datetime.strptime(
                created_at, "%Y-%m-%dT%H:%M:%SZ"
            ) + datetime.timedelta(hours=8)
        except ValueError:
            self.created_at: datetime.datetime = datetime.datetime.utcnow() + datetime.timedelta(hours=8)
        self.created_time: str = self.created_at.strftime("%Y-%m-%d %H:%M:%S")
        self.hash: str = get_hash()

    async def download(self, path: str = "Grasscutter.zip"):
        if exists("Grasscutter.zip"):
            remove("Grasscutter.zip")
        async with stream("GET", self.url, allow_redirects=True) as response:
            for chunk in response.iter_bytes():
                with open(path, "ab") as f:
                    f.write(chunk)


async def get_artifact() -> Optional[Artifact]:
    req = get("https://api.github.com/repos/Grasscutters/Grasscutter/actions/artifacts?per_page=1")
    if req.status_code == 200:
        data = req.json()
        if data.get("artifacts", []):
            return Artifact(data["artifacts"][0])
    return None


def get_hash() -> str:
    req = get("https://api.github.com/repos/Grasscutters/Grasscutter/actions/runs?per_page=1&status=success")
    if req.status_code == 200:
        data = req.json()
        if data.get("workflow_runs", []):
            return data["workflow_runs"][0]["head_commit"]["id"][:6] + "  " + \
                   data["workflow_runs"][0]["head_commit"]["message"][:50]
    return ""