matrix-service-state.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from os import environ

from matrix_client.api import MatrixHttpApi
from markdown2 import Markdown

# Format Icinag2 status to color
if environ["SERVICESTATE"] == "CRITICAL":
    COLOR="#ff0000"
elif environ["SERVICESTATE"] == "WARNING":
    COLOR="#ffcc00"
elif environ["SERVICESTATE"] == "OK":
    COLOR="#33cc33"
elif environ["SERVICESTATE"] == "UNKNOWN":
    COLOR="#9933ff"
else:
    COLOR="#e6e6e6"

# If notification has comment
if environ["NOTIFICATIONCOMMENT"]:
    COMMENT_PLAIN = """
Comment by {NOTIFICATIONAUTHORNAME}:
    {NOTIFICATIONCOMMENT}
""".format(**environ)

    COMMENT_MD = """>
> *Comment by {NOTIFICATIONAUTHORNAME}:*
>
> {NOTIFICATIONCOMMENT}
>""".format(**environ)
else:
    COMMENT_PLAIN = ""
    COMMENT_MD = ">"

DATA = {
    "COLOR": COLOR,
    "COMMENT_PLAIN": COMMENT_PLAIN,
    "COMMENT_MD": COMMENT_MD,
}
DATA.update(environ)

# Message without formating
MSG_PLAIN = """[{NOTIFICATIONTYPE}] Service {SERVICEDISPLAYNAME} on {HOSTDISPLAYNAME} is {SERVICESTATE}

Info: {SERVICEOUTPUT}
When: {LONGDATETIME}
{COMMENT_PLAIN}
{ICINGA_WEBURL}/monitoring/host/show?host={HOSTNAME}&service={SERVICEDESC}
""".format(**DATA)

# Message in markdown
MSG_MD = """**<font color="{COLOR}">[{NOTIFICATIONTYPE}] Service {SERVICEDISPLAYNAME} on {HOSTDISPLAYNAME} is {SERVICESTATE}</font>**

> *Info:*
>
> {SERVICEOUTPUT}
{COMMENT_MD}
> *{LONGDATETIME} - [Show in Icinga2]({ICINGA_WEBURL}/monitoring/host/show?host={HOSTNAME}&service={SERVICEDESC})*
""".format(**DATA)

# Init matrix API
matrix = MatrixHttpApi(environ['MATRIX_SERVER'], token=environ['MATRIX_TOKEN'])

# Send message in both formats to channel
response = matrix.send_message_event(
    room_id=environ['MATRIX_CHANNEL'],
    event_type="m.room.message",
    content={
        "msgtype": "m.text",
        "format": "org.matrix.custom.html",
        "body": MSG_PLAIN,
        "formatted_body": Markdown().convert(MSG_MD),
    }
)