PagerMaid-Pyro/pagermaid/web/api/command_alias.py

from fastapi import APIRouter
from fastapi.responses import JSONResponse

from pagermaid.common.alias import AliasManager
from pagermaid.web.api.utils import authentication

route = APIRouter()


@route.get(
    "/command_alias", response_class=JSONResponse, dependencies=[authentication()]
)
async def get_command_alias():
    alias = AliasManager()
    return {
        "status": 0,
        "msg": "ok",
        "data": {
            "items": alias.get_all_alias_dict(),
        },
    }


@route.post(
    "/command_alias", response_class=JSONResponse, dependencies=[authentication()]
)
async def add_command_alias(data: dict):
    data = data["items"]
    try:
        await AliasManager.save_from_web(data)
        return {"status": 0, "msg": "命令别名保存成功"}
    except Exception:
        return {"status": 1, "msg": "命令别名保存失败"}


@route.get(
    "/test_command_alias", response_class=JSONResponse, dependencies=[authentication()]
)
async def test_command_alias(message: str):
    alias = AliasManager()
    return {
        "status": 0,
        "msg": "测试成功",
        "data": {
            "new_msg": alias.test_alias(message),
        },
    }