1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
use pest::{
iterators::{Pair, Pairs},
Parser,
};
use pest_derive::Parser;
use crate::{html::escape, syntax_highlight::syntax_highlight};
#[derive(Parser)]
#[grammar = "src/markdown/parser.pest"]
struct Grammar;
pub fn render(s: &str) -> String {
match Grammar::parse(Rule::file, s) {
Ok(pairs) => {
eprintln!("{pairs:#?}");
render_pairs(pairs)
}
Err(e) => panic!("{e}"),
}
}
pub fn render_pairs(p: Pairs<Rule>) -> String {
p.map(|p| render_ast(p)).collect::<Vec<_>>().join("")
}
pub fn render_ast(p: Pair<Rule>) -> String {
match p.as_rule() {
Rule::block => render_pairs(p.into_inner()),
Rule::header => {
let mut level = 0;
while p.as_str()[level..].starts_with("#") {
level += 1
}
format!("<h{level}>{}</h{level}>", render_pairs(p.into_inner()))
}
Rule::paragraph => format!("<p>{}</p>", render_pairs(p.into_inner())),
Rule::style_italic => format!("<i>{}</i>", render_pairs(p.into_inner())),
Rule::style_bold => format!("<b>{}</b>", render_pairs(p.into_inner())),
Rule::style_code => format!("<code>{}</code>", p.into_inner().next().unwrap().as_str()),
Rule::unordered_list => format!("<ul>{}</ul>", render_pairs(p.into_inner())),
Rule::ordered_list => format!("<ol>{}</ol>", render_pairs(p.into_inner())),
Rule::unordered_list_item | Rule::ordered_list_item => {
format!("<li>{}</li>", render_pairs(p.into_inner()))
}
Rule::hyperlink => {
let k = p.into_inner().collect::<Vec<_>>();
let label = k[0].as_str();
let target = k[1].as_str();
format!("<a href=\"{}\">{}</a>", escape(target), escape(label))
}
Rule::span => render_pairs(p.into_inner()),
Rule::EOI => "".to_string(),
Rule::code_block => {
let k = p.into_inner().collect::<Vec<_>>();
let lang = k[0].as_str();
let inner = k[1].as_str();
format!(
"<pre>{}</pre>",
syntax_highlight(lang, &inner).unwrap_or_else(|| escape(&inner))
)
}
Rule::inline_latex => fix_katex(
&katex::render_with_opts(
&p.into_inner().as_str(),
&katex::OptsBuilder::default().build().unwrap(),
)
.unwrap(),
),
Rule::latex_block => fix_katex(
&katex::render_with_opts(
&p.into_inner().as_str(),
&katex::OptsBuilder::default()
.display_mode(true)
.build()
.unwrap(),
)
.unwrap(),
),
Rule::text => escape(p.as_str()),
_ => todo!("{:?}", p.as_rule()),
}
}
// TODO this is *really* bad fix
fn fix_katex<'a>(s: &str) -> String {
let e = s.find("<span class=\"katex-html\"").unwrap();
s[0..e].replace(
"<mspace linebreak=\"newline\"></mspace>",
"</mrow></semantics></math><math xmlns=\"http://www.w3.org/1998/Math/MathML\" display=\"block\"><semantics><mrow>",
)
}
|
