steep 1.7.0.dev.2 → 1.7.0.dev.3

data/lib/steep/type_inference/block_params.rb

@@ -57,8 +57,9 @@ module Steep
 
         def variable_types
           each_param.with_object({}) do |param, hash|
+            var_name = param.var || next
             # @type var hash: Hash[Symbol, AST::Types::t?]
-            hash[param.var] = param.type
+            hash[var_name] = param.type
           end
         end
 
@@ -133,10 +134,17 @@ module Steep
           else
             var = arg.children[0]
             type = annotations.var_type(lvar: var)
-
             case arg.type
-            when :arg, :procarg0
+            when :arg
               default_params << Param.new(var: var, type: type, value: nil, node: arg)
+            when :procarg0
+              var = arg.children[0]
+              if var.is_a?(Symbol)
+                default_params << Param.new(var: var, type: type, value: nil, node: arg)
+              else
+                var = var.children[0]
+                default_params << Param.new(var: var, type: type, value: nil, node: arg)
+              end
             when :optarg
               default_params = trailing_params
               optional_params << Param.new(var: var, type: type, value: arg.children.last, node: arg)

data/lib/steep/type_inference/context.rb

@@ -125,7 +125,7 @@ module Steep
         end
 
         def upper_bounds
-          table.each_value.with_object({}) do |type_param, bounds|
+          @upper_bounds ||= table.each_value.with_object({}) do |type_param, bounds|
             if type_param.upper_bound
               bounds[type_param.name] = type_param.upper_bound
             end

data/lib/steep/type_inference/logic_type_interpreter.rb

@@ -593,7 +593,7 @@ module Steep
       end
 
       def try_convert(type, method)
-        if shape = subtyping.builder.shape(type, public_only: true, config: config)
+        if shape = subtyping.builder.shape(type, config)
           if entry = shape.methods[method]
             method_type = entry.method_types.find do |method_type|
               method_type.type.params.optional?

data/lib/steep/version.rb

@@ -1,3 +1,3 @@
 module Steep
-  VERSION = "1.7.0.dev.2"
+  VERSION = "1.7.0.dev.3"
 end

data/lib/steep.rb

@@ -186,15 +186,19 @@ module Steep
   @ui_logger = nil
   self.log_output = STDERR
 
-  def self.measure(message, level: :warn)
+  def self.measure(message, level: :warn, threshold: 0.0)
     start = Time.now
-    yield
-  ensure
-    time = Time.now - start
-    if level.is_a?(Symbol)
-      level = Logger.const_get(level.to_s.upcase)
+    begin
+      yield
+    ensure
+      time = Time.now - start
+      if level.is_a?(Symbol)
+        level = Logger.const_get(level.to_s.upcase)
+      end
+      if time > threshold
+        self.logger.log(level) { "#{message} took #{time} seconds" }
+      end
     end
-    self.logger.log(level) { "#{message} took #{time} seconds" }
   end
 
   def self.log_error(exn, message: "Unexpected error: #{exn.inspect}")

data/sig/steep/ast/types/helper.rbs

@@ -15,6 +15,8 @@ module Steep
         module NoChild
           def each_child: () { (t) -> void } -> void
                         | () -> Enumerator[t, void]
+
+          def map_type: () { (t) -> t } -> self
         end
       end
     end

data/sig/steep/ast/types/intersection.rbs

@@ -29,6 +29,8 @@ module Steep
         def each_child: () { (t) -> void } -> void
                       | () -> Enumerator[t, void]
 
+        def map_type: () { (t) -> t } -> t
+
         def level: () -> Array[Integer]
 
         def with_location: (untyped new_location) -> Intersection

data/sig/steep/ast/types/name.rbs

@@ -51,6 +51,7 @@ module Steep
           def map_type: () { (t) -> t } -> self
         end
 
+        # Singleton of a class/module
         class Singleton < Base
           def ==: (untyped other) -> bool
 
@@ -65,13 +66,16 @@ module Steep
           include Helper::NoChild
         end
 
+        # An instance of a class/module
         class Instance < Applying
           def to_module: () -> Singleton
         end
 
+        # An interface type
         class Interface < Applying
         end
 
+        # Type alias
         class Alias < Applying
         end
       end

data/sig/steep/ast/types/record.rbs

