steep 0.30.0 → 0.34.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b5293cbc87834cee36e50cb66bab84e466a5182c1e868732b7bd3e685b2bd954
-  data.tar.gz: 7794e42573b5ac04c74780824213cfbe1ca255f1feaa4465a1066377d2647e0b
+  metadata.gz: 0ec77465913e4b2146e068c58f63b7ae8b5c51fd2391f6eb2776dba8372a81c2
+  data.tar.gz: 7528dbaf4873d9eb77728521837ecb0fe43a2010036572be2c3dc017c5b903cc
 SHA512:
-  metadata.gz: e723bf77a7452e0696c38d6d1797e71f246fb97d1b1f18082ba21b8eadb8174d6bcc4deda872898b5f39d162b91173c413cacfa28175018517170f8f2568bd70
-  data.tar.gz: 38dbe78df43706fe33d5350ad04524c25606955bcba5397ad18690717a60c59d361485ad2d5d3077787336d23902bd91ba9fa5b31e1933a93d1fe32747837688
+  metadata.gz: b0426373f8fb9e48ccec48d0676d70d5a674a389a83624df78dc3deacaba23840b02cc94becba92304de35f98c8d8bb3af12ef185161e4e6a6c37600cb61a82d
+  data.tar.gz: 5f3a32240541f116b6299c354fada0c8de398b6139b9d5e275e7b05490ccc401aa1b4f70978af5561e2ce9c6c7bb2587c1c8e43801276c0875000496ed332a5b

data/CHANGELOG.md

@@ -2,6 +2,32 @@
 
 ## master
 
