/*
 * AquaVM Workflow Engine
 *
 * Copyright (C) 2024 Fluence DAO
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation version 3 of the
 * License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

use crate::parser::lambda_parser::RawLambdaAST;
use crate::parser::va_lambda::RawLambdaASTParser;
use crate::ValueAccessor;
use air_lambda_ast::Functor;

thread_local!(static TEST_PARSER: RawLambdaASTParser = RawLambdaASTParser::new());

fn parse(source_code: &str) -> RawLambdaAST<'_> {
    TEST_PARSER.with(|parser| {
        let mut errors = Vec::new();
        let lexer = crate::parser::LambdaASTLexer::new(source_code);
        parser
            .parse(source_code, &mut errors, lexer)
            .expect("parsing should be successful")
    })
}

fn parse_to_accessors(source_code: &str) -> Vec<ValueAccessor<'_>> {
    let lambda_ast = parse(source_code);
    match lambda_ast {
        RawLambdaAST::ValuePath(accessors) => accessors,
        _ => panic!("it should be a value path"),
    }
}

fn parse_to_functor(source_code: &str) -> Functor {
    let lambda_ast = parse(source_code);
    match lambda_ast {
        RawLambdaAST::Functor(functor) => functor,
        _ => panic!("it should be a functor"),
    }
}

#[test]
fn field_access() {
    let field_name = "some_field_name";
    let lambda = format!(".$.{field_name}");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::FieldAccessByName { field_name }];
    assert_eq!(actual, expected);
}

#[test]
fn field_access_with_flattening() {
    let field_name = "some_field_name";
    let lambda = format!(".$.{field_name}!");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::FieldAccessByName { field_name }];
    assert_eq!(actual, expected);
}

#[test]
fn array_access() {
    let idx = 0;
    let lambda = format!(".$.[{idx}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::ArrayAccess { idx }];
    assert_eq!(actual, expected);
}

#[test]
fn array_access_with_flattening() {
    let idx = 0;
    let lambda = format!(".$.[{idx}]!");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::ArrayAccess { idx }];
    assert_eq!(actual, expected);
}

#[test]
fn scalar_access() {
    let scalar_name = "some_field_name";
    let lambda = format!(".$.[{scalar_name}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::FieldAccessByScalar { scalar_name }];
    assert_eq!(actual, expected);
}

#[test]
fn scalar_access_with_flattening() {
    let scalar_name = "some_scalar_name";
    let lambda = format!(".$.[{scalar_name}]!");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![ValueAccessor::FieldAccessByScalar { scalar_name }];
    assert_eq!(actual, expected);
}

#[test]
fn field_array_access() {
    let field_name = "some_field_name";
    let idx = 1;
    let lambda = format!(".$.{field_name}.[{idx}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::FieldAccessByName { field_name },
        ValueAccessor::ArrayAccess { idx },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn field_scalar_access() {
    let field_name = "some_field_name";
    let scalar_name = "some_scalar_name";
    let lambda = format!(".$.{field_name}.[{scalar_name}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::FieldAccessByName { field_name },
        ValueAccessor::FieldAccessByScalar { scalar_name },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn scalar_array_access() {
    let scalar_name = "some_scalar_name";
    let idx = 1;
    let lambda = format!(".$.[{scalar_name}].[{idx}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::FieldAccessByScalar { scalar_name },
        ValueAccessor::ArrayAccess { idx },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn field_array_access_without_dot() {
    let field_name = "some_field_name";
    let idx = 1;
    let lambda = format!(".$.{field_name}[{idx}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::FieldAccessByName { field_name },
        ValueAccessor::ArrayAccess { idx },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn array_field_access() {
    let field_name = "some_field_name";
    let idx = 1;
    let lambda = format!(".$.[{idx}].{field_name}");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::ArrayAccess { idx },
        ValueAccessor::FieldAccessByName { field_name },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn array_scalar_access() {
    let scalar_name = "some_scalar_name";
    let idx = 1;
    let lambda = format!(".$.[{idx}].[{scalar_name}]");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::ArrayAccess { idx },
        ValueAccessor::FieldAccessByScalar { scalar_name },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn many_array_field_access() {
    let field_name_1 = "some_field_name_1";
    let field_name_2 = "some_field_name_2";
    let idx_1 = 1;
    let idx_2 = u32::MAX;
    let lambda = format!(".$.[{idx_1}].{field_name_1}.[{idx_2}].{field_name_2}");

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::ArrayAccess { idx: idx_1 },
        ValueAccessor::FieldAccessByName {
            field_name: field_name_1,
        },
        ValueAccessor::ArrayAccess { idx: idx_2 },
        ValueAccessor::FieldAccessByName {
            field_name: field_name_2,
        },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn many_array_field_scalar_access() {
    let field_name_1 = "some_field_name_1";
    let field_name_2 = "some_field_name_2";
    let idx_1 = 1;
    let idx_2 = u32::MAX;
    let scalar_name_1 = "some_scalar_name_1";
    let scalar_name_2 = "some_scalar_name_2";
    let lambda = format!(
        ".$.[{idx_1}].[{scalar_name_1}].{field_name_1}.[{idx_2}].[{scalar_name_2}].{field_name_2}"
    );

    let actual = parse_to_accessors(&lambda);
    let expected = vec![
        ValueAccessor::ArrayAccess { idx: idx_1 },
        ValueAccessor::FieldAccessByScalar {
            scalar_name: scalar_name_1,
        },
        ValueAccessor::FieldAccessByName {
            field_name: field_name_1,
        },
        ValueAccessor::ArrayAccess { idx: idx_2 },
        ValueAccessor::FieldAccessByScalar {
            scalar_name: scalar_name_2,
        },
        ValueAccessor::FieldAccessByName {
            field_name: field_name_2,
        },
    ];
    assert_eq!(actual, expected);
}

#[test]
fn parse_length_functor() {
    let lambda = ".length";

    let actual = parse_to_functor(lambda);
    let expected = Functor::Length;
    assert_eq!(actual, expected);
}

#[test]
fn parse_length_functor_with_following_accessors() {
    let lambda = ".length.[0]";

    let actual = TEST_PARSER.with(|parser| {
        let mut errors = Vec::new();
        let lexer = crate::parser::LambdaASTLexer::new(lambda);
        parser.parse(lambda, &mut errors, lexer)
    });

    assert!(matches!(actual, Err(lalrpop_util::ParseError::User { .. })))
}
