// 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}}", "{{", "}}"));
    }
}