feat: track and forward events

Author: bpmct

.env.example

@@ -0,0 +1,2 @@
+GETDX_API_KEY=your_api_key_here
+EVENTS_TRACKED=Workspace Deleted,Workspace Manual Build Failure,Workspace Automatic Build Failure,Workspace Automatically Updated,Workspace Dormant,Workspace Marked For Deletion,User Account Created,User Account Deleted,User Account Suspended,User Account Activated,Template Deleted 

.gitignore

@@ -0,0 +1 @@
+.env

main.py

@@ -1,10 +1,23 @@
 import logging
+import os
+import requests
+from dotenv import load_dotenv
 from flask import Flask, jsonify, request
 
+# Load environment variables from .env file
+load_dotenv()
+
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Environment variables for configuration
+GETDX_API_KEY = os.getenv("GETDX_API_KEY")
+EVENTS_TRACKED = os.getenv("EVENTS_TRACKED", "").split(",")
+
+# Clean up event list and convert to lowercase for case-insensitive comparison
+EVENTS_TRACKED = [event.strip().lower() for event in EVENTS_TRACKED]
+
 app = Flask(__name__)
 
 @app.route("/", methods=['GET'])
@@ -15,11 +28,55 @@ def hello_world():
 def webhook_handler():
     payload = request.get_json()
     logger.info("Received webhook payload: %s", payload)
+    
+    event_name = payload.get("event", "").strip()
+    
+    if event_name.lower() in EVENTS_TRACKED:
+        logger.info("Forwarding event: %s", event_name)
+        forward_to_getdx(payload)
+    else:
+        logger.info("Event not tracked, ignoring: %s", event_name)
+    
     return jsonify({"status": "received"})
 
+def forward_to_getdx(payload):
+    if not GETDX_API_KEY:
+        logger.error("GETDX_API_KEY is not set")
+        return
+    
+    # Extract user information more flexibly
+    user_email = payload.get("user", {}).get("email") or payload.get("user_email")
+    
+    getdx_payload = {
+        "name": payload.get("event"),
+        "email": user_email,
+        "timestamp": payload.get("timestamp"),
+        "properties": {
+            "workspace_name": payload.get("workspace", {}).get("name"),
+            "template_name": payload.get("template", {}).get("name"),
+            "user_name": payload.get("user", {}).get("username"),
+            "organization": payload.get("organization", {}).get("name")
+        }
+    }
+    
+    # Remove None values from properties
+    getdx_payload["properties"] = {k: v for k, v in getdx_payload["properties"].items() if v is not None}
+    
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {GETDX_API_KEY}"
+    }
+    
+    try:
+        response = requests.post("https://api.getdx.com/events.track", json=getdx_payload, headers=headers)
+        response.raise_for_status()
+        logger.info("Successfully forwarded event to getDX")
+    except requests.exceptions.RequestException as e:
+        logger.error("Failed to forward event to getDX: %s", e)
+
 @app.route("/health")
 def health_check():
     return jsonify({"status": "healthy"})
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=8080, debug=True) 
+    app.run(host="0.0.0.0", port=8080, debug=True)

requirements.txt

@@ -1,2 +1,4 @@
 flask==3.0.2
-gunicorn==21.2.0 
+gunicorn==21.2.0
+python-dotenv==1.0.1
+requests==2.31.0