#!/usr/bin/env python3

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

from aiohttp import web
from common import config
from common import lxsecurity
from common import utils
from common import log
from common import Httpx
from apis import handle_api_request
import traceback
import time

logger = log.log("main")
aiologger = log.log('aiohttp_web')

Httpx.checkcn()

# check request info before start
async def handle_before_request(app, handler):
    async def handle_request(request):
        # nginx proxy header
        if (request.headers.get("X-Real-IP")):
            request.remote = request.headers.get("X-Real-IP")
        # check ip
        if (config.check_ip_banned(request.remote)):
            return utils.handle_response({"code": 1, "msg": "您的IP已被封禁", "data": None}, 403)
        # check global rate limit
        if (
            (time.time() - config.getRequestTime('global'))
            <
            (config.read_config("security.rate_limit.global"))
            ):
            return utils.handle_response({"code": 5, "msg": "全局限速", "data": None}, 429)
        if (
            (time.time() - config.getRequestTime(request.remote))
            <
            (config.read_config("security.rate_limit.ip"))
            ):
            return utils.handle_response({"code": 5, "msg": "IP限速", "data": None}, 429)
        # update request time
        config.updateRequestTime('global')
        config.updateRequestTime(request.remote)
        # check host
        if (config.read_config("security.allowed_host.enable")):
            if request.remote_host.split(":")[0] not in config.read_config("security.allowed_host.list"):
                if config.read_config("security.allowed_host.blacklist.enable"):
                    config.ban_ip(request.remote, int(config.read_config("security.allowed_host.blacklist.length")))
                return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
        try:
            resp = await handler(request)
            aiologger.info(f'{request.remote} - {request.method} "{request.path}", {resp.status}')
            return resp
        except web.HTTPException as ex:
            if ex.status == 500:  # 捕获500错误
                return utils.handle_response({"code": 4, "msg": "内部服务器错误", "data": None}, 500)
            else:
                logger.error(traceback.format_exc())
                return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
    return handle_request

async def main(request):
    return utils.handle_response({"code": 0, "msg": "success", "data": None})


async def handle(request):
    method = request.match_info.get('method')
    source = request.match_info.get('source')
    songId = request.match_info.get('songId')
    quality = request.match_info.get('quality')
    if (config.read_config("security.key.enable") and request.host.split(':')[0] not in config.read_config('security.whitelist_host')):
        if (request.headers.get("X-Request-Key")) != config.read_config("security.key.value"):
            if (config.read_config("security.key.ban")):
                config.ban_ip(request.remote)
            return utils.handle_response({"code": 1, "msg": "key验证失败", "data": None}, 403)
    if (config.read_config('security.check_lxm.enable') and request.host.split(':')[0] not in config.read_config('security.whitelist_host')):
        lxm = request.headers.get('lxm')
        if (not lxsecurity.checklxmheader(lxm, request.url)):
            if (config.read_config('security.lxm_ban.enable')):
                config.ban_ip(request.remote)
        return utils.handle_response({"code": 1, "msg": "lxm请求头验证失败", "data": None}, 403)
    
    try:
        return utils.handle_response(await handle_api_request(method, source, songId, quality))
    except Exception as e:
        logger.error(traceback.format_exc())
        return utils.handle_response({'code': 4, 'msg': '内部服务器错误', 'data': None}, 500)

async def handle_404(request):
    return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)

app = web.Application(middlewares=[handle_before_request])
# mainpage
app.router.add_get('/', main)

# api
app.router.add_get('/{method}/{source}/{songId}/{quality}', handle)

# 404
app.router.add_route('*', '/{tail:.*}', handle_404)

web.run_app(app, host=config.read_config('common.host'), port=config.read_config('common.port'))