152 lines
6.5 KiB
Python
152 lines
6.5 KiB
Python
"""Hello World 示例插件 — 新 SDK 版本
|
||
|
||
你的第一个 MaiCore 插件，包含问候功能、时间查询等基础示例。
|
||
"""
|
||
|
||
import datetime
|
||
import random
|
||
|
||
from maibot_sdk import MaiBotPlugin, Action, Command, Tool, EventHandler
|
||
from maibot_sdk.types import ActivationType, EventType, ToolParameterInfo, ToolParamType
|
||
|
||
|
||
class HelloWorldPlugin(MaiBotPlugin):
|
||
"""Hello World 示例插件"""
|
||
|
||
# ===== Tool 组件 =====
|
||
|
||
@Tool(
|
||
"compare_numbers",
|
||
description="使用工具比较两个数的大小，返回较大的数",
|
||
parameters=[
|
||
ToolParameterInfo(name="num1", param_type=ToolParamType.FLOAT, description="第一个数字", required=True),
|
||
ToolParameterInfo(name="num2", param_type=ToolParamType.FLOAT, description="第二个数字", required=True),
|
||
],
|
||
)
|
||
async def handle_compare_numbers(self, num1: float = 0, num2: float = 0, **kwargs):
|
||
"""比较两个数的大小"""
|
||
try:
|
||
if num1 > num2:
|
||
result = f"{num1} 大于 {num2}"
|
||
elif num1 < num2:
|
||
result = f"{num1} 小于 {num2}"
|
||
else:
|
||
result = f"{num1} 等于 {num2}"
|
||
return {"name": "compare_numbers", "content": result}
|
||
except Exception as e:
|
||
return {"name": "compare_numbers", "content": f"比较数字失败，炸了: {e}"}
|
||
|
||
# ===== Action 组件 =====
|
||
|
||
@Action(
|
||
"hello_greeting",
|
||
description="向用户发送问候消息",
|
||
activation_type=ActivationType.ALWAYS,
|
||
action_parameters={"greeting_message": "要发送的问候消息"},
|
||
action_require=["需要发送友好问候时使用", "当有人向你问好时使用", "当你遇见没有见过的人时使用"],
|
||
associated_types=["text"],
|
||
)
|
||
async def handle_hello(self, stream_id: str = "", greeting_message: str = "", **kwargs):
|
||
"""问候动作"""
|
||
config_result = await self.ctx.config.get("greeting.message")
|
||
base_message = config_result if isinstance(config_result, str) else "嗨！很开心见到你！😊"
|
||
message = base_message + greeting_message
|
||
await self.ctx.send.text(message, stream_id)
|
||
return True, "发送了问候消息"
|
||
|
||
@Action(
|
||
"bye_greeting",
|
||
description="向用户发送告别消息",
|
||
activation_type=ActivationType.KEYWORD,
|
||
activation_keywords=["再见", "bye", "88", "拜拜"],
|
||
action_parameters={"bye_message": "要发送的告别消息"},
|
||
action_require=["用户要告别时使用", "当有人要离开时使用", "当有人和你说再见时使用"],
|
||
associated_types=["text"],
|
||
)
|
||
async def handle_bye(self, stream_id: str = "", bye_message: str = "", **kwargs):
|
||
"""告别动作"""
|
||
message = f"再见！期待下次聊天！👋{bye_message}"
|
||
await self.ctx.send.text(message, stream_id)
|
||
return True, "发送了告别消息"
|
||
|
||
# ===== Command 组件 =====
|
||
|
||
@Command("time", description="查询当前时间", pattern=r"^/time$")
|
||
async def handle_time(self, stream_id: str = "", **kwargs):
|
||
"""时间查询命令"""
|
||
config_result = await self.ctx.config.get("time.format")
|
||
time_format = config_result if isinstance(config_result, str) else "%Y-%m-%d %H:%M:%S"
|
||
now = datetime.datetime.now()
|
||
time_str = now.strftime(time_format)
|
||
await self.ctx.send.text(f"⏰ 当前时间：{time_str}", stream_id)
|
||
return True, f"显示了当前时间: {time_str}", True
|
||
|
||
@Command("random_emojis", description="发送多张随机表情包", pattern=r"^/random_emojis$")
|
||
async def handle_random_emojis(self, stream_id: str = "", **kwargs):
|
||
"""发送多张随机表情包"""
|
||
emojis = await self.ctx.emoji.get_random(5)
|
||
if not emojis:
|
||
return False, "未找到表情包", False
|
||
# 用转发消息发送多张图片
|
||
messages = [
|
||
{"user_id": "0", "nickname": "神秘用户", "segments": [{"type": "image", "content": e.get("base64", "")}]}
|
||
for e in emojis
|
||
]
|
||
await self.ctx.send.forward(messages, stream_id)
|
||
return True, "已发送随机表情包", True
|
||
|
||
@Command("test", description="测试命令", pattern=r"^/test$")
|
||
async def handle_test(self, stream_id: str = "", **kwargs):
|
||
"""测试命令 — 发送简单测试消息"""
|
||
await self.ctx.send.text("测试正常！Bot 功能运行中 ✅", stream_id)
|
||
return True, "测试完成", True
|
||
|
||
# ===== EventHandler 组件 =====
|
||
|
||
@EventHandler("print_message_handler", description="打印接收到的消息", event_type=EventType.ON_MESSAGE)
|
||
async def handle_print_message(self, message=None, **kwargs):
|
||
"""打印消息事件"""
|
||
config_result = await self.ctx.config.get("print_message.enabled")
|
||
enabled = config_result if isinstance(config_result, bool) else False
|
||
if enabled and message:
|
||
raw = message.get("raw_message", "") if isinstance(message, dict) else str(message)
|
||
print(f"接收到消息: {raw}")
|
||
return True, True, "消息已打印", None, None
|
||
|
||
@EventHandler(
|
||
"forward_messages_handler", description="把接收到的消息转发到指定聊天ID", event_type=EventType.ON_MESSAGE
|
||
)
|
||
async def handle_forward_messages(self, message=None, stream_id: str = "", **kwargs):
|
||
"""收集消息并定期转发"""
|
||
if not message:
|
||
return True, True, None, None, None
|
||
plain_text = message.get("plain_text", "") if isinstance(message, dict) else ""
|
||
if not plain_text:
|
||
return True, True, None, None, None
|
||
|
||
# 使用插件级状态收集消息
|
||
if not hasattr(self, "_fwd_messages"):
|
||
self._fwd_messages: list[str] = []
|
||
self._fwd_counter: int = 0
|
||
|
||
self._fwd_messages.append(plain_text)
|
||
self._fwd_counter += 1
|
||
|
||
if self._fwd_counter % 10 == 0 and stream_id:
|
||
if random.random() < 0.01:
|
||
segments = [{"type": "text", "content": msg} for msg in self._fwd_messages]
|
||
await self.ctx.send.hybrid(segments, stream_id)
|
||
else:
|
||
messages = [
|
||
{"user_id": "0", "nickname": "转发", "segments": [{"type": "text", "content": msg}]}
|
||
for msg in self._fwd_messages
|
||
]
|
||
await self.ctx.send.forward(messages, stream_id)
|
||
self._fwd_messages = []
|
||
|
||
return True, True, None, None, None
|
||
|
||
|
||
def create_plugin():
|
||
return HelloWorldPlugin()
|