import aiohttp
import asyncio
import json
import re
import argparse
from bs4 import BeautifulSoup

parser = argparse.ArgumentParser()
parser.add_argument('-v', required=True)
parser.add_argument('-s', required=True)
parser.add_argument('-u', required=True)

args = parser.parse_args()
version = args.v
source = args.s
u = args.u


async def check_url(session, url, platform_name):
    try:
        async with session.get(url) as response:
            if response.status == 200:
                if platform_name:
                    return response.url, platform_name
                else:
                    return response.url
    except aiohttp.ClientError:
        pass
    return None


async def send_request(json_data):
    url = u + json_data
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as response:
            if response.status == 200:
                data = await response.text()
                soup = BeautifulSoup(data, 'html.parser')
                system_response = soup.find('div', style='text-align:center;font-family:monospace;margin:50px auto 0;max-width:600px').text
                print(f'Ответ от GAS: {system_response}')

def get_urls(find_url):
    platform_urls = {}
    version_pattern = re.compile(r'-([0-9]+)\.(exe|tbz)')

    for url, platform in find_url:
        match = version_pattern.search(str(url))
        if match:
            version_number = int(match.group(1))
            if platform in platform_urls:
                # Сравниваем и сохраняем URL (если найдено больше одного url для платформы, то выбираем с самой большой ревизией)
                current_version = int(version_pattern.search(platform_urls[platform]).group(1))
                if version_number > current_version:
                    platform_urls[platform] = str(url)
            else:
                platform_urls[platform] = str(url)

    return platform_urls

async def main():
    start_number = 0
    before_enter = 6000

    numbers = start_number
    find_url = []

    win32 = "https://upgrade.scdn.co/upgrade/client/win32-x86/spotify_installer-{version}-{numbers}.exe"
    win64 = "https://upgrade.scdn.co/upgrade/client/win32-x86_64/spotify_installer-{version}-{numbers}.exe"
    win_arm64 = "https://upgrade.scdn.co/upgrade/client/win32-arm64/spotify_installer-{version}-{numbers}.exe"
    osx = "https://upgrade.scdn.co/upgrade/client/osx-x86_64/spotify-autoupdate-{version}-{numbers}.tbz"
    osx_arm64 = "https://upgrade.scdn.co/upgrade/client/osx-arm64/spotify-autoupdate-{version}-{numbers}.tbz"

    async with aiohttp.ClientSession() as session:
        tasks = []
        url_templates = [win32, win64, win_arm64, osx, osx_arm64]
        platform_names = ["WIN32", "WIN64", "WIN-ARM64", "OSX", "OSX-ARM64"]

        for url_template, platform_name in zip(url_templates, platform_names):
            numbers = start_number
            while numbers <= before_enter:
                url = url_template.format(
                    version=version, numbers=numbers
                )
                tasks.append(check_url(session, url, platform_name))
                numbers += 1

        results = []
        for task in asyncio.as_completed(tasks):
            result = await task
            if result is not None:
                find_url.append(result)

    if find_url:
        latest_urls = get_urls(find_url)
        latest_urls["version"] = version
        latest_urls["source"] = source
        req_ver = json.dumps(latest_urls, ensure_ascii=False, indent=2)
        print(f'Версия {version} отправлена')
        await send_request(req_ver)
    else:
        print(f'Версия {version} не найдена')
        data = {
            "unknown": "unknown",
            "version": version
        }
        req_ver = json.dumps(data)
        await send_request(req_ver)


asyncio.run(main())