From a7d0d400f9eefc66da4044e85048effe283576d8 Mon Sep 17 00:00:00 2001
From: liuwei <liuwei@wdtrgf.com.cn>
Date: Thu, 20 Mar 2025 14:23:38 +0800
Subject: [PATCH] =?UTF-8?q?feature=EF=BC=9A1.=E5=8A=A0=E7=BE=A4=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD=E6=8F=92=E4=BB=B6=E5=8C=96=EF=BC=9B2.=E6=88=90?=
 =?UTF-8?q?=E5=91=98=E5=8F=98=E6=9B=B4=E4=B8=8D=E7=9B=91=E5=90=AC=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=EF=BC=8C=E4=BD=BF=E7=94=A8=E5=AE=9A=E6=97=B6=E6=A3=80?=
 =?UTF-8?q?=E6=9F=A5=E7=AD=96=E7=95=A5=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins/group_add/__init__.py |  7 +++
 plugins/group_add/config.toml |  2 +
 plugins/group_add/main.py     | 99 +++++++++++++++++++++++++++++++++++
 3 files changed, 108 insertions(+)
 create mode 100644 plugins/group_add/__init__.py
 create mode 100644 plugins/group_add/config.toml
 create mode 100644 plugins/group_add/main.py

diff --git a/plugins/group_add/__init__.py b/plugins/group_add/__init__.py
new file mode 100644
index 0000000..a8b1029
--- /dev/null
+++ b/plugins/group_add/__init__.py
@@ -0,0 +1,7 @@
+# 从当前包的main模块导入GroupAddPlugin类
+from .main import GroupAddPlugin
+
+# 提供get_plugin函数，返回插件实例
+def get_plugin():
+    """获取插件实例"""
+    return GroupAddPlugin()
\ No newline at end of file
diff --git a/plugins/group_add/config.toml b/plugins/group_add/config.toml
new file mode 100644
index 0000000..639e81a
--- /dev/null
+++ b/plugins/group_add/config.toml
@@ -0,0 +1,2 @@
+[GroupAdd]
+enable = false
\ No newline at end of file
diff --git a/plugins/group_add/main.py b/plugins/group_add/main.py
new file mode 100644
index 0000000..5821996
--- /dev/null
+++ b/plugins/group_add/main.py
@@ -0,0 +1,99 @@
+import logging
+import re
+from datetime import datetime
+from typing import Dict, Any, List, Optional
+
+from wcferry import WxMsg
+
+from plugin_common.message_plugin_interface import MessagePluginInterface
+from plugin_common.plugin_interface import PluginStatus
+from robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
+
+
+class GroupAddPlugin(MessagePluginInterface):
+    """群成员加入欢迎插件"""
+
+    def __init__(self):
+        super().__init__()
+        self.name = "群成员加入欢迎"
+        self.description = "当有新成员加入群聊时，自动发送欢迎消息"
+        self.version = "1.0.0"
+        self.author = "Trae AI"
+        
+        self.wcf = None
+        self.gbm = None
+        self.LOG = logging.getLogger(f"Plugin.{self.name}")
+        self.enable = True
+
+    def initialize(self, context: Dict[str, Any]) -> bool:
+        """初始化插件"""
+        self.LOG.info(f"正在初始化 {self.name} 插件...")
+        
+        # 保存上下文对象
+        self.wcf = context.get("wcf")
+        if not self.wcf:
+            self.LOG.error("无法获取wcf对象，插件初始化失败")
+            return False
+            
+        # 获取群管理器
+        self.gbm = context.get("gbm")
+        if not self.gbm:
+            self.LOG.error("无法获取群管理器对象，插件初始化失败")
+            return False
+            
+        # 从配置中获取启用状态
+        plugin_config = self._config.get("GroupAdd", {})
+        self.enable = plugin_config.get("enable", True)
+        
+        self.LOG.info(f"{self.name} 插件初始化完成，启用状态: {self.enable}")
+        return True
+
+    def on_receive_message(self, msg: WxMsg) -> bool:
+        """处理接收到的消息"""
+        if not self.enable:
+            return False
+            
+        # 只处理群系统消息
+        if msg.type != 10000 or not msg.from_group():
+            return False
+            
+        # 检查群权限
+        if self.gbm.get_group_permission(msg.roomid, Feature.GROUP_ADD) == PermissionStatus.DISABLED:
+            return False
+            
+        # 使用正则表达式提取双引号中的内容，判断是否为加入群聊消息
+        match = re.search(r'"(.*?)"', msg.content)
+        if not match:
+            return False
+            
+        # 获取当前时间
+        now_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+        
+        # 提取昵称并发送欢迎消息
+        nickname = match.group(1)
+        welcome_message = f"🎉 欢迎 【{nickname}】 加入群聊👋 \n 🕒 {now_time} 🕒 ！"
+        self.wcf.send_text(welcome_message, msg.roomid)
+        
+        self.LOG.info(f"已发送欢迎消息给新成员 {nickname} 在群 {msg.roomid}")
+        return True
+
+    @property
+    def commands(self) -> List[str]:
+        """插件支持的命令列表"""
+        return []
+
+    def get_help(self) -> str:
+        """获取插件帮助信息"""
+        return "群成员加入欢迎插件：当有新成员加入群聊时，自动发送欢迎消息。"
+
+    def start(self) -> bool:
+        """启动插件"""
+        self.enable = True
+        self.LOG.info(f"{self.name} 插件已启动")
+        return True
+
+    def stop(self) -> bool:
+        """停止插件"""
+        self.enable = False
+        self.LOG.info(f"{self.name} 插件已停止")
+        return True
\ No newline at end of file