luna-orm-trait 0.3.6

use nom::branch::alt;
use nom::bytes::complete::escaped;
use nom::bytes::complete::{tag, take_till, take_until, take_while};
use nom::character::complete::{
    alpha0, alphanumeric0, alphanumeric1, char, multispace0, multispace1, one_of,
};
use nom::character::is_alphanumeric;
use nom::combinator::{cut, iterator, map, opt, value};
use nom::error::{context, ContextError, ParseError, VerboseError};
use nom::multi::separated_list1;
use nom::sequence::Tuple;
use nom::sequence::{delimited, terminated};
use nom::sequence::{preceded, tuple};
use nom::Compare;
use nom::InputLength;
use nom::InputTake;

use nom::IResult;

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum TemplateValue {
    SingleQuoteString(String),
    DoubleQuoteString(String),
    BackQuoteString(String),
    Star(String),
    Segment(String),
    Variable(String),
}

impl ToString for TemplateValue {
    fn to_string(&self) -> String {
        match self {
            TemplateValue::SingleQuoteString(v)
            | TemplateValue::DoubleQuoteString(v)
            | TemplateValue::BackQuoteString(v)
            | TemplateValue::Star(v)
            | TemplateValue::Segment(v)
            | TemplateValue::Variable(v) => v.to_string(),
        }
    }
}

pub struct ParsedTemplateSql {
    pub sql: String,
    pub variables: Vec<String>,
}

impl ParsedTemplateSql {
    pub fn build(template_sql: &str) -> Result<Self, nom::Err<nom::error::Error<&str>>> {
        let (_, parsed) = parse_template_sql(template_sql)?;
        let parsed_template = ParsedTemplateSql::new(parsed);
        Ok(parsed_template)
    }

    pub fn new(values: Vec<TemplateValue>) -> Self {
        let has_question_mark: bool = values
            .iter()
            .any(|e| &TemplateValue::Segment("?".to_string()) == e);
        if has_question_mark {
            panic!("sql template should not contains ?");
        }

        let variables: Vec<String> = values
            .iter()
            .filter(|e| matches!(e, TemplateValue::Variable(_)))
            .map(|e| e.to_string())
            .collect();

        let result: Vec<TemplateValue> = values
            .into_iter()
            .map(|e| {
                let t: TemplateValue = if let TemplateValue::Variable(_) = e {
                    TemplateValue::Segment("?".to_string())
                } else {
                    e
                };
                t
            })
            .collect();

        let marked_sql = result
            .iter()
            .map(|e| e.to_string())
            .collect::<Vec<String>>()
            .join(" ");

        Self {
            sql: marked_sql,
            variables,
        }
    }
}

pub fn parse_back_quote_string<'a, E: ParseError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    let (remaining, parsed) = preceded(
        preceded(multispace0, char('`')),
        cut(terminated(take_until("`"), char('`'))),
    )(input)?;
    let value = TemplateValue::BackQuoteString(format!("`{}`", parsed));
    return Ok((remaining, value));
}

pub fn parse_single_quote_string<'a, E: ParseError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    let (remaining, parsed) = preceded(
        preceded(multispace0, char('\'')),
        cut(terminated(take_until("'"), char('\''))),
    )(input)?;
    let value = TemplateValue::SingleQuoteString(format!("'{}'", parsed));
    return Ok((remaining, value));
}

pub fn parse_double_quote_string<'a, E: ParseError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    let (remaining, parsed) = preceded(
        preceded(multispace0, char('"')),
        cut(terminated(take_until("\""), char('"'))),
    )(input)?;
    let value = TemplateValue::DoubleQuoteString(format!("\"{}\"", parsed));
    return Ok((remaining, value));
}

fn parse_string<'a, E: ParseError<&'a str> + ContextError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    context(
        "string",
        alt((
            parse_single_quote_string,
            parse_double_quote_string,
            parse_back_quote_string,
        )),
    )(input)
}

//pub fn parse_string(input: &str) -> IResult<&str, &str> {
//    alt((parse_single_quote_string, parse_double_quote_string))(input)
//}

pub fn parse_dot<'a, E: ParseError<&'a str>>(input: &'a str) -> IResult<&'a str, &'a str, E> {
    delimited(multispace0, tag("."), multispace0)(input)
}
pub fn parse_variable_seg<'a, E: ParseError<&'a str> + ContextError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, &'a str, E> {
    context("variable_seg", preceded(multispace0, alphanumeric1))(input)
}

