1 | #include <mruby.h>
|
---|
2 | #include <mruby/proc.h>
|
---|
3 | #include <mruby/class.h>
|
---|
4 |
|
---|
5 | static mrb_value
|
---|
6 | return_func_name(mrb_state *mrb, mrb_value self)
|
---|
7 | {
|
---|
8 | return mrb_cfunc_env_get(mrb, 0);
|
---|
9 | }
|
---|
10 |
|
---|
11 | static mrb_value
|
---|
12 | proc_new_cfunc_with_env(mrb_state *mrb, mrb_value self)
|
---|
13 | {
|
---|
14 | mrb_sym n;
|
---|
15 | mrb_value n_val;
|
---|
16 | mrb_get_args(mrb, "n", &n);
|
---|
17 | n_val = mrb_symbol_value(n);
|
---|
18 | mrb_define_method_raw(mrb, mrb_class_ptr(self), n,
|
---|
19 | mrb_proc_new_cfunc_with_env(mrb, return_func_name, 1, &n_val));
|
---|
20 | return self;
|
---|
21 | }
|
---|
22 |
|
---|
23 | static mrb_value
|
---|
24 | return_env(mrb_state *mrb, mrb_value self)
|
---|
25 | {
|
---|
26 | mrb_int idx;
|
---|
27 | mrb_get_args(mrb, "i", &idx);
|
---|
28 | return mrb_cfunc_env_get(mrb, idx);
|
---|
29 | }
|
---|
30 |
|
---|
31 | static mrb_value
|
---|
32 | cfunc_env_get(mrb_state *mrb, mrb_value self)
|
---|
33 | {
|
---|
34 | mrb_sym n;
|
---|
35 | mrb_value *argv; mrb_int argc;
|
---|
36 | mrb_get_args(mrb, "na", &n, &argv, &argc);
|
---|
37 | mrb_define_method_raw(mrb, mrb_class_ptr(self), n,
|
---|
38 | mrb_proc_new_cfunc_with_env(mrb, return_env, argc, argv));
|
---|
39 | return self;
|
---|
40 | }
|
---|
41 |
|
---|
42 | static mrb_value
|
---|
43 | cfunc_without_env(mrb_state *mrb, mrb_value self)
|
---|
44 | {
|
---|
45 | return mrb_cfunc_env_get(mrb, 0);
|
---|
46 | }
|
---|
47 |
|
---|
48 | void mrb_mruby_proc_ext_gem_test(mrb_state *mrb)
|
---|
49 | {
|
---|
50 | struct RClass *cls;
|
---|
51 |
|
---|
52 | cls = mrb_define_class(mrb, "ProcExtTest", mrb->object_class);
|
---|
53 | mrb_define_module_function(mrb, cls, "mrb_proc_new_cfunc_with_env", proc_new_cfunc_with_env, MRB_ARGS_REQ(1));
|
---|
54 | mrb_define_module_function(mrb, cls, "mrb_cfunc_env_get", cfunc_env_get, MRB_ARGS_REQ(2));
|
---|
55 | mrb_define_module_function(mrb, cls, "cfunc_without_env", cfunc_without_env, MRB_ARGS_NONE());
|
---|
56 | }
|
---|