aboutsummaryrefslogtreecommitdiff
path: root/code/src/markdown/mod.rs
blob: dee923c1c8274b4801e730a0346b28b8950225de (plain)
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
use pest::{
    iterators::{Pair, Pairs},
    Parser,
};
use pest_derive::Parser;

use crate::{html::escape, syntax_highlight::syntax_highlight};

#[derive(Parser)]
#[grammar = "grammar/markdown.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 => format!("<h1>{}</h1>", render_pairs(p.into_inner())),
        Rule::paragraph => format!("<p>{}</p>", render_pairs(p.into_inner())),
        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::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::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 => format!("<span>TODO: Inline latex</span>"),
        Rule::latex_block => format!("<p>TODO: Latex block</p>"),

        Rule::text => escape(p.as_str()),

        _ => todo!("{:?}", p.as_rule()),
    }
}