liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

临时调整权限模块，备份

Browse Source
This commit is contained in:
liuwei
2025-06-09 14:12:31 +08:00
parent cedab1cefd
commit 9d15bf965b
30 changed files with 882 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
plugins/ai_auto_response/main.py
View File

@@ -14,6 +14,10 @@ from .bot_ai import InterventionBot
class AIAutoResponsePlugin(MessagePluginInterface):
"""AI自动对话插件"""
# 功能权限常量
FEATURE_KEY = "AI_AUTO_RESPONSE"
FEATURE_DESCRIPTION = "🤖 AI自动对话功能 [自动对话]"
@property
def name(self) -> str:
return "AI自动对话"
@@ -38,11 +42,21 @@ class AIAutoResponsePlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.intervention_bot = None
self.group_messages = {} # 存储每个群的最近消息
self.max_messages = 20 # 每个群最多存储的消息数量
# 注册功能权限
self.feature = self.register_feature()
# DIFY API配置
self.dify_api_url = ""
@@ -89,7 +103,7 @@ class AIAutoResponsePlugin(MessagePluginInterface):
content = str(message.get("content", "")).strip()
roomid = message.get("roomid", "")
if GroupBotManager.get_group_permission(roomid, Feature.AI_AUTO) == PermissionStatus.DISABLED:
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False
# 如果是群消息，且该群启用了自动回复，则处理
if roomid:
@@ -133,7 +147,7 @@ class AIAutoResponsePlugin(MessagePluginInterface):
roomid = message.get("roomid", "")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and GroupBotManager.get_group_permission(roomid, Feature.AI_AUTO) == PermissionStatus.DISABLED:
if roomid and GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 处理自动回复
try:

18
plugins/beautyleg/main.py
View File

@@ -15,6 +15,10 @@ from wechat_ipad import WechatAPIClient
class BeautyLegPlugin(MessagePluginInterface):
"""美腿图片插件"""
# 功能权限常量
FEATURE_KEY = "BEAUTY_LEG"
FEATURE_DESCRIPTION = "👠 美腿图片功能 [美腿, 腿来]"
@property
def name(self) -> str:
return "美腿图片"
@@ -39,8 +43,18 @@ class BeautyLegPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 注册功能权限
self.feature = self.register_feature()
# 修改图片目录路径指向 resource/beauty_leg
self.image_folder = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))),
"resource", "beauty_leg")
@@ -90,7 +104,7 @@ class BeautyLegPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="美腿图片")
@plugin_points_cost(2, "美腿图片消耗积分", Feature.BEAUTY_LEG)
@plugin_points_cost(2, "美腿图片消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
@@ -101,7 +115,7 @@ class BeautyLegPlugin(MessagePluginInterface):
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.BEAUTY_LEG) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
try:

22
plugins/dify/main.py
View File

