ruby-lsp-typeprof 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd458cb056ff5e4f48058e864528f8db6d03a9d7709fb47cd81b387976346d77
-  data.tar.gz: fe8d2d2de7862ea0ce8dccf66a0ed6e01e259815d82e3c484f35e8d73f4f3d8f
+  metadata.gz: 1b0dc266b02ba9428bd88e6001b094436c17fe94407477e1e2f184825416a40a
+  data.tar.gz: 70fa07a521a222f5c9e1ac9a62e2458bd41308b1eb6559a7252d8cfe8298008c
 SHA512:
-  metadata.gz: '008e208221b72bb8db5908f01a4f74f30854f54d2a70648c62ad2b39185407014f982b00d52be93ba2a9ac8d47c5a84d1831f217210ec5e4ba49867937eb44e2'
-  data.tar.gz: 9e38f9e164764cd93f8da7e877a64262764abba9d74dbbf4221ece9403bbd4cade039daa0dd116f1a6789ab488a57c58be9bd2f9e46b5df406a0192568f91747
+  metadata.gz: 4ed5ddf9b0a88bd9ad17fa748225ad62d23b1ab5e99f181b21a5dea1f10c2fecab850403c39470447487550b11f7a03989b648f7d61aa8ab586b5f8b75d5d50c
+  data.tar.gz: 33ea5a60eeb2295adff2ad7f73828d17dabe17cbcc01be64209bca361e8bb8bcd36f61f5b6e84a40601be2e9299817652dae9915024f90a8c6738d1447cdf1c7

data/README.md

@@ -27,7 +27,8 @@ Configure them in VS Code's `settings.json`:
   "rubyLsp.addonSettings": {
     "TypeProf": {
       "enabled": false,
-      "enableCodeLens": false
+      "enableCodeLens": false,
+      "enableDocumentSymbol": false
     }
   }
 }
@@ -45,6 +46,7 @@ You can place this configuration in either:
 | --- | --- | --- | --- |
 | `enabled` | boolean | `true` | Master switch. When `false`, the add-on skips loading TypeProf entirely and disables every feature. |
 | `enableCodeLens` | boolean | `true` | Show inferred type signatures as code lens above method definitions. Has no effect when `enabled` is `false`. |
+| `enableDocumentSymbol` | boolean | `true` | Add inferred type signatures as child symbols of method definitions in the document symbol response, so clients that do not support code lens (for example Claude Code) can see them. Has no effect when `enabled` is `false`. |
 
 After changing the setting, restart Ruby LSP via the "Ruby LSP: Restart"
 command for it to take effect.

data/lib/ruby_lsp/ruby_lsp_typeprof/addon.rb

@@ -5,6 +5,7 @@ require "ruby_lsp/addon"
 require "uri"
 
 require_relative "code_lens_listener"
+require_relative "document_symbol_listener"
 require_relative "loggable"
 
 module RubyLsp
@@ -18,6 +19,7 @@ module RubyLsp
         @mutex = Mutex.new
         @enabled = true
         @code_lens_enabled = true
+        @document_symbol_enabled = true
       end
 
       def activate(global_state, outgoing_queue)
@@ -52,6 +54,13 @@ module RubyLsp
         CodeLensListener.new(response_builder, uri, dispatcher, @service, @mutex, @outgoing_queue)
       end
 
+      def create_document_symbol_listener(response_builder, dispatcher)
+        return unless @service
+        return unless @document_symbol_enabled
+
+        DocumentSymbolListener.new(response_builder, dispatcher, @service, @mutex, @outgoing_queue)
+      end
+
       def workspace_did_change_watched_files(changes)
         return unless @service
 
@@ -66,6 +75,7 @@ module RubyLsp
         settings = global_state.settings_for_addon(name)
         @enabled = settings&.dig(:enabled) != false
         @code_lens_enabled = settings&.dig(:enableCodeLens) != false
+        @document_symbol_enabled = settings&.dig(:enableDocumentSymbol) != false
       end
 
       def build_service(workspace_path)

data/lib/ruby_lsp/ruby_lsp_typeprof/code_lens_listener.rb

@@ -1,25 +1,20 @@
 # frozen_string_literal: true
 
-require_relative "loggable"
+require_relative "signature_cache"
 
 module RubyLsp
   module Typeprof
     class CodeLensListener
-      include Loggable
-
       def initialize(response_builder, uri, dispatcher, service, mutex, outgoing_queue)
         @response_builder = response_builder
-        @path = uri.to_standardized_path
-        @lens_cache = {}
-        @outgoing_queue = outgoing_queue
+        @signatures = SignatureCache.new(service, mutex, uri.to_standardized_path, outgoing_queue, feature: "code lens")
 
-        cache_code_lens_results(service, mutex)
-        dispatcher.register(self, :on_def_node_enter) unless @lens_cache.empty?
+        dispatcher.register(self, :on_def_node_enter) unless @signatures.empty?
       end
 
       def on_def_node_enter(node)
         line = node.location.start_line
-        hint = @lens_cache[line]
+        hint = @signatures[line]
         return unless hint
 
         @response_builder << build_code_lens(line, hint)
@@ -35,16 +30,6 @@ module RubyLsp
           command: LanguageServer::Protocol::Interface::Command.new(title: "#: #{hint}", command: "")
         )
       end
