cairo-lang-syntax 2.1.2

use cairo_lang_filesystem::span::{TextOffset, TextWidth};
use pretty_assertions::assert_eq;
use test_log::test;

use super::ast::{
    ExprBinary, ExprPath, PathSegmentGreen, PathSegmentSimple, SyntaxFileGreen, TerminalIdentifier,
    TerminalLiteralNumber, TerminalPlus, TokenIdentifier, TokenLiteralNumber, TokenPlus,
    TokenWhitespace, Trivia,
};
use super::kind::SyntaxKind;
use super::{SyntaxNode, Terminal, Token};
use crate::node::test_utils::DatabaseForTesting;

#[test]
fn test_ast() {
    let db_val = DatabaseForTesting::default();
    let db = &db_val;
    let root = setup(db);

    assert_eq!(
        root.descendants(db)
            .map(|node| (node.kind(db), node.text(db), node.offset(), node.width(db)))
            .collect::<Vec<_>>(),
        vec![
            (
                SyntaxKind::ExprBinary,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(7)
            ),
            (
                SyntaxKind::ExprPath,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(4)
            ),
            (
                SyntaxKind::PathSegmentSimple,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(4)
            ),
            (
                SyntaxKind::TerminalIdentifier,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(4)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(0)
            ),
            (
                SyntaxKind::TokenIdentifier,
                Some("foo".into()),
                TextOffset::default().add_width(TextWidth::new_for_testing(0)),
                TextWidth::new_for_testing(3)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(3)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::TokenWhitespace,
                Some(" ".into()),
                TextOffset::default().add_width(TextWidth::new_for_testing(3)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::TerminalPlus,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(4)),
                TextWidth::new_for_testing(2)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(4)),
                TextWidth::new_for_testing(0)
            ),
            (
                SyntaxKind::TokenPlus,
                Some("+".into()),
                TextOffset::default().add_width(TextWidth::new_for_testing(4)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(5)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::TokenWhitespace,
                Some(" ".into()),
                TextOffset::default().add_width(TextWidth::new_for_testing(5)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::TerminalLiteralNumber,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(6)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(6)),
                TextWidth::new_for_testing(0)
            ),
            (
                SyntaxKind::TokenLiteralNumber,
                Some("5".into()),
                TextOffset::default().add_width(TextWidth::new_for_testing(6)),
                TextWidth::new_for_testing(1)
            ),
            (
                SyntaxKind::Trivia,
                None,
                TextOffset::default().add_width(TextWidth::new_for_testing(7)),
                TextWidth::new_for_testing(0)
            )
        ]
    )
}

#[test]
fn test_stable_ptr() {
    let db_val = DatabaseForTesting::default();
    let db = &db_val;
    let root = setup(db);
    for node in root.descendants(db) {
        let ptr = node.stable_ptr();
        let looked_up_node = root.lookup_ptr(db, ptr);
        assert_eq!(node, looked_up_node);
    }
}

fn setup(db: &DatabaseForTesting) -> SyntaxNode {
    // TODO: Use a builder for easier construction of token.
    // Construct green nodes.
    let token_foo = TokenIdentifier::new_green(db, "foo".into());
    let token_whitespace1 = TokenWhitespace::new_green(db, " ".into());
    let token_plus = TokenPlus::new_green(db, "+".into());
    let token_whitespace2 = TokenWhitespace::new_green(db, " ".into());
    let token5 = TokenLiteralNumber::new_green(db, "5".into());
    assert_eq!(token_whitespace1, token_whitespace2);
    let no_trivia = Trivia::new_green(db, vec![]);
    let triviums = vec![token_whitespace1, token_whitespace2];
    assert_eq!(triviums[0], triviums[1]);
    let terminal_foo = TerminalIdentifier::new_green(
        db,
        no_trivia,
        token_foo,
        Trivia::new_green(db, vec![triviums[0].into()]),
    );
    let terminal_plus = TerminalPlus::new_green(
        db,
        no_trivia,
        token_plus,
        Trivia::new_green(db, vec![triviums[1].into()]),
    );
    let terminal5 = TerminalLiteralNumber::new_green(db, no_trivia, token5, no_trivia);
    let expr = ExprBinary::new_green(
        db,
        ExprPath::new_green(
            db,
            vec![PathSegmentGreen::from(PathSegmentSimple::new_green(db, terminal_foo)).into()],
        )
        .into(),
        terminal_plus.into(),
        terminal5.into(),
    );
    // SyntaxNode::new_root only accepts ast::SyntaxFileGreen, but we only have an expression.
    // This is a hack to crate a green id of "SyntaxFile" from "Expr".
    let root = SyntaxFileGreen(expr.0);
    SyntaxNode::new_root(db, root)
}