73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
#!/bin/env python3
|
|
from flask import Flask, request, jsonify
|
|
import websocket
|
|
import json
|
|
import yaml
|
|
import random
|
|
import threading
|
|
|
|
__version__ = "2.0.0b2"
|
|
versionstring='Taurix TellMe server v' + __version__
|
|
|
|
app = Flask(__name__)
|
|
lock = threading.Lock()
|
|
|
|
hooks = {}
|
|
with open(r'/etc/tellme/hooks.yml') as hooksfile:
|
|
hooks = yaml.load(hooksfile, Loader=yaml.FullLoader)
|
|
|
|
def sendmessage(target, message):
|
|
print("Sendmessage %s called to %s" % (message, target))
|
|
msg = ("%s %s" % (target, message))
|
|
|
|
# Create a unique correlation ID
|
|
command = {
|
|
"corrId": f"id{random.randint(0, 999999)}",
|
|
"cmd": msg,
|
|
}
|
|
json_command = json.dumps(command)
|
|
|
|
""" Connects to WebSocket server, sends a message, and returns the response """
|
|
uri = "ws://localhost:5080"
|
|
ws = websocket.create_connection(uri) # Blocking WebSocket connection
|
|
ws.send(json_command) # Send message to WebSocket
|
|
responsejson = ws.recv() # Receive response
|
|
response = json.loads(responsejson)
|
|
print(response)
|
|
chatitems = response['resp']['chatitems']
|
|
for chatitem in chatitems:
|
|
print(chatitem)
|
|
# print("Message %s sent over websocket to %s" % (chatitem['content'], target))
|
|
# ws.close() # Close WebSocket connection
|
|
# print(response)
|
|
ws.close()
|
|
return response
|
|
|
|
@app.route("/webhook/<id>", methods=['POST'])
|
|
def webhook_receiver(id):
|
|
print("Webhook id %s" % (id))
|
|
data = request.json # Get the JSON data from the incoming request
|
|
# Process the data and perform actions based on the event
|
|
print("Received webhook data:", data)
|
|
|
|
target = None
|
|
for key, value in hooks.items():
|
|
if str(key) == str(id):
|
|
target = value
|
|
|
|
if target is not None:
|
|
print(target)
|
|
with lock:
|
|
sendmessage(target, data.get('message'))
|
|
else:
|
|
print("No target found, dropping message")
|
|
|
|
return jsonify({'message': 'Webhook received successfully'}), 200
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print("Started %s" % (versionstring))
|
|
uri = "ws://localhost:5080"
|
|
# ws = websocket.create_connection(uri)
|
|
app.run()
|