fix: line breaks and link titles in gemini

This commit is contained in:
DarkCat09 2024-11-21 16:24:59 +04:00
parent be57431311
commit c7c94581b6
Signed by: DarkCat09
GPG key ID: BD3CE9B65916CD82
2 changed files with 127 additions and 49 deletions

View file

@ -9,7 +9,8 @@ mydata: 123
- ненумерованный список, элемент 1 - ненумерованный список, элемент 1,
даже с продолжением
- элемент 2 - элемент 2
- element 3 - element 3
@ -21,9 +22,13 @@ mydata: 123
- [ ] Todo - [ ] Todo
- [ ] Won't fix - [ ] Won't fix
[[x] Task inside link](https://dc09.ru)
> [!WARNING] > [!WARNING]
> >
> Some important quote > Some important quote
> continuing here \
> With line break
text text text `inline code` text text text text text text `inline code` text text text

View file

@ -16,12 +16,15 @@ pub fn compile_markdown(
let mut gmi = create_file(gmi)?; let mut gmi = create_file(gmi)?;
let mut state = State::Start; let mut state = State::Start;
let mut ol = false; // ordered list
let mut counter: u64 = 0; let mut counter: u64 = 0;
let mut links: Vec<GmiLink<'_>> = vec![]; let mut links: Vec<GmiLink<'_>> = vec![];
for event in Parser::new_ext(&src_text, Options::all()) { for event in Parser::new_ext(&src_text, Options::all()) {
use pulldown_cmark::Event::*; use pulldown_cmark::Event::*;
dbg!(&event);
match event { match event {
Start(tag) => { Start(tag) => {
use pulldown_cmark::{CodeBlockKind, HeadingLevel, Tag::*}; use pulldown_cmark::{CodeBlockKind, HeadingLevel, Tag::*};
@ -29,9 +32,7 @@ pub fn compile_markdown(
match tag { match tag {
Paragraph => { Paragraph => {
html.write_all(b"<p>")?; html.write_all(b"<p>")?;
if state != State::Start { write_paragraph_start(&mut gmi, state)?;
gmi.write_all(b"\r\n\r\n")?;
}
} }
Heading { level, id, .. } => { Heading { level, id, .. } => {
@ -41,95 +42,110 @@ pub fn compile_markdown(
html.write_fmt(format_args!("<{}>", level))?; html.write_fmt(format_args!("<{}>", level))?;
} }
write_paragraph_start(&mut gmi, state)?;
let hashes = match level { let hashes = match level {
HeadingLevel::H1 => "# ", HeadingLevel::H1 => "# ",
HeadingLevel::H2 => "## ", HeadingLevel::H2 => "## ",
_ => "### ", _ => "### ",
}; };
gmi.write_all(hashes.as_bytes())?; gmi.write_all(hashes.as_bytes())?;
state = State::GmiPrefix(hashes); state = State::Paragraph;
} }
BlockQuote(_kind) => { BlockQuote(_kind) => {
html.write_all(b"<blockquote>")?; html.write_all(b"<blockquote>")?;
write_paragraph_start(&mut gmi, state)?;
gmi.write_all(b"> ")?; gmi.write_all(b"> ")?;
state = State::GmiPrefix("> "); state = State::Quote;
} }
CodeBlock(CodeBlockKind::Fenced(lang)) => { CodeBlock(CodeBlockKind::Fenced(lang)) => {
html.write_all(b"<pre><code>")?;
// TODO: highlighting with syntect // TODO: highlighting with syntect
html.write_all(b"<pre><code>")?;
write_paragraph_start(&mut gmi, state)?;
gmi.write_fmt(format_args!("```{}\r\n", lang))?; gmi.write_fmt(format_args!("```{}\r\n", lang))?;
} }
CodeBlock(CodeBlockKind::Indented) => { CodeBlock(CodeBlockKind::Indented) => {
html.write_all(b"<pre><code>")?; html.write_all(b"<pre><code>")?;
write_paragraph_start(&mut gmi, state)?;
gmi.write_all(b"```")?; gmi.write_all(b"```")?;
} }
List(None) => { List(None) => {
html.write_all(b"<ul>")?; html.write_all(b"<ul>")?;
write_paragraph_start(&mut gmi, state)?;
state = State::Start;
} }
List(Some(counter_start)) => { List(Some(counter_start)) => {
counter = counter_start; counter = counter_start;
ol = true;
if counter_start == 1 { if counter_start == 1 {
html.write_all(b"<ol>")?; html.write_all(b"<ol>")?;
} else { } else {
html.write_fmt(format_args!("<ol start=\"{}\">", counter_start))?; html.write_fmt(format_args!("<ol start=\"{}\">", counter_start))?;
} }
write_paragraph_start(&mut gmi, state)?;
state = State::Start;
} }
Item => { Item => {
html.write_all(b"<li>")?; html.write_all(b"<li>")?;
gmi.write_fmt(format_args!("{}. ", counter))?;
counter += 1; if state != State::Start {
gmi.write_all(b"\r\n")?;
state = State::Paragraph;
}
if ol {
gmi.write_fmt(format_args!("{}. ", counter))?;
counter += 1;
} else {
gmi.write_all(b"* ")?;
}
} }
Strong => { Strong => {
html.write_all(b"<b>")?; html.write_all(b"<b>")?;
gmi.write_all(b"**")?; write_inline(&mut gmi, state, "**", &mut links)?;
} }
Emphasis => { Emphasis => {
html.write_all(b"<i>")?; html.write_all(b"<i>")?;
gmi.write_all(b"*")?; write_inline(&mut gmi, state, "*", &mut links)?;
} }
Strikethrough => { Strikethrough => {
html.write_all(b"<s>")?; html.write_all(b"<s>")?;
gmi.write_all(b"~")?; write_inline(&mut gmi, state, "~", &mut links)?;
} }
Link { Link { dest_url, .. } => {
dest_url, title, .. html.write_fmt(format_args!("<a href=\"{}\">", dest_url))?;
} => {
html.write_fmt(format_args!("<a href=\"{}\">{}</a>", dest_url, title))?;
gmi.write_fmt(format_args!("{}[{}]", &title, links.len() + 1))?;
links.push(GmiLink { links.push(GmiLink {
title, title: String::new(),
url: dest_url, url: dest_url,
}); });
state = State::Link;
} }
Image { Image { dest_url, .. } => {
dest_url, title, ..
} => {
html.write_fmt(format_args!(
"<img src=\"{}\" alt=\"{}\">",
dest_url, title
))?;
gmi.write_fmt(format_args!("{}[{}]", &title, links.len() + 1))?;
links.push(GmiLink { links.push(GmiLink {
title, title: String::new(),
url: dest_url, url: dest_url,
}); });
state = State::Link;
} }
Table(_align) => { Table(_align) => {
html.write_all(b"<table>")?; html.write_all(b"<table>")?;
write_paragraph_start(&mut gmi, state)?;
// gmi ?? // gmi ??
} }
@ -150,6 +166,10 @@ pub fn compile_markdown(
} }
} }
MetadataBlock(_) => {
state = State::Metadata(state != State::Start);
}
other => { other => {
eprintln!("Unsupported tag: {:?}", other); eprintln!("Unsupported tag: {:?}", other);
} }
@ -164,10 +184,10 @@ pub fn compile_markdown(
html.write_all(b"</p>")?; html.write_all(b"</p>")?;
if !links.is_empty() { if !links.is_empty() {
gmi.write_all(b"\r\n\r\n")?; gmi.write_all(b"\r\n")?;
for (i, link) in links.iter().enumerate() { for (i, link) in links.iter().enumerate() {
gmi.write_fmt(format_args!( gmi.write_fmt(format_args!(
"=> {} [{}]: {}", "\r\n=> {} [{}]: {}", // => https://... [1]: example
link.url, i, link.title, link.url, i, link.title,
))?; ))?;
} }
@ -183,6 +203,7 @@ pub fn compile_markdown(
List(ordered) => { List(ordered) => {
if ordered { if ordered {
html.write_all(b"</ol>")?; html.write_all(b"</ol>")?;
ol = false;
} else { } else {
html.write_all(b"</ul>")?; html.write_all(b"</ul>")?;
} }
@ -190,17 +211,23 @@ pub fn compile_markdown(
Item => html.write_all(b"</li>")?, Item => html.write_all(b"</li>")?,
Strong => { Strong => {
html.write_all(b"</b>")?; html.write_all(b"</b>")?;
gmi.write_all(b"**")?; write_inline(&mut gmi, state, "**", &mut links)?;
} }
Emphasis => { Emphasis => {
html.write_all(b"</i>")?; html.write_all(b"</i>")?;
gmi.write_all(b"*")?; write_inline(&mut gmi, state, "*", &mut links)?;
} }
Strikethrough => { Strikethrough => {
html.write_all(b"</s>")?; html.write_all(b"</s>")?;
gmi.write_all(b"~")?; write_inline(&mut gmi, state, "~", &mut links)?;
}
Link | Image => {
gmi.write_fmt(format_args!(
"{}[{}]", // example[1] ...\r\n => https://... [1]: example
links.last().unwrap().title,
links.len(),
))?;
} }
Link => {}
TableHead => html.write_all(b"</tr></thead><tbody>")?, TableHead => html.write_all(b"</tr></thead><tbody>")?,
TableRow => html.write_all(b"</tr>")?, TableRow => html.write_all(b"</tr>")?,
TableCell => { TableCell => {
@ -214,12 +241,27 @@ pub fn compile_markdown(
_ => {} _ => {}
} }
state = State::Paragraph; match state {
State::Metadata(false) => {
state = State::Start;
}
_ => {
state = State::Paragraph;
}
}
} }
Text(text) => { Text(text) => {
match state {
State::Metadata(_) => {
// TODO: parse yaml
continue;
}
_ => {}
}
html.write_all(text.as_bytes())?; html.write_all(text.as_bytes())?;
gmi.write_all(text.as_bytes())?; write_inline(&mut gmi, state, &text, &mut links)?;
} }
Code(code) => { Code(code) => {
@ -227,33 +269,39 @@ pub fn compile_markdown(
html.write_all(code.as_bytes())?; html.write_all(code.as_bytes())?;
html.write_all(b"</code>")?; html.write_all(b"</code>")?;
gmi.write_all(b"`")?; write_inline(&mut gmi, state, "`", &mut links)?;
gmi.write_all(code.as_bytes())?; write_inline(&mut gmi, state, &code, &mut links)?;
gmi.write_all(b"`")?; write_inline(&mut gmi, state, "`", &mut links)?;
} }
SoftBreak => {} SoftBreak => {
html.write_all(b" ")?;
write_inline(&mut gmi, state, " ", &mut links)?;
}
HardBreak => { HardBreak => {
html.write_all(b"<br>")?; html.write_all(b"<br>")?;
gmi.write_all(b"\r\n")?; gmi.write_all(b"\r\n")?;
if let State::GmiPrefix(prefix) = state { if state == State::Quote {
gmi.write_all(prefix.as_bytes())?; gmi.write_all(b"> ")?;
} }
} }
Rule => { Rule => {
html.write_all(b"<hr>")?; html.write_all(b"<hr>")?;
write_paragraph_start(&mut gmi, state)?;
gmi.write_all(b"---")?; gmi.write_all(b"---")?;
} }
TaskListMarker(state) => { TaskListMarker(done) => {
if state { if done {
html.write_all(b"<input type=checkbox checked disabled>")?; html.write_all(b"<input type=checkbox checked disabled>")?;
gmi.write_all(b"[x]")?; write_inline(&mut gmi, state, "[x] ", &mut links)?;
} else { } else {
html.write_all(b"<input type=checkbox disabled>")?; html.write_all(b"<input type=checkbox disabled>")?;
gmi.write_all(b"[ ]")?; write_inline(&mut gmi, state, "[ ] ", &mut links)?;
} }
} }
@ -278,15 +326,40 @@ fn create_file(path: impl AsRef<Path>) -> std::io::Result<BufWriter<std::fs::Fil
Ok(BufWriter::new(f)) Ok(BufWriter::new(f))
} }
#[inline(always)]
fn write_paragraph_start(gmi: &mut impl std::io::Write, state: State) -> std::io::Result<()> {
if state != State::Start && state != State::Quote {
gmi.write_all(b"\r\n\r\n")?;
}
Ok(())
}
#[inline(always)]
fn write_inline(
gmi: &mut impl std::io::Write,
state: State,
text: &str,
links: &mut Vec<GmiLink>,
) -> std::io::Result<()> {
if state == State::Link {
links.last_mut().unwrap().title.push_str(text);
} else {
gmi.write_all(text.as_bytes())?;
}
Ok(())
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)] #[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum State { enum State {
Start, Start,
Paragraph, Paragraph,
GmiPrefix(&'static str), Link,
Quote,
TableHead, TableHead,
Metadata(bool),
} }
struct GmiLink<'link> { struct GmiLink<'link> {
title: CowStr<'link>, title: String,
url: CowStr<'link>, url: CowStr<'link>,
} }