51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import re
|
|
from typing import Any
|
|
from nonebot import on_message
|
|
from nonebot.adapters import Event
|
|
from nonebot_plugin_alconna import UniMessage, UniMsg
|
|
from playwright.async_api import Page
|
|
|
|
from konabot.common.nb import match_keyword
|
|
from konabot.common.web_render import WebRenderer, konaweb
|
|
|
|
|
|
async def render_image(message: str, style: str = 'say') -> UniMessage[Any]:
|
|
"""
|
|
渲染文本为图片
|
|
"""
|
|
|
|
async def page_function(page: Page):
|
|
await page.wait_for_function("typeof setContent === 'function'")
|
|
await page.evaluate(
|
|
"([ message, style ]) => { return setContent(message, style); }",
|
|
[ message, style ],
|
|
)
|
|
|
|
img_data = await WebRenderer.render(
|
|
url=konaweb("krgsay"),
|
|
target="#main",
|
|
other_function=page_function,
|
|
)
|
|
return UniMessage.image(raw=img_data)
|
|
|
|
|
|
ALLOWED_STYLE = { "say", "cry", "hungry", "blush" }
|
|
|
|
|
|
cmd = on_message(
|
|
rule=match_keyword.match_keyword(
|
|
re.compile(r"^krg(" + '|'.join(ALLOWED_STYLE) + r")\s.+", re.I),
|
|
),
|
|
)
|
|
|
|
@cmd.handle()
|
|
async def _(event: Event, msg: UniMsg):
|
|
text = msg.extract_plain_text().lstrip()
|
|
command, content = text.split(maxsplit=1)
|
|
style = command.removeprefix("krg").lower()
|
|
if style not in ALLOWED_STYLE:
|
|
style = 'say'
|
|
msg = await render_image(content, style)
|
|
await msg.send(event)
|
|
|