use std::cell::RefCell;
use magnus::{function, method, prelude::*, wrap, Error, Ruby};
struct Point {
x: isize,
y: isize,
}
#[wrap(class = "Point")]
struct MutPoint(RefCell<Point>);
impl MutPoint {
fn new(x: isize, y: isize) -> Self {
Self(RefCell::new(Point { x, y }))
}
fn x(&self) -> isize {
self.0.borrow().x
}
fn set_x(&self, val: isize) {
self.0.borrow_mut().x = val;
}
fn add_x(ruby: &Ruby, rb_self: &Self, val: isize) -> Result<isize, Error> {
if let Some(sum) = rb_self.0.borrow().x.checked_add(val) {
rb_self.0.borrow_mut().x = sum;
Ok(sum)
} else {
Err(Error::new(
ruby.exception_range_error(),
"result out of range",
))
}
}
fn y(&self) -> isize {
self.0.borrow().y
}
fn set_y(&self, val: isize) {
self.0.borrow_mut().y = val;
}
fn add_y(ruby: &Ruby, rb_self: &Self, val: isize) -> Result<isize, Error> {
if let Some(sum) = rb_self.0.borrow().y.checked_add(val) {
rb_self.0.borrow_mut().y = sum;
Ok(sum)
} else {
Err(Error::new(
ruby.exception_range_error(),
"result out of range",
))
}
}
fn distance(&self, other: &MutPoint) -> f64 {
(((other.x() - self.x()).pow(2) + (other.y() - self.y()).pow(2)) as f64).sqrt()
}
}
fn main() -> Result<(), String> {
magnus::Ruby::init(|ruby| {
let class = ruby.define_class("Point", ruby.class_object())?;
class.define_singleton_method("new", function!(MutPoint::new, 2))?;
class.define_method("x", method!(MutPoint::x, 0))?;
class.define_method("x=", method!(MutPoint::set_x, 1))?;
class.define_method("add_x", method!(MutPoint::add_x, 1))?;
class.define_method("y", method!(MutPoint::y, 0))?;
class.define_method("y=", method!(MutPoint::set_y, 1))?;
class.define_method("add_y", method!(MutPoint::add_y, 1))?;
class.define_method("distance", method!(MutPoint::distance, 1))?;
let d: f64 = ruby.eval(
"a = Point.new(0, 0)
b = Point.new(0, 0)
b.x = 5
b.y = 10
a.distance(b)",
)?;
println!("{}", d);
Ok(())
})
}