furnace 0.1.0 → 0.1.1

data/lib/furnace/ast/visitor.rb

@@ -1,19 +1,31 @@
 module Furnace::AST
   module Visitor
     def visit(node)
-      node.children.map! do |child|
+      replacements = {}
+
+      node.children.each_with_index do |child, index|
         if child.is_a? Node
           visit child
 
           if child.type == :expand
-            child = child.children
+            replacements[index] = child.children
           end
         end
-
-        child
       end
 
-      node.children.flatten!
+      if replacements.any?
+        new_children = []
+
+        node.children.each_with_index do |child, index|
+          if replacements[index]
+            new_children.concat replacements[index]
+          else
+            new_children.push child
+          end
+        end
+
+        node.children.replace new_children
+      end
 
       node.children.delete_if do |child|
         if child.is_a? Node

data/lib/furnace/version.rb

@@ -1,3 +1,3 @@
 module Furnace
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: furnace
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-11 00:00:00.000000000 Z
+date: 2012-04-19 00:00:00.000000000 Z
 dependencies: []
 description: Furnace is a static code analysis framework for dynamic languages, aimed
   at efficient type and behavior inference.