Merge pull request #4 from Sajadorouji/master

All tested, is ok.

Enabled authentication and Output.log

Author: Carlos Augusto Malucelli

.gitignore

@@ -0,0 +1 @@
+flaskAlert.log

README.md

@@ -1,4 +1,4 @@
-# Alertmanager webhook for Telegram using Flask 
+# Alertmanager webhook for Telegram using Flask
 
 ## INSTALL
 
@@ -8,7 +8,7 @@ Change on flaskAlert.py
 =======================
 * botToken
 * chatID (without -)
-  
+
 Alertmanager configuration example
 ==================================
 
@@ -17,7 +17,7 @@ Alertmanager configuration example
 		  webhook_configs:
 		  - url: http://ipFlaskAlert:9119/alert
 		    send_resolved: true
-		    
+
 One way to get the chat ID
 ==========================
 1) Add bot on channel
@@ -30,8 +30,10 @@ Running
 
 Running on docker
 =================
-* docker container run -d -e bottoken="telegramBotToken" -e chatid="telegramChatID" -p 9119:9119 nopp/alertmanager-webhook-telegram:latest
+* docker run -d --name telegram-bot -e "bottoken=telegramBotToken" -e "chatid=telegramChatID" -e "username=<username>" -e "password=<password>"-p 9119:9119 nopp/alertmanager-webhook-telegram:latest
+
+> make sure set proper username and password when you exposing your app on internet
 
 Exemple to test
 ===============
-	curl -XPOST --data '{"status":"resolved","groupLabels":{"alertname":"instance_down"},"commonAnnotations":{"description":"i-0d7188fkl90bac100 of job ec2-sp-node_exporter has been down for more than 2 minutes.","summary":"Instance i-0d7188fkl90bac100 down"},"alerts":[{"status":"resolved","labels":{"name":"olokinho01-prod","instance":"i-0d7188fkl90bac100","job":"ec2-sp-node_exporter","alertname":"instance_down","os":"linux","severity":"page"},"endsAt":"2019-07-01T16:16:19.376244942-03:00","generatorURL":"http://pmts.io:9090","startsAt":"2019-07-01T16:02:19.376245319-03:00","annotations":{"description":"i-0d7188fkl90bac100 of job ec2-sp-node_exporter has been down for more than 2 minutes.","summary":"Instance i-0d7188fkl90bac100 down"}}],"version":"4","receiver":"infra-alert","externalURL":"http://alm.io:9093","commonLabels":{"name":"olokinho01-prod","instance":"i-0d7188fkl90bac100","job":"ec2-sp-node_exporter","alertname":"instance_down","os":"linux","severity":"page"}}' http://flaskAlert:9119/alert
+	curl -XPOST --data '{"status":"resolved","groupLabels":{"alertname":"instance_down"},"commonAnnotations":{"description":"i-0d7188fkl90bac100 of job ec2-sp-node_exporter has been down for more than 2 minutes.","summary":"Instance i-0d7188fkl90bac100 down"},"alerts":[{"status":"resolved","labels":{"name":"olokinho01-prod","instance":"i-0d7188fkl90bac100","job":"ec2-sp-node_exporter","alertname":"instance_down","os":"linux","severity":"page"},"endsAt":"2019-07-01T16:16:19.376244942-03:00","generatorURL":"http://pmts.io:9090","startsAt":"2019-07-01T16:02:19.376245319-03:00","annotations":{"description":"i-0d7188fkl90bac100 of job ec2-sp-node_exporter has been down for more than 2 minutes.","summary":"Instance i-0d7188fkl90bac100 down"}}],"version":"4","receiver":"infra-alert","externalURL":"http://alm.io:9093","commonLabels":{"name":"olokinho01-prod","instance":"i-0d7188fkl90bac100","job":"ec2-sp-node_exporter","alertname":"instance_down","os":"linux","severity":"page"}}' http://username:password@flaskAlert:9119/alert

docker/run.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 sed -i s/botToken/"$bottoken"/ flaskAlert.py
 sed -i s/xchatIDx/"$chatid"/ flaskAlert.py
+sed -i s/XXXUSERNAME/"$username"/ flaskAlert.py
+sed -i s/XXXPASSWORD/"$password"/ flaskAlert.py
 
 /usr/bin/gunicorn -w 4 -b 0.0.0.0:9119 flaskAlert:app

flaskAlert.py

@@ -3,26 +3,32 @@
 import json
 from flask import Flask
 from flask import request
+from flask_basicauth import BasicAuth
 
 app = Flask(__name__)
 app.secret_key = 'aYT>.L$kk2h>!'
+app.config['BASIC_AUTH_USERNAME'] = 'XXXUSERNAME'
+app.config['BASIC_AUTH_PASSWORD'] = 'XXXPASSWORD'
 
+basic_auth = BasicAuth(app)
+app.config['BASIC_AUTH_FORCE'] = True
 bot = telegram.Bot(token="botToken")
 chatID = "-xchatIDx"
 
 @app.route('/alert', methods = ['POST'])
 def postAlertmanager():
 
     content = json.loads(request.get_data())
-
+    with open("Output.txt", "w") as text_file:
+        text_file.write("{0}".format(content))
     try:
 
         for alert in content['alerts']:
 
             if 'name' in alert['labels']:
 
                 message = """
-Status """+alert['status']+"""   
+Status """+alert['status']+"""
 Alertname: """+alert['labels']['alertname']+"""
 Instance: """+alert['labels']['instance']+"""("""+alert['labels']['name']+""")
 """+alert['annotations']['description']+"""

requirements.txt

@@ -1,3 +1,4 @@
 flask
 python-telegram-bot
+Flask-BasicAuth
 gunicorn