feat: add engine lib modules (ufw, crowdsec, systemd, notify)
This commit is contained in:
1 parent
72eec40ef3
commit
9ad3c93644
5 files changed
+201
No files matched your search
@@ -0,0 +1,61 @@
|
||||
# engine/lib/notify.py
|
||||
"""ctx.notify — Telegram notifications (port of root_core_telegram*).
|
||||
|
||||
instance renamed com.ermnvldmr.root.<node> -> com.ermnvldmr.apex.<node>. POSTs
|
||||
application/x-www-form-urlencoded via urllib (stdlib) — functionally identical to
|
||||
the bash curl multipart for Telegram sendMessage.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
import html
|
||||
import urllib.parse
|
||||
import urllib.request
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
def build_text(title: str, message: str, node: str, level: str, ts: str) -> str:
|
||||
status, icon, severity = "firing", "🚨", "info"
|
||||
if level == "SUCCESS":
|
||||
status, icon = "resolved", "✅"
|
||||
elif level == "ERROR":
|
||||
severity = "critical"
|
||||
elif level == "WARN":
|
||||
severity = "warning"
|
||||
instance = f"com.ermnvldmr.apex.{node}"
|
||||
msg = message if len(message) <= 200 else message[:200] + "..."
|
||||
msg = html.escape(msg)
|
||||
safe_title = html.escape(title)
|
||||
return (f"{icon} <b>{safe_title}</b>\n"
|
||||
f"<b>Instance:</b> {instance}\n"
|
||||
f"<b>Timestamp:</b> {ts}\n"
|
||||
f"<b>Status:</b> {status.capitalize()}\n"
|
||||
f"<b>Escalation:</b> {severity}\n"
|
||||
f"<pre><code>{msg}</code></pre>\n"
|
||||
f"#status_{status} #escalation_{severity} #instance_{instance.replace('.', '_')}")
|
||||
|
||||
|
||||
class Notify:
|
||||
def __init__(self, log, node: str = "global"):
|
||||
self.log = log
|
||||
self.node = node
|
||||
|
||||
def telegram(self, title, bot_url, message, node=None, level="INFO") -> bool:
|
||||
node = node or self.node
|
||||
if not bot_url:
|
||||
self.log.warn("Telegram Bot URL not provided. Skipping notification.")
|
||||
return False
|
||||
ts = datetime.now().astimezone().strftime("%Y-%m-%d %H:%M:%S %Z")
|
||||
text = build_text(title, message, node, level, ts)
|
||||
data = urllib.parse.urlencode({"parse_mode": "HTML", "text": text}).encode()
|
||||
self.log.info(f"Sending Telegram notification ({level}): {title}")
|
||||
try:
|
||||
with urllib.request.urlopen(urllib.request.Request(bot_url, data=data), timeout=120) as r:
|
||||
r.read()
|
||||
return True
|
||||
except Exception as e:
|
||||
self.log.error(f"Failed to send Telegram notification: {e}")
|
||||
return False
|
||||
|
||||
def success(self, title, url, msg, node=None): return self.telegram(title, url, msg, node, "SUCCESS")
|
||||
def error(self, title, url, msg, node=None): return self.telegram(title, url, msg, node, "ERROR")
|
||||
def info(self, title, url, msg, node=None): return self.telegram(title, url, msg, node, "INFO")
|
||||
def warn(self, title, url, msg, node=None): return self.telegram(title, url, msg, node, "WARN")
|
||||
Reference in new issue
Block a user