tellme/server/src/tellmesrv.py

#!/bin/env python3
from flask import Flask, request, jsonify
import websocket
import json
import yaml
import random
import threading
import pprint

__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)
    pprint(response['resp'])
    # for chatitem in response['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()