@@ -29,6 +29,10 @@ VIDEO_EXTENSIONS = {'.mp4', '.avi', '.mov', '.wmv', '.flv', '.mkv', '.webm'}
class DifyPlugin(MessagePluginInterface):
"""Dify AI聊天插件"""
# 功能权限常量
FEATURE_KEY = "AI_CAPABILITY"
FEATURE_DESCRIPTION = "🤖 AI聊天功能 [ai, dify, 聊天, AI]"
@property
def name(self) -> str:
return "Dify聊天"
@@ -53,6 +57,14 @@ class DifyPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 会话上下文管理，格式: {group_id/wxid: [conversation_history]}
@@ -64,6 +76,8 @@ class DifyPlugin(MessagePluginInterface):
# 会话过期时间（秒）
self.conversation_timeout = 3600 # 1小时
self.last_activity: Dict[str, float] = {}
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -120,7 +134,7 @@ class DifyPlugin(MessagePluginInterface):
return False
@plugin_stats_decorator(plugin_name="Dify聊天")
@plugin_points_cost(2, "AI聊天消耗积分", Feature.AI_CAPABILITY)
@plugin_points_cost(2, "AI聊天消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
@@ -135,7 +149,7 @@ class DifyPlugin(MessagePluginInterface):
target = roomid if roomid else sender
# 检查权限
if roomid and gbm.get_group_permission(target, Feature.AI_CAPABILITY) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(target, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 处理被@的消息
@@ -184,8 +198,8 @@ class DifyPlugin(MessagePluginInterface):
revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5)
return False, f"处理出错: {e}"
async def _send_response(self, bot: WechatAPIClient, target: str, sender: str,
response: str, roomid: str) -> Tuple[bool, str]:
async def _send_response(self, bot: WechatAPIClient, target: str, sender: str,
response: str, roomid: str) -> Tuple[bool, str]:
"""发送响应消息的辅助方法"""
try:
# 判断是否为本地文件路径

16
plugins/douyin_parser/main.py
View File

@@ -22,6 +22,10 @@ class DouyinParserError(Exception):
class DouyinParserPlugin(MessagePluginInterface):
"""抖音无水印解析插件"""
# 功能权限常量
FEATURE_KEY = "DOUYIN_PARSER"
FEATURE_DESCRIPTION = "🎵 抖音解析功能 [自动解析抖音链接]"
@property
def name(self) -> str:
return "抖音解析"
@@ -46,10 +50,20 @@ class DouyinParserPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return [] # 不使用命令触发，而是通过消息内容匹配
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.LOG = logger
self.url_pattern = re.compile(r'https?://v\.douyin\.com/\w+/?')
# 注册功能权限
self.feature = self.register_feature()
# 修改为使用插件目录下的down_load_dir文件夹
self.download_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "down_load_dir")
# 确保下载目录存在
@@ -105,7 +119,7 @@ class DouyinParserPlugin(MessagePluginInterface):
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.DOUYIN_PARSER) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
try:

16
plugins/game_task/main.py
View File

@@ -19,6 +19,18 @@ import json
class GameTaskPlugin(MessagePluginInterface):
"""游戏任务插件"""
# 功能权限常量
FEATURE_KEY = "TASK_GAME"
FEATURE_DESCRIPTION = "🎮 百科问答功能 [/t, /a, /s, /r, /l, /h]"
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
@property
def name(self) -> str:
return "百科问答"
@@ -46,6 +58,8 @@ class GameTaskPlugin(MessagePluginInterface):
def __init__(self):
super().__init__()
self.LOG = logger
# 注册功能权限
self.feature = self.register_feature()
async_job.at_times(["17:58"])(self.run_random_task_assignment)
def initialize(self, context: Dict[str, Any]) -> bool:
@@ -123,7 +137,7 @@ class GameTaskPlugin(MessagePluginInterface):
self.LOG.debug(f"插件执行： {self.name}：{content}")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.TASK_GAME) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
try:

28
plugins/global_news/main.py
View File

@@ -18,11 +18,15 @@ from .news_crawler import nbc, cnn, abc, fox, bbc
class GlobalNewsPlugin(MessagePluginInterface):
"""全球政治经济新闻插件"""
"""全球新闻插件"""
# 功能权限常量
FEATURE_KEY = "GLOBAL_NEWS"
FEATURE_DESCRIPTION = "🌍 全球新闻功能 [全球新闻, 国际新闻, 环球新闻, 政经新闻]"
@property
def name(self) -> str:
return "全球政治经济新闻"
return "全球新闻"
@property
def version(self) -> str:
@@ -30,7 +34,7 @@ class GlobalNewsPlugin(MessagePluginInterface):
@property
def description(self) -> str:
return "提供全球政治经济新闻，支持多个国际新闻源"
return "提供全球政治经济新闻查询功能"
@property
def author(self) -> str:
@@ -44,11 +48,21 @@ class GlobalNewsPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.bot: WechatAPIClient = None
self._news_tasks = {} # 存储正在进行的新闻抓取任务
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -89,8 +103,8 @@ class GlobalNewsPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="全球政治经济新闻")
@plugin_points_cost(5, "全球新闻消耗积分", Feature.NEWS)
@plugin_stats_decorator(plugin_name="全球新闻")
@plugin_points_cost(1, "全球新闻消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
@@ -98,10 +112,10 @@ class GlobalNewsPlugin(MessagePluginInterface):
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
self.bot: WechatAPIClient = message.get("bot")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.NEWS) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 生成唯一任务ID

