Replace gemini links

[rbdr/page] / src / gemini_parser.rs

diff --git a/src/gemini_parser.rs b/src/gemini_parser.rs
index 5c204263bebede10c4c083ff70399ae1aa11b793..d0715f102246aef12dbeade159c285b9d856ccd8 100644 (file)
--- a/src/gemini_parser.rs
+++ b/src/gemini_parser.rs
@@ -8,9 +8,12 @@ pub fn parse(source: &str) -> String {
 
     for line in lines {
         let mut line_type = LineType::Text;
-        if line.len() > 2 {
-            let end = line.char_indices().map(|(i, _)| i).nth(2).unwrap();
-            line_type = identify_line(&(line[..end]), is_preformatted);
+        if line.char_indices().count() > 2 {
+            let mut end = line.len();
+            if line.char_indices().count() > 3 {
+                end = line.char_indices().map(|(i, _)| i).nth(3).unwrap();
+            }
+            line_type = identify_line(&line[..end], is_preformatted);
         }
         match line_type {
             LineType::PreformattedToggle => is_preformatted = !is_preformatted,
@@ -72,7 +75,14 @@ fn get_full_line_content(line_type: &LineType, line: &str) -> String {
      match line_type {
         LineType::Text => format!("<p>{}</p>\n", line.trim()),
         LineType::Blank => "<br/>\n".to_string(),
-        LineType::Link => format!("<div><a href=\"{}\">{}</a></div>\n", get_link_address(line), get_link_content(line)),
+        LineType::Link => {
+            let url = get_link_address(line);
+            if url.starts_with("gemini:") {
+                format!("<div><a href=\"{}\">{}</a></div>\n", url, get_link_content(line))
+            } else {
+                format!("<div><a href=\"{}\">{}</a></div>\n", url.replace(".gmi", ".html"), get_link_content(line))
+            }
+        },
         LineType::Heading1 => format!("<h1>{}</h1>\n", line[1..].trim()),
         LineType::Heading2 => format!("<h2>{}</h2>\n", line[2..].trim()),
         LineType::Heading3 => format!("<h3>{}</h3>\n", line[3..].trim()),