@@ -28,6 +28,8 @@ module Steep
         def each_child: () { (t) -> void } -> void
                       | () -> Enumerator[t, void]
 
+        def map_type: () { (t) -> t } -> Record
+
         def level: () -> Array[Integer]
 
         def with_location: (untyped new_location) -> Record

data/sig/steep/ast/types/tuple.rbs

@@ -26,6 +26,8 @@ module Steep
         def each_child: () { (t) -> void } -> void
                       | () -> Enumerator[t, void]
 
+        def map_type: () { (t) -> t } -> Tuple
+
         def level: () -> Array[Integer]
 
         def with_location: (untyped new_location) -> Tuple

data/sig/steep/ast/types/union.rbs

@@ -27,6 +27,8 @@ module Steep
         def each_child: () { (t) -> void } -> void
                       | () -> Enumerator[t, void]
 
+        def map_type: () { (t) -> t } -> t
+
         include Helper::ChildrenLevel
 
         def level: () -> Array[Integer]

data/sig/steep/expectations.rbs

@@ -19,7 +19,7 @@ module Steep
 
       def initialize: (start_position: position, end_position: position, severity: Steep::Diagnostic::LSPFormatter::severity, message: String, code: String) -> void
 
-      def self.from_hash: (untyped) -> instance
+      def self.from_hash: (untyped) -> Diagnostic
 
       def self.from_lsp: (untyped) -> Diagnostic
 

data/sig/steep/interface/block.rbs

@@ -25,9 +25,9 @@ module Steep
 
       def subst: (Substitution s) -> Block
 
-      @fvs: Set[Symbol]
+      @fvs: Set[AST::Types::variable]
 
-      def free_variables: () -> Set[Symbol]
+      def free_variables: () -> Set[AST::Types::variable]
 
       def to_s: () -> ::String
 

data/sig/steep/interface/builder.rbs

@@ -6,161 +6,147 @@ module Steep
       class Config
         # Type of `self` type included immediately in the type expression
         #
-        attr_reader self_type: AST::Types::t
+        attr_reader self_type: AST::Types::t?
 
         # Type of `class` type included immediately in the type expression
         #
-        attr_reader class_type: AST::Types::t | nil
+        attr_reader class_type: AST::Types::t?
 
         # Type of `instance` type included immediately in the type expression
         #
-        attr_reader instance_type: AST::Types::t | nil
+        attr_reader instance_type: AST::Types::t?
 
-        # Resolves `self` types included in shape members
-        #
-        # * `false` only when the type expression is `self`
-        #
-        attr_reader resolve_self: bool
+        # Upper bounds of type variables
+        attr_reader variable_bounds: Hash[Symbol, AST::Types::t?]
 
-        # Resolves `class` types included in shape members
-        #
-        # * `false` only when the type expression is `class`
-        #
-        attr_reader resolve_class: bool
+        def initialize: (self_type: AST::Types::t?, variable_bounds: Hash[Symbol, AST::Types::t?]) -> void
+                      | (self_type: AST::Types::t?, class_type: AST::Types::t, instance_type: AST::Types::t, variable_bounds: Hash[Symbol, AST::Types::t?]) -> void
 
-        # Resolves `instance` types included in shape members
-        #
-        # * `false` only when the type expression is `instance`
-        #
-        attr_reader resolve_instance: bool
+        def subst: () -> Substitution?
 
-        attr_reader variable_bounds: Hash[Symbol, AST::Types::t?]
+        def self.empty: () -> Config
 
-        def initialize: (
-          self_type: AST::Types::t,
-          class_type: AST::Types::t | nil,
-          instance_type: AST::Types::t | nil,
-          ?resolve_self: bool,
-          ?resolve_class: bool,
-          ?resolve_instance: bool,
-          variable_bounds: Hash[Symbol, AST::Types::t?]
-        ) -> void
+        def upper_bound: (Symbol) -> AST::Types::t?
 
-        def update: (
-          ?self_type: AST::Types::t,
-          ?class_type: AST::Types::t | nil,
-          ?instance_type: AST::Types::t | nil,
-          ?resolve_self: bool,
-          ?resolve_class: bool,
-          ?resolve_instance: bool,
-          ?variable_bounds: Hash[Symbol, AST::Types::t?]
-        ) -> self
+        private
 