19
plugins/group_auto_invite/main.py
View File

@@ -5,7 +5,7 @@ from typing import Dict, Any, List, Optional, Tuple
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.robot_cmd.robot_command import GroupBotManager
from utils.robot_cmd.robot_command import GroupBotManager, PermissionStatus
from utils.wechat.contact_manager import ContactManager
from wechat_ipad import WechatAPIClient
@@ -13,6 +13,10 @@ from wechat_ipad import WechatAPIClient
class GroupAutoInvitePlugin(MessagePluginInterface):
"""自动加群功能插件"""
# 功能权限常量
FEATURE_KEY = "GROUP_AUTO_INVITE"
FEATURE_DESCRIPTION = "🤝 自动加群功能 [#加群配置, #加群]"
@property
def name(self) -> str:
return "自动加群功能"
@@ -37,12 +41,22 @@ class GroupAutoInvitePlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# Redis 中存储群组映射的前缀
self.mapping_prefix = "group:group_mapping:"
self._commands = []
self.bot: WechatAPIClient = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -86,6 +100,9 @@ class GroupAutoInvitePlugin(MessagePluginInterface):
content = str(message.get("content", "")).strip()
roomid = message.get("roomid", "")
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False
# 处理加群配置命令
if content.startswith("#加群配置|"):
return True

26
plugins/group_member_change/main.py
View File

@@ -16,6 +16,10 @@ from wechat_ipad.models.appmsg_xml import LINK_XML
class GroupMemberChangePlugin(MessagePluginInterface):
"""群成员变更监控插件"""
# 功能权限常量
FEATURE_KEY = "GROUP_MEMBER_CHANGE"
FEATURE_DESCRIPTION = "👥 群成员变更监控 [自动监控群成员变动并发送通知]"
@property
def name(self) -> str:
return "群成员变更监控"
@@ -46,6 +50,8 @@ class GroupMemberChangePlugin(MessagePluginInterface):
def __init__(self):
super().__init__()
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -56,9 +62,15 @@ class GroupMemberChangePlugin(MessagePluginInterface):
def can_process(self, message: Dict[str, Any]) -> bool:
"""检查是否可以处理该消息"""
content = message.get("content")
if not content or "<sysmsg" not in content:
if not self.enable:
return False
content = str(message.get("content", "")).strip()
roomid = message.get("roomid", "")
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False
return True
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
@@ -74,7 +86,7 @@ class GroupMemberChangePlugin(MessagePluginInterface):
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.GROUP_MEMBER_CHANGE) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
xml_content = str(content).strip().replace("\n", "").replace("\t", "")
@@ -146,6 +158,14 @@ class GroupMemberChangePlugin(MessagePluginInterface):
"""插件支持的命令列表"""
return []
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def get_help(self) -> str:
"""获取插件帮助信息"""
return "群成员变更监控插件：自动监控群成员变动并发送通知。"

22
plugins/group_virtual/main.py
View File

@@ -2,6 +2,7 @@ from typing import Dict, Any, List, Optional, Tuple
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.robot_cmd.robot_command import GroupBotManager, PermissionStatus
from utils.wechat.contact_manager import ContactManager
from db.connection import DBConnectionManager
from db.group_virtual_redis import GroupVirtualRedisDB
@@ -12,6 +13,10 @@ from wechat_ipad.models.message import WxMessage
class GroupVirtualPlugin(MessagePluginInterface):
"""跨群聊天插件"""
# 功能权限常量
FEATURE_KEY = "GROUP_VIRTUAL"
FEATURE_DESCRIPTION = "🔄 跨群聊天功能 [自动转发群组间消息]"
@property
def name(self) -> str:
return "跨群聊天"
@@ -36,11 +41,21 @@ class GroupVirtualPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return [] # 不处理任何命令
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.data = None
self.message_cache = set() # 用于防止消息循环转发
self.group_virtual_redis = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -79,8 +94,13 @@ class GroupVirtualPlugin(MessagePluginInterface):
if not self.enable:
return False
# 只处理群消息
content = str(message.get("content", "")).strip()
roomid = message.get("roomid", "")
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False
# 只处理群消息
if not roomid:
return False

