AgentsMeeting/xmpp_zhiwei_bot.py

#!/usr/bin/env python3
"""XMPP Bot zhiwei@yoin.fun - Hermes API 版（修复版：ejabberd 26.4 兼容 + 稳定重连 + 群聊配置化）"""
import asyncio, logging, ssl, json, urllib.request, os, subprocess, time
from xml.sax.saxutils import escape

logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')
GATEWAY = "http://localhost:8643/v1/chat/completions"
API_KEY = "hermes123"
_opener = urllib.request.build_opener(urllib.request.ProxyHandler({}))

# 读取需要加入的MUC群聊房间列表
MUC_ROOMS_FILE = "/home/hmo/.hermes/zhiwei_rooms.txt"
MUC_ROOMS = []
if os.path.exists(MUC_ROOMS_FILE):
    with open(MUC_ROOMS_FILE) as f:
        MUC_ROOMS = [line.strip() for line in f if line.strip() and not line.strip().startswith('#')]
    logging.info(f"📋 知微待加入群聊: {MUC_ROOMS}")
else:
    logging.warning(f"⚠️ 群聊配置文件 {MUC_ROOMS_FILE} 不存在")

def send(from_jid, to_jid, body):
    safe = escape(body)
    subprocess.run(["docker","exec","ejabberd","ejabberdctl","send_message","chat",
        from_jid, to_jid, "", safe
    ], capture_output=True, timeout=10)

class ZhiweiBot:
    def __init__(self):
        import slixmpp
        self.xmpp = slixmpp.ClientXMPP('zhiwei@yoin.fun', 'hermes123')
        self.xmpp.add_event_handler('session_bind', self.on_bind)
        self.xmpp.add_event_handler('message', self.on_msg)
        self.xmpp.add_event_handler('disconnected', self.on_disconnect)
        self.xmpp.add_event_handler('connected', self.on_connected)
        # 不用 slixmpp 内置自动重连（不可靠），用手动重连
        self.xmpp.auto_reconnect = False
        ctx = ssl.create_default_context(); ctx.check_hostname = False; ctx.verify_mode = ssl.CERT_NONE
        self.xmpp.ssl_context = ctx
        self.ready = asyncio.Event()
        self._stopped = False
        self._call_seq = 0

    def stop(self):
        self._stopped = True
        self.xmpp.abort()

    async def on_connected(self, event):
        logging.info("🔗 知微TCP连接已建立")

    async def on_bind(self, event):
        self.xmpp.send_presence(); self.xmpp.get_roster()
        # 加入所有配置的MUC群聊
        for room in MUC_ROOMS:
            try:
                nick = room.split('@')[0]
                self.xmpp.plugin['xep_0045'].join_muc(room, nick)
                logging.info(f"✅ 知微加入群聊 {room}")
            except Exception as e:
                logging.warning(f"⚠️ 加入群聊 {room} 失败: {e}")
        self.ready.set()
        logging.info("✅ 知微上线")

    async def on_disconnect(self, event):
        self.ready.clear()
        logging.warning("⚠️ 知微断线")

    async def on_msg(self, msg):
        body = msg['body']; sender = str(msg['from']); mtype = msg['type']
        if not body:
            return

        # 群聊处理
        if mtype == 'groupchat':
            nickname = sender.split('/')[-1] if '/' in sender else ''
            if nickname in ('hmo', 'xxm'):
                room = sender.split('/')[0]
                logging.info(f"📩 知微群消息 [{room}/{nickname}]: {body[:80]}")
                self._call_seq += 1
                try:
                    payload = json.dumps({
                        "model":"hermes-agent",
                        "messages":[{"role":"user","content":f"[{room} {nickname}] {body}"}]
                    }).encode()
                    req = urllib.request.Request(GATEWAY, data=payload, method="POST")
                    req.add_header("Content-Type","application/json")
                    req.add_header("Authorization",f"Bearer {API_KEY}")
                    req.add_header("X-Hermes-Session-Id","xmpp-zhiwei")
                    loop = asyncio.get_event_loop()
                    result = await loop.run_in_executor(None, lambda: _opener.open(req, timeout=120))
                    data = json.loads(result.read())
                    reply = data.get("choices",[{}])[0].get("message",{}).get("content","")
                    finish = data.get("choices",[{}])[0].get("finish_reason","")
                    if reply.strip() and finish != "silent":
                        self.xmpp.send_message(mto=room, mbody=reply, mtype='groupchat')
                        logging.info(f"✅ 知微群回复: {reply[:80]}")
                except Exception as e:
                    logging.error(f"❌ 知微群错误: {e}")
            return

        # 私聊处理
        if mtype != 'chat': return
        if 'hmo@yoin.fun' in sender:
            self._call_seq += 1
            logging.info(f"📩 老爸(#{self._call_seq}): {body}")
            # 先秒回确认消息，让老爸知道bot活着
            send("zhiwei@yoin.fun", sender, "收到，正在思考...")
            try:
                payload = json.dumps({
                    "model":"hermes-agent",
                    "messages":[{"role":"user","content":f"[zhiwei] {body}"}]
                }).encode()
                req = urllib.request.Request(GATEWAY, data=payload, method="POST")
                req.add_header("Content-Type","application/json")
                req.add_header("Authorization",f"Bearer {API_KEY}")
                req.add_header("X-Hermes-Session-Id","xmpp-zhiwei")
                loop = asyncio.get_event_loop()
                result = await loop.run_in_executor(None, lambda: _opener.open(req, timeout=120))
                data = json.loads(result.read())
                reply = data.get("choices",[{}])[0].get("message",{}).get("content","")
                finish = data.get("choices",[{}])[0].get("finish_reason","")
                if reply.strip() and finish != "silent":
                    send("zhiwei@yoin.fun", sender, reply)
                    logging.info(f"✅ 知微回复: {reply[:80]}")
            except Exception as e:
                logging.error(f"❌ 知微错误: {e}")

