135 lines
5.5 KiB
Python
135 lines
5.5 KiB
Python
from owrx.controllers.status import StatusController
|
|
from owrx.controllers.template import IndexController, MapController, FeatureController
|
|
from owrx.controllers.assets import OwrxAssetsController, AprsSymbolsController, CompiledAssetsController
|
|
from owrx.controllers.websocket import WebSocketController
|
|
from owrx.controllers.api import ApiController
|
|
from owrx.controllers.metrics import MetricsController
|
|
from owrx.controllers.settings import SettingsController, GeneralSettingsController, SdrSettingsController
|
|
from owrx.controllers.bookmarks import BookmarksController
|
|
from owrx.controllers.session import SessionController
|
|
from owrx.controllers.profile import ProfileController
|
|
from owrx.controllers.imageupload import ImageUploadController
|
|
from http.server import BaseHTTPRequestHandler
|
|
from urllib.parse import urlparse, parse_qs
|
|
import re
|
|
from abc import ABC, abstractmethod
|
|
from http.cookies import SimpleCookie
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
class RequestHandler(BaseHTTPRequestHandler):
|
|
def __init__(self, request, client_address, server):
|
|
self.router = Router()
|
|
super().__init__(request, client_address, server)
|
|
|
|
def log_message(self, format, *args):
|
|
logger.debug("%s - - [%s] %s", self.address_string(), self.log_date_time_string(), format % args)
|
|
|
|
def do_GET(self):
|
|
self.router.route(self, self.get_request("GET"))
|
|
|
|
def do_POST(self):
|
|
self.router.route(self, self.get_request("POST"))
|
|
|
|
def get_request(self, method):
|
|
url = urlparse(self.path)
|
|
return Request(url, method, self.headers)
|
|
|
|
|
|
class Request(object):
|
|
def __init__(self, url, method, headers):
|
|
self.path = url.path
|
|
self.query = parse_qs(url.query)
|
|
self.matches = None
|
|
self.method = method
|
|
self.headers = headers
|
|
self.cookies = SimpleCookie()
|
|
if "Cookie" in headers:
|
|
self.cookies.load(headers["Cookie"])
|
|
|
|
def setMatches(self, matches):
|
|
self.matches = matches
|
|
|
|
|
|
class Route(ABC):
|
|
def __init__(self, controller, method="GET", options=None):
|
|
self.controller = controller
|
|
self.controllerOptions = options if options is not None else {}
|
|
self.method = method
|
|
|
|
@abstractmethod
|
|
def matches(self, request):
|
|
pass
|
|
|
|
|
|
class StaticRoute(Route):
|
|
def __init__(self, route, controller, method="GET", options=None):
|
|
self.route = route
|
|
super().__init__(controller, method, options)
|
|
|
|
def matches(self, request):
|
|
return request.path == self.route and self.method == request.method
|
|
|
|
|
|
class RegexRoute(Route):
|
|
def __init__(self, regex, controller, method="GET", options=None):
|
|
self.regex = re.compile(regex)
|
|
super().__init__(controller, method, options)
|
|
|
|
def matches(self, request):
|
|
matches = self.regex.match(request.path)
|
|
# this is probably not the cleanest way to do it...
|
|
request.setMatches(matches)
|
|
return matches is not None and self.method == request.method
|
|
|
|
|
|
class Router(object):
|
|
def __init__(self):
|
|
self.routes = [
|
|
StaticRoute("/", IndexController),
|
|
StaticRoute("/status.json", StatusController),
|
|
RegexRoute("/static/(.+)", OwrxAssetsController),
|
|
RegexRoute("/compiled/(.+)", CompiledAssetsController),
|
|
RegexRoute("/aprs-symbols/(.+)", AprsSymbolsController),
|
|
StaticRoute("/ws/", WebSocketController),
|
|
RegexRoute("(/favicon.ico)", OwrxAssetsController),
|
|
StaticRoute("/map", MapController),
|
|
StaticRoute("/features", FeatureController),
|
|
StaticRoute("/api/features", ApiController),
|
|
StaticRoute("/api/receiverdetails", ApiController, options={"action": "receiverDetails"}),
|
|
StaticRoute("/metrics", MetricsController, options={"action": "prometheusAction"}),
|
|
StaticRoute("/metrics.json", MetricsController),
|
|
StaticRoute("/settings", SettingsController),
|
|
StaticRoute("/settings/general", GeneralSettingsController),
|
|
StaticRoute(
|
|
"/settings/general", GeneralSettingsController, method="POST", options={"action": "processFormData"}
|
|
),
|
|
StaticRoute("/settings/sdr", SdrSettingsController),
|
|
StaticRoute("/settings/bookmarks", BookmarksController),
|
|
RegexRoute("/settings/bookmarks/(.+)", BookmarksController, method="POST", options={"action": "update"}),
|
|
StaticRoute("/login", SessionController, options={"action": "loginAction"}),
|
|
StaticRoute("/login", SessionController, method="POST", options={"action": "processLoginAction"}),
|
|
StaticRoute("/logout", SessionController, options={"action": "logoutAction"}),
|
|
StaticRoute("/pwchange", ProfileController),
|
|
StaticRoute("/pwchange", ProfileController, method="POST", options={"action": "processPwChange"}),
|
|
StaticRoute("/imageupload", ImageUploadController),
|
|
StaticRoute("/imageupload", ImageUploadController, method="POST", options={"action": "processImage"}),
|
|
]
|
|
|
|
def find_route(self, request):
|
|
for r in self.routes:
|
|
if r.matches(request):
|
|
return r
|
|
|
|
def route(self, handler, request):
|
|
route = self.find_route(request)
|
|
if route is not None:
|
|
controller = route.controller
|
|
controller(handler, request, route.controllerOptions).handle_request()
|
|
else:
|
|
handler.send_error(404, "Not Found", "The page you requested could not be found.")
|