You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
190 lines
6.3 KiB
Rust
190 lines
6.3 KiB
Rust
// Standard library imports
|
|
use std::collections::HashMap;
|
|
|
|
// Each line in template file can be of one of following types
|
|
#[derive(PartialEq, Debug)]
|
|
pub enum ContentType {
|
|
Literal(String),
|
|
TemplateVariable(ExpressionData),
|
|
Tag(TagType),
|
|
Unrecognized,
|
|
}
|
|
|
|
//If ContentType is TemplateVariable, the contents of the line are parsed and stored in the ExpressionData struct
|
|
#[derive(PartialEq, Debug, Clone)]
|
|
pub struct ExpressionData {
|
|
pub expression: String,
|
|
pub var_map: Vec<String>,
|
|
pub gen_html: String,
|
|
}
|
|
#[derive(PartialEq, Debug)]
|
|
pub enum TagType {
|
|
ForTag,
|
|
IfTag,
|
|
}
|
|
|
|
//this checks if a symbol string is contained within another string
|
|
pub fn check_symbol_string(input: &str, symbol: &str) -> bool {
|
|
let symbols = input.chars();
|
|
symbols.as_str().contains(symbol)
|
|
}
|
|
|
|
// This checks for matching symbol strings in a given input
|
|
pub fn check_matching_pair(input: &str, symbol1: &str, symbol2: &str) -> bool {
|
|
let input_iter = input.chars();
|
|
input_iter.as_str().contains(symbol1) && input_iter.as_str().contains(symbol2)
|
|
}
|
|
|
|
// this returns index of given char symbol, if symbol is present.
|
|
pub fn get_index_for_symbol(input: &str, symbol: char) -> (bool, usize) {
|
|
let mut characters = input.char_indices();
|
|
let mut does_exist = false;
|
|
let mut index = 0;
|
|
while let Some((c, d)) = characters.next() {
|
|
if d == symbol {
|
|
does_exist = true;
|
|
index = c;
|
|
break;
|
|
}
|
|
}
|
|
(does_exist, index)
|
|
}
|
|
|
|
// Reads one line of template file and returns type of Content. The types of content are defined in ContentType enum
|
|
// If contentType is TemplateVariable, it parses the line further to separate out the head, variable and tail components.
|
|
// IF ContentType is Literal, it returns the read input without any modifications.
|
|
|
|
pub fn get_content_type(input_line: &str) -> ContentType {
|
|
// Tag expressions are enclosed within {% and %}
|
|
let is_tag_expression = check_matching_pair(&input_line, "{%", "%}");
|
|
|
|
// ForTag expressions begin with keywords 'for' and 'in' enclosed within {% and %}
|
|
// ForTag expressions end with keyword 'endfor' enclosed within {% and %}
|
|
let is_for_tag = (check_symbol_string(&input_line, "for")
|
|
&& check_symbol_string(&input_line, "in"))
|
|
|| check_symbol_string(&input_line, "endfor");
|
|
// IfTag expressions begin with keyword 'if' enclosed within {% and %}
|
|
// IfTag expressions end with keyword 'endif' enclosed within {% and %}
|
|
let is_if_tag =
|
|
check_symbol_string(&input_line, "if") || check_symbol_string(&input_line, "endif");
|
|
// If template variable
|
|
let is_template_variable = check_matching_pair(&input_line, "{{", "}}");
|
|
let return_val;
|
|
// case: For Tag
|
|
if is_tag_expression && is_for_tag {
|
|
return_val = ContentType::Tag(TagType::ForTag);
|
|
//case: If Tag
|
|
} else if is_tag_expression && is_if_tag {
|
|
return_val = ContentType::Tag(TagType::IfTag);
|
|
// case: Template variable
|
|
} else if is_template_variable {
|
|
let content = get_expression_data(&input_line);
|
|
return_val = ContentType::TemplateVariable(content);
|
|
// case: Literal
|
|
} else if !is_tag_expression && !is_template_variable {
|
|
return_val = ContentType::Literal(input_line.to_string());
|
|
// Unknown type
|
|
} else {
|
|
return_val = ContentType::Unrecognized;
|
|
}
|
|
return_val
|
|
}
|
|
|
|
// Function to generate HTML for line containing template variable
|
|
// Supports multiple template vars in one line and no limit on number of string literals
|
|
pub fn generate_html_template_var(
|
|
content: &mut ExpressionData,
|
|
context: HashMap<String, String>,
|
|
) -> &mut ExpressionData {
|
|
content.gen_html = content.expression.clone();
|
|
for var in &content.var_map {
|
|
let (_h, i) = get_index_for_symbol(&var, '{');
|
|
let (_j, k) = get_index_for_symbol(&var, '}');
|
|
let var_without_braces = &var[i + 2..k];
|
|
let val = context.get(var_without_braces).unwrap();
|
|
content.gen_html = content.gen_html.replace(var, val);
|
|
}
|
|
content
|
|
}
|
|
|
|
// Helper function to parse template variable
|
|
pub fn get_expression_data(input_line: &str) -> ExpressionData {
|
|
let expression_iter = input_line.split_whitespace();
|
|
let mut template_var_map: Vec<String> = vec![];
|
|
for word in expression_iter {
|
|
if check_symbol_string(word, "{{") && check_symbol_string(word, "}}") {
|
|
template_var_map.push(word.to_string());
|
|
}
|
|
}
|
|
ExpressionData {
|
|
expression: input_line.into(),
|
|
var_map: template_var_map,
|
|
gen_html: "".into(),
|
|
}
|
|
}
|
|
|
|
//Unit tests
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
#[test]
|
|
fn check_get_index_for_symbol_test() {
|
|
assert_eq!((true, 3), get_index_for_symbol("Hi {name} bye", '{'));
|
|
}
|
|
#[test]
|
|
fn check_template_var_test() {
|
|
let content = "Hi {{name}} bye".into();
|
|
let expr_data = ExpressionData {
|
|
expression: content,
|
|
var_map: vec!["{{name}}".to_string()],
|
|
gen_html: "".into(),
|
|
};
|
|
assert_eq!(
|
|
ContentType::TemplateVariable(expr_data),
|
|
get_content_type("Hi {{name}} bye")
|
|
);
|
|
}
|
|
#[test]
|
|
fn check_for_tag_test() {
|
|
assert_eq!(
|
|
ContentType::Tag(TagType::ForTag),
|
|
get_content_type("{% for name in names %} bye")
|
|
);
|
|
}
|
|
#[test]
|
|
fn check_if_tag_test() {
|
|
assert_eq!(
|
|
ContentType::Tag(TagType::IfTag),
|
|
get_content_type("{% if name == 'Bob' %}")
|
|
);
|
|
}
|
|
#[test]
|
|
fn check_literal_test() {
|
|
let s = "<h1>Hello world</h1>";
|
|
assert_eq!(ContentType::Literal(s.to_string()), get_content_type(s));
|
|
}
|
|
|
|
#[test]
|
|
fn check_get_expression_data_test() {
|
|
let map_var = vec!["{{name}}".to_string(), "{{city}}".to_string()];
|
|
let expression_data = ExpressionData {
|
|
expression: "Hi {{name}} , welcome to {{city}}".into(),
|
|
var_map: map_var,
|
|
gen_html: "".into(),
|
|
};
|
|
|
|
assert_eq!(
|
|
expression_data,
|
|
get_expression_data("Hi {{name}} , welcome to {{city}}")
|
|
);
|
|
}
|
|
#[test]
|
|
fn check_symbol_string_test() {
|
|
assert_eq!(true, check_symbol_string("{{Hello}}", "{{"));
|
|
}
|
|
#[test]
|
|
fn check_symbol_pair_test() {
|
|
assert_eq!(true, check_matching_pair("{{Hello}}", "{{", "}}"));
|
|
}
|
|
}
|