+## 0.34.0 (2020-10-27)
+
+* Add `steep stats` command to show method call typing stats ([#246](https://github.com/soutaro/steep/pull/246))
+* Fix attribute assignment typing ([#243](https://github.com/soutaro/steep/pull/243))
+* Let `Range[T]` be covariant ([#242](https://github.com/soutaro/steep/pull/242))
+* Fix constant typing ([#247](https://github.com/soutaro/steep/pull/247), [#248](https://github.com/soutaro/steep/pull/248))
+
+## 0.33.0 (2020-10-13)
+
+* Make `!` typing flow sensitive ([#240](https://github.com/soutaro/steep/pull/240))
+
+## 0.32.0 (2020-10-09)
+
+* Let type-case support interface types ([#237](https://github.com/soutaro/steep/pull/237))
+
+## 0.31.1 (2020-10-07)
+
+* Fix `if-then-else` parsing ([#236](https://github.com/soutaro/steep/pull/236))
+
+## 0.31.0 (2020-10-04)
+
+* Fix type checking performance ([#230](https://github.com/soutaro/steep/pull/230))
+* Improve LSP completion/hover performance ([#232](https://github.com/soutaro/steep/pull/232))
+* Fix instance variable completion ([#234](https://github.com/soutaro/steep/pull/234))
+* Relax version requirements on Listen to allow installing on Ruby 3 ([#235](https://github.com/soutaro/steep/pull/235))
+
 ## 0.30.0 (2020-10-03)
 
 * Let top-level defs be methods of Object ([#227](https://github.com/soutaro/steep/pull/227))

data/bin/steep-prof

@@ -9,8 +9,7 @@ def exit(*)
 
 end
 
-
 STDERR.puts "Running profiler: mode=#{mode}, out=#{out}"
-StackProf.run(mode: mode, out: out) do
+StackProf.run(mode: mode, out: out, raw: true) do
   load File.join(__dir__, "../exe/steep")
 end

data/lib/steep.rb

@@ -16,7 +16,7 @@ require 'uri'
 
 require "rbs"
 
-require "steep/ast/location"
+require "steep/method_name"
 require "steep/ast/types/helper"
 require "steep/ast/types/any"
 require "steep/ast/types/instance"
@@ -39,7 +39,6 @@ require "steep/ast/types/logic"
 require "steep/ast/type_params"
 require "steep/ast/annotation"
 require "steep/ast/annotation/collection"
-require "steep/ast/buffer"
 require "steep/ast/builtin"
 require "steep/ast/types/factory"
 
@@ -72,6 +71,7 @@ require "steep/type_inference/type_env"
 require "steep/type_inference/local_variable_type_env"
 require "steep/type_inference/logic"
 require "steep/type_inference/logic_type_interpreter"
+require "steep/type_inference/method_call"
 require "steep/ast/types"
 
 require "steep/server/utils"
@@ -93,6 +93,7 @@ require "steep/project/hover_content"
 require "steep/project/completion_provider"
 require "steep/drivers/utils/driver_helper"
 require "steep/drivers/check"
+require "steep/drivers/stats"
 require "steep/drivers/validate"
 require "steep/drivers/annotations"
 require "steep/drivers/watch"

data/lib/steep/annotation_parser.rb

@@ -68,7 +68,7 @@ module Steep
           name = match[:name]
           type = match[:type]
 
-          method_type = factory.method_type(RBS::Parser.parse_method_type(type), self_type: AST::Types::Self.new)
+          method_type = factory.method_type(RBS::Parser.parse_method_type(type), self_type: AST::Types::Self.new, method_decls: Set[])
 
           AST::Annotation::MethodType.new(name: name.to_sym,
                                           type: method_type,

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

@@ -7,11 +7,14 @@ module Steep
         attr_reader :type_name_cache
         attr_reader :type_cache
 
+        attr_reader :type_interface_cache
+
         def initialize(builder:)
           @definition_builder = builder
 
           @type_name_cache = {}
           @type_cache = {}
+          @type_interface_cache = {}
         end
 
         def type_name_resolver
@@ -146,6 +149,8 @@ module Steep
               type: function_1(type.params, type.return_type),
               location: nil
             )
+          when Logic::Base
+            RBS::Types::Bases::Bool.new(location: nil)
           else
             raise "Unexpected type given: #{type} (#{type.class})"
           end
@@ -175,7 +180,7 @@ module Steep
           )
         end
 
-        def method_type(method_type, self_type:, subst2: nil, method_def: nil)
+        def method_type(method_type, self_type:, subst2: nil, method_decls:)
           fvs = self_type.free_variables()
 
           type_params = []
@@ -206,8 +211,7 @@ module Steep
                                return_type: type(block.type.return_type).subst(subst), location: nil)
               )
             end,
-            method_def: method_def,
-            location: method_def&.member&.location
+            method_decls: method_decls
           )
 
           if block_given?
@@ -339,48 +343,46 @@ module Steep
           end
         end
 
-        def setup_primitives(method_name, method_type)
-          if method_def = method_type.method_def
-            defined_in = method_def.defined_in
-            member = method_def.member
-
-            if member.is_a?(RBS::AST::Members::MethodDefinition)
-              case
-              when defined_in == RBS::BuiltinNames::Object.name && member.instance?
-                case method_name
-                when :is_a?, :kind_of?, :instance_of?
-                  return method_type.with(
-                    return_type: AST::Types::Logic::ReceiverIsArg.new(location: method_type.return_type.location)
-                  )
-                when :nil?
-                  return method_type.with(
-                    return_type: AST::Types::Logic::ReceiverIsNil.new(location: method_type.return_type.location)
-                  )
-                end
+        NilClassName = TypeName("::NilClass")
 
-              when defined_in == AST::Builtin::NilClass.module_name && member.instance?
-                case method_name
-                when :nil?
-                  return method_type.with(
-                    return_type: AST::Types::Logic::ReceiverIsNil.new(location: method_type.return_type.location)
-                  )
-                end
+        def setup_primitives(method_name, method_def, method_type)
+          defined_in = method_def.defined_in
+          member = method_def.member
 
-              when defined_in == RBS::BuiltinNames::BasicObject.name && member.instance?
-                case method_name
-                when :!
-                  return method_type.with(
-                    return_type: AST::Types::Logic::Not.new(location: method_type.return_type.location)
-                  )
-                end
+          if member.is_a?(RBS::AST::Members::MethodDefinition)
+            case method_name
+            when :is_a?, :kind_of?, :instance_of?
+              if defined_in == RBS::BuiltinNames::Object.name && member.instance?
+                return method_type.with(
+                  return_type: AST::Types::Logic::ReceiverIsArg.new(location: method_type.return_type.location)
+                )
+              end
 
-              when defined_in == RBS::BuiltinNames::Module.name && member.instance?
-                case method_name
-                when :===
-                  return method_type.with(
-                    return_type: AST::Types::Logic::ArgIsReceiver.new(location: method_type.return_type.location)
-                  )
-                end
+            when :nil?
+              case defined_in
+              when RBS::BuiltinNames::Object.name,
+                NilClassName
+                return method_type.with(
+                  return_type: AST::Types::Logic::ReceiverIsNil.new(location: method_type.return_type.location)
+                )
+              end
+
+            when :!
+              case defined_in
+              when RBS::BuiltinNames::BasicObject.name,
+                RBS::BuiltinNames::TrueClass.name,
+                RBS::BuiltinNames::FalseClass.name
+                return method_type.with(
+                  return_type: AST::Types::Logic::Not.new(location: method_type.return_type.location)
+                )
+              end
+
+            when :===
+              case defined_in
+              when RBS::BuiltinNames::Module.name
+                return method_type.with(
+                  return_type: AST::Types::Logic::ArgIsReceiver.new(location: method_type.return_type.location)
+                )
               end
             end
           end
@@ -390,15 +392,23 @@ module Steep
 
         def interface(type, private:, self_type: type)
           Steep.logger.debug { "Factory#interface: #{type}, private=#{private}, self_type=#{self_type}" }
-          type = expand_alias(type)
+
+          cache_key = [type, self_type, private]
+          if type_interface_cache.key?(cache_key)
+            return type_interface_cache[cache_key]
+          end
 
           case type
+          when Name::Alias
+            interface(expand_alias(type), private: private, self_type: self_type)
+
           when Self
             if self_type != type
               interface self_type, private: private, self_type: Self.new
             else
               raise "Unexpected `self` type interface"
             end
+
           when Name::Instance
             Interface::Interface.new(type: self_type, private: private).tap do |interface|
               definition = definition_builder.build_instance(type.name)
@@ -422,10 +432,13 @@ module Steep
 
                   interface.methods[name] = Interface::Interface::Entry.new(
                     method_types: method.defs.map do |type_def|
+                      method_name = InstanceMethodName.new(type_name: type_def.implemented_in || type_def.defined_in, method_name: name)
+                      decl = TypeInference::MethodCall::MethodDecl.new(method_name: method_name, method_def: type_def)
                       setup_primitives(
                         name,
+                        type_def,
                         method_type(type_def.type,
-                                    method_def: type_def,
+                                    method_decls: Set[decl],
                                     self_type: self_type,
                                     subst2: subst)
                       )
@@ -449,7 +462,11 @@ module Steep
               definition.methods.each do |name, method|
                 interface.methods[name] = Interface::Interface::Entry.new(
                   method_types: method.defs.map do |type_def|
-                    method_type(type_def.type, method_def: type_def, self_type: self_type, subst2: subst)
+                    decls = Set[TypeInference::MethodCall::MethodDecl.new(
+                      method_name: InstanceMethodName.new(type_name: type_def.implemented_in || type_def.defined_in, method_name: name),
+                      method_def: type_def
+                    )]
+                    method_type(type_def.type, method_decls: decls, self_type: self_type, subst2: subst)
                   end
                 )
               end
@@ -474,10 +491,16 @@ module Steep
 
                 interface.methods[name] = Interface::Interface::Entry.new(
                   method_types: method.defs.map do |type_def|
+                    decl = TypeInference::MethodCall::MethodDecl.new(
+                      method_name: SingletonMethodName.new(type_name: type_def.implemented_in || type_def.defined_in,
+                                                           method_name: name),
+                      method_def: type_def
+                    )
                     setup_primitives(
                       name,
+                      type_def,
                       method_type(type_def.type,
-                                  method_def: type_def,
+                                  method_decls: Set[decl],
                                   self_type: self_type,
                                   subst2: subst)
                     )
@@ -557,8 +580,7 @@ module Steep
                                                       rest_keywords: nil),
                         block: nil,
                         return_type: elem_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     } + aref.method_types
                   )
@@ -577,8 +599,7 @@ module Steep
                                                       rest_keywords: nil),
                         block: nil,
                         return_type: elem_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     } + update.method_types
                   )
@@ -592,8 +613,7 @@ module Steep
                         params: Interface::Params.empty,
                         block: nil,
                         return_type: type.types[0] || AST::Builtin.nil_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     ]
                   )
@@ -607,8 +627,7 @@ module Steep
                         params: Interface::Params.empty,
                         block: nil,
                         return_type: type.types.last || AST::Builtin.nil_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     ]
                   )
@@ -629,6 +648,7 @@ module Steep
                   Interface::Interface::Entry.new(
                     method_types: type.elements.map {|key_value, value_type|
                       key_type = Literal.new(value: key_value, location: nil)
+
                       Interface::MethodType.new(
                         type_params: [],
                         params: Interface::Params.new(required: [key_type],
@@ -639,8 +659,7 @@ module Steep
                                                       rest_keywords: nil),
                         block: nil,
                         return_type: value_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     } + ref.method_types
                   )
@@ -660,8 +679,7 @@ module Steep
                                                       rest_keywords: nil),
                         block: nil,
                         return_type: value_type,
-                        method_def: nil,
-                        location: nil
+                        method_decls: Set[]
                       )
                     } + update.method_types
                   )
@@ -676,8 +694,7 @@ module Steep
                 params: type.params,
                 return_type: type.return_type,
                 block: nil,
-                method_def: nil,
-                location: nil
+                method_decls: Set[]
               )
 
               interface.methods[:[]] = Interface::Interface::Entry.new(method_types: [method_type])
@@ -689,6 +706,8 @@ module Steep
 
           else
             raise "Unexpected type for interface: #{type}"
+          end.tap do |interface|
+            type_interface_cache[cache_key] = interface
           end
         end
 

data/lib/steep/cli.rb

@@ -16,7 +16,7 @@ module Steep
     end
 
     def self.available_commands
-      [:init, :check, :validate, :annotations, :version, :project, :watch, :langserver, :vendor]
+      [:init, :check, :validate, :annotations, :version, :project, :watch, :langserver, :vendor, :stats]
     end
 
     def process_global_options
@@ -91,6 +91,19 @@ module Steep
       end.run
     end
 
+    def process_stats
+      Drivers::Stats.new(stdout: stdout, stderr: stderr).tap do |check|
+        OptionParser.new do |opts|
+          opts.banner = "Usage: steep stats [options] [sources]"
+
+          opts.on("--steepfile=PATH") {|path| check.steepfile = Pathname(path) }
+          handle_logging_options opts
+        end.parse!(argv)
+
+        check.command_line_patterns.push *argv
+      end.run
+    end
+
     def process_validate
       Drivers::Validate.new(stdout: stdout, stderr: stderr).tap do |command|
         OptionParser.new do |opts|

data/lib/steep/drivers/stats.rb

@@ -0,0 +1,85 @@
+require "csv"
+
+module Steep
+  module Drivers
+    class Stats
+      attr_reader :stdout
+      attr_reader :stderr
+      attr_reader :command_line_patterns
+
+      include Utils::DriverHelper
+
+      def initialize(stdout:, stderr:)
+        @stdout = stdout
+        @stderr = stderr
+        @command_line_patterns = []
+      end
+
+      def run
+        project = load_config()
+
+        loader = Project::FileLoader.new(project: project)
+        loader.load_sources(command_line_patterns)
+        loader.load_signatures()
+
+        type_check(project)
+
+        stdout.puts(
+          CSV.generate do |csv|
+            csv << ["Target", "File", "Status", "Typed calls", "Untyped calls", "All calls", "Typed %"]
+
+            project.targets.each do |target|
+              case (status = target.status)
+              when Project::Target::TypeCheckStatus
+                status.type_check_sources.each do |source_file|
+                  case source_file.status
+                  when Project::SourceFile::TypeCheckStatus
+                    typing = source_file.status.typing
+
+                    typed = 0
+                    untyped = 0
+                    total = 0
+                    typing.method_calls.each_value do |call|
+                      case call
+                      when TypeInference::MethodCall::Typed
+                        typed += 1
+                      when TypeInference::MethodCall::Untyped
+                        untyped += 1
+                      end
+
+                      total += 1
+                    end
+
+                    csv << format_stats(target, source_file.path, "success", typed, untyped, total)
+                  when Project::SourceFile::TypeCheckErrorStatus
+                    csv << format_stats(target, source_file.path, "error", 0, 0, 0)
+                  else
+                    csv << format_stats(target, source_file.path, "unknown (#{source_file.status.class.to_s.split(/::/).last})", 0, 0, 0)
+                  end
+                end
+              end
+            end
+          end
+        )
+
+        0
+      end
+
+      def format_stats(target, path, status, typed, untyped, total)
+        [
+          target.name,
+          path.to_s,
+          status,
+          typed,
+          untyped,
+          total,
+          if total.nonzero?
+            format("%.2f", (typed.to_f/total)*100)
+          else
+            0
+          end
+        ]
+      end
+    end
+  end
+end