add ws client and update build

Author: yuqi20000915

selfagent/client/ws_client.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+"""
+WebSocket 客户端 - 接收服务器推送并弹出 Windows 通知
+运行: python ws_client.py
+"""
+
+import asyncio
+import websockets
+import json
+from datetime import datetime
+
+# Windows 通知
+try:
+    from win10toast import ToastNotifier
+    toaster = ToastNotifier()
+    HAS_TOAST = True
+except ImportError:
+    HAS_TOAST = False
+    print("提示: pip install win10toast 可启用 Windows 通知")
+
+WS_URL = "ws://111.170.6.103:10002/ws"
+
+def show_notification(title, body):
+    """显示 Windows 通知"""
+    print(f"\n🔔 [{datetime.now().strftime('%H:%M:%S')}] {title}: {body}")
+    if HAS_TOAST:
+        try:
+            toaster.show_toast(title, body, duration=5, threaded=True)
+        except:
+            pass
+
+async def connect():
+    """连接 WebSocket 并监听消息"""
+    while True:
+        try:
+            print(f"正在连接 {WS_URL} ...")
+            async with websockets.connect(WS_URL) as ws:
+                print("✅ 已连接，等待消息...\n")
+                async for message in ws:
+                    try:
+                        data = json.loads(message)
+                        title = data.get("title", "通知")
+                        body = data.get("body", message)
+                    except:
+                        title, body = "通知", message
+                    show_notification(title, body)
+        except Exception as e:
+            print(f"❌ 连接失败: {e}")
+        print("5秒后重连...")
+        await asyncio.sleep(5)
+
+if __name__ == "__main__":
+    print("="*40)
+    print("WebSocket 客户端 - 接收服务器推送")
+    print("="*40)
+    asyncio.run(connect())

selfagent/server/ws_server.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+"""
+WebSocket 服务器 - 向 SelfAgent App 推送消息
+部署到服务器后运行: python ws_server.py
+App 会自动连接 ws://111.170.6.103:9999/ws
+"""
+
+import asyncio
+import websockets
+import json
+from datetime import datetime
+
+# 存储所有连接的客户端
+clients = set()
+
+async def handler(websocket, path):
+    """处理客户端连接"""
+    clients.add(websocket)
+    client_ip = websocket.remote_address[0]
+    print(f"[{datetime.now().strftime('%H:%M:%S')}] 新连接: {client_ip} (在线: {len(clients)})")
+    
+    try:
+        async for message in websocket:
+            print(f"[{datetime.now().strftime('%H:%M:%S')}] 收到: {message}")
+    except websockets.exceptions.ConnectionClosed:
+        pass
+    finally:
+        clients.discard(websocket)
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] 断开: {client_ip} (在线: {len(clients)})")
+
+async def broadcast(message):
+    """向所有客户端广播消息"""
+    if clients:
+        msg = json.dumps({"title": "服务器通知", "body": message, "time": datetime.now().strftime('%H:%M:%S')})
+        await asyncio.gather(*[client.send(msg) for client in clients])
+        print(f"[{datetime.now().strftime('%H:%M:%S')}] 已推送给 {len(clients)} 个客户端")
+
+async def input_loop():
+    """命令行输入，输入消息后推送给所有客户端"""
+    print("\n" + "="*50)
+    print("WebSocket 服务器已启动 - ws://0.0.0.0:9999/ws")
+    print("输入消息后回车即可推送到所有 App")
+    print("="*50 + "\n")
+    
+    loop = asyncio.get_event_loop()
+    while True:
+        message = await loop.run_in_executor(None, input, "推送消息> ")
+        if message.strip():
+            await broadcast(message.strip())
+
+async def main():
+    server = await websockets.serve(handler, "0.0.0.0", 9999, path="/ws")
+    await asyncio.gather(server.wait_closed(), input_loop())
+
+if __name__ == "__main__":
+    asyncio.run(main())