async def main():
    retry_delay = 1
    max_delay = 60
    while True:
        try:
            z = ZhiweiBot()
            z.xmpp.register_plugin('xep_0030'); z.xmpp.register_plugin('xep_0199')
            z.xmpp.register_plugin('xep_0045')  # MUC群聊支持
            # XMPP Ping 每 60 秒发一次 keepalive，防服务器超时断线
            z.xmpp['xep_0199'].keepalive = True
            z.xmpp['xep_0199'].interval = 60
            z.xmpp['xep_0199'].timeout = 10
            z.xmpp.connect(host='127.0.0.1', port=5222)
            await asyncio.wait_for(z.ready.wait(), timeout=30)
            logging.info("知微就绪")
            retry_delay = 1

            # 保持运行，断线时自动跳出重连
            while not z._stopped:
                await asyncio.sleep(3)
                if not z.xmpp.is_connected():
                    logging.warning("知微连接丢失，重连中...")
                    break

        except asyncio.TimeoutError:
            logging.error("⏰ 知微连接超时")
        except Exception as e:
            logging.error(f"知微main错误: {e}")
        finally:
            try:
                if 'z' in dir() and z:
                    z.stop()
            except:
                pass

        # 指数退避重连
        logging.info(f"⏳ 知微等待 {retry_delay} 秒后重连...")
        await asyncio.sleep(retry_delay)
        retry_delay = min(retry_delay * 2, max_delay)

if __name__ == '__main__':
    import sys, os, signal

    # PID 文件防多进程：启动时检查，如果已有实例则杀旧启新
    PIDFILE = '/home/hmo/.hermes/zhiwei_bot.pid'

    if os.path.exists(PIDFILE):
        try:
            old_pid = int(open(PIDFILE).read().strip())
            os.kill(old_pid, signal.SIGTERM)
            print(f"Killed old bot process {old_pid}")
            import time
            time.sleep(2)
        except (ValueError, ProcessLookupError, OSError):
            pass

    with open(PIDFILE, 'w') as f:
        f.write(str(os.getpid()))

    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        pass
    finally:
        if os.path.exists(PIDFILE):
            os.remove(PIDFILE)