html, branding change and getting ready for pre-rel

idk anymore man
2026-03-24 22:50:26 +01:00 · 2026-03-13 22:13:59 +01:00
4 changed files with 99 additions and 74 deletions
@@ -0,0 +1,26 @@
+"""
+This is the Amethyst API mode Python interface whatevers.
+Docs will follow.
+"""
+
+# Below go imports.
+import sys
+import os
+
+if not os.getcwd() in sys.path:
+    sys.path.append(os.getcwd())
+import pywebsrv
+
+
+class API:
+    """
+    class
+    """
+
+    def __init__(self):
+        # DO NOT USE THIS CLASS FOR PROGRAM, ONLY ON_REQUEST PLEASE!!
+        # Below go definitions to get things working.
+        self.build_response = pywebsrv.WebServer.build_binary_response
+
+    def on_request(self, req):
+        return self.build_response(200, "This is a test", "text/html")
@@ -6,9 +6,9 @@
 <body>
    <center>
    <h1>Hello from Amethyst!</h1>
-    <h2>This page confirms Amethyst can read files from your PC and serve them to your browser!</h2>
-    <p>If you see this page and you're not the server owner, tell them they misconfigured something!</p>
-    <p>This server runs Amethyst Alpha Build 0039</p>
+    <h2>This page confirms Amethyst can read files from your PC or server and serve them to your browser!</h2>
+    <p>This is a test page, if you aren't the server owner, they might not have finished setting up their site, be patient. If this doesn't go away after a while, tell them they've made an oopsie</p>
+    <p>This server runs Amethyst Pre-Rel Build 0053</p>
    </center>
 </body>
 </html>
@@ -10,6 +10,7 @@ host localhost {
    directory:/home/nova/PyWebServer/html2
    allowed-methods:GET,PUT
    block-ip:10.1.100.2
+    apimode:0
    block-ua:match("Discordbot")
 }

@@ -50,33 +50,35 @@ import signal
 import sys

 try:
+    if not os.getcwd() in sys.path:
+        sys.path.append(f"{os.getcwd()}")
    from certgen import AutoCertGen
 except ImportError:
    # just do nothing, it's not working anyway.
-    # print(
-    #     "WARN: You need the AutoCertGen plugin! Please install it from\n"
-    #     "https://git.novacow.ch/Nova/AutoCertGen/"
-    # )
-    pass
+    print(
+        "WARN: You need the AutoCertGen plugin! Please install it from\n"
+        "https://git.novacow.ch/Nova/AutoCertGen/"
+    )
+    # pass

-AMETHYST_BUILD_NUMBER = "0046"
+AMETHYST_BUILD_NUMBER = "0053"
 AMETHYST_REPO = "https://git.novacow.ch/Nova/PyWebServer/"


 class ConfigParser:
    def __init__(self, text):
-        self.data = {"hosts": {}, "globals": {}}
+        self.data: dict = {"hosts": {}, "globals": {}}
        self._parse(text)

    def _parse(self, text):