-        @no_resolve: self?
-        def no_resolve: () -> self
+        def validate: () -> self
 
-        def resolve?: () -> bool
+        def validate_fvs: (Symbol name, AST::Types::t?) -> void
+      end
 
-        def ==: (untyped) -> bool
+      attr_reader factory: AST::Types::Factory
 
-        alias eql? ==
+      # # No type application (if generic), no self-instance-module resolution
+      attr_reader object_shape_cache: Hash[TypeName, Shape?]
 
-        def hash: () -> Integer
+      attr_reader union_shape_cache: Hash[AST::Types::Union, Shape?]
 
-        @subst: Substitution
+      attr_reader singleton_shape_cache: Hash[TypeName, Shape?]
 
-        # Substitution for immediate type expressions
-        def subst: () -> Substitution
+      def initialize: (AST::Types::Factory) -> void
 
-        # Returns `self_type`, or `nil` when it is `Types::Self`
-        #
-        def self_type?: () -> AST::Types::t?
+      def fetch_cache: [KEY] (Hash[KEY, Shape?], KEY) { () -> Shape? } -> Shape?
 
-        # Returns `class_type`, or `nil` when it is `Types::Class`
-        #
-        def class_type?: () -> AST::Types::t?
+      # Returns a shape of given type with respect to Config
+      #
+      # * If `self` occurs in the given type, it returns types with `self`
+      # * If `self` doesn't occur in the given type, it returns type withohut `self`, that is resolved to `config.self_type`
+      #
+      def shape: (AST::Types::t, Config) -> Shape?
 
-        # Returns `instanc_type`, or `nil` when it is `Types::Instance`
-        #
-        def instance_type?: () -> AST::Types::t?
-      end
+      # Returns a shape of given type with respect to Config
+      #
+      # The `self` types included in the returned Shape is the `self` type in the top level
+      #
+      # * `raw_shape(<self>)`        -> may return a shape with `self`
+      # * `raw_shape(<Array[self]>)  -> returns a shape that returns `self` from `#first` method
+      #
+      def raw_shape: (AST::Types::t, Config) -> Shape?
 
-      attr_reader factory: AST::Types::Factory
+      def self_shape: (AST::Types::t, Config) -> Shape?
 
-      type cache_key = [
-        AST::Types::t,                 # type
-        bool,                          # public_only
-        AST::Types::t | nil,           # self_type
-        AST::Types::t | nil,           # class_type
-        AST::Types::t | nil,           # instance_type
-        bool, bool, bool,              # resolve_self, resolve_class, resolve_instance
-        Hash[Symbol, AST::Types::t?]?  # variable_bounds
-      ]
-      attr_reader cache: Hash[cache_key, Shape?]
+      private
 
-      # No type application (if generic), no self-instance-module resolution
-      attr_reader raw_instance_object_shape_cache: Hash[[TypeName, bool], Shape]
+      def object_shape: (RBS::TypeName) -> Shape
 
-      attr_reader raw_singleton_object_shape_cache: Hash[[TypeName, bool], Shape]
+      def singleton_shape: (RBS::TypeName) -> Shape
 
-      attr_reader raw_interface_object_shape_cache: Hash[[TypeName, bool], Shape]
+      def union_shape: (AST::Types::t, Array[Shape]) -> Shape?
 
-      def initialize: (AST::Types::Factory) -> void
+      def intersection_shape: (AST::Types::t, Array[Shape]) -> Shape?
 
-      # Calculates the shape of given class, based on `public_only` and Config
-      #
-      # Returns `nil` if a type that cannot calculate Shape is given.
-      #
-      # * `public_only`: If false, returns a shape with private methods.
+      def proc_shape: (AST::Types::Proc, Shape) -> Shape
+
+      def tuple_shape: (AST::Types::Tuple) { (AST::Types::Name::Instance) -> Shape? } -> Shape?
+
+      def record_shape: (AST::Types::Record) { (AST::Types::Name::Instance) -> Shape? } -> Shape?
+
+      # Substitution for `self`/`instance`/`class` types, built from the name of given type
       #
-      def shape: (AST::Types::t, public_only: bool, config: Config) -> Shape?
+      def class_subst: (AST::Types::Name::Instance | AST::Types::Name::Singleton) -> Substitution
 
