diff options
Diffstat (limited to 'src/gemini_parser.rs')
| -rw-r--r-- | src/gemini_parser.rs | 18 |
1 files changed, 14 insertions, 4 deletions
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()), |