Update main.py

main.py

@@ -54,34 +54,20 @@ def duckduckgo_search(query: str):
 
 # ========== OPDS Feed Generators ==========
 
-def generate_root_feed():
+def generate_root_feed(base_url: str):
     ns = "http://www.w3.org/2005/Atom"
     ET.register_namespace("", ns)
     feed = ET.Element("feed", xmlns=ns)
     ET.SubElement(feed, "title").text = "DuckDuckGo OPDS Catalog"
     ET.SubElement(feed, "updated").text = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
 
-    # Link to search
-    entry = ET.SubElement(feed, "entry")
-    ET.SubElement(entry, "title").text = "Search"
-    ET.SubElement(entry, "id").text = "search"
-    ET.SubElement(entry, "link", {
-        "rel": "subsection",
-        "href": "/opds/search",
-        "type": "application/atom+xml;profile=opds-catalog;kind=acquisition"
-    })
-    ET.SubElement(entry, "updated").text = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
-
-    # Link to cached (not implemented yet)
-    entry2 = ET.SubElement(feed, "entry")
-    ET.SubElement(entry2, "title").text = "Cached"
-    ET.SubElement(entry2, "id").text = "cached"
-    ET.SubElement(entry2, "link", {
-        "rel": "subsection",
-        "href": "/opds/cached",
-        "type": "application/atom+xml;profile=opds-catalog;kind=navigation"
-    })
-    ET.SubElement(entry2, "updated").text = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
+    # Templated search link (OpenSearch-style)
+    feed.append(ET.Element("link", {
+        "rel": "search",
+        "type": "application/atom+xml",
+        "href": f"{base_url}/opds/search?q={{searchTerms}}",
+        "templated": "true"
+    }))
 
     return ET.tostring(feed, encoding="utf-8", xml_declaration=True)
 
@@ -109,7 +95,9 @@ def generate_search_feed(query: str, results):
 
 @app.get("/opds", include_in_schema=False)
 def opds_root() -> Response:
-    xml_data = generate_root_feed()
+    # derive base URL dynamically or configure
+    base_url = "http://localhost:8000"
+    xml_data = generate_root_feed(base_url)
     return Response(content=xml_data, media_type="application/atom+xml")
 
 @app.get("/opds/search")
@@ -131,4 +119,5 @@ def download_fb2(url: str) -> Response:
         content=fb2,
         media_type="application/fb2+xml",
         headers={"Content-Disposition": f"attachment; filename={filename}"}
-    )
+    )
+