36
plugins/guess_song/main.py
View File

@@ -76,6 +76,10 @@ class GuessSongRedisDB:
class GuessSongPlugin(MessagePluginInterface):
"""猜歌名游戏插件"""
# 功能权限常量
FEATURE_KEY = "GUESS_MUSIC"
FEATURE_DESCRIPTION = "🎵 猜歌名游戏 [猜歌名]"
@property
def name(self) -> str:
return "猜歌名游戏"
@@ -100,8 +104,17 @@ class GuessSongPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.feature = self.register_feature()
self.redis_db = None
def initialize(self, context: Dict[str, Any]) -> bool:
@@ -166,35 +179,24 @@ class GuessSongPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="猜歌名游戏")
@plugin_stats_decorator(plugin_name="猜歌名")
@plugin_points_cost(2, "猜歌名消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}")
# 简化命令处理，只检查开头是否包含命令前缀
command_found = False
for cmd in self._commands:
if content.startswith(cmd):
command_found = True
content = content[len(cmd):].strip()
break
if not command_found:
return False, "不匹配的命令"
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 使用roomid或sender作为游戏会话ID
session_id = roomid if roomid else sender
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.GUESS_MUSIC) == PermissionStatus.DISABLED:
return False, "没有权限"
# 获取当前游戏会话
current_game = None
if self.redis_db:

21
plugins/jd_sign_token/main.py
View File

@@ -151,6 +151,10 @@ class QL:
class JDTokenPlugin(MessagePluginInterface):
"""京东签到Token设置插件"""
# 功能权限常量
FEATURE_KEY = "JD_TOKEN"
FEATURE_DESCRIPTION = "🔑 京东签到Token设置 [设置京东]"
@property
def name(self) -> str:
@@ -176,8 +180,18 @@ class JDTokenPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -221,9 +235,12 @@ class JDTokenPlugin(MessagePluginInterface):
return False
content = str(message.get("content", "")).strip()
command = content.split(" ")[0]
roomid = message.get("roomid", "")
return command in self._commands
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False
return True
@plugin_stats_decorator(plugin_name="京东签到Token设置")
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:

10
plugins/message_sign/main.py
View File

@@ -45,6 +45,14 @@ class MessageSignPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return "SIGN_IN"
@property
def feature_description(self) -> Optional[str]:
return "📝 签到功能 [签到]"
def __init__(self):
super().__init__()
self.today_signin_count = {}
@@ -57,6 +65,8 @@ class MessageSignPlugin(MessagePluginInterface):
"resource", "6 托福-乱序.txt")
self.vocab_list = []
self.bot: WechatAPIClient = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""

33
plugins/message_summary/main.py
View File

@@ -24,6 +24,10 @@ from wechat_ipad import WechatAPIClient
class MessageSummaryPlugin(MessagePluginInterface):
"""消息总结插件，用于生成群聊消息总结"""
# 功能权限常量
FEATURE_KEY = "SUMMARY_CAPABILITY"
FEATURE_DESCRIPTION = "📝 群聊总结功能 [总结]"
@property
def name(self) -> str:
return "群聊总结"
@@ -46,12 +50,24 @@ class MessageSummaryPlugin(MessagePluginInterface):
@property
def commands(self) -> List[str]:
return ["总结", "summary"]
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.bot: WechatAPIClient = None
self.revoke: MessageAutoRevoke = None
self._commands = []
self.message_storage = None
self.revoke = None
self.bot = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -242,3 +258,14 @@ class MessageSummaryPlugin(MessagePluginInterface):
except Exception as e:
self.LOG.error(f"处理总结时出现未知错误: {e}")
return f"生成总结时出现未知错误: {str(e)}", None
def can_process(self, message: Dict[str, Any]) -> bool:
"""检查是否可以处理该消息"""
if not self.enable:
return False
content = str(message.get("content", "")).strip()
roomid = message.get("roomid", "")
if GroupBotManager.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False

27
plugins/music/main.py
View File

