Replace stateful cache with atomic C controller

2026-08-29 21:56:05 +10:00
parent a66945c245
commit bd2b623633
12 changed files with 1823 additions and 1199 deletions
@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+import json
+import os
+import sys
+from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
+from urllib.parse import urlparse
+
+view = os.environ["TEST_VIEW"]
+state_path = os.environ["TEST_STATE"]
+codec = "hevc"
+depth = 10
+refresh_attempts = 0
+
+
+def record_state():
+    with open(state_path, "w") as handle:
+        json.dump({"codec": codec, "depth": depth, "attempts": refresh_attempts}, handle)
+
+
+def refresh_media():
+    global codec, depth
+    link = os.path.join(view, "movies/Test/Test.mkv")
+    try:
+        target = os.readlink(link)
+    except OSError:
+        target = ""
+    codec, depth = ("h264", 8) if target.startswith("/data/cache/") else ("hevc", 10)
+    record_state()
+
+
+class Handler(BaseHTTPRequestHandler):
+    def send_json(self, value, status=200):
+        body = json.dumps(value, separators=(",", ":")).encode()
+        self.send_response(status)
+        self.send_header("Content-Type", "application/json")
+        self.send_header("Content-Length", str(len(body)))
+        self.end_headers()
+        self.wfile.write(body)
+
+    def do_GET(self):
+        path = urlparse(self.path).path
+        if path == "/Sessions":
+            self.send_json([])
+        elif path == "/Items/test-item":
+            self.send_json({"Id": "test-item", "Path": "/data/library/movies/Test/Test.mkv"})
+        elif path == "/Items":
+            self.send_json({
+                "Items": [{
+                    "Id": "test-item",
+                    "Path": "/data/library/movies/Test/Test.mkv",
+                    "RunTimeTicks": 6 * 10_000_000,
+                    "MediaStreams": [{
+                        "Type": "Video", "Codec": codec, "BitDepth": depth,
+                        "Profile": "Main 10" if depth == 10 else "High",
+                        "Width": 1280, "Height": 720,
+                    }],
+                }],
+                "TotalRecordCount": 1,
+            })
+        else:
+            self.send_json({"error": "not found"}, 404)
+
+    def do_POST(self):
+        global refresh_attempts
+        path = urlparse(self.path).path
+        if path.startswith("/Items/test-item/Refresh"):
+            refresh_attempts += 1
+            if refresh_attempts == 1:
+                record_state()
+                self.send_response(503)
+                self.end_headers()
+                return
+            refresh_media()
+            self.send_response(204)
+            self.end_headers()
+        elif path == "/Library/Refresh":
+            self.send_response(204)
+            self.end_headers()
+        else:
+            self.send_json({"error": "not found"}, 404)
+
+    def log_message(self, fmt, *args):
+        pass
+
+
+record_state()
+server = ThreadingHTTPServer(("0.0.0.0", int(sys.argv[1])), Handler)
+server.serve_forever()