1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

//! FFI glue between the Rust trampolines and the mruby C interpreter.

use std::ffi::CStr;

use crate::extn::core::artichoke;
use crate::extn::core::env::{self, trampoline};
use crate::extn::prelude::*;

const ENVIRON_CSTR: &CStr = qed::const_cstr_from_str!("Environ\0");
static ENV_RUBY_SOURCE: &[u8] = include_bytes!("env.rb");

pub fn init(interp: &mut Artichoke) -> InitializeResult<()> {
    if interp.is_class_defined::<env::Environ>() {
        return Ok(());
    }

    let scope = interp
        .module_spec::<artichoke::Artichoke>()?
        .map(EnclosingRubyScope::module)
        .ok_or_else(|| NotDefinedError::module("Artichoke"))?;
    let spec = class::Spec::new(
        "Environ",
        ENVIRON_CSTR,
        Some(scope),
        Some(def::box_unbox_free::<env::Environ>),
    )?;
    class::Builder::for_spec(interp, &spec)
        .value_is_rust_object()
        .add_method("[]", env_element_reference, sys::mrb_args_req(1))?
        .add_method("[]=", env_element_assignment, sys::mrb_args_req(2))?
        .add_method("initialize", env_initialize, sys::mrb_args_none())?
        .add_method("to_h", env_to_h, sys::mrb_args_none())?
        .define()?;
    interp.def_class::<env::Environ>(spec)?;
    interp.eval(ENV_RUBY_SOURCE)?;

    Ok(())
}

unsafe extern "C" fn env_initialize(mrb: *mut sys::mrb_state, slf: sys::mrb_value) -> sys::mrb_value {
    mrb_get_args!(mrb, none);
    unwrap_interpreter!(mrb, to => guard);
    let slf = Value::from(slf);
    let result = trampoline::initialize(&mut guard, slf);
    match result {
        Ok(value) => value.inner(),
        Err(exception) => error::raise(guard, exception),
    }
}

unsafe extern "C" fn env_element_reference(mrb: *mut sys::mrb_state, slf: sys::mrb_value) -> sys::mrb_value {
    let name = mrb_get_args!(mrb, required = 1);
    unwrap_interpreter!(mrb, to => guard);
    let obj = Value::from(slf);
    let name = Value::from(name);
    let result = trampoline::element_reference(&mut guard, obj, name);
    match result {
        Ok(value) => value.inner(),
        Err(exception) => error::raise(guard, exception),
    }
}

unsafe extern "C" fn env_element_assignment(mrb: *mut sys::mrb_state, slf: sys::mrb_value) -> sys::mrb_value {
    let (name, value) = mrb_get_args!(mrb, required = 2);
    unwrap_interpreter!(mrb, to => guard);
    let obj = Value::from(slf);
    let name = Value::from(name);
    let value = Value::from(value);
    let result = trampoline::element_assignment(&mut guard, obj, name, value);
    match result {
        Ok(value) => value.inner(),
        Err(exception) => error::raise(guard, exception),
    }
}

unsafe extern "C" fn env_to_h(mrb: *mut sys::mrb_state, slf: sys::mrb_value) -> sys::mrb_value {
    mrb_get_args!(mrb, none);
    unwrap_interpreter!(mrb, to => guard);
    let obj = Value::from(slf);
    let result = trampoline::to_h(&mut guard, obj);
    match result {
        Ok(value) => value.inner(),
        Err(exception) => error::raise(guard, exception),
    }
}