to_source 0.1.2 → 0.1.3

data/lib/to_source/core_ext/node.rb

@@ -9,8 +9,11 @@ module Rubinius
       #
       # Returns nothing.
       def lazy_visit(visitor, parent=nil, indent=false)
-        args = [self.node_name, self, parent]
-        args.push true if indent
+        name = node_name
+        name = "#{name}_def" if %w[ class module ].include?(name)
+
+        args = [name, self, parent]
+        args << true if indent
 
         visitor.__send__ *args
       end

data/lib/to_source/version.rb

@@ -1,3 +1,3 @@
 module ToSource
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/to_source/visitor.rb

@@ -17,6 +17,42 @@ module ToSource
       '  ' * @indentation
     end
 
+    def class_def(node, parent)
+      emit "class %s" % node.name.name
+
+      superclass = node.superclass
+      unless superclass.is_a?(Rubinius::AST::NilLiteral)
+        emit " < %s" % superclass.name
+      end
+
+      node.body.lazy_visit self, node, true
+
+      emit "\n"
+      emit "end"
+    end
+
+    def module_def(node, parent)
+      emit "module %s" % node.name.name
+
+      node.body.lazy_visit self, node, true
+
+      emit "\n"
+      emit "end"
+    end
+
+    def empty_body(*)
+      # do nothing
+    end
+
+    def class_scope(node, parent, indent)
+      emit "\n"
+      @indentation += 1 if indent
+      node.body.lazy_visit self, node, indent
+    ensure
+      @indentation -= 1 if indent
+    end
+    alias module_scope class_scope
+
     def local_variable_assignment(node, parent)
       emit "%s = " % node.name
       node.value.lazy_visit self, node

data/test/to_source/visitor_test.rb

@@ -15,6 +15,26 @@ module ToSource
       assert_equal expected, visit(code)
     end
 
+    def test_class
+      assert_source "class TestClass\nend"
+    end
+
+    def test_class_with_superclass
+      assert_source "class TestClass < Object\nend"
+    end
+
+    def test_class_with_body
+      assert_source "class TestClass\n  1\nend"
+    end
+
+    def test_module
+      assert_source "module TestModule\nend"
+    end
+
+    def test_module_with_body
+      assert_source "module TestModule\n  1\nend"
+    end
+
     def test_local_assignment
       assert_source "foo = 1"
     end

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: to_source
 version: !ruby/object:Gem::Version
+  version: 0.1.3
   prerelease: 
-  version: 0.1.2
 platform: ruby
 authors:
 - Josep M. Bach
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-11 00:00:00.000000000 Z
+date: 2012-04-12 00:00:00.000000000 Z
 dependencies: []
 description: Transform your Rubinius AST nodes back to source code. Reverse parsing!
 email:
@@ -37,17 +37,17 @@ rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
+required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+  none: false
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.12