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 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*?)%"), 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}