duby 0.0.1

data/lib/duby/ast/intrinsics.rb

@@ -0,0 +1,46 @@
+module Duby::AST
+  class Print < Node
+    attr_accessor :parameters
+    attr_accessor :println
+    
+    def initialize(parent, line_number, println, &block)
+      super(parent, line_number, &block)
+      @parameters = children
+      @println = println
+    end
+
+    def infer(typer)
+      if parameters.size > 0
+        resolved = parameters.select {|param| typer.infer(param); param.resolved?}
+        resolved! if resolved.size == parameters.size
+      else
+        resolved!
+      end
+      typer.no_type
+    end
+  end
+  
+  defmacro('puts') do |transformer, fcall, parent|
+    Print.new(parent, fcall.position, true) do |print|
+      if fcall.respond_to?(:args_node) && fcall.args_node
+        fcall.args_node.child_nodes.map do |arg|
+          transformer.transform(arg, print)
+        end
+      else
+        []
+      end
+    end
+  end
+
+  defmacro('print') do |transformer, fcall, parent|
+    Print.new(parent, fcall.position, false) do |print|
+      if fcall.respond_to?(:args_node) && fcall.args_node
+        fcall.args_node.child_nodes.map do |arg|
+          transformer.transform(arg, print)
+        end
+      else
+        []
+      end
+    end
+  end
+end

data/lib/duby/ast/literal.rb

@@ -0,0 +1,93 @@
+module Duby::AST
+  class Array < Node
+    def initialize(parent, line_number, &block)
+      super(parent, line_number, &block)
+    end
+  end
+  
+  class Fixnum < Node
+    include Literal
+    
+    def initialize(parent, line_number, literal)
+      super(parent, line_number)
+      @literal = literal
+    end
+    
+    def infer(typer)
+      return @inferred_type if resolved?
+      resolved!
+      @inferred_type = typer.fixnum_type
+    end
+    
+    def ==(other)
+      @literal == other.literal
+    end
+    
+    def eql?(other)
+      self.class == other.class && @literal.eql?(other.literal)
+    end
+  end
+  
+  class Float < Node
+    include Literal
+    
+    def initialize(parent, line_number, literal)
+      super(parent, line_number)
+      @literal = literal
+    end
+    
+    def infer(typer)
+      return @inferred_type if resolved?
+      resolved!
+      @inferred_type = typer.float_type
+    end
+  end
+  
+  class Hash < Node; end
+  
+  class String < Node
+    include Literal
+    
+    def initialize(parent, line_number, literal)
+      super(parent, line_number)
+      @literal = literal
+    end
+    
+    def infer(typer)
+      return @inferred_type if resolved?
+      resolved!
+      @inferred_type ||= typer.string_type
+    end
+  end
+  
+  class Symbol < Node; end
+  
+  class Boolean < Node
+    include Literal
+    
+    def initialize(parent, line_number, literal)
+      super(parent, line_number)
+      @literal = literal
+    end
+    
+    def infer(typer)
+      return @inferred_type if resolved?
+      resolved!
+      @inferred_type ||= typer.boolean_type
+    end
+  end
+
+  class Null < Node
+    include Literal
+
+    def initialize(parent, line_number)
+      super(parent, line_number)
+    end
+
+    def infer(typer)
+      return @inferred_type if resolved?
+      resolved!
+      @inferred_type ||= typer.null_type
+    end
+  end
+end

data/lib/duby/ast/local.rb

@@ -0,0 +1,77 @@
+module Duby::AST
+  class LocalDeclaration < Node
+    include Named
+    include Typed
+    include Scoped
+
+    def initialize(parent, line_number, name, &block)
+      super(parent, line_number, &block)
+      @name = name
+      @type = children[0]
+    end
+
+    def infer(typer)
+      unless resolved?
+        resolved!
+        @inferred_type = typer.known_types[type] || type
+        if @inferred_type
+          resolved!
+          typer.learn_local_type(scope, name, @inferred_type)
+        else
+          typer.defer(self)
+        end
+      end
+      @inferred_type
+    end
+  end
+  
+  class LocalAssignment < Node
+    include Named
+    include Valued
+    include Scoped
+    
+    def initialize(parent, line_number, name, &block)
+      super(parent, line_number, children, &block)
+      @value = children[0]
+      @name = name
+    end
+
+    def to_s
+      "LocalAssignment(name = #{name}, scope = #{scope})"
+    end
+    
+    def infer(typer)
+      unless @inferred_type
+        @inferred_type = typer.learn_local_type(scope, name, typer.infer(value))
+
+        @inferred_type ? resolved! : typer.defer(self)
+      end
+
+      @inferred_type
+    end
+  end
+
+  class Local < Node
+    include Named
+    include Scoped
+    
+    def initialize(parent, line_number, name)
+      super(parent, line_number, [])
+      @name = name
+    end
+
+    def to_s
+      "Local(name = #{name}, scope = #{scope})"
+    end
+    
+    def infer(typer)
+      unless @inferred_type
+        @inferred_type = typer.local_type(scope, name)
+
+        @inferred_type ? resolved! : typer.defer(self)
+      end
+
+      @inferred_type
+    end
+  end
+end

data/lib/duby/ast/method.rb

@@ -0,0 +1,187 @@
+module Duby::AST
+  class Arguments < Node
+    attr_accessor :args, :opt_args, :rest_arg, :block_arg
+    
+    def initialize(parent, line_number, &block)
+      super(parent, line_number, &block)
+      @args, @opt_args, @rest_arg, @block_arg = children
+    end
+    
+    def infer(typer)
+      unless @inferred_type
+        @inferred_type = args ? args.map {|arg| typer.infer(arg)} : []
+      end
+    end
+  end
+      
+  class Argument < Node
+    include Typed
+  end
+      
+  class RequiredArgument < Argument
+    include Named
+    include Scoped
+    
+    def initialize(parent, line_number, name)
+      super(parent, line_number)
+
+      @name = name
+    end
+    
+    def infer(typer)
+      unless @inferred_type
+        # if not already typed, check parent of parent (MethodDefinition) for signature info
+        method_def = parent.parent
+        signature = method_def.signature
+
+        # if signature, search for this argument
+        if signature[name.intern]
+          @inferred_type = typer.learn_local_type(scope, name, signature[name.intern])
+        else
+          @inferred_type = typer.local_type(scope, name)
+        end
+          
+        unless @inferred_type
+          typer.defer(self)
+        end
+      end
+        
+      @inferred_type
+    end
+  end
+      
+  class OptionalArgument < Argument
+    include Named
+    include Scoped
+    attr_accessor :child
+    
+    def initialize(parent, line_number, name, &block)
+      super(parent, line_number, &block)
+      @child = children[0]
+      @name = name
+    end
+
+    def infer(typer)
+      unless @inferred_type
+        # if not already typed, check parent of parent (MethodDefinition) for signature info
+        method_def = parent.parent
+        signature = method_def.signature
+
+        # if signature, search for this argument
+        @inferred_type = child.infer(typer)
+        if @inferred_type
+          typer.learn_local_type(scope, name, @inferred_type)
+          signature[name.intern] = @inferred_type
+        else
+          typer.defer(self)
+        end
+      end
+
+      @inferred_type
+    end
+  end
+      
+  class RestArgument < Argument
+    include Named
+    include Scoped
+    
+    def initialize(parent, line_number, name)
+      super(parent, line_number)
+
+      @name = name
+    end
+  end
+      
+  class BlockArgument < Argument
+    include Named
+    
+    def initialize(parent, line_number, name)
+      super(parent, line_number)
+
+      @name = name
+    end
+  end
+      
+  class MethodDefinition < Node
+    include Named
+    include Scope
+    attr_accessor :signature, :arguments, :body, :defining_class
+        
+    def initialize(parent, line_number, name, &block)
+      super(parent, line_number, &block)
+      @signature, @arguments, @body = children
+      @name = name
+    end
+    
+    def infer(typer)
+      @defining_class ||= typer.self_type
+      typer.infer(arguments)
+      typer.infer_signature(self)
+      forced_type = signature[:return]
+      inferred_type = body ? typer.infer(body) : typer.no_type
+        
+      if !inferred_type
+        typer.defer(self)
+      else
+        actual_type = if forced_type.nil?
+          inferred_type
+        else
+          forced_type
+        end
+        if actual_type.unreachable?
+          actual_type = typer.no_type
+        end
+        
+        if !abstract? &&
+            forced_type != typer.no_type &&
+            !actual_type.is_parent(inferred_type)
+          raise Duby::Typer::InferenceError.new(
+              "Inferred return type %s is incompatible with declared %s" %
+              [inferred_type, actual_type], self)
+        end
+
+        @inferred_type = typer.learn_method_type(defining_class, name, arguments.inferred_type, actual_type, signature[:throws])
+
+        # learn the other overloads as well
+        args_for_opt = []
+        if arguments.args
+          arguments.args.each do |arg|
+            if OptionalArgument === arg
+              arg_types_for_opt = args_for_opt.map do |arg_for_opt|
+                arg_for_opt.infer(typer)
+              end
+              typer.learn_method_type(defining_class, name, arg_types_for_opt, actual_type, signature[:throws])
+            end
+            args_for_opt << arg
+          end
+        end
+
+        signature[:return] = @inferred_type
+      end
+        
+      @inferred_type
+    end
+    
+    def abstract?
+      node = parent
+      while node && !node.kind_of?(Scope)
+        node = node.parent
+      end
+      InterfaceDeclaration === node
+    end
+    
+    def static?
+      false
+    end
+  end
+      
+  class StaticMethodDefinition < MethodDefinition
+    def defining_class
+      @defining_class.meta
+    end
+    
+    def static?
+      true
+    end
+  end
+end

data/lib/duby/ast/structure.rb

@@ -0,0 +1,44 @@
+module Duby::AST
+  class Body < Node
+    def initialize(parent, line_number, &block)
+      super(parent, line_number, &block)
+    end
+        
+    # Type of a block is the type of its final element
+    def infer(typer)
+      unless @inferred_type
+        if children.size == 0
+          @inferred_type = typer.default_type
+        else
+          children.each {|child| @inferred_type = typer.infer(child)}
+        end
+          
+        unless @inferred_type
+          typer.defer(self)
+        end
+      end
+
+      @inferred_type
+    end
+  end
+  
+  class Noop < Node
+    def infer(typer)
+      @inferred_type ||= typer.no_type
+    end
+  end
+  
+  class Script < Node
+    include Scope
+    attr_accessor :body
+    
+    def initialize(parent, line_number, &block)
+      super(parent, line_number, children, &block)
+      @body = children[0]
+    end
+    
+    def infer(typer)
+      @inferred_type ||= typer.infer(body) || (typer.defer(self); nil)
+    end
+  end
+end