mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-28 05:00:01 +08:00
137 lines
4.7 KiB
Python
137 lines
4.7 KiB
Python
"""宿主依赖止于接入层；保持旧版中文入口，不新增测试前缀。"""
|
|
|
|
import asyncio
|
|
import logging
|
|
import time
|
|
|
|
from nonebot import get_driver, on_command, require
|
|
from nonebot.adapters import Bot, Event, Message
|
|
from nonebot.params import CommandArg
|
|
|
|
require("nonebot_plugin_uninfo")
|
|
from nonebot_plugin_uninfo import Uninfo
|
|
from zhenxun.configs.path_config import DATA_PATH, FONT_PATH
|
|
from zhenxun.utils.message import MessageUtils
|
|
from zhenxun.utils.platform import PlatformUtils
|
|
|
|
from .runtime import COMMANDS, Runtime
|
|
from .data_paths import data_directory
|
|
from .config_migration import migrate_config
|
|
|
|
|
|
def image_quality():
|
|
from zhenxun.configs.config import Config
|
|
|
|
return Config.get(__package__.split('.')[-1]).get(
|
|
'农场图片质量', 75, build_model=False,
|
|
)
|
|
|
|
|
|
def resource_options():
|
|
from zhenxun.configs.config import Config
|
|
|
|
config = Config.get(__package__.split('.')[-1])
|
|
return (config.get('资源获取地址', 'Github', build_model=False),
|
|
config.get('farm_service 地址', '', build_model=False))
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
driver = get_driver()
|
|
data_dir = data_directory(
|
|
DATA_PATH, getattr(driver.config, 'farm_preview_data_dir', None),
|
|
)
|
|
runtime = Runtime(
|
|
data_dir,
|
|
getattr(driver.config, "farm_preview_font", None),
|
|
getattr(driver.config, "farm_preview_deployment", "zhenxun-preview"),
|
|
str(getattr(driver.config, "farm_preview_image_format", "AUTO")).upper(),
|
|
font_dir=FONT_PATH,
|
|
image_quality_provider=image_quality,
|
|
resource_options_provider=resource_options,
|
|
)
|
|
async def start_runtime():
|
|
from zhenxun.configs.config import Config
|
|
|
|
migrate_config(Config, __package__.split(".")[-1])
|
|
await runtime.start()
|
|
|
|
|
|
driver.on_startup(start_runtime)
|
|
driver.on_shutdown(runtime.close)
|
|
_avatars = {}
|
|
|
|
|
|
async def avatar_for(user_id, platform):
|
|
if platform not in ("OneBot V11", "QQ"):
|
|
return None
|
|
key = (platform, user_id)
|
|
now = time.monotonic()
|
|
cached = _avatars.get(key)
|
|
if cached and cached[0] > now:
|
|
return cached[1]
|
|
try:
|
|
content = await asyncio.wait_for(
|
|
PlatformUtils.get_user_avatar(user_id, "qq"), timeout=2,
|
|
)
|
|
if content and len(content) > 2 * 1024 * 1024:
|
|
content = None
|
|
except Exception:
|
|
content = None
|
|
logger.warning("farm preview avatar unavailable; using default")
|
|
while len(_avatars) >= 16:
|
|
_avatars.pop(next(iter(_avatars)))
|
|
_avatars[key] = (now + (300 if content else 30), content)
|
|
return content
|
|
|
|
|
|
def handler_for(action):
|
|
async def handle(
|
|
bot: Bot, event: Event, session: Uninfo,
|
|
args: Message = CommandArg(),
|
|
):
|
|
platform = bot.adapter.get_name()
|
|
user_id = str(session.user.id)
|
|
message_id = getattr(event, "message_id", None)
|
|
if message_id is None:
|
|
message_id = getattr(event, "id", None)
|
|
try:
|
|
target_user_id = None
|
|
if action == 'steal':
|
|
mentions = [str(seg.data.get('qq', seg.data.get(
|
|
'user_id', seg.data.get('id', ''),
|
|
))) for seg in args if seg.type in (
|
|
'at', 'mention', 'mention_user', 'mention_everyone',
|
|
)]
|
|
if len(mentions) > 1:
|
|
await MessageUtils.build_message(
|
|
'请只指定一个偷菜目标。',
|
|
).send(reply_to=True)
|
|
return
|
|
if mentions:
|
|
target_user_id = mentions[0]
|
|
if target_user_id in ('', 'all'):
|
|
await MessageUtils.build_message(
|
|
'请@一个具体用户。',
|
|
).send(reply_to=True)
|
|
return
|
|
avatar = (await avatar_for(user_id, platform)
|
|
if action in ("farm", "details", "plot_ids") else None)
|
|
response = await runtime.execute(
|
|
action, args.extract_plain_text(), bot_id=str(bot.self_id),
|
|
platform=platform, user_id=user_id, message_id=message_id,
|
|
display_name=session.user.name or "农场玩家", avatar=avatar,
|
|
target_user_id=target_user_id,
|
|
)
|
|
except Exception as error:
|
|
logger.error(
|
|
"farm preview request failed type=%s", type(error).__name__,
|
|
)
|
|
response = "农场服务异常，请查看服务器日志。"
|
|
await MessageUtils.build_message(response).send(reply_to=True)
|
|
return handle
|
|
|
|
|
|
for command, action in COMMANDS.items():
|
|
matcher = on_command(command, priority=5, block=True)
|
|
matcher.handle()(handler_for(action))
|