summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/tools/wave/network/static_handler.py
blob: 230af8da1a41cb3f155ec5f71d64fb60b93e3160 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# mypy: allow-untyped-defs

import os


class StaticHandler:
    def __init__(self, web_root, http_port, https_port):
        self.static_dir = os.path.join(
            os.getcwd(), "tools/wave/www")
        self._web_root = web_root
        self._http_port = http_port
        self._https_port = https_port

    def handle_request(self, request, response):
        file_path = request.request_path

        if self._web_root is not None:
            if not file_path.startswith(self._web_root):
                response.status = 404
                return
            file_path = file_path[len(self._web_root):]

        if file_path == "":
            file_path = "index.html"

        file_path = file_path.split("?")[0]
        file_path = os.path.join(self.static_dir, file_path)

        if not os.path.exists(file_path):
            response.status = 404
            return

        headers = []

        content_types = {
            "html": "text/html",
            "js": "text/javascript",
            "css": "text/css",
            "jpg": "image/jpeg",
            "jpeg": "image/jpeg",
            "ttf": "font/ttf",
            "woff": "font/woff",
            "woff2": "font/woff2"
        }

        headers.append(
            ("Content-Type", content_types[file_path.split(".")[-1]]))

        data = None
        with open(file_path, "rb") as file:
            data = file.read()

        if file_path.split("/")[-1] == "wave-service.js":
            data = data.decode("UTF-8")
            data = data.replace("{{WEB_ROOT}}", str(self._web_root))
            data = data.replace("{{HTTP_PORT}}", str(self._http_port))
            data = data.replace("{{HTTPS_PORT}}", str(self._https_port))

        response.content = data
        response.headers = headers