gobject-ast 0.1.2

use std::path::PathBuf;

use gobject_ast::{
    Parser,
    model::{
        Expression, ForInit, Parameter, Project, Statement, TopLevelItem, TypeDefItem,
        TypedefTarget,
    },
};

fn parse_fixture(name: &str) -> Project {
    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join(name);

    let mut parser = Parser::new().unwrap();
    parser.parse_file(&fixture_path).unwrap()
}

#[test]
fn test_parse_call_expressions() {
    let project = parse_fixture("call_expressions.c");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("call_expressions.c");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");

    let func = file
        .iter_function_definitions()
        .next()
        .expect("Should find a function");
    assert_eq!(func.name, "test_function");

    // Check we have statements parsed
    assert!(
        !func.body_statements.is_empty(),
        "Should have parsed body statements"
    );

    // Count call expressions
    let mut call_count = 0;
    for stmt in &func.body_statements {
        if let Statement::Expression(expr) = stmt
            && matches!(expr.as_ref(), Expression::Call(_))
        {
            call_count += 1;
        }
    }

    // We should find at least the function calls (not counting the variable
    // declaration)
    assert!(
        call_count >= 2,
        "Should find at least 2 call expressions (g_task_set_source_tag, g_object_unref), found {}",
        call_count
    );
}

#[test]
fn test_parse_assignments() {
    let project = parse_fixture("assignments.c");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("assignments.c");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");
    let func = file
        .iter_function_definitions()
        .next()
        .expect("Should find a function");

    // Count assignments
    let mut assignment_count = 0;
    for stmt in &func.body_statements {
        if let Statement::Expression(expr) = stmt
            && matches!(expr.as_ref(), Expression::Assignment(_))
        {
            assignment_count += 1;
        }
    }

    assert!(
        assignment_count >= 1,
        "Should find at least 1 assignment expression, found {}",
        assignment_count
    );
}

#[test]
fn test_parse_return_statement() {
    let project = parse_fixture("return_statement.c");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("return_statement.c");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");
    let func = file
        .iter_function_definitions()
        .next()
        .expect("Should find a function");

    // Should have a return statement
    assert!(!func.body_statements.is_empty(), "Should have statements");

    let has_return = func
        .body_statements
        .iter()
        .any(|stmt| matches!(stmt, Statement::Return(_)));

    assert!(has_return, "Should find return statement");
}

#[test]
fn test_parse_goto_statement() {
    let project = parse_fixture("goto_statement.c");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("goto_statement.c");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");

    let func = file
        .iter_function_definitions()
        .next()
        .expect("Should find a function");

    let has_goto = func.body_statements.iter().any(|s| {
        let mut found = false;
        s.walk(&mut |stmt| {
            if matches!(stmt, Statement::Goto(_)) {
                found = true;
            }
        });
        found
    });

    assert!(has_goto, "Should find goto statement");
}

#[test]
fn test_anonymous_union_field_types_collected() {
    let project = parse_fixture("anonymous_union.h");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("anonymous_union.h");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");

    let xdp_rule = file.iter_all_items().find_map(|item| match item {
        TopLevelItem::TypeDefinition(td @ TypeDefItem::Typedef { name, .. })
            if name == "XdpUsbRule" =>
        {
            Some(td)
        }
        _ => None,
    });

    let TypeDefItem::Typedef { struct_fields, .. } =
        xdp_rule.expect("XdpUsbRule typedef not found")
    else {
        panic!("XdpUsbRule should be a Typedef");
    };

    assert_eq!(
        struct_fields.len(),
        2,
        "XdpUsbRule should have 2 top-level fields"
    );

    let rule_type = &struct_fields[0];
    assert_eq!(rule_type.field_name.as_deref(), Some("rule_type"));
    assert_eq!(rule_type.field_type.base_type, "int");
    assert!(rule_type.inner_fields.is_empty());

    // The anonymous union `union { ... } d` is stored as field `d` with
    // inner_fields.
    let union_d = &struct_fields[1];
    assert_eq!(union_d.field_name.as_deref(), Some("d"));
    assert!(
        union_d.field_type.base_type.is_empty(),
        "anonymous union has no base type"
    );

    let inner: Vec<(&str, &str)> = union_d
        .inner_fields
        .iter()
        .map(|f| {
            (
                f.field_name.as_deref().unwrap_or(""),
                f.field_type.base_type.as_str(),
            )
        })
        .collect();
    assert_eq!(
        inner,
        vec![
            ("device_class", "int"),
            ("product", "UsbProduct"),
            ("vendor", "UsbVendor")
        ],
        "anonymous union inner fields mismatch"
    );
}

