import aiohttp
from utils.utils import get_local_proxy
from utils.user_agent import get_user_agent

url = f"http://fanyi.youdao.com/translate?smartresult=dict&smartresult=rule&smartresult=ugc&sessionFrom=null"


async def translate_msg(language_type, msg):
    data = {
        "type": parse_language(language_type),
        "i": msg,
        "doctype": "json",
        "version": "2.1",
        "keyfrom": "fanyi.web",
        "ue": "UTF-8",
        "action": "FY_BY_CLICKBUTTON",
        "typoResult": "true",
    }
    async with aiohttp.ClientSession(headers=get_user_agent()) as session:
        async with session.post(url, data=data, proxy=get_local_proxy()) as res:
            data = await res.json()
            if data["errorCode"] == 0:
                return data["translateResult"][0][0]["tgt"]
    return "翻译惜败.."


# ZH_CN2EN 中文　»　英语
# ZH_CN2JA 中文　»　日语
# ZH_CN2KR 中文　»　韩语
# ZH_CN2FR 中文　»　法语
# ZH_CN2RU 中文　»　俄语
# ZH_CN2SP 中文　»　西语
# EN2ZH_CN 英语　»　中文
# JA2ZH_CN 日语　»　中文
# KR2ZH_CN 韩语　»　中文
# FR2ZH_CN 法语　»　中文
# RU2ZH_CN 俄语　»　中文
# SP2ZH_CN 西语　»　中文


def parse_language(language_type):
    if language_type == "英翻":
        return "EN2ZH_CN"
    if language_type == "日翻":
        return "JA2ZH_CN"
    if language_type == "韩翻":
        return "KR2ZH_CN"
    # if language_type == '法翻':
    #     return 'FR2ZH_CN'
    # if language_type == '俄翻':
    #     return 'RU2ZH_CN'
    if language_type == "翻英":
        return "ZH_CN2EN"
    if language_type == "翻日":
        return "ZH_CN2JA"
    if language_type == "翻韩":
        return "ZH_CN2KR"
    # if language_type == '翻法':
    #     return 'ZH_CN2FR'
    # if language_type == '翻俄':
    #     return 'ZH_CN2RU'