eavi 2.0.0 → 2.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 89417f14ff76e2a1a025877ffc8f421da4efe828
-  data.tar.gz: f80ca41d54257abd0c2abd75943b29effce8fb2d
+  metadata.gz: cd59d004569dc526eda05b7ff0cfea3dab4d87aa
+  data.tar.gz: eea265db8b1a62cc99199d5e24da12de21804316
 SHA512:
-  metadata.gz: 616c403446a6fae16c540b058bee183a57de7884b9b749eba4b6aaac78786e4503b5dc438e3650a5d63ee8ddf76a4c07d20d59ceecb2c4f3bd3fa9b89e798aea
-  data.tar.gz: beb4d9dff7a5833de44466f2ad383b3774cd3a6d0b0c06bfc5ceb12dd83e423e43751f7270b2ab5b026bf8dbc5f364db55c631a06650b1c3f15e37ab1cdecb7b
+  metadata.gz: ef978d38107d68845f29d377f78b0ac7858513fb1d98efa2e0ef278d9c5407eef165c95ab3223c46e8831c73b754c9d33a2e3435520d414ede4d6f763f02ac65
+  data.tar.gz: 00f83214a0489d1c2e7b224d97b67b9ddfa91f131bf67d06ee43b83955e17481b13bb5c416d60c2ea33e843d1402a1d3c0bd8970cfb13a65df13140f62633f42

data/lib/eavi/visitor.rb

@@ -31,6 +31,8 @@ module Eavi
 
     # Domain-Specific Language for the module/class
     module ModuleDSL
+      # protected
+
       # DSL method to add visit methods on types +types+.
       def def_visit(*types, &block)
         add_visit_method(*types, &block)
@@ -51,9 +53,12 @@ module Eavi
 
       # Add/override a visit method for the types +types+.
       def add_visit_method(*types, &block)
-        block = block.curry(1) if block.arity.zero?
+        if block.arity.zero?
+          original_block = block
+          block = proc { |_| instance_exec(&original_block) }
+        end
         types.each do |type|
-          specialized_add_visit_method(type, &block)
+          specialized_add_visit_method(type, block)
         end
       end
 
@@ -73,9 +78,7 @@ module Eavi
 
       # Return a list of the visit method.
       def visit_methods
-        return methods.select do |method|
-          VisitMethodHelper.match(method)
-        end
+        specialized_visit_methods
       end
 
       # Return a list of the types with a visit method.
@@ -94,7 +97,7 @@ module Eavi
         define_method(visit_method_alias, instance_method(:visit))
       end
 
-      def specialized_add_visit_method(type, &block)
+      def specialized_add_visit_method(type, block)
         define_method(VisitMethodHelper.gen_name(type), block)
       end
 
@@ -105,6 +108,12 @@ module Eavi
       def specialized_remove_method(visit_method)
         remove_method(visit_method)
       end
+
+      def specialized_visit_methods
+        return instance_methods.select do |method|
+          VisitMethodHelper.match(method)
+        end
+      end
     end
 
     # Extends only when extended
@@ -115,7 +124,7 @@ module Eavi
         define_singleton_method(visit_method_alias, method(:visit))
       end
 
-      def specialized_add_visit_method(type, &block)
+      def specialized_add_visit_method(type, block)
         define_singleton_method(VisitMethodHelper.gen_name(type), block)
       end
 
@@ -126,6 +135,12 @@ module Eavi
       def specialized_remove_method(visit_method)
         singleton_class.send(:remove_method, visit_method)
       end
+
+      def specialized_visit_methods
+        return methods.select do |method|
+          VisitMethodHelper.match(method)
+        end
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: eavi
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.0.1
 platform: ruby
 authors:
 - Pierre Le Gall
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-02-11 00:00:00.000000000 Z
+date: 2017-03-05 00:00:00.000000000 Z
 dependencies: []
 description: Make the visitor pattern accessible and flexible in Ruby.
 email: pierre@legall.im