Initial commit with translated description
This commit is contained in:
113
scripts/send_status_websocket.py
Normal file
113
scripts/send_status_websocket.py
Normal file
@@ -0,0 +1,113 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Send status messages using Clawdbot WebSocket API (fast).
|
||||
"""
|
||||
|
||||
import sys
|
||||
sys.stdout.reconfigure(encoding='utf-8')
|
||||
|
||||
import os
|
||||
import json
|
||||
import websocket
|
||||
|
||||
# Status type to emoji mapping
|
||||
STATUS_EMOJIS = {
|
||||
"progress": "🔄",
|
||||
"success": "✅",
|
||||
"error": "❌",
|
||||
"warning": "⚠️"
|
||||
}
|
||||
|
||||
def send_status(message: str, status_type: str, step_name: str, details: str = None):
|
||||
"""
|
||||
Format and send a status message to Telegram via WebSocket.
|
||||
|
||||
Args:
|
||||
message: Short status message (< 140 chars)
|
||||
status_type: Type of status (progress, success, error, warning)
|
||||
step_name: Name of the step being reported
|
||||
details: Optional additional context
|
||||
"""
|
||||
if status_type not in STATUS_EMOJIS:
|
||||
raise ValueError(f"Invalid status_type: {status_type}")
|
||||
|
||||
emoji = STATUS_EMOJIS[status_type]
|
||||
|
||||
# Build the message
|
||||
formatted = f"{emoji} [{step_name}] {message}"
|
||||
|
||||
if details:
|
||||
formatted += f" ({details})"
|
||||
|
||||
# Keep it concise (under 140 chars)
|
||||
if len(formatted) > 140:
|
||||
formatted = formatted[:137] + "..."
|
||||
|
||||
# Send via WebSocket
|
||||
gateway_port = os.environ.get("CLAWDBOT_GATEWAY_PORT", "18789")
|
||||
gateway_token = os.environ.get("CLAWDBOT_GATEWAY_TOKEN")
|
||||
target = os.environ.get("TELEGRAM_TARGET", "7590912486")
|
||||
|
||||
if not gateway_token:
|
||||
print(f"✗ CLAWDBOT_GATEWAY_TOKEN not found", file=sys.stderr)
|
||||
return formatted
|
||||
|
||||
ws_url = f"ws://127.0.0.1:{gateway_port}/ws"
|
||||
|
||||
try:
|
||||
# Connect and send
|
||||
ws = websocket.create_connection(ws_url, timeout=10)
|
||||
|
||||
# Handshake
|
||||
handshake = {"type": "handshake", "token": gateway_token}
|
||||
ws.send(json.dumps(handshake))
|
||||
|
||||
# Send message
|
||||
msg = {
|
||||
"type": "message",
|
||||
"action": "send",
|
||||
"target": target,
|
||||
"message": formatted,
|
||||
"channel": "telegram"
|
||||
}
|
||||
ws.send(json.dumps(msg))
|
||||
|
||||
# Get response
|
||||
response = ws.recv()
|
||||
result = json.loads(response)
|
||||
|
||||
ws.close()
|
||||
|
||||
# Check if message was sent
|
||||
if result.get("event") == "message.sent" or result.get("type") == "ack":
|
||||
return formatted
|
||||
else:
|
||||
print(f"⚠️ Response: {result}", file=sys.stderr)
|
||||
return formatted
|
||||
|
||||
except Exception as e:
|
||||
print(f"✗ WebSocket error: {e}", file=sys.stderr)
|
||||
# Fallback: print to console
|
||||
print(formatted, file=sys.stderr)
|
||||
return formatted
|
||||
|
||||
def main():
|
||||
if len(sys.argv) < 4:
|
||||
print("Usage: send_status_websocket.py <message> <status_type> <step_name>")
|
||||
print("\nStatus Types: progress, success, error, warning")
|
||||
sys.exit(1)
|
||||
|
||||
message = sys.argv[1]
|
||||
status_type = sys.argv[2]
|
||||
step_name = sys.argv[3]
|
||||
details = sys.argv[4] if len(sys.argv) > 4 else None
|
||||
|
||||
try:
|
||||
result = send_status(message, status_type, step_name, details)
|
||||
print(f"✓ Status sent: {result}")
|
||||
except Exception as e:
|
||||
print(f"✗ Error: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user