Merge pull request #45 from Cassius0924/improve-notifier

[Improvement] 优化收到命令提醒

README.md

@@ -29,12 +29,12 @@ docker run -d \
 --name wxBotWebhook \
 -p 3001:3001 \
 -e LOGIN_API_TOKEN="<Token>" \
--e RECVD_MSG_API="http://<内网IP>:<接收消息端口>/receive_msg" \
+-e RECVD_MSG_API="http://<宿主机IP>:<接收消息端口>/receive_msg" \
 dannicool/docker-wechatbot-webhook
 ```
 
-- `<Token>`：登录令牌（不是密码），自己设置一个好记的。
-- `<内网IP>`：填入服务器的**内网IP**。如果是在自己电脑，则填入 `127.0.0.1`。
+- `<Token>`：登录令牌（可选）。
+- `<宿主机IP>`：填入 Docekr 的宿主机地址。
 - `<接收消息端口>`：设置一个接收消息的端口，此项目中默认为 `4000`。
 
 3. 登录微信
@@ -45,7 +45,7 @@ dannicool/docker-wechatbot-webhook
 docker logs -f wxBotWebhook
 ```
 
-### 启动服务器
+### 启动 WeChatter
 
 1. 下载源代码
 

wechatter/app/routers/wechat.py

@@ -10,7 +10,7 @@
 from wechatter.message import MessageHandler
 from wechatter.message_forwarder import MessageForwarder
 from wechatter.models.message import Message
-from wechatter.notifier import Notifier
+from wechatter.sender import notify_logged_in, notify_logged_out
 from wechatter.sqlite.sqlite_manager import SqliteManager
 
 router = APIRouter()
@@ -79,10 +79,10 @@ def handle_system_event(content: str) -> None:
     # 判断是否为机器人登录消息
     if content_dict["event"] == "login":
         print("机器人登录成功")
-        Notifier.notify_logged_in()
+        notify_logged_in()
     elif content_dict["event"] == "logout":
         print("机器人已退出登录")
-        Notifier.notify_logged_out()
+        notify_logged_out()
     elif content_dict["event"] == "error":
         pass
     else:

wechatter/commands/_commands/copilot_gpt4.py

@@ -106,6 +106,7 @@ def _gptx(model: str, to: SendTo, message: str = "") -> None:
     chat_info = CopilotGPT4.get_chating_chat_info(wx_id, model)
     if message == "":  # /gpt4
         # 判断对话是否有效
+        _send_text_msg(to, "正在创建新对话...")
         if chat_info is None or CopilotGPT4.is_chat_valid(chat_info):
             CopilotGPT4.create_chat(wx_id=wx_id, model=model)
             logger.info("创建新对话成功")
@@ -115,6 +116,7 @@ def _gptx(model: str, to: SendTo, message: str = "") -> None:
         _send_text_msg(to, "对话未开始，继续上一次对话")
     else:  # /gpt4 <message>
         # 如果没有对话记录，则创建新对话
+        _send_text_msg(to, f"正在调用 {model} 进行对话...")
         if chat_info is None:
             chat_info = CopilotGPT4.create_chat(wx_id=wx_id, model=model)
             logger.info("无历史对话记录，创建新对话成功")
@@ -159,6 +161,7 @@ def _gptx_continue(model: str, to: SendTo, message: str = "") -> None:
         logger.info("请输入对话记录编号")
         _send_text_msg(to, "请输入对话记录编号")
         return
+    _send_text_msg(to, f"正在切换到对话记录 {message}...")
     chat_info = CopilotGPT4.continue_chat(
         wx_id=wx_id, model=model, chat_index=int(message)
     )

wechatter/message/message_parser.py

@@ -6,7 +6,6 @@
 import wechatter.config as config
 from wechatter.bot.bot_info import BotInfo
 from wechatter.models.message import Message, SendTo
-from wechatter.notifier import Notifier
 
 
 class MessageHandler:
@@ -41,8 +40,6 @@ def handle_message(self, message: Message) -> None:
         to = SendTo(message.source)
 
         # 是命令消息
-        # 回复消息已收到
-        Notifier.notify_received(to)
 
         # 开始处理命令
         cmd_handler = cmd_dict["handler"]

wechatter/sender/__init__.py

@@ -1,3 +1,4 @@
+from .notifier import notify_logged_in, notify_logged_out, notify_received
 from .sender import Sender
 
-__all__ = ["Sender"]
+__all__ = ["Sender", "notify_received", "notify_logged_in", "notify_logged_out"]

wechatter/sender/notifier.py

@@ -0,0 +1,21 @@
+# 消息通知器
+from wechatter.models.message import SendMessage, SendMessageType, SendTo
+from wechatter.sender import Sender
+
+
+def notify_received(to: SendTo) -> None:
+    """通知收到命令请求"""
+    msg = "收到命令请求"
+    Sender.send_msg(to, SendMessage(SendMessageType.TEXT, msg))
+
+
+def notify_logged_in() -> None:
+    """通知登录成功"""
+    msg = "微信机器人启动成功"
+    Sender.send_msg_to_admins(msg)
+
+
+def notify_logged_out() -> None:
+    """通知已退出登录"""
+    msg = "微信机器人已退出"
+    Sender.send_msg_to_admins(msg)