summaryrefslogtreecommitdiff
path: root/src/wikitext.rs
blob: 54570faadab8a702fa5e732522c0fc5bedb35f32 (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
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
use crate::directive::Directive;
use crate::error::SiteError;
use crate::page::PageMeta;
use crate::site::Site;
use log::trace;
use std::collections::HashMap;
use std::path::Path;
#[derive(Debug, Eq, PartialEq)]
pub enum Snippet {
    Markdown(String),
    WikiLink(WikiLink),
    Directive(Directive),
}

impl Snippet {
    pub fn process(&self, site: &mut Site, meta: &mut PageMeta) -> Result<String, SiteError> {
        trace!("Snippet::process: self={:?}", self);
        let s = match self {
            Snippet::Markdown(text) => text.into(),
            Snippet::WikiLink(w) => {
                let resolved = site.resolve(meta.path(), Path::new(w.target()))?;
                trace!("resolved {} to {}", w.target(), resolved.display());
                format!("[{}]({})", w.link_text(), resolved.display())
            }
            Snippet::Directive(d) => d.process(site, meta)?,
        };
        Ok(s)
    }
}

#[derive(Debug, Eq, PartialEq)]
pub struct WikiLink {
    link_text: String,
    target: String,
}

impl WikiLink {
    pub fn new(link_text: &str, target: &str) -> Self {
        Self {
            link_text: link_text.into(),
            target: target.into(),
        }
    }

    fn link_text(&self) -> &str {
        &self.link_text
    }

    fn target(&self) -> &str {
        &self.target
    }
}

#[derive(Debug, Eq, PartialEq)]
pub struct ParsedDirective {
    name: String,
    args: HashMap<String, String>,
}

impl ParsedDirective {
    pub fn new(name: &str, args: HashMap<String, String>) -> Result<Self, SiteError> {
        trace!("ParsedDirective::new: name={:?} args={:?}", name, args);
        Ok(Self {
            name: name.into(),
            args,
        })
    }

    pub fn name(&self) -> &str {
        &self.name
    }

    pub fn args(&self) -> HashMap<&str, &str> {
        self.args
            .iter()
            .map(|(k, v)| (k.as_str(), v.as_str()))
            .collect()
    }

    pub fn unnamed_args(&self) -> Vec<&str> {
        self.args
            .iter()
            .filter_map(|(k, v)| {
                if v.is_empty() {
                    Some(k.as_str())
                } else {
                    None
                }
            })
            .collect()
    }
}