-        lines = [
+        lines: list = [
            line.strip()
            for line in text.splitlines()
            if line.strip() and not line.strip().startswith("#")
        ]

-        current_block = None
-        current_name = None
+        current_block: tuple | None = None
+        current_name: str | None = None

        for line in lines:
            if line.startswith("host ") and line.endswith("{"):
@@ -96,8 +98,8 @@ class ConfigParser:

            if ":" in line and current_block:
                key, value = line.split(":", 1)
-                key = key.strip()
-                value = value.strip()
+                key: str = key.strip()
+                value: str = value.strip()

                if "," in value:
                    value = [v.strip() for v in value.split(",")]
@@ -139,6 +141,8 @@ class FileHandler:
    def read_file(self, file_path, directory=None):
        if "../" in file_path or "%" in file_path:
            return 403, None
+        if file_path == "api.py":
+            return 404, None

        if directory is not None:
            full_path = os.path.join(directory, file_path.lstrip("/"))
@@ -239,8 +243,6 @@ class FileHandler:
        Generate some self-signed certificates using AutoCertGen
        TODO: doesn't work, need to fix. probably add `./` to $PATH
        """
-        if not os.getcwd() in sys.path:
-            sys.path.append(f"{os.getcwd()}")
        autocert = AutoCertGen()
        autocert.gen_cert()

@@ -251,14 +253,17 @@ class RequestParser:
        self.hosts = self.file_handler.read_new_config("hosts")
        print(f"Hosts: {self.hosts}")

-    def parse_request_line(self, line):
+    def parse_request_line(self, line, host):
        """Parses the HTTP request line."""
        try:
            method, path, version = line.split(" ")
        except ValueError:
            return "DELETE", "/this/is/a/bogus/request", "HTTP/1.0"
-        if path.endswith("/") and "." not in path:
-            path += "index.html"
+        if path.endswith("/") or ("." not in path):
+            if not path.endswith("/"):
+                path += "/"
+            index = self.file_handler.read_new_config("index", host) or "index.html"
+            path += f"{index}"
        return method, path, version

    def ua_is_allowed(self, ua, host=None):
@@ -276,18 +281,20 @@ class RequestParser:
        #         return False
        # return True

-    def is_method_allowed(self, method):
+    def is_method_allowed(self, method, host=None):
        """
        Checks if the HTTP method is allowed.
        Reads allowed methods from a configuration file.
        Falls back to allowing only 'GET' if the file does not exist.
        Should (for now) only be GET as I haven't implemented the logic for PUT
        """
-        allowed_methods = ["GET"]
+        # allowed_methods = ["GET"]
        # While the logic for PUT, DELETE, etc. is not added, we shouldn't
        # allow for it to attempt it.
        # Prepatched for new update.
-        # allowed_methods = self.file_handler.read_config("allowed-methods")
+        allowed_methods = self.file_handler.read_new_config("allowed-methods", host)
+        if allowed_methods is None:
+            allowed_methods = ["GET"]
        return method in allowed_methods

    def host_parser(self, host):
@@ -313,13 +320,6 @@ class RequestParser:
            return True


-#
-# class ProxyServer:
-#     def __init__(
-#         self,
-#     ):
-
-
 class WebServer:
    def __init__(
        self, http_port=8080, https_port=8443, cert_file="cert.pem", key_file="key.pem"
@@ -356,17 +356,6 @@ class WebServer:
            "This host cannot be reached without sending a `Host` header."
        )

-        # TODO: enable experimental ipv6 support in config
-
-        # ipv6 when????/??//?????//?
-        # self.http_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        # self.http_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        # self.http_socket.bind(("0.0.0.0", self.http_port))
-        #
-        # self.https_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        # self.https_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        # self.https_socket.bind(("0.0.0.0", self.https_port))
-
        self.http_socket = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
        self.http_socket.bind(("::", self.http_port))

@@ -384,18 +373,18 @@ class WebServer:
            )

        self.http_404_html = (
-            "<html><head><title>HTTP 404 - PyWebServer</title></head>"
-            f"<body><center><h1>HTTP 404 - Not Found!</h1><p>Running PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}</p>"
+            "<html><head><title>HTTP 404 - Amethyst</title></head>"
+            f"<body><center><h1>HTTP 404 - Not Found!</h1><p>Running Amethyst/build-{AMETHYST_BUILD_NUMBER}</p>"
            "</center></body></html>"
        )
        self.http_403_html = (
-            "<html><head><title>HTTP 403 - PyWebServer</title></head>"
-            f"<body><center><h1>HTTP 403 - Forbidden</h1><p>Running PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}</p>"
+            "<html><head><title>HTTP 403 - Amethyst</title></head>"
+            f"<body><center><h1>HTTP 403 - Forbidden</h1><p>Running Amethyst/build-{AMETHYST_BUILD_NUMBER}</p>"
            "</center></body></html>"
        )
        self.http_405_html = (
-            "<html><head><title>HTTP 405 - PyWebServer</title></head>"
-            f"<body><center><h1>HTTP 405 - Method not allowed</h1><p>Running PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}</p>"
+            "<html><head><title>HTTP 405 - Amethyst</title></head>"
+            f"<body><center><h1>HTTP 405 - Method not allowed</h1><p>Running Amethyst/build-{AMETHYST_BUILD_NUMBER}</p>"
            "</center></body></html>"
        )

@@ -505,7 +494,12 @@ class WebServer:
        else:
            return self.build_response(400, "You cannot connect without a User-Agent.")

-        method, path, version = self.parser.parse_request_line(request_line)
+        if ":" in host:
+            host2 = host.rsplit(":", 1)[0]
+        else:
+            host2 = host
+
+        method, path, version = self.parser.parse_request_line(request_line, host2)

        if not all([method, path, version]):
            return self.build_response(400, "Bad Request")
@@ -515,21 +509,24 @@ class WebServer:
            print("Got reload command! Reloading configuration...")
            self.file_handler = FileHandler()
            self.parser = RequestParser()
-            return self.build_response(302, "")
+            return self.build_response(302, "", host=host2)

        if not self.parser.is_method_allowed(method):
            return self.build_response(405, self.http_405_html)

-        if ":" in host:
-            host2 = host.rsplit(":", 1)[0]
-        else:
-            host2 = host
-
        directory = (
            self.file_handler.read_new_config("directory", host2)
            or self.file_handler.base_dir
        )

+        if self.file_handler.read_new_config("apimode", host2) is True:
+            if not os.path.join(os.getcwd(), directory) in sys.path:
+                sys.path.append(f"{os.path.join(os.getcwd(), directory)}")
+            import api
+
+            apiclass = api.API()
+            return apiclass.on_request(data)
+
        file_content, mimetype = self.file_handler.read_file(path, directory)

        if file_content == 403:
@@ -568,7 +565,7 @@ class WebServer:
        status_message = messages.get(status_code)
        headers = (
            f"HTTP/1.1 {status_code} {status_message}\r\n"
-            f"Server: PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}\r\n"
+            f"Server: Amethyst/amethyst-build-{AMETHYST_BUILD_NUMBER}\r\n"
            f"Content-Type: {content_type}\r\n"
            f"Content-Length: {len(binary_data)}\r\n"
            f"Connection: close\r\n\r\n"
@@ -577,7 +574,8 @@ class WebServer:
        )
        return headers.encode() + binary_data

-    def build_response(self, status_code, body):
+    @staticmethod
+    def build_response(status_code, body, host=None):
        """
        For textfiles we'll not have to guess MIME-types, though the other function
        build_binary_response will be merged in here anyway.
@@ -593,7 +591,7 @@ class WebServer:
            405: "Method Not Allowed",
            413: "Payload Too Large",
            500: "Internal Server Error",
-            635: "Go Away",
+            621: "fuck off! :3",
        }
        status_message = messages.get(status_code)

