77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
import typing as t
|
|
from collections import defaultdict
|
|
from flask import Flask, request
|
|
from . import signal
|
|
from . import configuration
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route("/")
|
|
def root() -> t.Tuple[str, int]:
|
|
return "Not supported.", 400
|
|
|
|
|
|
@app.route("/alertmanager", methods=["POST"])
|
|
@app.route("/alertmanager/<recipient>", methods=["POST"])
|
|
def alertmanager(recipient=None):
|
|
data = request.get_json(cache=False)
|
|
|
|
if "version" not in data or int(data["version"]) != 4:
|
|
return "Bad API version", 400
|
|
if "alerts" not in data:
|
|
return "No alerts", 400
|
|
if "status" not in data:
|
|
return "No status", 400
|
|
if data["status"] != "firing":
|
|
# Ignore
|
|
return "OK", 200
|
|
|
|
severities = defaultdict(int)
|
|
alert_lines = []
|
|
for alert in data["alerts"]:
|
|
try:
|
|
severities[alert["labels"]["severity"]] += 1
|
|
alert_lines.append(
|
|
f"<{alert['labels']['alertname']}> {alert['annotations']['description']}"
|
|
)
|
|
except KeyError as exn:
|
|
alert_lines.append(f"(malformed alert — {exn})")
|
|
|
|
msg = f"[PROMETHEUS] {len(data['alerts'])} firing: "
|
|
msg += ", ".join([f"{count} {typ}" for typ, count in severities.items()])
|
|
msg += "\n\n"
|
|
msg += "\n".join(["* " + line for line in alert_lines])
|
|
|
|
if recipient is None:
|
|
recipient = configuration.DEFAULT_RECIPIENT
|
|
try:
|
|
recipient_nums = configuration.RECIPIENTS[recipient]
|
|
except KeyError:
|
|
return "Bad recipient", 400
|
|
|
|
signal.signal_send(recipient_nums, msg)
|
|
|
|
return "OK", 200
|
|
|
|
|
|
@app.route("/plain", methods=["POST"])
|
|
def plain():
|
|
"""Send a message through Signal, plain POST args.
|
|
|
|
Expects:
|
|
* message: plaintext message to be sent;
|
|
* recipient: ID of recipient. If absent, default recipient is used.
|
|
"""
|
|
|
|
if "message" not in request.form:
|
|
return "Missing message", 400
|
|
recipient = request.form.get("recipient", configuration.DEFAULT_RECIPIENT)
|
|
try:
|
|
recipient_num = configuration.RECIPIENTS[recipient]
|
|
except KeyError:
|
|
return "Unauthorized recipient", 400
|
|
|
|
signal.signal_send(recipient_num, request.form["message"])
|
|
return "OK", 200
|