Add RSS links to HTML

1 files changed, 18 insertions, 2 deletions

diff --git a/gemini-to-web/src/gemini_to_web/html.py b/gemini-to-web/src/gemini_to_web/html.py
index 9594155..ec2a640 100644
--- a/gemini-to-web/src/gemini_to_web/html.py
+++ b/gemini-to-web/src/gemini_to_web/html.py
@@ -12,7 +12,7 @@ def first_header_title_extractor(parsed: list[parser.GemElement]):
         return heading_lines[0].heading_text
 
 
-def to_html(parsed: list[parser.GemElement], title_extractor=first_header_title_extractor):
+def to_html(parsed: list[parser.GemElement], title_extractor=first_header_title_extractor, extra_head=None):
     body = []
     building_element = None
     building_content = None
@@ -27,6 +27,9 @@ def to_html(parsed: list[parser.GemElement], title_extractor=first_header_title_
     if title_extractor:
        head.append(htmlgenerator.TITLE(title_extractor(parsed)))
 
+    if extra_head:
+        head += extra_head
+
     for item in parsed:
         match item:
             case parser.HeadingLine(level, heading_text):
@@ -96,11 +99,24 @@ def pretty(s):
 
 
 def cli_to_html():
+    import argparse
     import sys
+
+    argument_parser = argparse.ArgumentParser()
+    argument_parser.add_argument("--feed-title")
+    argument_parser.add_argument("--feed-href")
+    args = argument_parser.parse_args()
+
+    assert not bool(args.feed_title) ^ bool(args.feed_href), "--feed-title and --feed-href must be both present or both absent"
+
+    extra_head = []
+    if args.feed_href:
+        extra_head.append(htmlgenerator.LINK(rel="alternate", type="application/rss+xml", title=args.feed_title, href=args.feed_href))
+
     input_ = sys.stdin.read()
     gemtext = parser.parse(input_)
     gemtext = list(gemtext)
-    html = to_html(gemtext)
+    html = to_html(gemtext, extra_head=extra_head)
     rendered = htmlgenerator.render(html, {})
     rendered = pretty(rendered)
     print(rendered)