-
-      def cache_code_lens_results(service, mutex)
-        mutex.synchronize do
-          service.code_lens(@path) do |code_range, hint|
-            @lens_cache[code_range.first.lineno] = hint
-          end
-        end
-      rescue StandardError => e
-        log_error("Ruby LSP TypeProf failed to compute code lens for #{@path}: #{e.full_message(highlight: false)}")
-      end
     end
   end
 end

data/lib/ruby_lsp/ruby_lsp_typeprof/document_symbol_listener.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require_relative "signature_cache"
+
+module RubyLsp
+  module Typeprof
+    # Appends the inferred method signature as a child symbol of each method
+    # definition symbol produced by Ruby LSP's own DocumentSymbol listener.
+    #
+    # Ruby LSP does not pass the document URI to `create_document_symbol_listener`,
+    # so the URI is read from the core listener already registered on the dispatcher.
+    class DocumentSymbolListener
+      def initialize(response_builder, dispatcher, service, mutex, outgoing_queue)
+        @response_builder = response_builder
+        path = document_path(dispatcher)
+        @signatures = SignatureCache.new(service, mutex, path, outgoing_queue, feature: "document symbol")
+
+        dispatcher.register(self, :on_def_node_enter) unless @signatures.empty?
+      end
+
+      def on_def_node_enter(node)
+        hint = @signatures[node.location.start_line]
+        return unless hint
+
+        parent = @response_builder.last
+        return unless parent.is_a?(LanguageServer::Protocol::Interface::DocumentSymbol)
+
+        parent.children << build_signature_symbol(node, hint)
+      end
+
+      private
+
+      def document_path(dispatcher)
+        core_listener = dispatcher.listeners.each_value.flat_map(&:itself).find do |listener|
+          listener.is_a?(::RubyLsp::Listeners::DocumentSymbol)
+        end
+        core_listener&.instance_variable_get(:@uri)&.to_standardized_path
+      end
+
+      def build_signature_symbol(node, hint)
+        range = range_from_location(node.name_loc)
+
+        LanguageServer::Protocol::Interface::DocumentSymbol.new(
+          name: hint,
+          kind: ::RubyLsp::Constant::SymbolKind::TYPE_PARAMETER,
+          range: range,
+          selection_range: range
+        )
+      end
+
+      def range_from_location(location)
+        LanguageServer::Protocol::Interface::Range.new(
+          start: LanguageServer::Protocol::Interface::Position.new(
+            line: location.start_line - 1, character: location.start_column
+          ),
+          end: LanguageServer::Protocol::Interface::Position.new(
+            line: location.end_line - 1, character: location.end_column
+          )
+        )
+      end
+    end
+  end
+end

data/lib/ruby_lsp/ruby_lsp_typeprof/signature_cache.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require_relative "loggable"
+
+module RubyLsp
+  module Typeprof
+    # Collects the method signatures TypeProf inferred for a single file,
+    # keyed by the 1-based line number of the method definition.
+    class SignatureCache
+      include Loggable
+
+      def initialize(service, mutex, path, outgoing_queue, feature:)
+        @outgoing_queue = outgoing_queue
+        @hints = {}
+        return unless path
+
+        mutex.synchronize do
+          service.code_lens(path) do |code_range, hint|
+            @hints[code_range.first.lineno] = hint
+          end
+        end
+      rescue StandardError => e
+        log_error("Ruby LSP TypeProf failed to compute #{feature} for #{path}: #{e.full_message(highlight: false)}")
+      end
+
+      def [](line)
+        @hints[line]
+      end
+
+      def empty?
+        @hints.empty?
+      end
+    end
+  end
+end

data/lib/ruby_lsp_typeprof/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLsp
   module Typeprof
-    VERSION = "0.4.0"
+    VERSION = "0.5.0"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-lsp-typeprof
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Takumi Shotoku
@@ -51,7 +51,9 @@ files:
 - lib/ruby-lsp-typeprof.rb
 - lib/ruby_lsp/ruby_lsp_typeprof/addon.rb
 - lib/ruby_lsp/ruby_lsp_typeprof/code_lens_listener.rb
+- lib/ruby_lsp/ruby_lsp_typeprof/document_symbol_listener.rb
 - lib/ruby_lsp/ruby_lsp_typeprof/loggable.rb
+- lib/ruby_lsp/ruby_lsp_typeprof/signature_cache.rb
 - lib/ruby_lsp_typeprof/version.rb
 - sig/ruby_lsp_typeprof/version.rbs
 homepage: https://github.com/sinsoku/ruby-lsp-typeprof
@@ -59,7 +61,7 @@ licenses:
 - MIT
 metadata:
   homepage_uri: https://github.com/sinsoku/ruby-lsp-typeprof
-  source_code_uri: https://github.com/sinsoku/ruby-lsp-typeprof/tree/v0.4.0
+  source_code_uri: https://github.com/sinsoku/ruby-lsp-typeprof/tree/v0.5.0
   changelog_uri: https://github.com/sinsoku/ruby-lsp-typeprof/releases
   rubygems_mfa_required: 'true'
 rdoc_options: []
@@ -76,7 +78,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.10
+rubygems_version: 4.0.16
 specification_version: 4
 summary: A Ruby LSP addon for TypeProf
 test_files: []