steep 0.15.0 → 0.16.0

data/lib/steep/server/worker_process.rb

@@ -0,0 +1,62 @@
+module Steep
+  module Server
+    class WorkerProcess
+      attr_reader :reader
+      attr_reader :writer
+      attr_reader :stderr
+
+      attr_reader :wait_thread
+
+      def initialize(reader:, writer:, stderr:, wait_thread:)
+        @reader = reader
+        @writer = writer
+        @stderr = stderr
+        @wait_thread = wait_thread
+      end
+
+      def self.spawn_worker(type, name:, steepfile:)
+        log_level = %w(debug info warn error fatal unknown)[Steep.logger.level]
+        command = case type
+                  when :code
+                    ["steep", "worker", "--code", "--name=#{name}", "--log-level=#{log_level}", "--steepfile=#{steepfile}"]
+                  when :signature
+                    ["steep", "worker", "--signature", "--name=#{name}", "--log-level=#{log_level}", "--steepfile=#{steepfile}"]
+                  when :interaction
+                    ["steep", "worker", "--interaction", "--name=#{name}", "--log-level=#{log_level}", "--steepfile=#{steepfile}"]
+                  else
+                    raise
+                  end
+
+        stdin, stdout, thread = Open3.popen2(*command, pgroup: true)
+        stderr = nil
+
+        writer = LanguageServer::Protocol::Transport::Io::Writer.new(stdin)
+        reader = LanguageServer::Protocol::Transport::Io::Reader.new(stdout)
+
+        new(reader: reader, writer: writer, stderr: stderr, wait_thread: thread)
+      end
+
+      def self.spawn_code_workers(steepfile:, count: [Etc.nprocessors-3, 1].max)
+        count.times.map do |i|
+          spawn_worker(:code, name: "code@#{i}", steepfile: steepfile)
+        end
+      end
+
+      def <<(message)
+        writer.write(message)
+      end
+
+      def read(&block)
+        reader.read(&block)
+      end
+
+      def shutdown
+        self << { method: :shutdown, params: nil }
+        self << { method: :exit, params: nil }
+
+        writer.io.close()
+        @wait_thread.join()
+      end
+    end
+  end
+end

data/lib/steep/signature/validator.rb

@@ -51,7 +51,7 @@ module Steep
         case decl
         when Declarations::Class
           rescue_validation_errors do
-            Steep.logger.info "#{Location.to_string decl.location}:\tValidating class definition `#{name}`..."
+            Steep.logger.debug "#{Location.to_string decl.location}:\tValidating class definition `#{name}`..."
             builder.build_instance(decl.name.absolute!).each_type do |type|
               env.validate type, namespace: RBS::Namespace.root
             end
@@ -61,7 +61,7 @@ module Steep
           end
         when Declarations::Interface
           rescue_validation_errors do
-            Steep.logger.info "#{Location.to_string decl.location}:\tValidating interface `#{name}`..."
+            Steep.logger.debug "#{Location.to_string decl.location}:\tValidating interface `#{name}`..."
             builder.build_interface(decl.name.absolute!, decl).each_type do |type|
               env.validate type, namespace: RBS::Namespace.root
             end
@@ -78,7 +78,7 @@ module Steep
       def validate_const
         env.each_constant do |name, decl|
           rescue_validation_errors do
-            Steep.logger.info "#{Location.to_string decl.location}:\tValidating constant `#{name}`..."
+            Steep.logger.debug "#{Location.to_string decl.location}:\tValidating constant `#{name}`..."
             env.validate(decl.type, namespace: name.namespace)
           end
         end
@@ -87,7 +87,7 @@ module Steep
       def validate_global
         env.each_global do |name, decl|
           rescue_validation_errors do
-            Steep.logger.info "#{Location.to_string decl.location}:\tValidating global `#{name}`..."
+            Steep.logger.debug "#{Location.to_string decl.location}:\tValidating global `#{name}`..."
             env.validate(decl.type, namespace: RBS::Namespace.root)
           end
         end
@@ -96,7 +96,7 @@ module Steep
       def validate_alias
         env.each_alias do |name, decl|
           rescue_validation_errors do
-            Steep.logger.info "#{Location.to_string decl.location}:\tValidating alias `#{name}`..."
+            Steep.logger.debug "#{Location.to_string decl.location}:\tValidating alias `#{name}`..."
             env.validate(decl.type, namespace: name.namespace)
           end
         end

data/lib/steep/version.rb

@@ -1,3 +1,3 @@
 module Steep
-  VERSION = "0.15.0"
+  VERSION = "0.16.0"
 end

data/lib/steep.rb

@@ -10,6 +10,8 @@ require "rainbow"
 require "listen"
 require 'pry'
 require 'language_server-protocol'
+require "etc"
+require "open3"
 
 require "rbs"
 
@@ -70,6 +72,14 @@ require "steep/type_inference/local_variable_type_env"
 require "steep/type_inference/logic"
 require "steep/ast/types"
 
+require "steep/server/utils"
+require "steep/server/base_worker"
+require "steep/server/code_worker"
+require "steep/server/signature_worker"
+require "steep/server/worker_process"
+require "steep/server/interaction_worker"
+require "steep/server/master"
+
 require "steep/project"
 require "steep/project/file"
 require "steep/project/options"
@@ -89,6 +99,7 @@ require "steep/drivers/trace_printer"
 require "steep/drivers/print_project"
 require "steep/drivers/init"
 require "steep/drivers/vendor"
+require "steep/drivers/worker"
 
 if ENV["NO_COLOR"]
   Rainbow.enabled = false

data/steep.gemspec

@@ -42,5 +42,5 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency "rainbow", ">= 2.2.2", "< 4.0"
   spec.add_runtime_dependency "listen", "~> 3.1"
   spec.add_runtime_dependency "pry", "~> 0.12.2"
-  spec.add_runtime_dependency "language_server-protocol", "~> 3.14.0.1"
+  spec.add_runtime_dependency "language_server-protocol", "~> 3.14.0.2"
 end

data/vendor/ruby-signature/README.md

@@ -10,51 +10,51 @@ You need to install the dependencies, and build its parser with `bin/setup`.
 
 ```
 $ bin/setup
-$ bundle exec exe/ruby-signature
+$ bundle exec exe/rbs
 ```
 
 ## Usage
 
 ```
-$ ruby-signature list
-$ ruby-signature ancestors ::Object
-$ ruby-signature methods ::Object
-$ ruby-signature method ::Object tap
+$ rbs list
+$ rbs ancestors ::Object
+$ rbs methods ::Object
+$ rbs method ::Object tap
 ```
 
-### ruby-signature [--class|--module|interface] list
+### rbs [--class|--module|interface] list
 
 ```
-$ ruby-signature list
+$ rbs list
 ```
 
 This command lists all of the classes/modules/interfaces defined in `.rbs` files.
 
-### ruby-signature ancestors [--singleton|--instance] CLASS
+### rbs ancestors [--singleton|--instance] CLASS
 
 ```
-$ ruby-signature ancestors Array                    # ([].class.ancestors)
-$ ruby-signature ancestors --singleton Array        # (Array.class.ancestors)
+$ rbs ancestors Array                    # ([].class.ancestors)
+$ rbs ancestors --singleton Array        # (Array.class.ancestors)
 ```
 
 This command prints the _ancestors_ of the class.
 The name of the command is borrowed from `Class#ancestors`, but the semantics is a bit different.
 The `ancestors` command is more precise (I believe).
 
-### ruby-signature methods [--singleton|--instance] CLASS
+### rbs methods [--singleton|--instance] CLASS
 
 ```
-$ ruby-signature methods ::Integer                  # 1.methods
-$ ruby-signature methods --singleton ::Object       # Object.methods
+$ rbs methods ::Integer                  # 1.methods
+$ rbs methods --singleton ::Object       # Object.methods
 ```
 
 This command prints all methods provided for the class.
 
-### ruby-signature method [--singleton|--instance] CLASS METHOD
+### rbs method [--singleton|--instance] CLASS METHOD
 
 ```
-$ ruby-signature method ::Integer '+'               # 1+2
-$ ruby-signature method --singleton ::Object tap    # Object.tap { ... }
+$ rbs method ::Integer '+'               # 1+2
+$ rbs method --singleton ::Object tap    # Object.tap { ... }
 ```
 
 This command prints type and properties of the method.
@@ -66,13 +66,13 @@ It accepts two global options, `-r` and `-I`.
 `-r` is for libraries. You can specify the names of libraries.
 
 ```
-$ ruby-signature -r set list
+$ rbs -r set list
 ```
 
 `-I` is for application signatures. You can specify the name of directory.
 
 ```
-$ ruby-signature -I sig list
+$ rbs -I sig list
 ```
 
 ## Guides

data/vendor/ruby-signature/Rakefile

@@ -15,7 +15,7 @@ task :validate => :parser do
   sh "rbs validate"
 end
 
-FileList["test/stdlib/*_test.rb"].each do |test|
+FileList["test/stdlib/**/*_test.rb"].each do |test|
   multitask test => :parser do
     sh "ruby bin/test_runner.rb #{test}"
   end
@@ -44,21 +44,96 @@ namespace :generate do
     path = Pathname("test/stdlib/#{klass}_test.rb")
     raise "#{path} already exists!" if path.exist?
 
