52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
|
import json
|
||
|
|
import re
|
||
|
|
import socketserver
|
||
|
|
from dataclasses import asdict
|
||
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
||
|
|
|
||
|
|
from Code.Communication.Direction import Direction
|
||
|
|
from Code.Communication.Neighbours import Neighbours
|
||
|
|
|
||
|
|
|
||
|
|
class RequestHandler(BaseHTTPRequestHandler):
|
||
|
|
neighbours: Neighbours = None
|
||
|
|
|
||
|
|
def do_GET(self):
|
||
|
|
print("got Get request")
|
||
|
|
if self.path == "/":
|
||
|
|
self.send_response(200, "running")
|
||
|
|
self.end_headers()
|
||
|
|
|
||
|
|
def handle(self) -> None:
|
||
|
|
super().handle()
|
||
|
|
|
||
|
|
|
||
|
|
"""
|
||
|
|
/connect/right
|
||
|
|
/connect/left
|
||
|
|
|
||
|
|
with body : {ip:"string",port:number}
|
||
|
|
"""
|
||
|
|
|
||
|
|
def do_POST(self):
|
||
|
|
print("Got post request")
|
||
|
|
if self.path.startswith("/connect/"):
|
||
|
|
direction = re.findall("/connect/(left|right)", self.path, flags=re.IGNORECASE)[0]
|
||
|
|
data_string = self.rfile.read(int(self.headers['Content-Length']))
|
||
|
|
data = json.loads(data_string)
|
||
|
|
neighbour, accepted = self.neighbours.acceptConnection(Direction[direction.upper()], data["ip"], data["port"])
|
||
|
|
|
||
|
|
print(f"Sending neighbour: {neighbour}")
|
||
|
|
if accepted:
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-Type', 'application/json')
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write(json.dumps(asdict(neighbour)).encode('utf8'))
|
||
|
|
else:
|
||
|
|
self.send_response(303)
|
||
|
|
self.send_header('Location', f"http://{neighbour.ip}:{neighbour.port}")
|
||
|
|
self.end_headers()
|
||
|
|
|
||
|
|
self.end_headers()
|
||
|
|
|