X-Git-Url: https://git.r.bdr.sh/rbdr/page/blobdiff_plain/68fa37d61babe8d990ace5f9cb5a7de81eb57ea0..0a7a1f274b3fa8469e7078332aa493f3ae653f08:/src/gemini_parser.rs

diff --git a/src/gemini_parser.rs b/src/gemini_parser.rs
index 5c20426..d0715f1 100644
--- 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()),