import aiohttp
import httpx
import shutil
import aiofiles
import asyncio
from bs4 import BeautifulSoup
import re
import os
import logging


async def get_latest_version(session):
    feeds_url = "http://feeds.braiins-os.com"

    resp = await session.get(feeds_url)
    data = resp.text

    soup = BeautifulSoup(data, "html.parser")

    versions = []

    for link in soup.find_all("td", {"class": "link"}):
        link_title = link.text.strip("/")
        if re.match("(\d+)\.(\d+)(\.\d+)?", link_title):
            versions.append(link_title)

    versions = sorted(versions, reverse=True)

    latest_version = versions[0]
    return latest_version


async def get_feeds_file(session, version):
    feeds_url = "http://feeds.braiins-os.com"

    resp = await session.get(feeds_url + "/" + version, follow_redirects=True)
    data = resp.text

    soup = BeautifulSoup(data, "html.parser")

    file = None

    for link in soup.find_all("a", href=True):
        href = link["href"]
        if re.match("braiins-os_am1-s9_ssh_.+\.tar.gz", href):
            if not href.endswith(".asc"):
                file = href

    if file:
        return file


async def get_update_file(session, version):
    feeds_url = "http://feeds.braiins-os.com"

    resp = await session.get(feeds_url + "/am1-s9")
    data = resp.text

    soup = BeautifulSoup(data, "html.parser")

    file = None

    for link in soup.find_all("a", href=True):
        href = link["href"]
        if re.match(f"firmware_(.+)-{version}-plus_arm_cortex-a9_neon\.tar", href):
            if not href.endswith(".asc"):
                file = href

    if file:
        return file


async def get_latest_update_file(session, update_file):
    update_file_loc = f"http://feeds.braiins-os.com/am1-s9/{update_file}"

    update_file_dir = os.path.join(os.path.dirname(__file__), "files", "update.tar")

    if os.path.exists(update_file_dir):
        os.remove(update_file_dir)

    update_file_data = await session.get(update_file_loc)
    if update_file_data.status_code == 200:
        f = await aiofiles.open(
            os.path.join(os.path.dirname(__file__), "files", "update.tar"),
            mode="wb",
        )
        await f.write(update_file_data.text)
        await f.close()


async def get_latest_install_file(session, version, feeds_path, install_file):
    install_file_loc = f"http://feeds.braiins-os.com/{version}/{install_file}"
    feeds_file_path = os.path.join(feeds_path, "toolbox_bos_install_am1-s9")

    with open(feeds_file_path, "a+") as feeds_file:
        feeds_file.write(version + "\t" + install_file.strip() + "\n")

    install_file_folder = os.path.join(feeds_path, version)
    if os.path.exists(install_file_folder):
        shutil.rmtree(install_file_folder)
    os.mkdir(install_file_folder)

    install_file_data = await session.get(install_file_loc)
    if install_file_data.status_code == 200:
        f = await aiofiles.open(
            os.path.join(install_file_folder, install_file), mode="wb"
        )
        for chunk in install_file_data.iter_bytes():
            await f.write(chunk)
        await f.close()


async def update_installer_files():
    feeds_path = os.path.join(
        os.path.dirname(__file__), "files", "bos-toolbox", "feeds"
    )
    feeds_versions = await get_local_versions()
    async with httpx.AsyncClient() as session:
        version = await get_latest_version(session)

        if version not in feeds_versions:
            update_file = await get_update_file(session, version)
            install_file = await get_feeds_file(session, version)
            await get_latest_update_file(session, update_file)
            await get_latest_install_file(session, version, feeds_path, install_file)
        else:
            logging.info("Feeds are up to date.")


async def get_local_versions():
    feeds_versions = []
    feeds_path = os.path.join(
        os.path.dirname(__file__), "files", "bos-toolbox", "feeds"
    )
    if not os.path.exists(feeds_path):
        os.mkdir(feeds_path)

    feeds_file_path = os.path.join(feeds_path, "toolbox_bos_install_am1-s9")

    if not os.path.exists(feeds_file_path):
        feeds_file = open(feeds_file_path, "w+")
        feeds_file.close()

    with open(feeds_file_path) as feeds_file:
        for line in feeds_file.readlines():
            ver = line.strip().split("\t")[0]
            feeds_versions.append(ver)

    return feeds_versions


if __name__ == "__main__":
    asyncio.run(update_installer_files())