to_source 0.0.1 → 0.1.0

data/Readme.md

@@ -18,7 +18,14 @@ transform themselves into source code. It's the reverse of Rubinius' builtin
 
 to_source needs Rubinius 2.0 to run, in either 1.8 or 1.9 mode.
 
-Put this in your Gemfile:
+To install it as a gem:
+
+    $ gem install to_source
+
+And `require 'to_source'` from your code. Automatically, your AST nodes respond
+to the `#to_source` method.
+
+But if you're using Bundler, just put this in your Gemfile:
 
     gem 'to_source'
 

data/lib/to_source/version.rb

@@ -1,3 +1,3 @@
 module ToSource
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/lib/to_source/visitor.rb

@@ -87,6 +87,12 @@ module ToSource
       node.finish.lazy_visit self, node
     end
 
+    def range_exclude(node, parent)
+      node.start.lazy_visit self, node
+      emit '...'
+      node.finish.lazy_visit self, node
+    end
+
     def regex_literal(node, parent)
       emit '/'
       emit node.source
@@ -217,6 +223,90 @@ module ToSource
       emit node.name
     end
 
+    def if(node, parent)
+      body, else_body = node.body, node.else
+      keyword = 'if'
+
+      if node.body.is_a?(Rubinius::AST::NilLiteral) && !node.else.is_a?(Rubinius::AST::NilLiteral)
+
+        body, else_body = else_body, body
+        keyword = 'unless'
+      end
+
+      emit keyword << ' '
+      node.condition.lazy_visit self, node
+      emit "\n"
+
+      @indentation += 1
+
+      if body.is_a?(Rubinius::AST::Block)
+        body.lazy_visit self, parent, true
+      else
+        emit current_indentation
+        body.lazy_visit self, parent
+      end
+
+      emit "\n"
+
+      if else_body.is_a?(Rubinius::AST::NilLiteral)
+        emit 'end'
+        return
+      end
+
+      emit "else\n"
+
+      if else_body.is_a?(Rubinius::AST::Block)
+        else_body.lazy_visit self, parent, true
+      else
+        emit current_indentation
+        else_body.lazy_visit self, parent
+      end
+
+      emit "\n"
+      emit 'end'
+    end
+
+    def while(node, parent)
+      emit 'while '
+      node.condition.lazy_visit self, node
+      emit "\n"
+
+      @indentation += 1
+
+      if node.body.is_a?(Rubinius::AST::Block)
+        node.body.lazy_visit self, parent, true
+      else
+        emit current_indentation
+        node.body.lazy_visit self, parent
+      end
+
+      emit "\n"
+      emit "end"
+    end
+
+    def until(node, parent)
+      emit 'until '
+      node.condition.lazy_visit self, node
+      emit "\n"
+
+      @indentation += 1
+
+      if node.body.is_a?(Rubinius::AST::Block)
+        node.body.lazy_visit self, parent, true
+      else
+        emit current_indentation
+        node.body.lazy_visit self, parent
+      end
+
+      emit "\n"
+      emit "end"
+    end
+
+    def return(node, parent)
+      emit 'return '
+      node.value.lazy_visit self, parent
+    end
+
     private
 
     def process_binary_operator(node, parent)

data/test/to_source/visitor_test.rb

@@ -71,6 +71,10 @@ module ToSource
       assert_source '20..34'
     end
 
+    def test_range_exclude
+      assert_source '20...34'
+    end
+
     def test_regex
       assert_source '/.*/'
     end
@@ -122,5 +126,45 @@ module ToSource
       assert_source "!1"
       assert_source "!!1"
     end
+
+    def test_if
+      assert_source "if 3\n  9\nend"
+    end
+
+    def test_if_with_multiple_blocks
+      assert_source "if 3\n  9\n  8\nend"
+    end
+
+    def test_else
+      assert_source "if 3\n  9\nelse\n  9\nend"
+    end
+
+    def test_else_with_multiple_blocks
+      assert_source "if 3\n  9\n  8\nelse\n  9\n  8\nend"
+    end
+
+    def test_unless
+      assert_source "unless 3\n  9\nend"
+    end
+
+    def test_while
+      assert_source "while false\n  3\nend"
+    end
+
+    def test_while_with_multiple_blocks
+      assert_source "while false\n  3\n  5\nend"
+    end
+
+    def test_until
+      assert_source "until false\n  3\nend"
+    end
+
+    def test_until_with_multiple_blocks
+      assert_source "while false\n  3\n  5\nend"
+    end
+
+    def test_return
+      assert_source "return 9"
+    end
   end
 end

metadata

@@ -2,14 +2,14 @@
 name: to_source
 version: !ruby/object:Gem::Version
   prerelease: 
-  version: 0.0.1
+  version: 0.1.0
 platform: ruby
 authors:
 - Josep M. Bach
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-02-23 00:00:00.000000000 Z
+date: 2012-02-24 00:00:00.000000000 Z
 dependencies: []
 description: Transform your Rubinius AST nodes back to source code. Reverse parsing!
 email: