Update main.py

main.py

@@ -6,6 +6,7 @@ import requests
 from bs4 import BeautifulSoup
 from urllib.parse import quote
 import xml.etree.ElementTree as ET
+from datetime import datetime
 
 app = FastAPI()
 
@@ -51,18 +52,52 @@ def duckduckgo_search(query: str):
             break
     return results
 
-# ========== OPDS Feed Generator ==========
-def generate_opds(query: str, results):
+# ========== OPDS Feed Generators ==========
+
+def generate_root_feed():
+    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")
+
+    return ET.tostring(feed, encoding="utf-8", xml_declaration=True)
+
+
+def generate_search_feed(query: str, results):
     ns = "http://www.w3.org/2005/Atom"
     ET.register_namespace("", ns)
     feed = ET.Element("feed", xmlns=ns)
     ET.SubElement(feed, "title").text = f"Search results for '{query}'"
-    ET.SubElement(feed, "updated").text = "2025-07-31T12:00:00Z"
+    ET.SubElement(feed, "updated").text = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
+
     for title, url in results:
         entry = ET.SubElement(feed, "entry")
         ET.SubElement(entry, "title").text = title
         ET.SubElement(entry, "id").text = url
-        ET.SubElement(entry, "updated").text = "2025-07-31T12:00:00Z"
+        ET.SubElement(entry, "updated").text = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
         ET.SubElement(entry, "link", {
             "rel": "http://opds-spec.org/acquisition",
             "href": f"/download?url={quote(url, safe='')}",
@@ -70,10 +105,17 @@ def generate_opds(query: str, results):
         })
     return ET.tostring(feed, encoding="utf-8", xml_declaration=True)
 
-@app.get("/opds")
-def opds_catalog(q: str = Query(..., description="Search query")) -> Response:
+# ========== Routes ==========
+
+@app.get("/opds", include_in_schema=False)
+def opds_root() -> Response:
+    xml_data = generate_root_feed()
+    return Response(content=xml_data, media_type="application/atom+xml")
+
+@app.get("/opds/search")
+def opds_search(q: str = Query(..., description="Search query")) -> Response:
     results = duckduckgo_search(q)
-    xml_data = generate_opds(q, results)
+    xml_data = generate_search_feed(q, results)
     return Response(content=xml_data, media_type="application/atom+xml")
 
 @app.get("/download")
@@ -89,4 +131,4 @@ def download_fb2(url: str) -> Response:
         content=fb2,
         media_type="application/fb2+xml",
         headers={"Content-Disposition": f"attachment; filename={filename}"}
-    )
+    )