@@ -19,6 +19,10 @@ from wechat_ipad.models.appmsg_xml import MUSIC_XML
class MusicPlugin(MessagePluginInterface):
"""音乐点播插件"""
# 功能权限常量
FEATURE_KEY = "MUSIC"
FEATURE_DESCRIPTION = "🎵 点歌功能 [点歌, 音乐, 音乐点播, 点播音乐, 音乐点歌]"
@property
def name(self) -> str:
return "音乐点播"
@@ -43,8 +47,18 @@ class MusicPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -83,28 +97,27 @@ class MusicPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="音乐点播")
@plugin_points_cost(2, "音乐点播消耗积分", Feature.MUSIC)
@plugin_stats_decorator(plugin_name="点歌")
@plugin_points_cost(2, "点歌消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}")
command = content.split(" ")[0]
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 检查命令格式
if len(content.split(" ")) == 1:
await bot.send_text_message((roomid if roomid else sender), f"❌命令格式错误！\n{self.command_format}"
, sender)
return False, "命令格式错误"
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.MUSIC) == PermissionStatus.DISABLED:
return False, "没有权限"
# 提取歌曲名
user_song_name = content[len(command):].strip()

16
plugins/plugin_manager/main.py
View File

@@ -37,6 +37,14 @@ class PluginManagerPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return "PLUGIN_MANAGER"
@property
def feature_description(self) -> Optional[str]:
return "🔧 插件管理功能 [插件管理]"
def start(self) -> bool:
"""启动插件"""
self.LOG.info(f"[{self.name}] 插件已启动")
@@ -52,6 +60,8 @@ class PluginManagerPlugin(MessagePluginInterface):
def __init__(self):
super().__init__()
self.bot: WechatAPIClient = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -93,6 +103,10 @@ class PluginManagerPlugin(MessagePluginInterface):
self.LOG.error("WechatAPIClient 未初始化")
return False, "Bot 未初始化"
# 检查功能权限
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 检查命令格式
parts = content.split(" ")
if len(parts) == 1:
@@ -103,7 +117,7 @@ class PluginManagerPlugin(MessagePluginInterface):
self.plugin_registry = PluginRegistry()
self.plugin_manager = PluginManager().get_instance()
# 检查权限 (只允许管理员操作)
# 检查管理员权限
if not self._is_admin(sender, gbm):
await self.bot.send_at_message(target, f"❌权限不足，只有管理员可以管理插件", [sender])
return True, "权限不足"

14
plugins/point_trade/main.py
View File

@@ -21,6 +21,10 @@ from wechat_ipad.models.message import WxMessage
class PointTradePlugin(MessagePluginInterface):
"""积分交易插件"""
# 功能权限常量
FEATURE_KEY = "POINT_TRADE"
FEATURE_DESCRIPTION = "💱 积分交易功能 [积分交易, 积分转账, 我的积分, 积分排行, 打劫, 保释]"
@property
def name(self) -> str:
return "积分交易"
@@ -45,11 +49,21 @@ class PointTradePlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.db_pool = None
self.bot: WechatAPIClient = None
self.revoke: MessageAutoRevoke = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""

25
plugins/robot_menu/main.py
View File

@@ -4,7 +4,7 @@ from typing import Dict, Any, List, Optional, Tuple
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.plugin_decorators import plugin_stats_decorator, plugin_points_cost
from utils.revoke.message_auto_revoke import MessageAutoRevoke
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from wechat_ipad import WechatAPIClient
@@ -12,6 +12,10 @@ from wechat_ipad import WechatAPIClient
class RobotMenuPlugin(MessagePluginInterface):
"""功能菜单插件"""
# 功能权限常量
FEATURE_KEY = "ROBOT_MENU"
FEATURE_DESCRIPTION = "📋 功能菜单 [菜单, 功能]"
@property
def name(self) -> str:
@@ -37,8 +41,18 @@ class RobotMenuPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -133,17 +147,22 @@ class RobotMenuPlugin(MessagePluginInterface):
return "当前没有启用机器人的群组"
return "\n".join(group_list)
@plugin_stats_decorator(plugin_name="功能菜单")
@plugin_stats_decorator(plugin_name="机器人菜单")
@plugin_points_cost(1, "机器人菜单消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}")
command = content.split(" ")[0]
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
revoke: MessageAutoRevoke = message.get("revoke")
# 检查权限
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 检查命令格式
if len(content.split(" ")) == 1:
# 显示功能菜单

