proc_to_ast 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 91c5fe173fcebfbe9b300cae9ec8c670fe5faafe
-  data.tar.gz: 79c1ff86ee61c98ea86e6c5fcb09c651a461adb8
+  metadata.gz: 08e16687b46d8a606c8490cfdf2046302a98c490
+  data.tar.gz: 8c1d95a9b8fc423b77ea79d5a8132e3d49a496d2
 SHA512:
-  metadata.gz: 09568ba9f1b51ffde7d256d08f2e145a548379f38bba040ee40debb5a80a66a79ecaf7e60a0cc0358c3185d58da4ff6fdd7014c60ba193f9e9f06c750bb644e5
-  data.tar.gz: 4d077ea1bdb11c48a0e518d3252bc5a763e6306033c4f6cc22fe42102bb4fb66cfa7f3b0b246d0887b411292673e408f643486a5b77ccbe269a400f2624afe15
+  metadata.gz: 4126703d5e1e93a62a64dc7fe795bcd91ea181939d2db7455b1f7b3fd29fc53d9b100db51c4f1e68f692f9d607e193b54880ac39e8a5761202fb8d47550abc61
+  data.tar.gz: b57c0598263cd2bdfeb9e03e47dae6158510ca723159de749ac935939290dd45c702f441e90719e96d984e0c28eb422c497b2e2c7f7157039680051bc50888f1

data/lib/proc_to_ast.rb

@@ -5,6 +5,8 @@ require 'unparser'
 require 'coderay'
 
 module ProcToAst
+  class MultiMatchError < StandardError; end
+
   class Traverser
     def traverse_node(node)
       if node.type != :block
@@ -14,9 +16,7 @@ module ProcToAst
           end
         }.compact
       else
-        if proc_block?(node)
-          node
-        end
+        node
       end
     end
 
@@ -62,7 +62,12 @@ class Proc
       source_buffer = Parser::Source::Buffer.new(filename, linenum)
       source_buffer.source = source
       node = parser.parse(source_buffer)
-      ProcToAst::Traverser.new.traverse_node(node).first
+      block_nodes = ProcToAst::Traverser.new.traverse_node(node)
+      if block_nodes.length == 1
+        block_nodes.first
+      else
+        raise ProcToAst::MultiMatchError
+      end
     rescue Parser::SyntaxError
       retry if try_count < retry_limit
     end

data/lib/proc_to_ast/version.rb

@@ -1,3 +1,3 @@
 module ProcToAst
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/proc_to_ast_spec.rb

@@ -8,27 +8,52 @@ describe Proc do
       expect(ast.type).to eq(:block)
     end
 
-    it "converts proc variation" do
+    context "proc variation" do
+      it "converts Kernel#proc" do
+        pr = proc { p 1 }
 
-      hoge = proc { p 1 }
+        expect(pr.to_ast).to be_a(AST::Node)
+      end
 
-      _ = [1].map {|i| i * 2}; fuga = ->(a) {
-        p a
-      }
+      it "converts Proc.new" do
+        pr = Proc.new do |b|
+          puts b
+        end
+
+        expect(pr.to_ast).to be_a(AST::Node)
+      end
 
-      foo = Proc.new do |b|
-        puts b
+      it "converts block passing method" do
+        def receive_block(&block)
+          block.to_ast
+        end
+
+        block_pass = receive_block do |n|
+          puts n
+          [1, 2, 3].map do |i|
+            i * 2
+          end
+        end
+
+        block_pass2 = receive_block { %w(a b c).map(&:upcase) }
+
+        expect(block_pass).to be_a(AST::Node)
+        expect(block_pass2).to be_a(AST::Node)
       end
 
-      expect(hoge.to_ast).to be_a(AST::Node)
-      expect(fuga.to_ast).to be_a(AST::Node)
-      expect(foo.to_ast).to be_a(AST::Node)
+      it "raise ProcToAst::MultiMatchError, when other block exists on same line " do
+        _ = [1].map {|i| i * 2}; fuga = ->(a) {
+          p a
+        }
+
+        expect{ fuga.to_ast }.to raise_error(ProcToAst::MultiMatchError)
+      end
     end
   end
 
   describe "#to_source" do
     it "return source code string" do
-      _ = [1].map {|i| i * 2}; fuga = ->(a) {
+      fuga = ->(a) {
         p a
       }
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: proc_to_ast
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - joker1007