nmd_core/codex/modifier/
standard_text_modifier.rs

1use std::collections::HashMap;
2
3use once_cell::sync::Lazy;
4use regex::Regex;
5
6use super::{base_modifier::BaseModifier, constants::{ABRIDGED_STYLE_PATTERN, IDENTIFIER_PATTERN, NEW_LINE_PATTERN, STYLE_PATTERN}, ModifierIdentifier, ModifiersBucket};
7
8
9static MODIFIER_PATTERNS_REGEX: Lazy<HashMap<ModifierIdentifier, Regex>> = Lazy::new(|| {
10    let mut regex: HashMap<ModifierIdentifier, Regex> = HashMap::new();
11
12    StandardTextModifier::ordered().into_iter().for_each(|m| {
13        regex.insert(m.identifier(), Regex::new(&m.modifier_pattern()).unwrap());
14    });
15
16    regex
17});
18
19
20#[derive(Debug, PartialEq, Clone)]
21pub enum StandardTextModifier {
22    
23    BoldStarVersion,
24    BoldUnderscoreVersion,
25    ItalicStarVersion,
26    ItalicUnderscoreVersion,
27    Strikethrough,
28    Underlined,
29    Link,
30    AbridgedEmbeddedStyle,
31    EmbeddedStyle,
32    Identifier,
33    Highlight,
34    Emoji,
35    Superscript,
36    Subscript,
37    InlineCode,
38    InlineMath,
39    Comment,
40    AbridgedBookmark,
41    Bookmark,
42    Todo,
43    Checkbox,
44    CheckboxChecked,
45    GreekLetter,
46    Escape,
47    Reference,
48    Cite,
49}
50
51impl StandardTextModifier {
52
53    pub fn ordered() -> Vec<Self> {
54
55        //! they must have the compatibility order
56        vec![
57            Self::InlineCode,
58            Self::InlineMath,
59            Self::Comment,
60            Self::GreekLetter,
61            Self::Todo,
62            Self::Bookmark,
63            Self::AbridgedBookmark,
64            Self::EmbeddedStyle,
65            Self::AbridgedEmbeddedStyle,
66            Self::Identifier,
67            Self::Highlight,
68            Self::BoldStarVersion,
69            Self::BoldUnderscoreVersion,
70            Self::ItalicStarVersion,
71            Self::ItalicUnderscoreVersion,
72            Self::Strikethrough,
73            Self::Underlined,
74            Self::Superscript,
75            Self::Subscript,
76            Self::Link,
77            Self::Checkbox,
78            Self::CheckboxChecked,
79            Self::Emoji,
80            Self::Escape,
81            Self::Reference,
82            Self::Cite,
83        ]
84    }
85
86    pub fn identifier(&self) -> ModifierIdentifier {
87        match self {
88            Self::AbridgedBookmark => String::from("abridged-bookmark"),
89            Self::Bookmark => String::from("bookmark"),
90            Self::Todo => String::from("todo"),
91            Self::AbridgedEmbeddedStyle => String::from("abridged-embedded-style"),
92            Self::Identifier => String::from("identifier"),
93            Self::EmbeddedStyle => String::from("embedded-style"),
94            Self::Highlight => String::from("highlight"),
95            Self::Comment => String::from("comment"),
96            Self::Emoji => String::from("emoji"),
97            Self::Checkbox => String::from("checkbox"),
98            Self::CheckboxChecked => String::from("checkbox-checked"),
99            Self::Superscript => String::from("superscript"),
100            Self::Subscript => String::from("subscript"),
101            Self::BoldStarVersion => String::from("bold-star-version"),
102            Self::BoldUnderscoreVersion => String::from("bold-underscore-version"),
103            Self::ItalicStarVersion => String::from("italic-star-version"),
104            Self::ItalicUnderscoreVersion => String::from("italic-underscore-version"),
105            Self::Strikethrough => String::from("strikethrough"),
106            Self::Underlined => String::from("underlined"),
107            Self::Link => String::from("link"),
108            Self::InlineCode => String::from("inline-code"),
109            Self::InlineMath => String::from("inline-math"),
110            Self::GreekLetter => String::from("greek-letter"),
111            Self::Escape => String::from("escape"),
112            Self::Reference => String::from("reference"),
113            Self::Cite => String::from("cite"),
114        }
115    }
116    
117    pub fn modifier_pattern(&self) -> String {
118        match *self {
119            Self::AbridgedBookmark => format!(r"@\[([^\]]*?)\](?:{})?", IDENTIFIER_PATTERN),
120            Self::Bookmark => format!(r"@\[([^\]]*?)\](?:{})?\((?s:(.*?))\)", IDENTIFIER_PATTERN),
121            Self::Todo => String::from(r"@\[(?i:TODO)\]\((?s:(.*?))\)"),
122            Self::AbridgedEmbeddedStyle => format!(r"\[([^\]\n]*?)\]{}?(?:{})?{}?\{{{}\}}", NEW_LINE_PATTERN, IDENTIFIER_PATTERN, NEW_LINE_PATTERN, ABRIDGED_STYLE_PATTERN),
123            Self::Identifier => format!(r"\[(.*?)\]{}?#([\w-]*)", NEW_LINE_PATTERN),
124            Self::EmbeddedStyle => format!(r"\[([^\]\n]*?)\]{}?(?:{})?{}?\{{\{{{}\}}\}}", NEW_LINE_PATTERN, IDENTIFIER_PATTERN, NEW_LINE_PATTERN, STYLE_PATTERN),
125            Self::Highlight => String::from(r"==(.*)=="),
126            Self::Comment => String::from(r"^//(.*)"),
127            Self::Emoji => String::from(r":(\w*):"),
128            Self::Checkbox => String::from(r"(\[\]|\[ \])"),
129            Self::CheckboxChecked => String::from(r"(\[x\]|\[X\])"),
130            Self::Superscript => String::from(r"\^(.*)\^"),
131            Self::Subscript => String::from(r"~(.*)~"),
132            Self::BoldStarVersion => String::from(r"\*\*(.*?)\*\*"),
133            Self::BoldUnderscoreVersion => String::from(r"__(.*?)__"),
134            Self::ItalicStarVersion => String::from(r"\*(.*?)\*"),
135            Self::ItalicUnderscoreVersion => String::from(r"_(.*?)_"),
136            Self::Strikethrough => String::from(r"~~(.*?)~~"),
137            Self::Underlined => String::from(r"\+\+(.*?)\+\+"),
138            Self::Link => String::from(r"\[([^\]]+)\]\(([^)]+)\)"),
139            Self::InlineCode => String::from(r"`(.*?)`"),
140            Self::InlineMath => format!(r#"\$([^$]+)\$"#),
141            Self::GreekLetter => String::from(r"%(\w*?)%"),        // if it changes, fix greek letters rules
142            Self::Escape => String::from(r"\\([\*\+\\~%\^\$@=\[\]!<>\{\}\(\)#-_\|\?&]+)"),
143            Self::Reference => String::from(r"&([\w-]+)&"),
144            Self::Cite => String::from(r"\^\[([\w_]+)\]"),
145        }
146    }
147
148    pub fn incompatible_modifiers(&self) -> ModifiersBucket {
149        match self {
150
151            Self::InlineCode => ModifiersBucket::All,
152            Self::InlineMath => ModifiersBucket::All,
153            Self::Emoji => ModifiersBucket::All,
154            Self::GreekLetter => ModifiersBucket::All,
155            Self::Escape => ModifiersBucket::All,
156            Self::Reference => ModifiersBucket::All,
157            Self::Cite => ModifiersBucket::All,
158            _ => ModifiersBucket::None
159        }
160    }
161
162    pub fn modifier_pattern_regex(&self) -> &Regex {
163        MODIFIER_PATTERNS_REGEX.get(&self.identifier()).unwrap()
164    }
165}
166
167
168impl Into<BaseModifier> for StandardTextModifier {
169    fn into(self) -> BaseModifier {
170        BaseModifier::new(self.modifier_pattern(), self.modifier_pattern_regex().clone(), self.incompatible_modifiers())
171    }
172}