-      private
+      def interface_subst: (AST::Types::Name::Interface) -> Substitution
+
+      # Substitution for type application of class instance, type alias, or interface types
+      def app_subst: (AST::Types::Name::Instance | AST::Types::Name::Alias | AST::Types::Name::Interface) -> Substitution
+
+      def method_name_for: (RBS::Definition::Method::TypeDef, Symbol name) -> method_name
+
+      def replace_primitive_method: (method_name, RBS::Definition::Method::TypeDef, MethodType) -> MethodType
+
+      # # Calculates the shape of given class, based on `public_only` and Config
+      # #
+      # # Returns `nil` if a type that cannot calculate Shape is given.
+      # #
+      # # * `public_only`: If false, returns a shape with private methods.
+      # #
+      # def shape: (AST::Types::t, public_only: bool, config: Config) -> Shape?
+
+      # private
 
       @subtyping: Subtyping::Check?
 
       def subtyping: () -> Subtyping::Check
 
-      # Fetch and update cache
-      #
-      # Cache if given type is cacheable:
-      #
-      # * `self`, `instance`, `class` is not cacheable
-      # * Type variables are not cacheable
-      #
-      def fetch_cache: (AST::Types::t, bool public_only, Config) { () -> Shape? } -> Shape?
+      # # Fetch and update cache
+      # #
+      # # Cache if given type is cacheable:
+      # #
+      # # * `self`, `instance`, `class` is not cacheable
+      # # * Type variables are not cacheable
+      # #
+      # def fetch_cache: (AST::Types::t, bool public_only, Config) { () -> Shape? } -> Shape?
 
-      def include_self?: (AST::Types::t) -> bool
+      # def include_self?: (AST::Types::t) -> bool
 
-      def definition_builder: () -> RBS::DefinitionBuilder
+      # def definition_builder: () -> RBS::DefinitionBuilder
 
-      def object_shape: (
-        AST::Types::Name::Instance | AST::Types::Name::Singleton | AST::Types::Name::Interface,
-        bool public_only,
-        boolish keep_self,
-        boolish keep_instance,
-        boolish keep_singleton
-      ) -> Shape
+      # def object_shape: (
+      #   AST::Types::Name::Instance | AST::Types::Name::Singleton | AST::Types::Name::Interface,
+      #   bool public_only,
+      #   boolish keep_self,
+      #   boolish keep_instance,
+      #   boolish keep_singleton
+      # ) -> Shape
 
-      def raw_object_shape: (
-        AST::Types::Name::Instance | AST::Types::Name::Singleton | AST::Types::Name::Interface,
-        bool public_only,
-        Substitution subst
-      ) -> Shape
+      # def raw_object_shape: (
+      #   AST::Types::Name::Instance | AST::Types::Name::Singleton | AST::Types::Name::Interface,
+      #   bool public_only,
+      #   Substitution subst
+      # ) -> Shape
 
-      def union_shape: (AST::Types::t, Array[Shape], bool public_only) -> Shape
+      # def union_shape: (AST::Types::t, Array[Shape], bool public_only) -> Shape
 
-      def intersection_shape: (AST::Types::t, Array[Shape], bool public_only) -> Shape
+      # def intersection_shape: (AST::Types::t, Array[Shape], bool public_only) -> Shape
 
-      def tuple_shape: (AST::Types::Tuple, bool public_only, Config) -> Shape
+      # def tuple_shape: (AST::Types::Tuple, bool public_only, Config) -> Shape
 
-      def record_shape: (AST::Types::Record, bool public_only, Config) -> Shape?
+      # def record_shape: (AST::Types::Record, bool public_only, Config) -> Shape?
 
-      def proc_shape: (AST::Types::Proc, bool public_only, Config) -> Shape?
+      # def proc_shape: (AST::Types::Proc, bool public_only, Config) -> Shape?
 
-      def replace_primitive_method: (method_name, RBS::Definition::Method::TypeDef, MethodType) -> MethodType
 
-      def method_name_for: (RBS::Definition::Method::TypeDef, Symbol name) -> method_name
     end
   end
 end

data/sig/steep/interface/function.rbs