#[test]
fn test_statement_order() {
    let project = parse_fixture("statement_order.c");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("statement_order.c");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");
    let func = file
        .iter_function_definitions()
        .next()
        .expect("Should find a function");

    // Verify order: should have declaration/call first, then call
    assert!(
        func.body_statements.len() >= 2,
        "Should have at least 2 statements, found {}",
        func.body_statements.len()
    );

    // Second statement should be a call to g_bytes_unref
    let mut found_pattern = false;
    for i in 0..func.body_statements.len() - 1 {
        if let Statement::Expression(expr) = &func.body_statements[i + 1]
            && let Expression::Call(call2) = expr.as_ref()
            && call2.is_function("g_bytes_unref")
        {
            found_pattern = true;
        }
    }

    assert!(
        found_pattern,
        "Should find consecutive g_bytes_get_data and g_bytes_unref calls in order"
    );
}

#[test]
fn test_bitfield_struct_parsing() {
    let project = parse_fixture("bitfield_struct.h");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("bitfield_struct.h");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");

    let typedef = file
        .iter_all_items()
        .find_map(|item| match item {
            TopLevelItem::TypeDefinition(td @ TypeDefItem::Typedef { name, .. })
                if name == "MyBitStruct" =>
            {
                Some(td)
            }
            _ => None,
        })
        .expect("MyBitStruct typedef not found");

    let TypeDefItem::Typedef { struct_fields, .. } = typedef else {
        panic!("expected Typedef");
    };

    assert_eq!(struct_fields.len(), 4, "expected 4 fields");

    let flags = &struct_fields[0];
    assert_eq!(flags.field_name.as_deref(), Some("flags"));
    assert_eq!(flags.bit_width, Some(1));

    let count = &struct_fields[1];
    assert_eq!(count.field_name.as_deref(), Some("count"));
    assert_eq!(count.bit_width, Some(4));

    let padding = &struct_fields[2];
    assert_eq!(padding.field_name.as_deref(), Some("padding"));
    assert_eq!(padding.bit_width, Some(27));

    let normal = &struct_fields[3];
    assert_eq!(normal.field_name.as_deref(), Some("normal_field"));
    assert_eq!(normal.bit_width, None);
}

#[test]
fn test_callback_typedef_parsing() {
    let project = parse_fixture("typedef_callback.h");

    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .join("tests")
        .join("fixtures")
        .join("typedef_callback.h");

    let file = project
        .get_file(&fixture_path)
        .expect("File should be parsed");

    let typedefs: Vec<&TypeDefItem> = file
        .iter_all_items()
        .filter_map(|item| match item {
            TopLevelItem::TypeDefinition(td) => Some(td),
            _ => None,
        })
        .collect();

    // Plain type alias: `typedef struct _MyObject MyObject`
    let plain = typedefs
        .iter()
        .find(|td| matches!(td, TypeDefItem::Typedef { name, .. } if name == "MyObject"))
        .expect("MyObject typedef not found");
    let TypeDefItem::Typedef { target, .. } = plain else {
        panic!("expected Typedef");
    };
    assert!(
        matches!(target, TypedefTarget::Type(_)),
        "MyObject should be a plain type alias"
    );
    let TypedefTarget::Type(ti) = target else {
        unreachable!()
    };
    assert_eq!(ti.base_type, "_MyObject");
    assert!(ti.is_struct);

    // `typedef void (*MyCallback)(MyObject *obj, gpointer user_data)`
    let cb = typedefs
        .iter()
        .find(|td| matches!(td, TypeDefItem::Typedef { name, .. } if name == "MyCallback"))
        .expect("MyCallback typedef not found");
    let TypeDefItem::Typedef { target, .. } = cb else {
        panic!("expected Typedef");
    };
    let TypedefTarget::Callback {
        return_type,
        parameters,
    } = target
    else {
        panic!(
            "MyCallback should be TypedefTarget::Callback, got {:?}",
            target
        );
    };
    assert_eq!(return_type.base_type, "void");
    assert_eq!(return_type.pointer_depth, 0);
    assert_eq!(parameters.len(), 2);
    let Parameter::Regular { type_info: p0, .. } = &parameters[0] else {
        panic!("expected Regular")
    };
    let Parameter::Regular { type_info: p1, .. } = &parameters[1] else {
        panic!("expected Regular")
    };
    assert_eq!(p0.base_type, "MyObject");
    assert_eq!(p1.base_type, "gpointer");

    // `typedef gboolean (*MyPredicate)(const gchar *name, guint index)`
    let pred = typedefs
        .iter()
        .find(|td| matches!(td, TypeDefItem::Typedef { name, .. } if name == "MyPredicate"))
        .expect("MyPredicate typedef not found");
    let TypeDefItem::Typedef { target, .. } = pred else {
        panic!("expected Typedef");
    };
    let TypedefTarget::Callback {
        return_type,
        parameters,
    } = target
    else {
        panic!("MyPredicate should be TypedefTarget::Callback");
    };
    assert_eq!(return_type.base_type, "gboolean");
    assert_eq!(parameters.len(), 2);

    // `typedef const gchar *(*MyGetNameFunc)(MyObject *obj)`
    let getter = typedefs
        .iter()
        .find(|td| matches!(td, TypeDefItem::Typedef { name, .. } if name == "MyGetNameFunc"))
        .expect("MyGetNameFunc typedef not found");
    let TypeDefItem::Typedef { target, .. } = getter else {
        panic!("expected Typedef");
    };
    let TypedefTarget::Callback {
        return_type,
        parameters,
    } = target
    else {
        panic!("MyGetNameFunc should be TypedefTarget::Callback");
    };
    assert_eq!(return_type.base_type, "gchar");
    assert_eq!(return_type.pointer_depth, 1);
    assert!(return_type.is_const);
    assert_eq!(parameters.len(), 1);
    let Parameter::Regular { type_info: p0, .. } = &parameters[0] else {
        panic!("expected Regular")
    };
    assert_eq!(p0.base_type, "MyObject");
}

