Change track through web page

ac-server-scripts-api.py

@@ -0,0 +1,132 @@
+from http.server import HTTPServer, BaseHTTPRequestHandler
+import os
+import json
+from urllib.parse import parse_qs
+from change_track import change_track, get_all_tracks, get_configs, get_preview_image, get_outline_image
+
+class Handler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        if (self.handle_GET_path()):
+            return
+
+        self.send_error(404)
+
+    def do_POST(self):
+        if (self.handle_POST_path()):
+            return
+        
+        self.send_error(404)
+    
+    def handle_GET_path(self) -> bool:
+        """
+        handles the path entered in the GET request.
+        
+        Returns: true if the calling function should return, false otherwise
+        """
+        # Serve preview images
+        if self.path.startswith("/img/"):
+            parts = self.path.split("/")
+            parts = [p for p in parts if p]  # remove empty
+
+            img_type = parts[1] if len(parts) > 1 else ""
+            
+            track, config = self.extract_track_and_config(parts)
+            
+            img_path = ""
+            if (img_type == "preview"):
+                img_path = get_preview_image(track, config)
+            elif (img_type == "outline"):
+                img_path = get_outline_image(track, config)
+            else:
+                self.send_error(404)
+                return True
+                
+
+            self.send_image(img_path)
+
+            return True
+        
+                # JSON track info
+        if self.path.startswith("/track/"):
+            track = self.path.replace("/track/", "").strip("/")
+            configs = get_configs(track)
+
+            data = {
+                "track": track,
+                "configs": configs,
+                "image": f"/img/preview/{track}",
+                "outline": f"/img/outline/{track}"
+            }
+
+            self.send_response(200)
+            self.send_header("Content-type", "application/json")
+            self.end_headers()
+            self.wfile.write(json.dumps(data).encode())
+            return True
+        
+        if self.path == "/" or self.path == "/index.html":
+            with open("index.html", "r") as f:
+                html = f.read()
+
+            tracks = get_all_tracks()
+            track_options = "".join([f'<option value="{t}">{t}</option>' for t in tracks])
+
+            html = html.replace("{{tracks}}", track_options)
+            html = html.replace("{{configs}}", "")  # empty at load
+
+            self.send_response(200)
+            self.send_header("Content-type", "text/html")
+            self.end_headers()
+            self.wfile.write(html.encode())
+            return True
+        
+        return False
+    
+    def handle_POST_path(self):
+        if (self.path.startswith("/changetrack/")):
+            parts = self.path.split("/")
+            parts = [p for p in parts if p]  # remove empty
+            track, config = self.extract_track_and_config(parts, 1)
+            print(f"Changing track to '{track}' with config '{config}'")
+            
+            success, message = change_track(track, config)
+            print(message)
+            if success:
+                self.send_response(200)
+                self.send_header("Content-type", "application/json")
+                self.end_headers()
+                response = {"status": "success", "message": message}
+                self.wfile.write(json.dumps(response).encode())
+                return True
+            else:
+                self.send_response(400)
+                self.send_header("Content-type", "application/json")
+                self.end_headers()
+                response = {"status": "error", "message": message}
+                self.wfile.write(json.dumps(response).encode())
+                return True
+        return False
+    
+    def send_image(self, img_path: str):
+            if os.path.exists(img_path):
+                self.send_response(200)
+                self.send_header("Content-type", "image/png")
+                self.end_headers()
+                with open(img_path, "rb") as f:
+                    self.wfile.write(f.read())
+            else:
+                self.send_error(404)
+                
+    def extract_track_and_config(self, parts: list, track_index: int = 2) -> tuple[str, str]:
+        print(parts)
+        track = parts[track_index] if len(parts) > track_index else ""
+        config = parts[track_index + 1] if len(parts) > track_index + 1 else ""
+        
+        return track, config
+        
+        
+
+
+server = HTTPServer(("0.0.0.0", 10303), Handler)
+print("Server running on port 10303")
+server.serve_forever()