-    path.write <<~RUBY
-      require_relative "test_helper"
-      
-      class #{klass}Test < StdlibTest
-        target #{klass}
-        # library "pathname", "set", "securerandom"     # Declare library signatures to load
-        using hook.refinement
-
-        # def test_method_name
-        #   # Call the method
-        #   method_name(arg)
-        #   method_name(arg, arg2)
-        # end
+    require "erb"
+    require "ruby/signature"
+
+    class TestTemplateBuilder
+      attr_reader :klass, :env
+
+      def initialize(klass)
+        @klass = klass
+
+        @env = Ruby::Signature::Environment.new
+        Ruby::Signature::EnvironmentLoader.new.load(env: @env)
+      end
+
+      def call
+        ERB.new(<<~ERB, trim_mode: "-").result(binding)
+          require_relative "test_helper"
+
+          class <%= klass %>Test < StdlibTest
+            target <%= klass %>
+            # library "pathname", "set", "securerandom"     # Declare library signatures to load
+            using hook.refinement
+          <%- class_methods.each do |method_name, definition| %>
+            def test_class_method_<%= test_name_for(method_name) %>
+            <%- definition.method_types.each do |method_type| -%>
+              # <%= method_type %>
+              <%= klass %>.<%= method_name %>
+            <%- end -%>
+            end
+          <%- end -%>
+          <%- instance_methods.each do |method_name, definition| %>
+            def test_<%= test_name_for(method_name) %>
+            <%- definition.method_types.each do |method_type| -%>
+              # <%= method_type %>
+              <%= klass %>.new.<%= method_name %>
+            <%- end -%>
+            end
+          <%- end -%>
+          end
+        ERB
+      end
+
+      private
+
+      def test_name_for(method_name)
+        {
+          :==  => 'double_equal',
+          :!=  => 'not_equal',
+          :=== => 'triple_equal',
+          :[]  => 'square_bracket',
+          :[]= => 'square_bracket_assign',
+          :>   => 'greater_than',
+          :<   => 'less_than',
+          :>=  => 'greater_than_equal_to',
+          :<=  => 'less_than_equal_to',
+          :<=> => 'spaceship',
+          :+   => 'plus',
+          :-   => 'minus',
+          :*   => 'multiply',
+          :/   => 'divide',
+          :**  => 'power',
+          :%   => 'modulus',
+          :&   => 'and',
+          :|   => 'or',
+          :^   => 'xor',
+          :>>  => 'right_shift',
+          :<<  => 'left_shift',
+          :=~  => 'pattern_match',
+          :!~  => 'does_not_match',
+          :~   => 'tilde'
+        }.fetch(method_name, method_name)
       end
-    RUBY
+
+      def type_name
+        @type_name ||= Ruby::Signature::TypeName.new(name: klass.to_sym, namespace: Ruby::Signature::Namespace.new(path: [], absolute: true))
+      end
+
+      def class_methods
+        @class_methods ||= Ruby::Signature::DefinitionBuilder.new(env: env).build_singleton(type_name).methods.select {|_, definition|
+          definition.implemented_in.name.absolute! == type_name
+        }
+      end
+
+      def instance_methods
+        @instance_methods ||= Ruby::Signature::DefinitionBuilder.new(env: env).build_instance(type_name).methods.select {|_, definition|
+          definition.implemented_in.name.absolute! == type_name
+        }
+      end
+    end
+
+    path.write TestTemplateBuilder.new(klass).call
 
     puts "Created: #{path}"
   end

data/vendor/ruby-signature/rbs.gemspec

@@ -42,4 +42,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "rubocop"
   spec.add_development_dependency "rubocop-rubycw"
   spec.add_development_dependency "minitest-reporters", "~> 1.3.6"
+  spec.add_development_dependency "json", "~> 2.3.0"
 end

data/vendor/ruby-signature/stdlib/builtin/builtin.rbs

@@ -42,3 +42,4 @@ type int = Integer | _ToInt
 type real = Integer | Float | Rational
 
 type string = String | _ToStr
+type encoding = Encoding | string

data/vendor/ruby-signature/stdlib/builtin/enumerable.rbs

@@ -133,7 +133,7 @@ module Enumerable[unchecked out Elem, out Return]: _Each[Elem, Return]
 
   def inject: (untyped init, Symbol method) -> untyped
             | (Symbol method) -> untyped
-            | [A] (A initial) { (Elem, A) -> A } -> Elem
+            | [A] (A initial) { (A, Elem) -> A } -> A
             | () { (Elem, Elem) -> Elem } -> Elem
 
   # Returns the object in *enum* with the maximum value. The first form

data/vendor/ruby-signature/stdlib/builtin/module.rbs

@@ -1034,7 +1034,7 @@ class Module < Object
   # Removes the method identified by *symbol* from the current class. For an
   # example, see Module#undef_method. String arguments are converted to symbols.
   #
-  def remove_method: (Symbol | String arg0) -> self
+  def remove_method: (*Symbol | String arg0) -> self
 
   # Returns `true` if *mod* is a singleton class or `false` if it is an ordinary
   # class or module.
@@ -1087,7 +1087,7 @@ class Module < Object
   #     In parent
   #     prog.rb:23: undefined method `hello' for #<Child:0x401b3bb4> (NoMethodError)
   #
-  def undef_method: (Symbol | String arg0) -> self
+  def undef_method: (*Symbol | String arg0) -> self
 
   # Import class refinements from *module* into the current class or module
   # definition.