U 2i@sbdZddlZddlZddlZddlZddlZddlmZm Z m Z dZ e dddZ ee eefe ee fe dd d Zdeeeed d d ZdeeeedddZdeeeedddZd eeeedddZedkr^ddlZeejdkredededejdZejdZeejdkr@ejdndZeeeeZededS)!z Notification module - Sends messages to Discord and Telegram channels. Uses urllib (built-in) instead of requests to avoid SSL/URL parsing issues. N)OptionalDictAnyz5/home/ubuntu/.openclaw/workspace/config/notifier.json)returnc CsXtjtr2tt}t|W5QRSQRXtdtdddtdidS)z Load notification configuration.ZDISCORD_BOT_TOKENZDISCORD_WEBHOOK_URL) bot_tokenZ webhook_urlrZTELEGRAM_BOT_TOKEN)discordtelegram)ospathexists CONFIG_PATHopenjsonloadgetenv)fr0/home/ubuntu/.openclaw/workspace/tools/notify.py load_configs  r)urlheadersdatarc Csdd|d<tjj|t|d|dd}tjj|dd"}t| dW5QRSQRXdS) z$Make HTTP POST request using urllib.z2Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36z User-Agentzutf-8POST)rrmethod )timeoutN) urllibZrequestZRequestrdumpsencodeZurlopenloadsreaddecode)rrrZreqZresponserrr http_post!sr") channel_idmessagementionc Cst}|did}|s$tdd|d}|}|rHd|d|}|dgid }d d d d |dd|dddt|g }tj|dddd} | jdkrtd| j t | j S)zRSend message to Discord channel via bot API using curl (Cloudflare blocks urllib).rrz Discord bot token not configuredz%https://discord.com/api/v10/channels/z /messages@ Zparse)contentZallowed_mentionsZcurlz-sz-Xrz-HzAuthorization: Bot zContent-Type: application/jsonz-dTr)Zcapture_outputtextrrz curl failed: ) rget ValueErrorrr subprocessrun returncode RuntimeErrorstderrrstdout) r#r$r%configtokenrr(payloadcmdresultrrrsend_discord_message0s6  r7)chat_idr$r%c Csht}|did}|s$tdd|d}|}|rHd|d|}||dd }d d i}t|||S) z*Send message to Telegram chat via bot API.rrz!Telegram bot token not configuredzhttps://api.telegram.org/botz /sendMessager&r'ZMarkdown)r8r)Z parse_modez Content-Typezapplication/json)rr*r+r") r8r$r%r2r3rr)r4rrrrsend_telegram_messageRs r9channelr$r%cCsV|dr"|dd}t|||S|drD|dd}t|||Std|dS)z{ Send notification to specified channel. Channel format: - discord:123456789 - telegram:123456789 zdiscord:z telegram:zUnknown channel format: N) startswithreplacer7r9r+)r;r$r%r#r8rrrsend_notificationis      r?cCs t|||S)z Send a resume command that will trigger the agent to continue work. This is the same as send_notification but named clearly for intent. )r?r:rrrsend_resume_command|sr@__main__z.Usage: notify.py [mention]z9Example: notify.py discord:123456789 'Hello world' taro83zSent: )N)N)N)N)__doc__rr Zurllib.requestrZ urllib.errorr,typingrrrr dictrstrr"r7r9r?r@__name__syslenargvprintexitr;r$r%r6rrrrs0$"