move ast to tree, remove seperate helper file

lib/typed/ast/code.rb

@@ -1,27 +0,0 @@
-
-# Base class for Expresssion and Statement
-module Typed
-
-  class Code
-
-  end
-
-  class Statement < Code
-  end
-  class Expression < Code
-  end
-
-end
-
-require_relative "while_statement"
-require_relative "if_statement"
-require_relative "return_statement"
-require_relative "statements"
-require_relative "operator_expression"
-require_relative "field_access"
-require_relative "call_site"
-require_relative "basic_values"
-require_relative "assignment"
-require_relative "class_statement"
-require_relative "function_statement"
-require_relative "to_code"

lib/typed/ast/while_statement.rb

@@ -1,5 +0,0 @@
-module Typed
-  class WhileStatement < Statement
-    attr_accessor :branch_type , :condition , :statements
-  end
-end

lib/typed/ast_helper.rb

@@ -1,23 +0,0 @@
-AST::Node.class_eval do
-
-  def [](name)
-    #puts self.inspect
-    children.each do |child|
-      if child.is_a?(AST::Node)
-        #puts child.type
-        if (child.type == name)
-          return child.children
-        end
-      else
-        #puts child.class
-      end
-    end
-    nil
-  end
-
-  def first_from( node_name )
-    from = self[node_name]
-    return nil unless from
-    from.first
-  end
-end

lib/typed/compiler.rb

@@ -1,3 +1,5 @@
+require_relative "tree"
+
 module Typed
 
   CompilerModules = [ "assignment" , "basic_values" , "call_site", "class_field" ,
@@ -187,8 +189,6 @@ module Typed
   end
 end
 
-require_relative "ast_helper"
-require_relative "ast/code"
 require_relative "compiler/collections"
 require_relative "compiler/field_def"
 require_relative "compiler/field_access"

lib/typed/tree.rb

@@ -0,0 +1,38 @@
+# Base class for Expresssion and Statement
+module Typed
+
+  class Code              ; end
+  class Statement < Code  ; end
+  class Expression < Code ; end
+
+end
+
+["while_statement", "if_statement" , "return_statement" , "statements",
+  "operator_expression" , "field_access" , "call_site" , "basic_values",
+  "assignment" , "class_statement" , "function_statement" , "to_code"].each do |code|
+    require_relative "tree/" + code
+end
+
+AST::Node.class_eval do
+
+  def [](name)
+    #puts self.inspect
+    children.each do |child|
+      if child.is_a?(AST::Node)
+        #puts child.type
+        if (child.type == name)
+          return child.children
+        end
+      else
+        #puts child.class
+      end
+    end
+    nil
+  end
+
+  def first_from( node_name )
+    from = self[node_name]
+    return nil unless from
+    from.first
+  end
+end

lib/typed/tree/while_statement.rb

@@ -0,0 +1,7 @@
+module Typed
+  module Tree
+    class WhileStatement < Statement
+      attr_accessor :branch_type , :condition , :statements
+    end
+  end
+end