use magnus::{
embed::init, function, gc, method, prelude::*, typed_data::Obj, value::Opaque,
DataTypeFunctions, TypedData,
};
#[magnus::wrap(class = "Point", free_immediately)]
struct Point {
x: isize,
y: isize,
}
impl Point {
fn new(x: isize, y: isize) -> Self {
Self { x, y }
}
}
#[derive(TypedData)]
#[magnus(class = "Line", free_immediately, mark)]
struct Line {
#[magnus(opaque_attr_reader)]
start: Opaque<Obj<Point>>,
#[magnus(opaque_attr_reader)]
end: Opaque<Obj<Point>>,
}
impl Line {
fn new(start: Obj<Point>, end: Obj<Point>) -> Self {
Self {
start: start.into(),
end: end.into(),
}
}
fn length(&self) -> f64 {
let start = self.start();
let end = self.end();
(((end.x - start.x).pow(2) + (end.y - start.y).pow(2)) as f64).sqrt()
}
}
impl DataTypeFunctions for Line {
fn mark(&self, marker: &gc::Marker) {
marker.mark(self.start);
marker.mark(self.end);
}
}
#[test]
fn it_can_nest_wrapped_structs() {
let ruby = unsafe { init() };
let class = ruby.define_class("Point", ruby.class_object()).unwrap();
class
.define_singleton_method("new", function!(Point::new, 2))
.unwrap();
let class = ruby.define_class("Line", ruby.class_object()).unwrap();
class
.define_singleton_method("new", function!(Line::new, 2))
.unwrap();
class
.define_method("length", method!(Line::length, 0))
.unwrap();
let result: f64 = ruby
.eval(
r#"
start = Point.new(0, 0)
finish = Point.new(10, 10)
line = Line.new(start, finish)
line.length
"#,
)
.unwrap();
assert!(result - 14.14213 < 0.00001);
}