use std::cell::{Cell, RefCell};
use rust_cc::*;

#[derive(Finalize)]
struct ToTrace {
    has_been_traced: Cell<bool>,
}

unsafe impl Trace for ToTrace {
    fn trace(&self, _: &mut Context<'_>) {
        self.has_been_traced.set(true);
    }
}

impl ToTrace {
    fn new() -> Cc<ToTrace> {
        Cc::new(ToTrace {
            has_been_traced: Cell::new(false),
        })
    }
}

#[derive(Trace, Finalize)] // Finalize is required by Trace
struct MyStruct {
    cyclic: RefCell<Option<Cc<MyStruct>>>,
    traced: Cc<ToTrace>,
    #[rust_cc(ignore)]
    ignored: Cc<ToTrace>,
}

fn main() {
    let my_struct = Cc::new(MyStruct {
        cyclic: RefCell::new(None),
        traced: ToTrace::new(),
        ignored: ToTrace::new(),
    });

    *my_struct.cyclic.borrow_mut() = Some(my_struct.clone());

    // Drop an instance and collect
    let _ = my_struct.clone();
    collect_cycles();

    assert!(my_struct.traced.has_been_traced.get());
    assert!(!my_struct.ignored.has_been_traced.get());
}