src/directive/shortcut.rs


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

use crate::directive::DirectiveImplementation;
use crate::error::SiteError;
use crate::page::PageMeta;
use crate::site::{Shortcut as S, Site};
use crate::wikitext::ParsedDirective;

use log::trace;

#[derive(Debug, Eq, PartialEq)]
pub struct Shortcut {
    shortcut: S,
}

impl DirectiveImplementation for Shortcut {
    const REQUIRED: &'static [&'static str] = &["name", "url"];
    const ALLOWED: &'static [&'static str] = &["desc"];
    const ALLOW_ANY_UNNAMED: bool = false;

    fn from_parsed(p: &ParsedDirective) -> Self {
        let args = p.args();
        let name = args.get("name").unwrap();
        let desc = args.get("desc").unwrap_or(&"");
        let url = args.get("url").unwrap();
        Self::new(S::new(name, desc, url))
    }

    fn prepare(&self, site: &mut Site) -> Result<(), SiteError> {
        trace!("shortcut: prepare");
        site.add_shortcut(self.shortcut.clone());
        Ok(())
    }

    fn process(&self, _site: &Site, _meta: &mut PageMeta) -> Result<String, SiteError> {
        Ok("".into())
    }
}

impl Shortcut {
    pub fn new(shortcut: S) -> Self {
        Self { shortcut }
    }
}