@@ -604,7 +602,7 @@ class WebServer:
        # Don't encode yet, if 302 status code we have to include location.
        headers = (
            f"HTTP/1.1 {status_code} {status_message}\r\n"
-            f"Server: PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}\r\n"
+            f"Server: Amethyst/build-{AMETHYST_BUILD_NUMBER}\r\n"
            f"Content-Length: {len(body)}\r\n"
            f"Connection: close\r\n\r\n"
        ).encode()
@@ -613,28 +611,28 @@ class WebServer:
            # 302 currently only happens when the reload is triggered.
            # Why not 307, Moved Permanently? Because browsers will cache the
            # response and not send the reload command.
-            host = self.file_handler.read_config("host")[0]
-            port = self.file_handler.read_config(
-                "port-https"
-            ) or self.file_handler.read_config("port")
-            if port != 80 and port != 443:
-                if port == 8443:
-                    host = f"https://{host}:{port}/"
-                else:
-                    host = f"http://{host}:{port}/"
-            else:
-                if port == 443:
-                    host = f"https://{host}/"
-                else:
-                    host = f"http://{host}/"
+            # if port == 443:
+            #     host = f"https://{host}/"
+            # else:
+            #     host = f"http://{host}/"
            headers = (
                f"HTTP/1.1 {status_code} {status_message}\r\n"
                f"Location: {host}\r\n"
-                f"Server: PyWebServer/amethyst-build-{AMETHYST_BUILD_NUMBER}\r\n"
+                f"Server: Amethyst/build-{AMETHYST_BUILD_NUMBER}\r\n"
                f"Content-Length: {len(body)}\r\n"
                f"Connection: close\r\n\r\n"
            ).encode()

+        if status_code == 621:
+            headers = (
+                f"HTTP/1.1 {status_code} {status_message}\r\n"
+                "Server: Amethyst/build-0621\r\n"
+                "Content-Length: 30\r\n"
+                f"Connection: close\r\n\r\n"
+            )
+            body = "https://e621.net/posts/6155664"
+
+        print(f"{headers + body}")
        return headers + body

    def shutdown(self, signum, frame):
@@ -660,7 +658,7 @@ def main():
    file_handler = FileHandler()
    file_handler.base_dir = file_handler.read_config("directory")
    http_port = file_handler.read_new_config("port") or 8080
-    https_port = file_handler.read_new_config("port-https") or 8443
+    https_port = file_handler.read_new_config("https-port") or 8443
    http_enabled = bool(file_handler.read_new_config("http")) or True
    print(http_enabled)
    https_enabled = bool(file_handler.read_new_config("https")) or False
Author	SHA1	Message	Date
Nova	b48b682da7	html, branding change and getting ready for pre-rel	2026-03-24 22:50:26 +01:00
Nova	97896c87a7	idk anymore man	2026-03-13 22:13:59 +01:00