17
plugins/stats_collector/main.py
View File

@@ -13,6 +13,18 @@ from db.connection import DBConnectionManager
class StatsCollectorPlugin(PluginInterface):
"""统计收集插件"""
# 功能权限常量
FEATURE_KEY = "STATS_COLLECTOR"
FEATURE_DESCRIPTION = "📊 指令记录功能 [自动记录指令使用情况]"
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
@property
def name(self) -> str:
return "指令记录"
@@ -38,10 +50,8 @@ class StatsCollectorPlugin(PluginInterface):
return []
def __init__(self):
super().__init__()
self.LOG = logger
self.LOG.info(f"正在初始化 {self.name} 插件...")
# 默认配置
self.config = {
@@ -49,10 +59,11 @@ class StatsCollectorPlugin(PluginInterface):
"record_all_plugins": True, # 是否记录所有插件的调用
"excluded_plugins": [], # 排除的插件列表
}
self.event_manager = EventManager.get_instance()
self.db_manager = DBConnectionManager.get_instance()
self.stats_db = StatsDBOperator(self.db_manager)
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, config: Dict[str, Any]) -> bool:
"""初始化插件"""

65
plugins/system_updater/main.py
View File

@@ -5,13 +5,20 @@ from typing import Dict, Any, List, Optional, Tuple
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.robot_cmd.robot_command import Feature, PermissionStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.points_decorator import plugin_points_cost
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from wechat_ipad import WechatAPIClient
class SystemUpdaterPlugin(MessagePluginInterface):
"""系统更新插件"""
# 功能权限常量
FEATURE_KEY = "SYSTEM_UPDATER"
FEATURE_DESCRIPTION = "🔄 系统更新功能 [更新系统, 系统更新]"
@property
def name(self) -> str:
return "系统更新"
@@ -36,11 +43,21 @@ class SystemUpdaterPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.admin_wxids = []
self.wait_time = 5 # 默认等待15秒
self.bot: WechatAPIClient = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -87,47 +104,31 @@ class SystemUpdaterPlugin(MessagePluginInterface):
return False
@plugin_stats_decorator(plugin_name="系统更新")
@plugin_points_cost(1, "系统更新消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}")
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm = message.get("gbm", None)
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
self.bot: WechatAPIClient = message.get("bot")
# 检查权限
if self.admin_wxids and sender not in self.admin_wxids:
await self.bot.send_text_message((roomid if roomid else sender), "⚠️ 您没有执行此操作的权限",
sender)
return True, "无权限"
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 如果是群消息，检查群权限
if roomid and gbm and hasattr(gbm, 'get_group_permission'):
if gbm.get_group_permission(roomid, Feature.ROBOT) == PermissionStatus.DISABLED:
return False, "机器人功能已禁用"
# 检查是否是命令
if content in self._commands:
return True, "命令已接收"
# 提取等待时间参数
wait_time = self.wait_time
command = content.split(" ")[0]
# 检查是否是带参数的更新命令
for cmd in self._commands:
if content.startswith(f"{cmd} "):
return True, "带参数的命令已接收"
if len(content.split(" ")) > 1:
try:
param = content.split(" ")[1].strip()
if param.isdigit():
wait_time = int(param)
self.LOG.info(f"使用自定义等待时间: {wait_time}秒")
except:
pass
# 发送更新通知
await self.bot.send_text_message((roomid if roomid else sender),
f"🔄 系统即将更新并重启，等待时间设置为{wait_time}秒...",
sender)
# 执行系统更新脚本
self._execute_system_update()
return True, "系统更新中"
return False, "无法处理的消息"
def _execute_system_update(self):
"""执行系统更新操作"""

12
plugins/video/main.py
View File

@@ -18,6 +18,18 @@ from wechat_ipad import WechatAPIClient
class VideoPlugin(MessagePluginInterface):
"""视频插件"""
# 功能权限常量
FEATURE_KEY = "VIDEO"
FEATURE_DESCRIPTION = "🎥 黑丝视频功能 [黑丝视频, 黑丝, 来个黑丝, 搞个黑丝]"
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
@property
def name(self) -> str:
return "黑丝视频"

