binding_ninja 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f1c7723d881ecc904a3b67b20881a9be6d6fac52
-  data.tar.gz: bde453b73c9ffee4d969579d7019ffe468e0a5a5
+  metadata.gz: 3d3854a6100c403b0db08ad5f0c4c6c4e4b2c527
+  data.tar.gz: b2b6748aa0b7c8195c94f69c5d80798908d87d69
 SHA512:
-  metadata.gz: 23871995930d39725660c653138eb9069598efa27530539ba213db193e7de118792316252d4e7e8e658c84491e1b959a34c2af0da6afcbf2663922f3b5d7f587
-  data.tar.gz: a2e3f3dac226dd124805fb289a1db3d8af137cc422282eb761fcb7b87f62af71b80c31ed7fc3a9c01f87c73f552f6344698831bd0c7f965df8a6d04198f92aaa
+  metadata.gz: e62aebf9862aa927ef2112aa8f6abac2f4b85ac93bef9a4c15417e2bc720000ee21051c033acfc17a0585d302949d9aed8c56352e9df33aced3741a4efb41776
+  data.tar.gz: af4d430af39734eaa1f63205aa4f115dc258eb4a724c2e662828d381d892056b29d3304ed1f3c626fb9baefc09058467202885029c1d40e826a7fe546aee3dba

data/.travis.yml

@@ -1,5 +1,6 @@
 sudo: false
 language: ruby
 rvm:
+  - 2.3.2
   - 2.4.1
-before_install: gem install bundler -v 1.15.3
+before_install: gem install bundler -v 1.15.4

data/README.md

@@ -1,4 +1,5 @@
 # BindingNinja
