From 362b6fa165451d3f52bb5bb0a160c76db92b726d Mon Sep 17 00:00:00 2001
From: Timothy Farrell <tim@thecookiejar.me>
Date: Thu, 12 Mar 2026 08:38:27 +0000
Subject: [PATCH] Add refresh and sequence

---
 frontend.html |  1 +
 main.py       | 78 +++++++++++++++++++++++++++++++++++++++------------
 2 files changed, 61 insertions(+), 18 deletions(-)

diff --git a/frontend.html b/frontend.html
index 9e4f968..e5891cc 100644
--- a/frontend.html
+++ b/frontend.html
@@ -4,6 +4,7 @@
     <meta charset="utf-8">
     <meta name="viewport" content="width=device-width, initial-scale=1">
     <title>Image Server</title>
+    $extra_meta
     <style>
         * { margin: 0; padding: 0; box-sizing: border-box; }
         html, body { height: 100%; overflow: hidden; background: #1a1a1a; }
diff --git a/main.py b/main.py
index c8db8fb..90a4831 100644
--- a/main.py
+++ b/main.py
@@ -129,34 +129,76 @@ async def root():
     return RedirectResponse(url="/{hash}".format(hash=random_hash))
 
 
+def _get_navigation_data(file_hash: str):
+    """Get navigation data for a file hash"""
+    keys = list(file_mapping.keys())
+    idx = keys.index(file_hash)
+    next_hash = keys[(idx + 1) % len(keys)]
+    prev_hash = keys[idx - 1] if idx > 0 else keys[-1]
+    next_random_hash = _get_random_hash()
+    indexer = _find_indexer_for_hash(file_hash)
+    filename = indexer.get_filename_by_hash(file_hash) if indexer else ""
+
+    return {
+        "file_hash": file_hash,
+        "next_hash": next_hash,
+        "prev_hash": prev_hash,
+        "next_random_hash": next_random_hash,
+        "filename": filename,
+    }
+
+
+def _render_page(navigation_data: dict, extra_meta: str = "") -> HTMLResponse:
+    """Render the frontend page with navigation data"""
+    with open("frontend.html", "r") as f:
+        content = f.read()
+
+    template = string.Template(content)
+    content = template.substitute(
+        img_url="/api/{file_hash}/data".format(file_hash=navigation_data["file_hash"]),
+        next_random_url="/{next_random_hash}".format(
+            next_random_hash=navigation_data["next_random_hash"]
+        ),
+        next_url="/{next_hash}".format(next_hash=navigation_data["next_hash"]),
+        prev_url="/{prev_hash}".format(prev_hash=navigation_data["prev_hash"]),
+        filename=navigation_data["filename"],
+        extra_meta=extra_meta,
+    )
+
+    return HTMLResponse(content=content)
+
+
 @app.get("/{file_hash}")
 async def hash_page(file_hash: str):
     """Serve a page for a specific file hash with navigation"""
     if file_hash not in file_mapping:
         raise HTTPException(status_code=404, detail="File not found")
 
-    keys = list(file_mapping.keys())
-    idx = keys.index(file_hash)
-    next_hash = keys[(idx + 1) % len(keys)]
-    prev_hash = keys[idx - 1] if idx > 0 else keys[-1]
-    next_random_hash=_get_random_hash()
+    navigation_data = _get_navigation_data(file_hash)
+    return _render_page(navigation_data)
 
-    indexer = _find_indexer_for_hash(file_hash)
-    filename = indexer.get_filename_by_hash(file_hash) if indexer else ""
 
-    with open("frontend.html", "r") as f:
-        content = f.read()
+@app.get("/{order}/{delay}/{file_hash}")
+async def hash_page_with_refresh(order: str, delay: int, file_hash: str):
+    """Serve a page for a specific file hash with auto-refresh navigation"""
+    if file_hash not in file_mapping:
+        raise HTTPException(status_code=404, detail="File not found")
 
-    template = string.Template(content)
-    content = template.substitute(
-        img_url="/api/{file_hash}/data".format(file_hash=file_hash),
-        next_random_url="/{next_random_hash}".format(next_random_hash=next_random_hash),
-        next_url="/{next_hash}".format(next_hash=next_hash),
-        prev_url="/{prev_hash}".format(prev_hash=prev_hash),
-        filename=filename,
-    )
+    if order not in ("next", "random"):
+        raise HTTPException(status_code=400, detail="Invalid order. Must be 'next' or 'random'")
 
-    return HTMLResponse(content=content)
+    navigation_data = _get_navigation_data(file_hash)
+
+    if order == "next":
+        refresh_url = "/{next_hash}".format(next_hash=navigation_data["next_hash"])
+    else:
+        refresh_url = "/{next_random_hash}".format(
+            next_random_hash=navigation_data["next_random_hash"]
+        )
+
+    refresh_meta = f'<meta http-equiv="refresh" content="{delay};url={refresh_url}">'
+
+    return _render_page(navigation_data, refresh_meta)
 
 
 def _find_indexer_for_hash(file_hash: str):