/*
pub fn check_following_dot<'a, E: ParseError<&'a str> + ContextError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, (), E> {
    if input.starts_with(".") {
        let error = VerboseError::from_char(input, '.');
        return Err(nom::Err::Failure(error));
    }
    return Ok((input, ()));
}
*/

pub fn parse_variable<'a, E: ParseError<&'a str> + ContextError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    //let (remaining, parsed) = parse_variable_seg(input)?;
    //let first_seg: &str = parsed;

    let segs_result = delimited(
        preceded(multispace0, tag("#{")),
        separated_list1(parse_dot, parse_variable_seg),
        preceded(multispace0, tag("}")),
    )(input)?;
    let (remaining, parsed) = segs_result;
    let following_segs: Vec<&str> = parsed;
    let mut segs: Vec<&str> = Vec::new();
    //segs.push(first_seg);
    segs.extend(following_segs);
    let variable = segs.join(".");
    return Ok((remaining, TemplateValue::Variable(variable.to_string())));
}

pub fn parse_segment<'a, E: ParseError<&'a str>>(
    input: &'a str,
) -> IResult<&'a str, TemplateValue, E> {
    let (remaining, parsed) = alt((
        preceded(multispace0, alphanumeric1),
        preceded(multispace0, tag("*")),
        preceded(multispace0, tag(",")),
        preceded(multispace0, tag(";")),
        preceded(multispace0, tag("=")),
        preceded(multispace0, tag("?")),
        preceded(multispace0, tag("<")),
        preceded(multispace0, tag(">")),
        preceded(multispace0, tag("%")),
        preceded(multispace0, tag("(")),
        preceded(multispace0, tag(")")),
    ))(input)?;
    return Ok((remaining, TemplateValue::Segment(parsed.to_string())));
}

/*
pub fn parse_variable(input: &str) -> IResult<&str, &str> {
    delimited(
        tuple((tag("#{"), multispace0)),
        parse_dot_variable,
        tuple((multispace0, tag("}"))),
    )(input)
}
*/

pub fn parese_template_value(input: &str) -> IResult<&str, TemplateValue> {
    alt((parse_variable, parse_string, parse_segment))(input)
}
pub fn parse_template_sql(input: &str) -> IResult<&str, Vec<TemplateValue>> {
    let mut values: Vec<TemplateValue> = Vec::new();
    let (mut remaining, mut parsed) = parese_template_value(input)?;
    values.push(parsed);
    while !remaining.is_empty() {
        (remaining, parsed) = parese_template_value(remaining)?;
        values.push(parsed);
    }
    return Ok((remaining, values));
}

#[cfg(test)]
mod test {

    use nom::error::ErrorKind;

    use super::TemplateValue;
    use super::*;

    #[test]
    pub fn test_string() {
        let (remaining, parsed) =
            parse_string::<(&str, ErrorKind)>("'this is single string'").unwrap();
        assert_eq!(
            parsed,
            TemplateValue::SingleQuoteString("'this is single string'".to_string())
        );
        let (remaining, parsed) =
            parse_string::<(&str, ErrorKind)>("\"this is double string\"").unwrap();
        assert_eq!(
            parsed,
            TemplateValue::DoubleQuoteString("\"this is double string\"".to_string())
        );
    }

    #[test]
    pub fn test_variable() {
        let (remaining, parsed) = parse_variable::<(&str, ErrorKind)>("#{ var1 . var2 }").unwrap();
        assert_eq!(parsed, TemplateValue::Variable("var1.var2".to_string()));
    }

    #[test]
    pub fn test_template() {
        let (remaining, parsed) =
            parse_template_sql("SELECT * `test` user #{v1. v2. v3} where id = 23").unwrap();
        let result_vec = vec![
            TemplateValue::Segment("SELECT".to_string()),
            TemplateValue::Segment("*".to_string()),
            TemplateValue::BackQuoteString("`test`".to_string()),
            TemplateValue::Segment("user".to_string()),
            TemplateValue::Variable("v1.v2.v3".to_string()),
            TemplateValue::Segment("where".to_string()),
            TemplateValue::Segment("id".to_string()),
            TemplateValue::Segment("=".to_string()),
            TemplateValue::Segment("23".to_string()),
        ];
        assert_eq!(parsed, result_vec);

        let parsed_sql = ParsedTemplateSql::new(parsed);

        assert_eq!(parsed_sql.sql, "SELECT * `test` user ? where id = 23");
        let variables = vec!["v1.v2.v3".to_string()];
        assert_eq!(parsed_sql.variables, variables);
    }
}