sheppy/atlantis-event-dispatcher
1
0
Fork 0
mirror of https://github.com/FAUSheppy/atlantis-event-dispatcher synced 2025-12-08 15:28:32 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: inbuilt basic auth decoration

Browse Source
This commit is contained in:
Johnathen Sheppard
2022-12-29 11:34:07 +01:00
parent 73e211fb68
commit 881bc96213
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
interface.py
View File

@@ -4,6 +4,7 @@ import argparse
import flask import flask
import subprocess import subprocess
import os import os
from functools import wraps
HOST = "icinga.atlantishq.de" HOST = "icinga.atlantishq.de"
SIGNAL_USER_FILE = "signal_targets.txt" SIGNAL_USER_FILE = "signal_targets.txt"
@@ -18,6 +19,15 @@ def dbReadSignalUserFile():
users.append(user) users.append(user)
return users return users
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
auth = flask.request.authorization
if not auth or not auth.password == app.config["PASSWORD"]:
return (flask.jsonify({ 'message' : 'Authentication required' }), 401)
return f(*args, **kwargs)
return decorated_function
def signalSend(user, msg): def signalSend(user, msg):
signalCliBin = "signal-cli" signalCliBin = "signal-cli"
if app.config["SIGNAL_CLI_BIN"]: if app.config["SIGNAL_CLI_BIN"]:
@@ -30,17 +40,20 @@ def sendMessageToAllClients(msg):
signalSend(number, msg) signalSend(number, msg)
@app.route('/send-to-clients', methods=["POST"]) @app.route('/send-to-clients', methods=["POST"])
@login_required
def sendToNumbers(): def sendToNumbers():
for numberOrUser in flask.request.json["users"]: for numberOrUser in flask.request.json["users"]:
signalSend(numberOrUser, flask.request.json["message"]) signalSend(numberOrUser, flask.request.json["message"])
return ("","204") return ("","204")
@app.route('/send-all', methods=["POST"]) @app.route('/send-all', methods=["POST"])
@login_required
def sendToAll(): def sendToAll():
sendMessageToAllClients(flask.request.json["message"]) sendMessageToAllClients(flask.request.json["message"])
return ("","204") return ("","204")
@app.route('/send-all-grafana', methods=["POST"]) @app.route('/send-all-grafana', methods=["POST"])
@login_required
def sendToAllGrafana(): def sendToAllGrafana():
j = flask.request.json j = flask.request.json
state = j["state"] state = j["state"]
@@ -53,6 +66,7 @@ def sendToAllGrafana():
return ("","204") return ("","204")
@app.route('/send-all-icinga', methods=["POST"]) @app.route('/send-all-icinga', methods=["POST"])
@login_required
def sendToAllIcinga(): def sendToAllIcinga():
args = flask.request.json args = flask.request.json
@@ -72,6 +86,11 @@ def sendToAllIcinga():
sendMessageToAllClients(message) sendMessageToAllClients(message)
return ("","204") return ("","204")
@app.before_first_request
def init():
app.config["PASSWORD"] = os.environ["SIGNAL_API_PASS"]
app.config["SIGNAL_CLI_BIN"] = os.environ["SIGNAL_CLI_BIN"]
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Simple Telegram Notification Interface', parser = argparse.ArgumentParser(description='Simple Telegram Notification Interface',
@@ -83,5 +102,6 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
app.config["SIGNAL_CLI_BIN"] = os.path.expanduser(args.signal_cli_bin) app.config["SIGNAL_CLI_BIN"] = os.path.expanduser(args.signal_cli_bin)
app.config["PASSWORD"] = os.environ["SIGNAL_API_PASS"]
app.run(host=args.interface, port=args.port) app.run(host=args.interface, port=args.port)