PagerMaid_Plugins_Pyro/code/main.py

# python3
# -*- coding: utf-8 -*-
# @Time    : 2021/11/22 14:17
# @Author  : yzyyz
# @Email   :  youzyyz1384@qq.com
# @File    : run.py
# @Software: PyCharm
import re
import httpx

from pagermaid.listener import listener
from pagermaid.enums import Message, AsyncClient

codeType = {
    "py": ["python", "py"],
    "cpp": ["cpp", "cpp"],
    "java": ["java", "java"],
    "php": ["php", "php"],
    "js": ["javascript", "js"],
    "c": ["c", "c"],
    "c#": ["csharp", "cs"],
    "go": ["go", "go"],
    "asm": ["assembly", "asm"],
}


async def run(string: str, client: AsyncClient):
    string = string.replace("&", "&").replace("[", "[").replace("]", "]")
    try:
        a = re.findall(
            r"(py|php|java|cpp|js|c#|c|go|asm)\s?(-i)?\s?(\w*)?(\n|\r)((?:.|\n)+)",
            string,
        )[0]
        print(a)
    except Exception:
        return "输入有误汪\n目前仅支持c/cpp/c#/py/php/go/java/js"
    lang, code_str = a[0], a[4]
    if "-i" in string:
        data_json = {
            "files": [{"name": f"main.{codeType[lang][1]}", "content": code_str}],
            "stdin": a[2],
            "command": "",
        }
    else:
        data_json = {
            "files": [{"name": f"main.{codeType[lang][1]}", "content": code_str}],
            "stdin": "",
            "command": "",
        }
    headers = {
        "Authorization": "Token 0123456-789a-bcde-f012-3456789abcde",
        "content-type": "application/",
    }
    res = await client.post(
        url=f"https://glot.io/run/{codeType[lang][0]}?version=latest",
        headers=headers,
        json=data_json,
    )
    if res.status_code != 200:
        return "请求失败了呐~~~"
    if res.json()["stdout"] == "":
        return res.json()["stderr"].strip()
    return f"<b>>>></b> <code>{code_str}</code> \n{res.json()['stdout']}"


@listener(command="code", description="运行代码", parameters="[语言] [-i] [inputText]\n[代码]")
async def code(message: Message, client: AsyncClient):
    await message.edit(await run(message.arguments, client))