@@ -27,6 +27,7 @@ module Steep
             def var_type: () -> AST::Types::t
 
             def map_type: () { (AST::Types::t) -> AST::Types::t } -> self
+                        | () -> Enumerator[AST::Types::t, void]
           end
 
           class Required < Base
@@ -57,6 +58,7 @@ module Steep
           def map: () { (param) -> param } -> PositionalParams
 
           def map_type: () { (AST::Types::t) -> AST::Types::t } -> PositionalParams
+                      | () -> Enumerator[AST::Types::t, void]
 
           def subst: (Substitution s) -> PositionalParams
 
@@ -74,13 +76,13 @@ module Steep
 
           def size: () -> Integer
 
-          def self.build: (required: Array[AST::Types::t], optional: Array[AST::Types::t], rest: AST::Types::t?) -> PositionalParams
+          def self.build: (required: Array[AST::Types::t], optional: Array[AST::Types::t], rest: AST::Types::t?) -> PositionalParams?
 
           extend Utils
 
           # Calculates xs + ys.
           # Never fails.
-          def self.merge_for_overload: (PositionalParams? xs, PositionalParams? ys) -> PositionalParams
+          def self.merge_for_overload: (PositionalParams? xs, PositionalParams? ys) -> PositionalParams?
 
           # xs | ys
           def self.merge_for_union: (PositionalParams? xs, PositionalParams? ys) -> PositionalParams?
@@ -98,15 +100,15 @@ module Steep
 
           attr_reader rest: AST::Types::t?
 
-          def initialize: (?requireds: ::Hash[untyped, untyped], ?optionals: ::Hash[untyped, untyped], ?rest: untyped?) -> void
+          def initialize: (?requireds: ::Hash[Symbol, AST::Types::t], ?optionals: ::Hash[Symbol, AST::Types::t], ?rest: AST::Types::t?) -> void
 
-          def ==: (untyped other) -> untyped
+          def ==: (untyped other) -> bool
 
           alias eql? ==
 
-          def hash: () -> untyped
+          def hash: () -> Integer
 
-          def update: (?requireds: untyped, ?optionals: untyped, ?rest: untyped) -> untyped
+          def update: (?requireds: Hash[Symbol, AST::Types::t], ?optionals: Hash[Symbol, AST::Types::t], ?rest: AST::Types::t?) -> KeywordParams
 
           def empty?: () -> bool
 
@@ -117,6 +119,7 @@ module Steep
                        | () -> Enumerator[AST::Types::t, void]
 
           def map_type: () { (AST::Types::t) -> AST::Types::t } -> KeywordParams
+                      | () -> Enumerator[AST::Types::t, KeywordParams]
 
           def subst: (Substitution s) -> KeywordParams
 
@@ -127,13 +130,13 @@ module Steep
           include Utils
 
           # For overloading
-          def +: (untyped other) -> untyped
+          def +: (KeywordParams other) -> KeywordParams
 
           # For union
-          def |: (untyped other) -> untyped
+          def |: (KeywordParams other) -> KeywordParams
 
           # For intersection
-          def &: (untyped other) -> (nil | untyped)
+          def &: (KeywordParams other) -> KeywordParams?
         end
 
         def required: () -> Array[AST::Types::t]
@@ -146,54 +149,55 @@ module Steep
 
         attr_reader keyword_params: KeywordParams
 
-        def self.build: (?required: untyped, ?optional: untyped, ?rest: untyped?, ?required_keywords: ::Hash[untyped, untyped], ?optional_keywords: ::Hash[untyped, untyped], ?rest_keywords: untyped?) -> untyped
+        def self.build: (?required: Array[AST::Types::t], ?optional: Array[AST::Types::t], ?rest: AST::Types::t?, ?required_keywords: ::Hash[Symbol, AST::Types::t], ?optional_keywords: ::Hash[Symbol, AST::Types::t], ?rest_keywords: AST::Types::t?) -> Params
 
-        def initialize: (positional_params: untyped, keyword_params: untyped) -> void
+        def initialize: (positional_params: PositionalParams?, keyword_params: KeywordParams) -> void
 
-        def update: (?positional_params: untyped, ?keyword_params: untyped) -> untyped
+        def update: (?positional_params: PositionalParams?, ?keyword_params: KeywordParams) -> Params
 