24
plugins/video_man/main.py
View File

@@ -13,11 +13,16 @@ from base.plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.points_decorator import plugin_points_cost
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from base.wechat_api.wechat_api_client import WechatAPIClient
class VideoManPlugin(MessagePluginInterface):
"""猛男视频插件"""
# 功能权限常量
FEATURE_KEY = "VIDEO_MAN"
FEATURE_DESCRIPTION = "💪 猛男视频功能 [猛男视频, 猛男, 来个猛男, 搞个猛男]"
@property
def name(self) -> str:
return "猛男视频"
@@ -42,10 +47,20 @@ class VideoManPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 使用插件目录下的down_load_dir文件夹
self.download_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "down_load_dir")
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -90,7 +105,7 @@ class VideoManPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="猛男视频")
@plugin_points_cost(2, "猛男视频消耗积分", Feature.VIDEO_MAN)
@plugin_points_cost(2, "猛男视频消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
@@ -98,9 +113,10 @@ class VideoManPlugin(MessagePluginInterface):
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.VIDEO_MAN) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
try:
@@ -108,12 +124,12 @@ class VideoManPlugin(MessagePluginInterface):
file_abspath, first_frame = await self._download_video("https://api.52vmy.cn/api/video/boy?type=json")
# FIXME 需要换成web容器地址。否则无法获取。
if not file_abspath:
await self.bot.send_text_message((roomid if roomid else sender), f"\n❌视频下载失败，请稍后再试",
await bot.send_text_message((roomid if roomid else sender), f"\n❌视频下载失败，请稍后再试",
sender)
return False, "视频下载失败"
# 发送视频
result = await self.bot.send_video_message((roomid if roomid else sender), Path(file_abspath),
result = await bot.send_video_message((roomid if roomid else sender), Path(file_abspath),
Path(first_frame))
self.LOG.info(f"发送视频结果: {result}")
return True, "发送成功"

19
plugins/weather/main.py
View File

@@ -14,6 +14,10 @@ from wechat_ipad import WechatAPIClient
class WeatherPlugin(MessagePluginInterface):
"""天气查询插件"""
# 功能权限常量
FEATURE_KEY = "WEATHER"
FEATURE_DESCRIPTION = "🌤️ 天气查询功能 [天气]"
@property
def name(self) -> str:
return "天气查询"
@@ -38,9 +42,19 @@ class WeatherPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.plugin_dir = os.path.dirname(os.path.abspath(__file__))
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -88,17 +102,18 @@ class WeatherPlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="天气查询")
@plugin_points_cost(2, "天气查询消耗积分", Feature.WEATHER)
@plugin_points_cost(1, "天气查询消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}")
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and GroupBotManager.get_group_permission(roomid, Feature.WEATHER) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 处理消息内容 - 不再使用jieba分词

20
plugins/xiuren_image/main.py
View File

@@ -18,6 +18,10 @@ from wechat_ipad import WechatAPIClient
class XiurenImagePlugin(MessagePluginInterface):
"""秀人图片插件"""
# 功能权限常量
FEATURE_KEY = "XIUREN_IMAGE"
FEATURE_DESCRIPTION = "🖼️ 秀人图片功能 [秀人]"
@property
def name(self) -> str:
return "秀人图片"
@@ -42,10 +46,20 @@ class XiurenImagePlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
# 使用Path对象处理路径，自动适应不同操作系统
self.image_folder = str(Path(Path(__file__).parent.parent.parent, "xiuren"))
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -97,8 +111,8 @@ class XiurenImagePlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="秀人图片")
@plugin_points_cost(9, "秀人图片消耗积分", Feature.PIC)
@plugin_stats_decorator(plugin_name="秀人网图片")
@plugin_points_cost(2, "秀人网图片消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
@@ -109,7 +123,7 @@ class XiurenImagePlugin(MessagePluginInterface):
bot: WechatAPIClient = message.get("bot")
revoke: MessageAutoRevoke = message.get("revoke")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.PIC) == PermissionStatus.DISABLED:
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
try: