Refactor template variable collecting and freezing
This change extracts template variable collectiong and freezing into a different module to keep the elements module clean. This commit also changes all Mutex'es to RwLocks to get better performance with read-only locking (the increase is measurable even with small files).pull/1/head
trivernis
4 years ago
parent
792d991fb2
commit
815bd26e51
@ -0,0 +1,291 @@
|
||||
use crate::parsing::elements::{Block, Element, Inline, Line, ListItem};
|
||||
use std::collections::HashMap;
|
||||
use std::sync::{Arc, RwLock};
|
||||
|
||||
pub trait FreezeVariables {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>>;
|
||||
}
|
||||
|
||||
pub trait GetTemplateVariables {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>>;
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct Template {
|
||||
pub(crate) text: Vec<Element>,
|
||||
pub(crate) variables: HashMap<String, Arc<RwLock<TemplateVariable>>>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct TemplateVariable {
|
||||
pub(crate) prefix: String,
|
||||
pub(crate) name: String,
|
||||
pub(crate) suffix: String,
|
||||
pub(crate) value: Option<Element>,
|
||||
}
|
||||
|
||||
impl Template {
|
||||
pub fn render(&self, replacements: HashMap<String, Element>) -> Vec<Element> {
|
||||
replacements.iter().for_each(|(k, r)| {
|
||||
if let Some(v) = self.variables.get(k) {
|
||||
v.write().unwrap().set_value(r.clone())
|
||||
}
|
||||
});
|
||||
let elements = self
|
||||
.text
|
||||
.iter()
|
||||
.map(|e| {
|
||||
let mut e = e.clone();
|
||||
if let Some(template) = e.freeze_variables() {
|
||||
Element::Inline(Box::new(Inline::TemplateVar(template)))
|
||||
} else {
|
||||
e
|
||||
}
|
||||
})
|
||||
.collect();
|
||||
self.variables
|
||||
.iter()
|
||||
.for_each(|(_, v)| v.write().unwrap().reset());
|
||||
|
||||
elements
|
||||
}
|
||||
}
|
||||
|
||||
impl TemplateVariable {
|
||||
pub fn set_value(&mut self, value: Element) {
|
||||
self.value = Some(value)
|
||||
}
|
||||
pub fn reset(&mut self) {
|
||||
self.value = None
|
||||
}
|
||||
}
|
||||
|
||||
impl GetTemplateVariables for Inline {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Inline::TemplateVar(temp) => vec![Arc::clone(temp)],
|
||||
Inline::Colored(col) => col.value.get_template_variables(),
|
||||
Inline::Superscript(sup) => sup.value.get_template_variables(),
|
||||
Inline::Striked(striked) => striked.value.get_template_variables(),
|
||||
Inline::Underlined(under) => under.value.get_template_variables(),
|
||||
Inline::Italic(it) => it.value.get_template_variables(),
|
||||
Inline::Bold(bo) => bo.value.get_template_variables(),
|
||||
_ => Vec::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl GetTemplateVariables for Line {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Line::Text(line) => line
|
||||
.subtext
|
||||
.iter()
|
||||
.map(|s| s.get_template_variables())
|
||||
.flatten()
|
||||
.collect(),
|
||||
Line::Centered(center) => center
|
||||
.line
|
||||
.subtext
|
||||
.iter()
|
||||
.map(|s| s.get_template_variables())
|
||||
.flatten()
|
||||
.collect(),
|
||||
_ => Vec::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl GetTemplateVariables for Block {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Block::Section(sec) => sec
|
||||
.elements
|
||||
.iter()
|
||||
.map(|e| e.get_template_variables())
|
||||
.flatten()
|
||||
.collect(),
|
||||
Block::Paragraph(par) => par
|
||||
.elements
|
||||
.iter()
|
||||
.map(|l| l.get_template_variables())
|
||||
.flatten()
|
||||
.collect(),
|
||||
Block::Quote(q) => q
|
||||
.text
|
||||
.iter()
|
||||
.map(|t| {
|
||||
t.subtext
|
||||
.iter()
|
||||
.map(|i| i.get_template_variables())
|
||||
.flatten()
|
||||
})
|
||||
.flatten()
|
||||
.collect(),
|
||||
Block::List(list) => list
|
||||
.items
|
||||
.iter()
|
||||
.map(|item| item.get_template_variables())
|
||||
.flatten()
|
||||
.collect(),
|
||||
_ => Vec::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl GetTemplateVariables for Element {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Element::Block(block) => block.get_template_variables(),
|
||||
Element::Inline(inline) => inline.get_template_variables(),
|
||||
Element::Line(line) => line.get_template_variables(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl FreezeVariables for Inline {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Inline::TemplateVar(temp) => {
|
||||
let temp = temp.read().unwrap();
|
||||
return Some(Arc::new(RwLock::new((*temp).clone())));
|
||||
}
|
||||
Inline::Colored(col) => {
|
||||
if let Some(temp) = col.value.freeze_variables() {
|
||||
col.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
Inline::Superscript(sup) => {
|
||||
if let Some(temp) = sup.value.freeze_variables() {
|
||||
sup.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
Inline::Striked(striked) => {
|
||||
if let Some(temp) = striked.value.freeze_variables() {
|
||||
striked.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
Inline::Underlined(under) => {
|
||||
if let Some(temp) = under.value.freeze_variables() {
|
||||
under.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
Inline::Italic(it) => {
|
||||
if let Some(temp) = it.value.freeze_variables() {
|
||||
it.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
Inline::Bold(bo) => {
|
||||
if let Some(temp) = bo.value.freeze_variables() {
|
||||
bo.value = Box::new(Inline::TemplateVar(temp))
|
||||
}
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
impl GetTemplateVariables for ListItem {
|
||||
fn get_template_variables(&self) -> Vec<Arc<RwLock<TemplateVariable>>> {
|
||||
let mut inner_vars: Vec<Arc<RwLock<TemplateVariable>>> = self
|
||||
.children
|
||||
.iter()
|
||||
.map(|child| child.get_template_variables())
|
||||
.flatten()
|
||||
.collect();
|
||||
inner_vars.append(&mut self.text.get_template_variables());
|
||||
|
||||
inner_vars
|
||||
}
|
||||
}
|
||||
|
||||
impl FreezeVariables for Line {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Line::Text(text) => {
|
||||
text.subtext = text
|
||||
.subtext
|
||||
.iter_mut()
|
||||
.map(|i| {
|
||||
if let Some(t) = i.freeze_variables() {
|
||||
Inline::TemplateVar(t)
|
||||
} else {
|
||||
(*i).clone()
|
||||
}
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
Line::Centered(center) => {
|
||||
center.line.subtext = center
|
||||
.line
|
||||
.subtext
|
||||
.iter_mut()
|
||||
.map(|i| {
|
||||
if let Some(t) = i.freeze_variables() {
|
||||
Inline::TemplateVar(t)
|
||||
} else {
|
||||
(*i).clone()
|
||||
}
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
impl FreezeVariables for Block {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Block::Section(s) => s.elements.iter_mut().for_each(|b| {
|
||||
b.freeze_variables();
|
||||
}),
|
||||
Block::Paragraph(p) => p.elements.iter_mut().for_each(|l| {
|
||||
l.freeze_variables();
|
||||
}),
|
||||
Block::Quote(q) => q.text.iter_mut().for_each(|t| {
|
||||
t.subtext = t
|
||||
.subtext
|
||||
.iter_mut()
|
||||
.map(|i| {
|
||||
if let Some(t) = i.freeze_variables() {
|
||||
Inline::TemplateVar(t)
|
||||
} else {
|
||||
(*i).clone()
|
||||
}
|
||||
})
|
||||
.collect()
|
||||
}),
|
||||
Block::List(list) => list.items.iter_mut().for_each(|item| {
|
||||
item.freeze_variables();
|
||||
}),
|
||||
_ => {}
|
||||
};
|
||||
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
impl FreezeVariables for Element {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>> {
|
||||
match self {
|
||||
Element::Block(b) => b.freeze_variables(),
|
||||
Element::Line(l) => l.freeze_variables(),
|
||||
Element::Inline(i) => return i.freeze_variables(),
|
||||
};
|
||||
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
impl FreezeVariables for ListItem {
|
||||
fn freeze_variables(&mut self) -> Option<Arc<RwLock<TemplateVariable>>> {
|
||||
self.children.iter_mut().for_each(|child| {
|
||||
child.freeze_variables();
|
||||
});
|
||||
self.text.freeze_variables();
|
||||
None
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue