Add TypeCheckGroups command

lib/steep/server/custom_methods.rb

@@ -29,6 +29,14 @@ def self.response(id, result)
         end
       end
 
+      module TypeCheckGroups
+        METHOD = "$/steep/typecheck/groups"
+
+        def self.notification(params)
+          { method: METHOD, params: params }
+        end
+      end
+
       module TypeCheck__Start
         METHOD = "$/steep/typecheck/start"
 

lib/steep/server/master.rb

@@ -657,6 +657,18 @@ def process_message_from_client(message)
 
           start_type_check(request: request, last_request: nil)
 
+        when CustomMethods::TypeCheckGroups::METHOD
+          params = message[:params] #: CustomMethods::TypeCheckGroups::params
+
+          groups = params.fetch(:groups)
+
+          progress = work_done_progress(SecureRandom.uuid)
+          progress.begin("Type checking #{groups.empty? ? "project" : groups.join(", ")}", request_id: fresh_request_id)
+
+          request = controller.make_group_request(groups, progress: progress)
+          request.needs_response = false
+          start_type_check(request: request, last_request: current_type_check_request, report_progress_threshold: 0)
+
         when "$/ping"
           enqueue_write_job SendMessageJob.to_client(
             message: {

lib/steep/server/type_check_controller.rb

@@ -254,6 +254,50 @@ def update_priority(open: nil, close: nil)
         end
       end
 
+      def make_group_request(groups, progress:)
+        TypeCheckController::Request.new(guid: progress.guid, progress: progress).tap do |request|
+          if groups.empty?
+            files.signature_paths.each do |path, target_group|
+              target_group = target_group.target if target_group.is_a?(Project::Group)
+              request.signature_paths << [target_group.name, path]
+            end
+            files.source_paths.each do |path, target_group|
+              target_group = target_group.target if target_group.is_a?(Project::Group)
+              request.code_paths << [target_group.name, path]
+            end
+          else
+            group_set = groups.map do |group_name|
+              target_name, group_name = group_name.split(".", 2)
+              target_name or raise
+
+              target_name = target_name.to_sym
+              group_name = group_name.to_sym if group_name
+
+              if group_name
+                if target = project.targets.find {|target| target.name == target_name }
+                  target.groups.find {|group| group.name == group_name }
+                end
+              else
+                project.targets.find {|target| target.name == target_name }
+              end
+            end.compact.to_set
+
+            files.signature_paths.each do |path, target_group|
+              if group_set.include?(target_group)
+                target_group = target_group.target if target_group.is_a?(Project::Group)
+                request.signature_paths << [target_group.name, path]
+              end
+            end
+            files.source_paths.each do |path, target_group|
+              if group_set.include?(target_group)
+                target_group = target_group.target if target_group.is_a?(Project::Group)
+                request.code_paths << [target_group.name, path]
+              end
+            end
+          end
+        end
+      end
+
       def make_request(guid: SecureRandom.uuid, include_unchanged: false, progress:)
         TypeCheckController::Request.new(guid: guid, progress: progress).tap do |request|
           if include_unchanged

sig/steep/server/custom_methods.rbs

@@ -55,6 +55,24 @@ module Steep
         def self.response: (String id, result) -> untyped
       end
 
+      # Notification to start type checking groups
+      #
+      # This notification is assumed to be sent from the LSP client -- an editor.
+      #
+      module TypeCheckGroups
+        METHOD: String
+
+        # Name of groups and targets to type check
+        #
+        # An empty array means every targets.
+        #
+        type params = {
+          groups: Array[String]
+        }
+
+        def self.notification: (params) -> untyped
+      end
+
       # Notification to start type checking, sent from the master to workers
       #
       module TypeCheck__Start

sig/steep/server/type_check_controller.rbs

@@ -127,7 +127,7 @@ module Steep
                          | (close: Pathname) -> void
 
       def active_target?: (Target | Group) -> bool
-                        
+
       def push_changes_for_target: (Target | Group) -> void
 
       # Returns a TypeCheckRequest that contains all paths to be type checked
@@ -141,6 +141,9 @@ module Steep
       #
       def make_request: (?guid: String, ?include_unchanged: bool, progress: WorkDoneProgress) -> Request?
 
+      # Returns a request to type check Ruby code and RBS type signatures of given groups
+      #
+      def make_group_request: (Array[String], progress: WorkDoneProgress) -> Request
     end
   end
 end