38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from owrx.controllers import StatusController, IndexController, AssetsController, WebSocketController
|
|
from http.server import BaseHTTPRequestHandler
|
|
import re
|
|
|
|
class RequestHandler(BaseHTTPRequestHandler):
|
|
def __init__(self, request, client_address, server):
|
|
self.router = Router()
|
|
super().__init__(request, client_address, server)
|
|
def do_GET(self):
|
|
self.router.route(self)
|
|
|
|
class Router(object):
|
|
mappings = [
|
|
{"route": "/", "controller": IndexController},
|
|
{"route": "/status", "controller": StatusController},
|
|
{"regex": "/static/(.+)", "controller": AssetsController},
|
|
{"route": "/ws/", "controller": WebSocketController},
|
|
{"regex": "(/favicon.ico)", "controller": AssetsController}
|
|
]
|
|
def find_controller(self, path):
|
|
for m in Router.mappings:
|
|
if "route" in m:
|
|
if m["route"] == path:
|
|
return (m["controller"], None)
|
|
if "regex" in m:
|
|
regex = re.compile(m["regex"])
|
|
matches = regex.match(path)
|
|
if matches:
|
|
return (m["controller"], matches)
|
|
def route(self, handler):
|
|
res = self.find_controller(handler.path)
|
|
#print("path: {0}, controller: {1}, matches: {2}".format(handler.path, controller, matches))
|
|
if res is not None:
|
|
(controller, matches) = res
|
|
controller(handler, matches).handle_request()
|
|
else:
|
|
handler.send_error(404, "Not Found", "The page you requested could not be found.")
|