-        def first_param: () -> untyped
+        def first_param: () -> PositionalParams::param?
 
-        def with_first_param: (untyped param) -> untyped
+        def with_first_param: (PositionalParams::param) -> Params
 
-        def has_positional?: () -> (true | false)
+        def has_positional?: () -> bool
 
-        def self.empty: () -> untyped
+        def self.empty: () -> Params
 
-        def ==: (untyped other) -> untyped
+        def ==: (untyped other) -> bool
 
         alias eql? ==
 
-        def hash: () -> untyped
+        def hash: () -> Integer
 
         def flat_unnamed_params: () -> Array[[:required | :optional, AST::Types::t]]
 
-        def flat_keywords: () -> untyped
+        def flat_keywords: () -> Hash[Symbol, AST::Types::t]
 
-        def required_keywords: () -> untyped
+        def required_keywords: () -> Hash[Symbol, AST::Types::t]
 
-        def optional_keywords: () -> untyped
+        def optional_keywords: () -> Hash[Symbol, AST::Types::t]
 
-        def rest_keywords: () -> untyped
+        def rest_keywords: () -> AST::Types::t?
 
-        def has_keywords?: () -> untyped
+        def has_keywords?: () -> bool
 
         def each_positional_param: () { (PositionalParams::Base) -> void } -> void
 
-        def without_keywords: () -> untyped
+        def without_keywords: () -> Params
 
-        def drop_first: () -> untyped
+        def drop_first: () -> Params
 
         def each_type: () { (AST::Types::t) -> void } -> void
                      | () -> Enumerator[AST::Types::t, void]
 
-        def free_variables: () -> untyped
+        @fvs: Set[AST::Types::variable]?
+        def free_variables: () -> Set[AST::Types::variable]
 
-        def closed?: () -> untyped
+        def closed?: () -> bool
 
         def subst: (Substitution s) -> Params
 
-        def size: () -> untyped
+        def size: () -> Integer
 
         def to_s: () -> ::String
 
@@ -244,7 +248,8 @@ module Steep
 
       def hash: () -> Integer
 
-      def free_variables: () -> Set[Symbol]
+      @fvs: Set[AST::Types::variable]?
+      def free_variables: () -> Set[AST::Types::variable]
 
       def subst: (Substitution s) -> Function
 

data/sig/steep/interface/shape.rbs

@@ -2,11 +2,26 @@ module Steep
   module Interface
     class Shape
       class Entry
-        attr_reader method_types: Array[MethodType]
+        @private_method: bool
 
-        def initialize: (method_types: Array[MethodType]) -> void
+        @method_types: Array[MethodType]?
+
+        @generator: (^() -> Array[MethodType]?)?
+
+        def initialize: (method_types: Array[MethodType], private_method: bool) -> void
+                      | (private_method: bool) { () -> Array[MethodType]? } -> void
+
+        def has_method_type?: () -> bool
 
         def to_s: () -> String
+
+        def private_method?: () -> bool
+
+        def public_method?: () -> bool
+
+        def method_types: () -> Array[MethodType]
+
+        def force: () -> void
       end
 
       class Methods
@@ -36,9 +51,9 @@ module Steep
 
         def push_substitution: (Substitution) -> Methods
 
-        def merge!: (Methods other) -> void
+        def merge!: (Methods other) ?{ (Symbol name, Entry old_entry, Entry new_entry) -> Entry } -> void
 
-        # def +: (Methods other) -> Methods
+        def public_methods: () -> Methods
       end
 
       attr_reader type: AST::Types::t
@@ -56,6 +71,10 @@ module Steep
       def public?: () -> bool
 
       def subst: (Substitution, ?type: AST::Types::t?) -> Shape
+
+      def public_shape: () -> Shape
+
+      @public_shape: Shape?
     end
   end
 end

data/sig/steep/interface/substitution.rbs

@@ -41,6 +41,8 @@ module Steep
 
       def merge: (Substitution s) -> Substitution
 
+      def update: (?self_type: AST::Types::t?, ?instance_type: AST::Types::t?, ?module_type: AST::Types::t?) -> Substitution
+
       def apply?: (AST::Types::t) -> bool
 
       def add!: (Symbol v, AST::Types::t ty) -> self