# ----------------------------------------
# - mode: python -
# - author: helloplhm-qwq -
# - name: lx.py -
# - project: lx-music-api-server -
# - license: MIT -
# ----------------------------------------
# This file is part of the "lx-music-api-server" project.

from . import Httpx
from . import config
from . import scheduler
from .variable import iscn
from .log import log
from aiohttp.web import Response

logger = log('lx_script')

async def get_response(retry = 0):
    if (retry > 10):
        raise Exception('请求源脚本内容失败')
    baseurl = 'https://raw.githubusercontent.com/lxmusics/lx-music-api-server/main/lx-music-source-example.js'
    try:
        if (iscn and (retry % 2) == 0):
            req = await Httpx.AsyncRequest('https://mirror.ghproxy.com/' + baseurl)
        else:
            req = await Httpx.AsyncRequest(baseurl)
    except: 
        return await get_response(retry + 1)
    return req

async def get_script():
    req = await get_response()
    if (req.status == 200):
        with open('./lx-music-source-example.js', 'w') as f:
            f.write(req.text)
            f.close()
        logger.info('更新源脚本成功')
    else:
        raise Exception('请求源脚本内容失败')

async def generate_script_response(request):
    if (request.query.get('key') != config.read_config('security.key.value') and config.read_config('security.key.enable')):
        return 'key验证失败'
    try:
        with open('./lx-music-source-example.js', 'r') as f:
            script = f.read()
    except:
        return '本地无源脚本'
    scriptLines = script.split('\n')
    newScriptLines = []
    for line in scriptLines:
        line = line.strip()
        if (line.startswith('const API_URL')):
            newScriptLines.append(f'const API_URL = "{request.scheme}://{request.host}/"')
        elif (line.startswith('const API_KEY')):
            newScriptLines.append(f'const API_KEY = "{config.read_config("security.key.value")}"')
        else:
            newScriptLines.append(line)
    
    return Response(text = '\n'.join(newScriptLines), content_type = 'text/javascript',
                    headers = {
                        'Content-Disposition': 'attachment; filename=lx-music-source.js'
                    })

if (config.read_config('common.allow_download_script')):
    scheduler.append('update_script', get_script)