41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
|
"""Probe-only proxy: records each MAS event as delivered by Postfix, then
|
||
|
|
forwards it to the ticket service and records the app's answer."""
|
||
|
|
|
||
|
|
import json
|
||
|
|
import sys
|
||
|
|
import urllib.error
|
||
|
|
import urllib.request
|
||
|
|
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||
|
|
|
||
|
|
UPSTREAM = sys.argv[1]
|
||
|
|
EVENTS = []
|
||
|
|
|
||
|
|
|
||
|
|
class H(BaseHTTPRequestHandler):
|
||
|
|
def do_POST(self):
|
||
|
|
body = self.rfile.read(int(self.headers.get("Content-Length", "0")))
|
||
|
|
req = urllib.request.Request(UPSTREAM, body, {"Content-Type": "application/json"})
|
||
|
|
try:
|
||
|
|
with urllib.request.urlopen(req) as r:
|
||
|
|
status, resp = r.status, r.read()
|
||
|
|
except urllib.error.HTTPError as e:
|
||
|
|
status, resp = e.code, e.read()
|
||
|
|
EVENTS.append({"event": json.loads(body), "app_status": status, "app": json.loads(resp or b"null")})
|
||
|
|
self._send(status, resp)
|
||
|
|
|
||
|
|
def do_GET(self):
|
||
|
|
self._send(200, json.dumps(EVENTS).encode())
|
||
|
|
|
||
|
|
def _send(self, status, body):
|
||
|
|
self.send_response(status)
|
||
|
|
self.send_header("Content-Type", "application/json")
|
||
|
|
self.send_header("Content-Length", str(len(body)))
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write(body)
|
||
|
|
|
||
|
|
def log_message(self, *a):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
ThreadingHTTPServer(("0.0.0.0", 8024), H).serve_forever()
|