#[test]
fn test_variadic_parameter_parsing() {
    let project = parse_fixture("variadic_func.h");
    let file = project.files.values().next().expect("No files parsed");

    let decls: Vec<_> = file.iter_function_declarations().collect();
    assert_eq!(decls.len(), 3);

    // foo(int n, ...) — 2 params: Regular + Variadic
    let foo = decls
        .iter()
        .find(|d| d.name == "foo")
        .expect("foo not found");
    assert_eq!(foo.parameters.len(), 2);
    assert!(matches!(foo.parameters[0], Parameter::Regular { .. }));
    assert!(matches!(foo.parameters[1], Parameter::Variadic));

    // bar(const gchar *format, ...) — 2 params: Regular + Variadic
    let bar = decls
        .iter()
        .find(|d| d.name == "bar")
        .expect("bar not found");
    assert_eq!(bar.parameters.len(), 2);
    assert!(matches!(bar.parameters[0], Parameter::Regular { .. }));
    assert!(matches!(bar.parameters[1], Parameter::Variadic));

    // baz(void) — not variadic
    let baz = decls
        .iter()
        .find(|d| d.name == "baz")
        .expect("baz not found");
    assert!(
        baz.parameters
            .iter()
            .all(|p| !matches!(p, Parameter::Variadic))
    );
}

#[test]
fn test_for_statement_init_variants() {
    let project = parse_fixture("for_init.c");
    let fixture_path = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests/fixtures/for_init.c");
    let file = project.get_file(&fixture_path).expect("file not parsed");

    let func = file
        .iter_function_definitions()
        .next()
        .expect("no function");

    let for_stmts: Vec<_> = func
        .body_statements
        .iter()
        .filter_map(|s| {
            if let Statement::For(f) = s {
                Some(f)
            } else {
                None
            }
        })
        .collect();

    assert_eq!(for_stmts.len(), 4, "expected 4 for loops");

    // expression initializer: `for (i = 0; ...)`
    assert!(
        matches!(&for_stmts[0].initializer, Some(ForInit::Expr(_))),
        "first loop should have Expr initializer"
    );

    // C99 int declaration: `for (int j = 0; ...)`
    let Some(ForInit::Decl(decl)) = &for_stmts[1].initializer else {
        panic!("second loop should have Decl initializer");
    };
    assert_eq!(decl.name, "j");
    assert_eq!(decl.type_info.base_type, "int");

    // pointer declaration: `for (GList *l = list; ...)`
    let Some(ForInit::Decl(decl)) = &for_stmts[2].initializer else {
        panic!("third loop should have Decl initializer");
    };
    assert_eq!(decl.name, "l");
    assert_eq!(decl.type_info.base_type, "GList");

    // no initializer: `for (; i < 20; ...)`
    assert!(
        for_stmts[3].initializer.is_none(),
        "fourth loop should have no initializer"
    );
}