openwebrx-clone/owrx/http.py

71 lines
2.3 KiB
Python
Raw Normal View History

from owrx.controllers import (
StatusController,
IndexController,
AssetsController,
WebSocketController,
MapController,
FeatureController,
ApiController,
MetricsController,
)
2019-05-03 20:59:24 +00:00
from http.server import BaseHTTPRequestHandler
import re
from urllib.parse import urlparse, parse_qs
2019-05-03 20:59:24 +00:00
2019-05-10 19:50:58 +00:00
import logging
2019-05-10 19:50:58 +00:00
logger = logging.getLogger(__name__)
2019-05-03 20:59:24 +00:00
class RequestHandler(BaseHTTPRequestHandler):
def __init__(self, request, client_address, server):
self.router = Router()
super().__init__(request, client_address, server)
2019-05-03 20:59:24 +00:00
def do_GET(self):
self.router.route(self)
class Request(object):
def __init__(self, query=None, matches=None):
self.query = query
self.matches = matches
2019-05-03 20:59:24 +00:00
class Router(object):
mappings = [
{"route": "/", "controller": IndexController},
{"route": "/status", "controller": StatusController},
2019-05-04 14:56:23 +00:00
{"regex": "/static/(.+)", "controller": AssetsController},
2019-05-09 14:10:58 +00:00
{"route": "/ws/", "controller": WebSocketController},
{"regex": "(/favicon.ico)", "controller": AssetsController},
# backwards compatibility for the sdr.hu portal
2019-07-01 14:49:39 +00:00
{"regex": "/(gfx/openwebrx-avatar.png)", "controller": AssetsController},
{"route": "/map", "controller": MapController},
{"route": "/features", "controller": FeatureController},
{"route": "/api/features", "controller": ApiController},
{"route": "/metrics", "controller": MetricsController},
2019-05-03 20:59:24 +00:00
]
2019-05-03 20:59:24 +00:00
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)
2019-05-03 20:59:24 +00:00
def route(self, handler):
url = urlparse(handler.path)
res = self.find_controller(url.path)
2019-05-03 20:59:24 +00:00
if res is not None:
(controller, matches) = res
query = parse_qs(url.query)
request = Request(query, matches)
controller(handler, request).handle_request()
2019-05-03 20:59:24 +00:00
else:
handler.send_error(404, "Not Found", "The page you requested could not be found.")