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_method_t m;
|
---|
17 | struct RProc *p;
|
---|
18 | mrb_get_args(mrb, "n", &n);
|
---|
19 | n_val = mrb_symbol_value(n);
|
---|
20 | p = mrb_proc_new_cfunc_with_env(mrb, return_func_name, 1, &n_val);
|
---|
21 | MRB_METHOD_FROM_PROC(m, p);
|
---|
22 | mrb_define_method_raw(mrb, mrb_class_ptr(self), n, m);
|
---|
23 | return self;
|
---|
24 | }
|
---|
25 |
|
---|
26 | static mrb_value
|
---|
27 | return_env(mrb_state *mrb, mrb_value self)
|
---|
28 | {
|
---|
29 | mrb_int idx;
|
---|
30 | mrb_get_args(mrb, "i", &idx);
|
---|
31 | return mrb_cfunc_env_get(mrb, idx);
|
---|
32 | }
|
---|
33 |
|
---|
34 | static mrb_value
|
---|
35 | cfunc_env_get(mrb_state *mrb, mrb_value self)
|
---|
36 | {
|
---|
37 | mrb_sym n;
|
---|
38 | mrb_value *argv; mrb_int argc;
|
---|
39 | mrb_method_t m;
|
---|
40 | struct RProc *p;
|
---|
41 | mrb_get_args(mrb, "na", &n, &argv, &argc);
|
---|
42 | p = mrb_proc_new_cfunc_with_env(mrb, return_env, argc, argv);
|
---|
43 | MRB_METHOD_FROM_PROC(m, p);
|
---|
44 | mrb_define_method_raw(mrb, mrb_class_ptr(self), n, m);
|
---|
45 | return self;
|
---|
46 | }
|
---|
47 |
|
---|
48 | static mrb_value
|
---|
49 | cfunc_without_env(mrb_state *mrb, mrb_value self)
|
---|
50 | {
|
---|
51 | return mrb_cfunc_env_get(mrb, 0);
|
---|
52 | }
|
---|
53 |
|
---|
54 | void mrb_mruby_proc_ext_gem_test(mrb_state *mrb)
|
---|
55 | {
|
---|
56 | struct RClass *cls;
|
---|
57 |
|
---|
58 | cls = mrb_define_class(mrb, "ProcExtTest", mrb->object_class);
|
---|
59 | mrb_define_module_function(mrb, cls, "mrb_proc_new_cfunc_with_env", proc_new_cfunc_with_env, MRB_ARGS_REQ(1));
|
---|
60 | mrb_define_module_function(mrb, cls, "mrb_cfunc_env_get", cfunc_env_get, MRB_ARGS_REQ(2));
|
---|
61 | mrb_define_module_function(mrb, cls, "cfunc_without_env", cfunc_without_env, MRB_ARGS_NONE());
|
---|
62 | }
|
---|