mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-30 06:09:59 +08:00
88 lines
3.5 KiB
Python
88 lines
3.5 KiB
Python
"""AstrBot薄接入；业务、交易及绘图复用共享运行时。"""
|
|
|
|
import asyncio
|
|
import json
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from astrbot.api import AstrBotConfig
|
|
from astrbot.api.event import AstrMessageEvent, filter
|
|
from astrbot.api.message_components import At, Image
|
|
from astrbot.api.star import Context, Star, StarTools
|
|
|
|
from .runtime import COMMANDS, Runtime
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class FarmPlugin(Star):
|
|
def __init__(self, context: Context, config: AstrBotConfig):
|
|
super().__init__(context)
|
|
self.config = config
|
|
self._lifecycle_lock = asyncio.Lock()
|
|
self._closed = False
|
|
configured = config.get('data_dir', '')
|
|
data_dir = (Path(configured).expanduser() if configured else
|
|
StarTools.get_data_dir('astrbot_plugin_farm'))
|
|
image_format = str(config.get('image_format', 'AUTO')).upper()
|
|
if image_format not in ('AUTO', 'JPEG', 'PNG'):
|
|
raise ValueError('image_format must be AUTO, JPEG or PNG')
|
|
self.runtime = Runtime(
|
|
data_dir, config.get('font_path') or None, 'astrbot',
|
|
image_format, font_setting='font_path',
|
|
image_quality=config.get('image_quality', 75),
|
|
image_quality_provider=lambda: self.config.get('image_quality', 75),
|
|
)
|
|
|
|
async def initialize(self):
|
|
async with self._lifecycle_lock:
|
|
if not self._closed:
|
|
await self.runtime.start()
|
|
|
|
@filter.command('农场帮助', alias=set(COMMANDS) - {'农场帮助'})
|
|
async def farm(self, event: AstrMessageEvent):
|
|
"""农场帮助、注册、种植、仓库、交易及图片查询。"""
|
|
if self._closed:
|
|
yield event.plain_result('农场插件已关闭。')
|
|
return
|
|
await self.initialize()
|
|
parts = event.get_message_str().strip().split(maxsplit=1)
|
|
if not parts or parts[0] not in COMMANDS:
|
|
return
|
|
action = COMMANDS[parts[0]]
|
|
arguments = parts[1] if len(parts) == 2 else ''
|
|
target = None
|
|
if action == 'steal':
|
|
mentions = [str(segment.qq) for segment in event.get_messages()
|
|
if isinstance(segment, At)
|
|
and str(segment.qq) != str(event.get_self_id())]
|
|
if len(mentions) != 1 or mentions[0] in ('', 'all', '0'):
|
|
yield event.plain_result('请@一个具体的偷菜目标。')
|
|
return
|
|
target = mentions[0]
|
|
platform_id = event.get_platform_id()
|
|
if not platform_id or not event.get_sender_id():
|
|
yield event.plain_result('无法取得稳定的平台或用户标识，未执行。')
|
|
return
|
|
bot_id = json.dumps([str(platform_id), str(event.get_self_id())])
|
|
response = await self.runtime.execute(
|
|
action, arguments, bot_id=bot_id,
|
|
platform=event.get_platform_name(), user_id=event.get_sender_id(),
|
|
message_id=getattr(event.message_obj, 'message_id', None),
|
|
display_name=event.get_sender_name() or '农场玩家',
|
|
target_user_id=target,
|
|
)
|
|
if isinstance(response, bytes):
|
|
yield event.chain_result([Image.fromBytes(response)])
|
|
else:
|
|
yield event.plain_result(response)
|
|
|
|
async def terminate(self):
|
|
async with self._lifecycle_lock:
|
|
if self._closed:
|
|
return
|
|
self._closed = True
|
|
await self.runtime.close()
|
|
logger.info('farm astrbot plugin terminated')
|