X-Git-Url: https://git.r.bdr.sh/rbdr/gema_texto/blobdiff_plain/8beb2bfa1af988442550c9e6999d924dd14b5236..086ffc1eeecc98091adf894e01bfb60536149cb3:/src/html_renderer.rs?ds=sidebyside

diff --git a/src/html_renderer.rs b/src/html_renderer.rs
index e513533..5251d0a 100644
--- a/src/html_renderer.rs
+++ b/src/html_renderer.rs
@@ -1,4 +1,5 @@
 use crate::gemini_parser::GeminiLine;
+use std::path::Path;
 
 /// Renders HTML from a vector of `GeminiLine` elements.
 ///
@@ -99,8 +100,23 @@ fn line_content(line: &GeminiLine) -> String {
     match line {
         GeminiLine::Text(content, false) => format!("<p>{content}</p>"),
         GeminiLine::Link(url, text) => {
-            let display = if text.is_empty() { url } else { text };
-            format!("<p class=\"a\"><a href=\"{url}\">{display}</a></p>")
+            let path = Path::new(url);
+            let processed_url = if !url.starts_with("gemini:")
+                && path
+                    .extension()
+                    .map_or(false, |ext| ext.eq_ignore_ascii_case("gmi"))
+            {
+                url.replace(".gmi", ".html")
+            } else {
+                url.to_string()
+            };
+
+            let display = if text.is_empty() {
+                &processed_url
+            } else {
+                text
+            };
+            format!("<p class=\"a\"><a href=\"{processed_url}\">{display}</a></p>")
         }
         GeminiLine::Heading(level, content) => format!("<h{level}>{content}</h{level}>"),
         GeminiLine::ListItem(content) => format!("<li>{content}</li>"),
@@ -108,7 +124,10 @@ fn line_content(line: &GeminiLine) -> String {
             format!("<pre aria-label=\"{alt_text}\">")
         }
         GeminiLine::PreformattedToggle(false, _) => "</pre>".to_string(),
-        GeminiLine::Text(content, true) | GeminiLine::Quote(content) => content.to_string(),
+        GeminiLine::Text(content, true) | GeminiLine::Quote(content) => content
+            .replace('<', "&lt;")
+            .replace('>', "&gt;")
+            .to_string(),
     }
 }
 
@@ -223,12 +242,18 @@ mod tests {
     #[test]
     fn test_links() {
         let input = vec![
+            GeminiLine::Link("gemini://hi.gmi".to_string(), "Example".to_string()),
+            GeminiLine::Link("/hi.gmi/kidding".to_string(), "Example".to_string()),
+            GeminiLine::Link("/hi.gmi".to_string(), "Example".to_string()),
             GeminiLine::Link("https://example.com".to_string(), "Example".to_string()),
             GeminiLine::Link("https://rust-lang.org".to_string(), String::new()),
         ];
         assert_eq!(
             render_html(&input),
-            "<p class=\"a\"><a href=\"https://example.com\">Example</a></p>\n\
+            "<p class=\"a\"><a href=\"gemini://hi.gmi\">Example</a></p>\n\
+             <p class=\"a\"><a href=\"/hi.gmi/kidding\">Example</a></p>\n\
+             <p class=\"a\"><a href=\"/hi.html\">Example</a></p>\n\
+             <p class=\"a\"><a href=\"https://example.com\">Example</a></p>\n\
              <p class=\"a\"><a href=\"https://rust-lang.org\">https://rust-lang.org</a></p>\n"
         );
     }