+[![Build Status](https://travis-ci.org/joker1007/binding_ninja.svg?branch=master)](https://travis-ci.org/joker1007/binding_ninja)
 
 This is method wrapper for passing binding of method caller implcitly.
 
@@ -30,14 +31,39 @@ class Foo
     p arg2
   end
   auto_inject_binding :foo
+
+  def foo2(binding, arg1, arg2)
+    p binding
+    p arg1
+    p arg2
+  end
+  auto_inject_binding :foo2, if: ENV["ENABLE_BINDING_NINJA"]
+  # or
+  auto_inject_binding :foo2, if: ->(obj) { obj.enable_auto_inject_binding? }
+  # or
+  auto_inject_binding :foo2, if: :enable_auto_inject_binding?
+
+  def enable_auto_inject_binding?
+    true
+  end
 end
 
 Foo.new.foo(1, 2) 
 # => <Binding of toplevel>
 # => 1
 # => 2
+
+# if ENABLE_BINDING_NINJA environment variable is nil or false,
+# binding arguments is nil.
+Foo.new.foo2(1, 2) 
+# => nil
+# => 1
+# => 2
 ```
 
+`:if` option can accept Proc object and Symbol object.
+If option accepts a proc or symbol, uses result of evaluating the proc or method named by the symbol.
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/ext/binding_ninja/binding_ninja.c

@@ -3,30 +3,55 @@
 VALUE rb_mBindingNinja;
 static VALUE auto_inject_binding_invoke(int argc, VALUE *argv, VALUE self);
 
+static ID options_id;
+
 static VALUE
-auto_inject_binding(VALUE mod, VALUE method_sym)
+auto_inject_binding(int argc, VALUE *argv, VALUE mod)
 {
     ID mid;
-    VALUE mod_name, extensions, ext_mod;
+    static ID extensions_id;
+    static ID keyword_ids[1];
+    VALUE extensions, ext_mod, method_sym, options, opt, cond;
+
+    if (!keyword_ids[0]) {
+      keyword_ids[0] = rb_intern("if");
+    }
+
+    if (!extensions_id) {
+      extensions_id = rb_intern("@auto_inject_binding_extensions");
+    }
+
+    if (rb_cvar_defined(mod, options_id)) {
+      options = rb_cvar_get(mod, options_id);
+    } else {
+      options = rb_hash_new();
+      rb_cvar_set(mod, options_id, options);
+    }
+
+    rb_scan_args(argc, argv, "1:", &method_sym, &opt);
+    if (!NIL_P(opt)) {
+      rb_get_kwargs(opt, keyword_ids, 0, 1, &cond);
+      if (cond != Qundef) {
+        rb_hash_aset(options, method_sym, cond);
+      }
+    }
 
     mid = SYM2ID(method_sym);
-    int arity = rb_mod_method_arity(mod, mid);
-    if (abs(arity) < 1) {
+    if (abs(rb_mod_method_arity(mod, mid)) < 1) {
       rb_raise(rb_eArgError, "target method receives 1 or more arguments");
     }
 
-    mod_name = rb_mod_name(mod);
-    extensions = rb_ivar_get(rb_mBindingNinja, rb_intern("@auto_inject_binding_extensions"));
-    ext_mod = rb_hash_aref(extensions, mod_name);
-    if (ext_mod == Qnil) {
+    extensions = rb_ivar_get(rb_mBindingNinja, extensions_id);
+    ext_mod = rb_hash_aref(extensions, mod);
+    if (NIL_P(ext_mod)) {
       ext_mod = rb_module_new();
-      rb_hash_aset(extensions, mod_name, ext_mod);
+      rb_hash_aset(extensions, mod, ext_mod);
     }
+
     if (rb_mod_include_p(mod, ext_mod) == Qfalse) {
       rb_prepend_module(mod, ext_mod);
     }
-
-    rb_define_method_id(ext_mod, SYM2ID(method_sym), auto_inject_binding_invoke, -1);
+    rb_define_method_id(ext_mod, mid, auto_inject_binding_invoke, -1);
 
     return method_sym;
 }
@@ -34,12 +59,36 @@ auto_inject_binding(VALUE mod, VALUE method_sym)
 static VALUE
 auto_inject_binding_invoke(int argc, VALUE *argv, VALUE self)
 {
-  VALUE binding, args_ary;
+  VALUE method_sym, options, binding, args_ary, cond;
+  static VALUE dummy_proc_args, dummy_method_arg[0];
 
-  binding = rb_binding_new();
-  args_ary = rb_ary_new_from_values(argc, argv);
-  rb_ary_unshift(args_ary, binding);
+  if (!dummy_proc_args) {
+    dummy_proc_args = rb_ary_new();
+    rb_obj_freeze(dummy_proc_args);
+  }
+
+  method_sym = ID2SYM(rb_frame_this_func());
+  options = rb_cvar_get(CLASS_OF(self), options_id);
 
+  cond = rb_hash_lookup2(options, method_sym, Qtrue);
+
+  if (rb_obj_is_proc(cond)) {
+    if (abs(rb_proc_arity(cond)) > 0) {
+      cond = rb_proc_call(cond, rb_ary_new_from_args(1, self));
+    } else {
+      cond = rb_proc_call(cond, dummy_proc_args);
+    }
+  } else if (SYMBOL_P(cond)) {
+    cond = rb_method_call(0, dummy_method_arg, rb_obj_method(self, cond));
+  }
+
+  args_ary = rb_ary_new_from_values(argc, argv);
+  if (RTEST(cond)) {
+    binding = rb_binding_new();
+    rb_ary_unshift(args_ary, binding);
+  } else {
+    rb_ary_unshift(args_ary, Qnil);
+  }
   return rb_call_super(argc + 1, RARRAY_CONST_PTR(args_ary));
 }
 
@@ -47,6 +96,7 @@ void
 Init_binding_ninja(void)
 {
   rb_mBindingNinja = rb_define_module("BindingNinja");
+  options_id = rb_intern("@@__auto_inject_binding_options");
   rb_ivar_set(rb_mBindingNinja, rb_intern("@auto_inject_binding_extensions"), rb_hash_new());
-  rb_define_private_method(rb_mBindingNinja, "auto_inject_binding", auto_inject_binding, 1);
+  rb_define_private_method(rb_mBindingNinja, "auto_inject_binding", auto_inject_binding, -1);
 }

data/lib/binding_ninja/version.rb

@@ -1,3 +1,3 @@
 module BindingNinja
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: binding_ninja
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - joker1007
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-08-07 00:00:00.000000000 Z
+date: 2017-09-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler