steep 0.1.0.pre2 → 0.1.0

data/lib/steep/ast/location.rb

@@ -0,0 +1,82 @@
+module Steep
+  module AST
+    class Location
+      attr_reader :buffer
+      attr_reader :start_pos
+      attr_reader :end_pos
+
+      def initialize(buffer:, start_pos:, end_pos:)
+        @buffer = buffer
+        @start_pos = start_pos
+        @end_pos = end_pos
+      end
+
+      def inspect
+        "#<#{self.class}:#{self.__id__} @buffer=..., @start_pos=#{start_pos}, @end_pos=#{end_pos}, source='#{source.lines.first}', start_line=#{start_line}, start_column=#{start_column}>"
+      end
+
+      def name
+        buffer.name
+      end
+
+      def start_line
+        start_loc[0]
+      end
+
+      def start_column
+        start_loc[1]
+      end
+
+      def end_line
+        end_loc[0]
+      end
+
+      def end_column
+        end_loc[1]
+      end
+
+      def start_loc
+        @start_loc ||= buffer.pos_to_loc(start_pos)
+      end
+
+      def end_loc
+        @end_loc ||= buffer.pos_to_loc(end_pos)
+      end
+
+      def source
+        @source ||= buffer.source(start_pos...end_pos)
+      end
+
+      def to_s
+        "#{start_line}:#{start_column}...#{end_line}:#{end_column}"
+      end
+
+      def ==(other)
+        other.is_a?(Location) &&
+          other.buffer == buffer &&
+          other.start_pos == start_pos &&
+          other.end_pos == end_pos
+      end
+
+      def +(other)
+        raise "Invalid concat: buffer=#{buffer.name}, other.buffer=#{other.buffer.name}" unless other.buffer == buffer
+
+        self.class.new(buffer: buffer,
+                       start_pos: start_pos,
+                       end_pos: other.end_pos)
+      end
+
+      def self.concat(*locations)
+        locations.inject {|l1, l2|
+          l1 + l2
+        }
+      end
+
+      def pred?(loc)
+        loc.is_a?(Location) &&
+          loc.name == name &&
+          loc.start_pos == end_pos
+      end
+    end
+  end
+end

data/lib/steep/ast/method_type.rb

@@ -0,0 +1,116 @@
+module Steep
+  module AST
+    class MethodType
+      module Params
+        class Base
+          attr_reader :location
+
+          def initialize(location:)
+            @location = location
+          end
+
+          def update_location(location)
+            dup.instance_eval do
+              @location = location
+              self
+            end
+          end
+        end
+
+        class Required < Base
+          attr_reader :type
+          attr_reader :next_params
+
+          def initialize(location:, type:, next_params: nil)
+            super(location: location)
+            @type = type
+            @next_params = next_params
+          end
+        end
+
+        class Optional < Base
+          attr_reader :type
+          attr_reader :next_params
+
+          def initialize(location:, type:, next_params: nil)
+            super(location: location)
+            @type = type
+            @next_params = next_params
+          end
+        end
+
+        class Rest < Base
+          attr_reader :type
+          attr_reader :next_params
+
+          def initialize(location:, type:, next_params: nil)
+            super(location: location)
+            @type = type
+            @next_params = next_params
+          end
+        end
+
+        class RequiredKeyword < Base
+          attr_reader :name
+          attr_reader :type
+          attr_reader :next_params
+
+          def initialize(location:, name:, type:, next_params: nil)
+            super(location: location)
+            @name = name
+            @type = type
+            @next_params = next_params
+          end
+        end
+
+        class OptionalKeyword < Base
+          attr_reader :name
+          attr_reader :type
+          attr_reader :next_params
+
+          def initialize(location:, name:, type:, next_params: nil)
+            super(location: location)
+            @name = name
+            @type = type
+            @next_params = next_params
+          end
+        end
+
+        class RestKeyword < Base
+          attr_reader :type
+
+          def initialize(location:, type:)
+            super(location: location)
+            @type = type
+          end
+        end
+      end
+
+      class Block
+        attr_reader :location
+        attr_reader :params
+        attr_reader :return_type
+
+        def initialize(location:, params:, return_type:)
+          @location = location
+          @params = params
+          @return_type = return_type
+        end
+      end
+
+      attr_reader :location
+      attr_reader :type_params
+      attr_reader :params
+      attr_reader :block
+      attr_reader :return_type
+
+      def initialize(location:, type_params:, params:, block:, return_type:)
+        @location = location
+        @type_params = type_params
+        @params = params
+        @block = block
+        @return_type = return_type
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/class.rb

@@ -0,0 +1,33 @@
+module Steep
+  module AST
+    module Signature
+      class SuperClass
+        attr_reader :location
+        attr_reader :name
+        attr_reader :args
+
+        def initialize(name:, args:, location:)
+          @name = name
+          @args = args
+          @location = location
+        end
+      end
+
+      class Class
+        attr_reader :location
+        attr_reader :name
+        attr_reader :params
+        attr_reader :super_class
+        attr_reader :members
+
+        def initialize(name:, params:, super_class:, location:, members:)
+          @name = name
+          @params = params
+          @super_class = super_class
+          @location = location
+          @members = members
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/const.rb

@@ -0,0 +1,17 @@
+module Steep
+  module AST
+    module Signature
+      class Const
+        attr_reader :location
+        attr_reader :name
+        attr_reader :type
+
+        def initialize(location:, name:, type:)
+          @location = location
+          @name = name
+          @type = type
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/env.rb

@@ -0,0 +1,123 @@
+module Steep
+  module AST
+    module Signature
+      class Env
+        attr_reader :modules
+        attr_reader :classes
+        attr_reader :extensions
+        attr_reader :interfaces
+        attr_reader :constants
+        attr_reader :globals
+
+        def initialize()
+          @modules = {}
+          @classes = {}
+          @extensions = {}
+          @interfaces = {}
+          @constants = {}
+          @globals = {}
+        end
+
+        def add(sig)
+          case sig
+          when Signature::Class
+            raise "Duplicated class: #{sig.name}" if classes.key?(sig.name.absolute!) || modules.key?(sig.name.absolute!)
+            classes[sig.name.absolute!] = sig
+          when Signature::Module
+            raise "Duplicated module: #{sig.name}" if classes.key?(sig.name.absolute!) || modules.key?(sig.name.absolute!)
+            modules[sig.name.absolute!] = sig
+          when Signature::Interface
+            raise "Duplicated interface: #{sig.name}" if interfaces.key?(sig.name)
+            interfaces[sig.name] = sig
+          when Signature::Extension
+            extensions[sig.module_name.absolute!] ||= []
+            if extensions[sig.module_name.absolute!].any? {|ext| ext.name == sig.name }
+              raise "Duplicated extension: #{sig.module_name.absolute!} (#{sig.name})"
+            end
+            extensions[sig.module_name.absolute!] << sig
+          when Signature::Const
+            constants[sig.name.absolute!] = sig
+          when Signature::Gvar
+            raise "Duplicated global: #{sig.name}" if globals.key?(sig.name)
+            globals[sig.name] = sig
+          else
+            raise "Unknown signature:: #{sig}"
+          end
+        end
+
+        def find_module(name, current_module: nil)
+          find_name(modules, name, current_module: current_module) or raise "Unknown module: #{name}"
+        end
+
+        def find_class(name, current_module: nil)
+          find_name(classes, name, current_module: current_module) or raise "Unknown class: #{name}"
+        end
+
+        def find_class_or_module(name, current_module: nil)
+          sig =
+            find_name(modules, name, current_module: current_module) ||
+              find_name(classes, name, current_module: current_module)
+
+          sig or raise "Unknown class/module: #{name}}"
+        end
+
+        def find_extensions(name, current_module: nil)
+          find_name(extensions, name, current_module: current_module) || []
+        end
+
+        def find_const(name, current_module: nil)
+          find_name(constants, name, current_module: current_module)
+        end
+
+        def find_gvar(name)
+          globals[name]
+        end
+
+        def find_name(hash, name, current_module:)
+          if current_module
+            hash[current_module + name] || find_name(hash, name, current_module: current_module.parent)
+          else
+            hash[name.absolute!]
+          end
+        end
+
+        def find_interface(name)
+          interfaces[name] or raise "Unknown interface: #{name}"
+        end
+
+        def module?(type_name, current_module: nil)
+          name = type_name.map_module_name {|m| current_module ? current_module + m : m.absolute! }.name
+          modules.key?(name)
+        end
+
+        def class?(type_name, current_module: nil)
+          name = type_name.map_module_name {|m| current_module ? current_name + m : m.absolute! }.name
+          classes.key?(name)
+        end
+
+        def class_name?(name, current_module: nil)
+          classes.key?(current_module ? current_module + name : name.absolute!)
+        end
+
+        def module_name?(name, current_module: nil)
+          modules.key?(current_module ? current_module + name : name.absolute!)
+        end
+
+        def const_name?(name, current_module: nil)
+          constants.key?(current_module ? current_module + name : name.absolute!)
+        end
+
+        def each(&block)
+          if block_given?
+            classes.each_value(&block)
+            modules.each_value(&block)
+            interfaces.each_value(&block)
+            constants.each_value(&block)
+          else
+            enum_for :each
+          end
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/extension.rb

@@ -0,0 +1,21 @@
+module Steep
+  module AST
+    module Signature
+      class Extension
+        attr_reader :location
+        attr_reader :module_name
+        attr_reader :name
+        attr_reader :members
+        attr_reader :params
+
+        def initialize(location:, module_name:, params:, name:, members:)
+          @location = location
+          @module_name = module_name
+          @params = params
+          @name = name
+          @members = members
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/gvar.rb

@@ -0,0 +1,17 @@
+module Steep
+  module AST
+    module Signature
+      class Gvar
+        attr_reader :location
+        attr_reader :name
+        attr_reader :type
+
+        def initialize(location:, name:, type:)
+          @location = location
+          @name = name
+          @type = type
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/interface.rb

@@ -0,0 +1,31 @@
+module Steep
+  module AST
+    module Signature
+      class Interface
+        class Method
+          attr_reader :location
+          attr_reader :name
+          attr_reader :types
+
+          def initialize(location:, name:, types:)
+            @location = location
+            @name = name
+            @types = types
+          end
+        end
+
+        attr_reader :location
+        attr_reader :name
+        attr_reader :params
+        attr_reader :methods
+
+        def initialize(location:, name:, params:, methods:)
+          @location = location
+          @name = name
+          @params = params
+          @methods = methods
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/members.rb

@@ -0,0 +1,71 @@
+module Steep
+  module AST
+    module Signature
+      module Members
+        class Include
+          attr_reader :location
+          attr_reader :name
+          attr_reader :args
+
+          def initialize(location:, name:, args:)
+            @location = location
+            @name = name
+            @args = args
+          end
+        end
+
+        class Extend
+          attr_reader :location
+          attr_reader :name
+          attr_reader :args
+
+          def initialize(location:, name:, args:)
+            @location = location
+            @name = name
+            @args = args
+          end
+        end
+
+        class Method
+          attr_reader :location
+          attr_reader :name
+          attr_reader :kind
+          attr_reader :types
+          attr_reader :attributes
+
+          def initialize(location:, name:, kind:, types:, attributes:)
+            @location = location
+            @name = name
+            @kind = kind
+            @types = types
+            @attributes = attributes
+          end
+
+          def module_method?
+            kind == :module || kind == :module_instance
+          end
+
+          def instance_method?
+            kind == :instance || kind == :module_instance
+          end
+
+          def constructor?
+            attributes.include?(:constructor)
+          end
+        end
+
+        class Ivar
+          attr_reader :location
+          attr_reader :name
+          attr_reader :type
+
+          def initialize(location:, name:, type:)
+            @location = location
+            @name = name
+            @type = type
+          end
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/signature/module.rb

@@ -0,0 +1,21 @@
+module Steep
+  module AST
+    module Signature
+      class Module
+        attr_reader :location
+        attr_reader :name
+        attr_reader :params
+        attr_reader :self_type
+        attr_reader :members
+
+        def initialize(name:, location:, params:, self_type:, members:)
+          @name = name
+          @location = location
+          @params = params
+          @self_type = self_type
+          @members = members
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/type_params.rb

@@ -0,0 +1,13 @@
+module Steep
+  module AST
+    class TypeParams
+      attr_reader :location
+      attr_reader :variables
+
+      def initialize(location: nil, variables:)
+        @location = location
+        @variables = variables
+      end
+    end
+  end
+end

data/lib/steep/ast/types/any.rb

@@ -0,0 +1,39 @@
+module Steep
+  module AST
+    module Types
+      class Any
+        attr_reader :location
+
+        def initialize(location: nil)
+          @location = location
+        end
+
+        def ==(other)
+          other.is_a?(Any)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          self
+        end
+
+        def to_s
+          "any"
+        end
+
+        def free_variables
+          Set.new
+        end
+
+        def level
+          [1]
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/bot.rb

@@ -0,0 +1,39 @@
+module Steep
+  module AST
+    module Types
+      class Bot
+        attr_reader :location
+
+        def initialize(location: nil)
+          @location = location
+        end
+
+        def ==(other)
+          other.is_a?(Bot)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          self
+        end
+
+        def to_s
+          "⟘"
+        end
+
+        def free_variables
+          Set.new
+        end
+
+        def level
+          [2]
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/class.rb

@@ -0,0 +1,35 @@
+module Steep
+  module AST
+    module Types
+      class Class
+        attr_reader :location
+
+        def initialize(location: nil)
+          @location = location
+        end
+
+        def ==(other)
+          other.is_a?(Class)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          s.module_type or raise "Unexpected substitution: #{inspect}"
+        end
+
+        def free_variables
+          Set.new
+        end
+
+        def level
+          [0]
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/helper.rb

@@ -0,0 +1,21 @@
+module Steep
+  module AST
+    module Types
+      module Helper
+        module ChildrenLevel
+          def level_of_children(children)
+            children.map(&:level).sort {|a, b| b.size <=> a.size }.inject() do |a, b|
+              a.zip(b).map do |(x, y)|
+                if x && y
+                  x + y
+                else
+                  x || y
+                end
+              end
+            end || []
+          end
+        end
+      end
+    end
+  end
+end