ruby-lsp-ree 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8eb134e8d6b0ac3bf5321789ed6617ea1347ee096b1e642f953885b398524030
-  data.tar.gz: 460f2cc171155affbafe236535604b116d0af15bbcf14c3925470a8ff3616211
+  metadata.gz: a1d126a7d0c8462f33b63113cf86ae134f8cb029da147273bdb2f8cb76cb448e
+  data.tar.gz: dbb25c53fc012b487a3ed9555ad470c10445603e808aa3f44442070b4fc57851
 SHA512:
-  metadata.gz: cf49632399253b97f3b3df77ae7767008141900552461de8bca6b7395ac0d48fa4dbc5a353d2245f50147d7d119059b3f4d5a3f55aa41ae4fd11521d58b7930d
-  data.tar.gz: 0f392c77bbe2bdee9e9f613bc856c1d93b9bb9a2c87a93078017388fb253b27325d244ba289b36c24ff7e71e9f72b4f9c11abb663a7a2f675b5a9661aae29554
+  metadata.gz: 50d85fbda0c61e33709823c7e8b5f263d94eba2d6822b1e84f011ad0b058adf6d5766a5fe27b6613c537b2e36f878a8ef50ddf7fad7dd41cfd954011fe90c156
+  data.tar.gz: 66c76a82263258d76fcc7cc604e6aa499deb209ae80c6ae8a5bad1dd486f7ee70c9e0af896277c0cc01631ad99c91f5b5b827e18337b0d95e4d18622d1c2c389

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+## [0.1.4] - 2025-02-25
+
+- autocompletion in spec files
+- Add Link in spec files
+- improved support of new files and unsaved documents
+
 ## [0.1.3] - 2025-02-21
 
 - improved Go To Definition for ree object methods and imported constants

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-lsp-ree (0.1.3)
+    ruby-lsp-ree (0.1.4)
 
 GEM
   remote: https://rubygems.org/

data/lib/ruby_lsp/ruby_lsp_ree/handlers/completion_handler.rb

@@ -14,6 +14,7 @@ module RubyLsp
         @index = index
         @uri = uri
         @node_context = node_context
+        @root_node = @node_context.instance_variable_get(:@nesting_nodes).first
         @finder = ReeObjectFinder.new(@index)
       end
 
@@ -145,7 +146,7 @@ module RubyLsp
         
         return [] if class_name_objects.size == 0
 
-        parsed_doc = RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@node_context.parent, @uri)
+        parsed_doc = RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@root_node, @uri)
 
         imported_consts = []
         not_imported_consts = []
@@ -206,7 +207,7 @@ module RubyLsp
 
         return [] if ree_objects.size == 0
   
-        parsed_doc = RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@node_context.parent, @uri)
+        parsed_doc = RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@root_node, @uri)
 
         ree_objects.map do |ree_object|
           ree_object_name = ree_object.name
@@ -306,6 +307,8 @@ module RubyLsp
       end
 
       def get_additional_text_edits_for_method(parsed_doc, fn_name, package_name)
+        return [] unless parsed_doc
+
         if parsed_doc.includes_linked_object?(fn_name)
           return []
         end

data/lib/ruby_lsp/ruby_lsp_ree/handlers/definition_handler.rb

@@ -0,0 +1,132 @@
+require_relative "../utils/ree_lsp_utils"
+require_relative "../ree_object_finder"
+require_relative "../parsing/parsed_link_node"
+require_relative "../parsing/parsed_document_builder"
+
+module RubyLsp
+  module Ree
+    class DefinitionHandler
+      include Requests::Support::Common
+      include RubyLsp::Ree::ReeLspUtils
+
+      def initialize(index, uri, node_context)
+        @index = index
+        @uri = uri
+        @node_context = node_context
+        @finder = ReeObjectFinder.new(@index)
+      end
+
+      def get_constant_definition_items(node)
+        result = []
+
+        link_nodes = if @node_context.parent.is_a?(Prism::CallNode) && @node_context.parent.name == :link
+          # inside link node
+          link_node = RubyLsp::Ree::ParsedLinkNode.new(@node_context.parent)
+          link_node.parse_imports
+          [link_node]
+        else
+          parsed_doc = if @node_context.parent.is_a?(Prism::CallNode)
+            RubyLsp::Ree::ParsedDocumentBuilder.build_from_uri(@uri)
+          else
+            RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@node_context.parent, @uri)
+          end
+
+          parsed_doc.link_nodes
+        end
+        
+        link_nodes.each do |link_node|
+          if link_node.imports.include?(node.name.to_s)
+            uri = ''
+            if link_node.file_path_type?
+              path = find_local_file_path(link_node.name)  
+              next unless path
+
+              uri = File.join(Dir.pwd, path)
+            else
+              package_name = link_node.link_package_name || package_name_from_uri(@uri)
+
+              method_candidates = @index[link_node.name]
+              next if !method_candidates || method_candidates.size == 0
+        
+              method = method_candidates.detect{ package_name_from_uri(_1.uri) == package_name }
+              next unless method
+
+              uri = method.uri.to_s
+            end
+
+            result << Interface::Location.new(
+              uri: uri,
+              range: Interface::Range.new(
+                start: Interface::Position.new(line: 0, character: 0),
+                end: Interface::Position.new(line: 0, character: 0),
+              ),
+            )
+          end
+        end
+
+        result
+      end
+
+      def get_ree_objects_definition_items(node)
+        message = node.message
+        result = []
+        
+        definition_item = ReeObjectFinder.new(@index).find_object(message)
+        return [] unless definition_item
+
+        definition_uri = definition_item.uri.to_s
+        return [] unless definition_uri
+
+        result << Interface::Location.new(
+          uri: definition_uri,
+          range: Interface::Range.new(
+            start: Interface::Position.new(line: 0, character: 0),
+            end: Interface::Position.new(line: 0, character: 0),
+          ),
+        )
+
+        result
+      end
+
+      def get_linked_object_definition_items(node)
+        result = []
+        parent_node = @node_context.parent
+        return [] unless parent_node.name == :link
+
+        link_node = RubyLsp::Ree::ParsedLinkNode.new(parent_node, package_name_from_uri(@uri))
+        package_name = link_node.link_package_name
+
+        method_candidates = @index[node.unescaped]
+        return [] if !method_candidates || method_candidates.size == 0
+        
+        method = method_candidates.detect{ package_name_from_uri(_1.uri) == package_name }
+        return [] unless method
+
+        result << Interface::Location.new(
+          uri: method.uri.to_s,
+          range: Interface::Range.new(
+            start: Interface::Position.new(line: 0, character: 0),
+            end: Interface::Position.new(line: 0, character: 0),
+          ),
+        )
+
+        result
+      end
+
+      def get_linked_filepath_definition_items(node)
+        result = []
+        local_path = find_local_file_path(node.unescaped)
+
+        return [] unless local_path
+
+        result << Interface::Location.new(
+          uri: File.join(Dir.pwd, local_path),
+          range: Interface::Range.new(
+            start: Interface::Position.new(line: 0, character: 0),
+            end: Interface::Position.new(line: 0, character: 0),
+          ),
+        )
+      end
+    end
+  end
+end

data/lib/ruby_lsp/ruby_lsp_ree/handlers/hover_handler.rb

@@ -0,0 +1,49 @@
+require_relative "../ree_object_finder"
+require_relative "../parsing/parsed_document_builder"
+require_relative "../parsing/parsed_link_node"
+require_relative "../utils/ree_lsp_utils"
+
+module RubyLsp
+  module Ree
+    class HoverHandler
+      include Requests::Support::Common
+      include RubyLsp::Ree::ReeLspUtils
+
+      def initialize(index, node_context)
+        @index = index
+        @node_context = node_context
+        @finder = ReeObjectFinder.new(@index)
+      end
+
+      def get_ree_object_hover_items(node)
+        ree_object = @finder.find_object(node.name.to_s)
+
+        return [] unless ree_object
+
+        documentation =  <<~DOC
+        Ree object, type: :#{@finder.object_type(ree_object)}
+
+        usage: #{node.name.to_s}#{get_detail_string(ree_object)}
+        
+        package: #{package_name_from_uri(ree_object.uri)}
+
+        file: #{path_from_package_folder(ree_object.uri)}
+        DOC
+
+        [documentation]
+      end
+
+      def get_detail_string(ree_object)
+        return '' if ree_object.signatures.size == 0
+
+        "(#{get_parameters_string(ree_object.signatures.first)})"
+      end
+
+      def get_parameters_string(signature)
+        return '' unless signature
+
+        signature.parameters.map(&:decorated_name).join(', ')
+      end
+    end
+  end
+end

data/lib/ruby_lsp/ruby_lsp_ree/listeners/completion_listener.rb

@@ -21,6 +21,8 @@ module RubyLsp
 
         completion_items = @handler.get_class_name_completion_items(node)
         put_items_into_response(completion_items)
+      rescue => e
+        $stderr.puts("error in completion listener(on_constant_read_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
       def on_call_node_enter(node)
@@ -37,6 +39,8 @@ module RubyLsp
         end
         
         put_items_into_response(completion_items)
+      rescue => e
+        $stderr.puts("error in completion listener(on_call_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
       def put_items_into_response(items)

data/lib/ruby_lsp/ruby_lsp_ree/listeners/definition_listener.rb

@@ -1,136 +1,62 @@
-require_relative "../utils/ree_lsp_utils"
-require_relative "../parsing/parsed_link_node"
-require_relative "../parsing/parsed_document_builder"
-require_relative "../ree_object_finder"
+require_relative "../handlers/definition_handler"
 
 module RubyLsp
   module Ree
     class DefinitionListener
       include Requests::Support::Common
-      include RubyLsp::Ree::ReeLspUtils
 
       def initialize(response_builder, node_context, index, dispatcher, uri)
         @response_builder = response_builder
-        @node_context = node_context
-        @nesting = node_context.nesting
-        @index = index
-        @uri = uri
-
-        dispatcher.register(self, :on_call_node_enter, :on_symbol_node_enter, :on_string_node_enter, :on_constant_read_node_enter)
+        @handler = RubyLsp::Ree::DefinitionHandler.new(index, uri, node_context)
+
+        dispatcher.register(
+          self, 
+          :on_call_node_enter, 
+          :on_symbol_node_enter, 
+          :on_string_node_enter, 
+          :on_constant_read_node_enter
+        )
       end
 
       def on_constant_read_node_enter(node)
-        link_nodes = if @node_context.parent.is_a?(Prism::CallNode) && @node_context.parent.name == :link
-          # inside link node
-          link_node = RubyLsp::Ree::ParsedLinkNode.new(@node_context.parent)
-          link_node.parse_imports
-          [link_node]
-        else
-          parsed_doc = if @node_context.parent.is_a?(Prism::CallNode)
-            RubyLsp::Ree::ParsedDocumentBuilder.build_from_uri(@uri)
-          else
-            RubyLsp::Ree::ParsedDocumentBuilder.build_from_ast(@node_context.parent, @uri)
-          end
-
-          parsed_doc.link_nodes
-        end
-        
-        link_nodes.each do |link_node|
-          if link_node.imports.include?(node.name.to_s)
-            uri = ''
-            if link_node.file_path_type?
-              path = find_local_file_path(link_node.name)  
-              next unless path
-
-              uri = File.join(Dir.pwd, path)
-            else
-              package_name = link_node.link_package_name || package_name_from_uri(@uri)
-
-              method_candidates = @index[link_node.name]
-              next if !method_candidates || method_candidates.size == 0
-        
-              method = method_candidates.detect{ package_name_from_uri(_1.uri) == package_name }
-              next unless method
-
-              uri = method.uri.to_s
-            end
-
-            @response_builder << Interface::Location.new(
-              uri: uri,
-              range: Interface::Range.new(
-                start: Interface::Position.new(line: 0, character: 0),
-                end: Interface::Position.new(line: 0, character: 0),
-              ),
-            )
-          end
-        end
+        definition_items = @handler.get_constant_definition_items(node)
+        put_items_into_response(definition_items)
+      rescue => e
+        $stderr.puts("error in definition listener(on_constant_read_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
       def on_call_node_enter(node)
-        message = node.message
-        return unless message
-
-        definition_uri = nil
+        return unless node.message
 
         if node.receiver
           # ruby lsp handles such cases itself
           return
         else
-          definition_item = ReeObjectFinder.new(@index).find_object(message)
-          definition_uri = definition_item.uri.to_s
+          definition_items = @handler.get_ree_objects_definition_items(node)
+          put_items_into_response(definition_items)
         end
-
-        return unless definition_uri
-
-        @response_builder << Interface::Location.new(
-          uri: definition_uri,
-          range: Interface::Range.new(
-            start: Interface::Position.new(line: 0, character: 0),
-            end: Interface::Position.new(line: 0, character: 0),
-          ),
-        )
-
-        nil
+      rescue => e
+        $stderr.puts("error in definition listener(on_call_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
       def on_symbol_node_enter(node)
-        parent_node = @node_context.parent
-        return unless parent_node.name == :link
-
-        link_node = RubyLsp::Ree::ParsedLinkNode.new(parent_node, package_name_from_uri(@uri))
-        package_name = link_node.link_package_name
-
-        method_candidates = @index[node.unescaped]
-        return if !method_candidates || method_candidates.size == 0
-        
-        method = method_candidates.detect{ package_name_from_uri(_1.uri) == package_name }
-        return unless method
-
-        @response_builder << Interface::Location.new(
-          uri: method.uri.to_s,
-          range: Interface::Range.new(
-            start: Interface::Position.new(line: 0, character: 0),
-            end: Interface::Position.new(line: 0, character: 0),
-          ),
-        )
-
-        nil
+        definition_items = @handler.get_linked_object_definition_items(node)
+        put_items_into_response(definition_items)
+      rescue => e
+        $stderr.puts("error in definition listener(on_symbol_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
       def on_string_node_enter(node)
-        local_path = find_local_file_path(node.unescaped)
+        definition_items = @handler.get_linked_filepath_definition_items(node)
+        put_items_into_response(definition_items)
+      rescue => e
+        $stderr.puts("error in definition listener(on_string_node_enter): #{e.message} : #{e.backtrace.first}")
+      end
 
-        if local_path
-          @response_builder << Interface::Location.new(
-            uri: File.join(Dir.pwd, local_path),
-            range: Interface::Range.new(
-              start: Interface::Position.new(line: 0, character: 0),
-              end: Interface::Position.new(line: 0, character: 0),
-            ),
-          )
+      def put_items_into_response(items)
+        items.each do |item|
+          @response_builder << item
         end
-
-        nil
       end
     end
   end

data/lib/ruby_lsp/ruby_lsp_ree/listeners/hover_listener.rb

@@ -1,7 +1,4 @@
-require_relative "../ree_object_finder"
-require_relative "../parsing/parsed_document_builder"
-require_relative "../parsing/parsed_link_node"
-require_relative "../utils/ree_lsp_utils"
+require_relative "../handlers/hover_handler"
 
 module RubyLsp
   module Ree
@@ -10,8 +7,7 @@ module RubyLsp
 
       def initialize(response_builder, node_context, index, dispatcher)
         @response_builder = response_builder
-        @node_context = node_context
-        @finder = ReeObjectFinder.new(index)
+        @handler = RubyLsp::Ree::HoverHandler.new(index, node_context)
 
         dispatcher.register(self, :on_call_node_enter)
         dispatcher.register(self, :on_constant_read_node_enter)
@@ -30,35 +26,16 @@ module RubyLsp
       end
 
       def on_call_node_enter(node)
-        ree_object = @finder.find_object(node.name.to_s)
-
-        return unless ree_object
-
-        documentation =  <<~DOC
-        Ree object, type: :#{@finder.object_type(ree_object)}
-
-        usage: #{node.name.to_s}#{get_detail_string(ree_object)}
-        
-        package: #{package_name_from_uri(ree_object.uri)}
-
-        file: #{path_from_package_folder(ree_object.uri)}
-        DOC
-
-        $stderr.puts(documentation)
-
-        @response_builder.push(documentation, category: :documentation)
+        hover_items = @handler.get_ree_object_hover_items(node)
+        put_items_into_response(hover_items)
+      rescue => e
+        $stderr.puts("error in hover listener(on_call_node_enter): #{e.message} : #{e.backtrace.first}")
       end
 
-      def get_detail_string(ree_object)
-        return '' if ree_object.signatures.size == 0
-
-        "(#{get_parameters_string(ree_object.signatures.first)})"
-      end
-
-      def get_parameters_string(signature)
-        return '' unless signature
-
-        signature.parameters.map(&:decorated_name).join(', ')
+      def put_items_into_response(items)
+        items.each do |item|
+          @response_builder.push(item, category: :documentation)
+        end
       end
     end
   end

data/lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_document.rb

@@ -1,4 +1,5 @@
 require_relative 'parsed_link_node'
+require 'ostruct'
 
 class RubyLsp::Ree::ParsedDocument
   include RubyLsp::Ree::ReeLspUtils
@@ -18,8 +19,12 @@ class RubyLsp::Ree::ParsedDocument
     @fn_node || @action_node || @dao_node || @bean_node || @mapper_node || @aggregate_node
   end
 
+  def allows_root_links?
+    false
+  end
+
   def includes_link_dsl?
-    @class_includes.any?{ _1.name == LINK_DSL_MODULE }
+    @class_includes.any?{ node_name(_1) == LINK_DSL_MODULE }
   end
 
   def includes_linked_constant?(const_name)
@@ -27,11 +32,11 @@ class RubyLsp::Ree::ParsedDocument
   end
 
   def includes_linked_object?(obj_name)
-    @link_nodes.map(&:name).include?(obj_name)
+    @link_nodes.map{ node_name(_1) }.include?(obj_name)
   end
 
   def find_link_node(name)
-    @link_nodes.detect{ _1.name == name }
+    @link_nodes.detect{ node_name(_1) == name }
   end
 
   def find_link_with_imported_object(name)
@@ -61,49 +66,49 @@ class RubyLsp::Ree::ParsedDocument
   def parse_fn_node
     return unless class_node
 
-    @fn_node ||= class_node.body.body.detect{ |node| node.name == :fn }
+    @fn_node ||= class_node.body.body.detect{ |node| node_name(node) == :fn }
     @links_container_block_node ||= @fn_node&.block
   end
 
   def parse_action_node
     return unless class_node
 
-    @action_node ||= class_node.body.body.detect{ |node| node.name == :action }
+    @action_node ||= class_node.body.body.detect{ |node| node_name(node) == :action }
     @links_container_block_node ||= @action_node&.block
   end
 
   def parse_dao_node
     return unless class_node
 
-    @dao_node ||= class_node.body.body.detect{ |node| node.name == :dao }
+    @dao_node ||= class_node.body.body.detect{ |node| node_name(node) == :dao }
     @links_container_block_node ||= @dao_node&.block
   end
 
   def parse_bean_node
     return unless class_node
 
-    @bean_node ||= class_node.body.body.detect{ |node| node.name == :bean }
+    @bean_node ||= class_node.body.body.detect{ |node| node_name(node) == :bean }
     @links_container_block_node ||= @bean_node&.block
   end
 
   def parse_mapper_node
     return unless class_node
 
-    @mapper_node ||= class_node.body.body.detect{ |node| node.name == :mapper }
+    @mapper_node ||= class_node.body.body.detect{ |node| node_name(node) == :mapper }
     @links_container_block_node ||= @mapper_node&.block
   end
 
   def parse_aggregate_node
     return unless class_node
 
-    @aggregate_node ||= class_node.body.body.detect{ |node| node.name == :aggregate }
+    @aggregate_node ||= class_node.body.body.detect{ |node| node_name(node) == :aggregate }
     @links_container_block_node ||= @aggregate_node&.block
   end
 
   def parse_class_includes
     return unless class_node
 
-    @class_includes ||= class_node.body.body.select{ _1.name == :include }.map do |class_include|
+    @class_includes ||= class_node.body.body.select{ node_name(_1) == :include }.map do |class_include|
       parent_name = class_include.arguments.arguments.first.parent.name.to_s
       module_name = class_include.arguments.arguments.first.name
     
@@ -116,10 +121,10 @@ class RubyLsp::Ree::ParsedDocument
   def parse_links
     return unless class_node
 
-    nodes = if links_container_node && @links_container_block_node.body
-      @links_container_block_node.body.body.select{ |node| node.name == :link }
-    elsif class_includes.any?{ _1.name == LINK_DSL_MODULE }
-      class_node.body.body.select{ |node| node.name == :link }
+    nodes = if links_container_node && @links_container_block_node && @links_container_block_node.body
+      @links_container_block_node.body.body.select{ |node| node_name(node) == :link }
+    elsif class_includes.any?{ node_name(_1) == LINK_DSL_MODULE }
+      class_node.body.body.select{ |node| node_name(node) == :link }
     else
       []
     end
@@ -135,7 +140,7 @@ class RubyLsp::Ree::ParsedDocument
     return unless class_node
     
     @values ||= class_node.body.body
-      .select{ _1.name == :val }
+      .select{ node_name(_1) == :val }
       .map{ OpenStruct.new(name: _1.arguments.arguments.first.unescaped) }
   end
 
@@ -143,7 +148,7 @@ class RubyLsp::Ree::ParsedDocument
     return unless class_node
     
     @filters ||= class_node.body.body
-      .select{ _1.name == :filter }
+      .select{ node_name(_1) == :filter }
       .map{ OpenStruct.new(name: _1.arguments.arguments.first.unescaped, signatures: parse_filter_signature(_1)) }
 
   end
@@ -153,7 +158,7 @@ class RubyLsp::Ree::ParsedDocument
     
     @bean_methods ||= class_node.body.body
       .select{ _1.is_a?(Prism::DefNode) }
-      .map{ OpenStruct.new(name: _1.name.to_s, signatures: parse_signatures_from_params(_1.parameters)) }
+      .map{ OpenStruct.new(name: node_name(_1).to_s, signatures: parse_signatures_from_params(_1.parameters)) }
   end
 
   def parse_filter_signature(filter_node)
@@ -174,4 +179,10 @@ class RubyLsp::Ree::ParsedDocument
     name_parts = [class_node.constant_path&.parent&.name, class_node.constant_path.name]
     name_parts.compact.map(&:to_s).join('::')
   end
+
+  def node_name(node)
+    return nil unless node.respond_to?(:name)
+
+    node.name
+  end
 end

data/lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_document_builder.rb

@@ -1,5 +1,6 @@
 require 'prism'
 require_relative 'parsed_document'
+require_relative 'parsed_rspec_document'
 
 class RubyLsp::Ree::ParsedDocumentBuilder
   extend RubyLsp::Ree::ReeLspUtils
@@ -7,6 +8,7 @@ class RubyLsp::Ree::ParsedDocumentBuilder
   def self.build_from_uri(uri, type = nil)
     ast = Prism.parse_file(uri.path).value
     document = build_document(ast, type)
+    return unless document
 
     document.set_package_name(package_name_from_uri(uri))
 
@@ -15,6 +17,7 @@ class RubyLsp::Ree::ParsedDocumentBuilder
 
   def self.build_from_ast(ast, uri, type = nil)
     document = build_document(ast, type)
+    return unless document
 
     document.set_package_name(package_name_from_uri(uri))
 
@@ -35,10 +38,37 @@ class RubyLsp::Ree::ParsedDocumentBuilder
     when :bean
       build_bean_document(ast)
     else
+      build_detected_doument_type(ast)
+    end
+  end
+
+  def self.build_detected_doument_type(ast)
+    if has_root_class?(ast)
       build_regular_document(ast)
+    elsif has_root_rspec_call?(ast)
+      build_rspec_document(ast)
+    else 
+      nil
     end
   end
 
+  def self.has_root_class?(ast)
+    ast.statements.body.detect{ |node| node.is_a?(Prism::ClassNode) }
+  end
+
+  def self.has_root_rspec_call?(ast)
+    ast.statements.body.detect{ |node| node.is_a?(Prism::CallNode) && node.name == :describe }
+  end
+
+  def self.build_rspec_document(ast)
+    document = RubyLsp::Ree::ParsedRspecDocument.new(ast)
+
+    document.parse_describe_node    
+    document.parse_links
+    
+    document
+  end
+
   def self.build_regular_document(ast)
     document = RubyLsp::Ree::ParsedDocument.new(ast)
     

data/lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_rspec_document.rb

@@ -0,0 +1,57 @@
+require_relative 'parsed_link_node'
+require 'ostruct'
+
+class RubyLsp::Ree::ParsedRspecDocument
+  include RubyLsp::Ree::ReeLspUtils
+
+  attr_reader :ast, :package_name, :describe_node
+
+  def initialize(ast)
+    @ast = ast
+  end
+
+  def set_package_name(package_name)
+    @package_name = package_name
+  end
+
+  def allows_root_links?
+    true
+  end
+
+  def includes_linked_object?(obj_name)
+    @link_nodes.map(&:name).include?(obj_name)
+  end
+
+  def links_container_node
+    nil
+  end
+
+  def root_node_line_location
+    OpenStruct.new(
+      start_line: @describe_node.location.start_line,
+      end_column: @describe_node.block.opening_loc.end_column
+    )
+  end
+
+  def has_blank_links_container?
+    false
+  end
+
+  def includes_link_dsl?
+    false
+  end
+
+  def parse_describe_node
+    @describe_node ||= @ast.statements.body.detect{ |node| node.is_a?(Prism::CallNode) && node.name == :describe }
+  end
+
+  def parse_links
+    nodes = @describe_node.block.body.body.select{ |node| node.name == :link }
+
+    @link_nodes = nodes.map do |link_node|
+      link_node = RubyLsp::Ree::ParsedLinkNode.new(link_node, package_name)
+      link_node.parse_imports
+      link_node
+    end
+  end
+end

data/lib/ruby_lsp/ruby_lsp_ree/ree_formatter.rb

@@ -10,6 +10,8 @@ module RubyLsp
       def run_formatting(uri, document)
         source = document.source
         sort_links(source)
+      rescue => e
+        $stderr.puts("error in ree_formatter: #{e.message} : #{e.backtrace.first}")
       end
 
       private

data/lib/ruby_lsp/ruby_lsp_ree/ree_indexing_enhancement.rb

@@ -33,8 +33,8 @@ module RubyLsp
       private
 
       def parse_signatures(fn_name)
-        uri = @listener.instance_variable_get(:@uri)
-        ast = Prism.parse_file(uri.path).value
+        source = @listener.instance_variable_get(:@source_lines).join
+        ast = Prism.parse(source).value
 
         class_node = ast.statements.body.detect{ |node| node.is_a?(Prism::ClassNode) }
         return [] unless class_node

data/lib/ruby_lsp/ruby_lsp_ree/utils/ree_lsp_utils.rb

@@ -49,6 +49,11 @@ module RubyLsp
         elsif parsed_doc.includes_link_dsl?
           fn_line = parsed_doc.link_nodes.first.location.start_line - 1
           position = parsed_doc.link_nodes.first.location.start_column
+        elsif parsed_doc.allows_root_links?
+          root_node_location = parsed_doc.root_node_line_location
+         
+          fn_line = root_node_location.start_line
+          position = root_node_location.end_column + 1
         else
           return nil
         end

data/lib/ruby_lsp_ree/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLsp
   module Ree
-    VERSION = "0.1.3"
+    VERSION = "0.1.4"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-lsp-ree
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Ruslan Gatiyatov
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2025-02-21 00:00:00.000000000 Z
+date: 2025-02-25 00:00:00.000000000 Z
 dependencies: []
 description: A Ruby LSP addon that adds extra editor functionality for Ree applications
 email:
@@ -27,12 +27,15 @@ files:
 - Rakefile
 - lib/ruby_lsp/ruby_lsp_ree/addon.rb
 - lib/ruby_lsp/ruby_lsp_ree/handlers/completion_handler.rb
+- lib/ruby_lsp/ruby_lsp_ree/handlers/definition_handler.rb
+- lib/ruby_lsp/ruby_lsp_ree/handlers/hover_handler.rb
 - lib/ruby_lsp/ruby_lsp_ree/listeners/completion_listener.rb
 - lib/ruby_lsp/ruby_lsp_ree/listeners/definition_listener.rb
 - lib/ruby_lsp/ruby_lsp_ree/listeners/hover_listener.rb
 - lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_document.rb
 - lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_document_builder.rb
 - lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_link_node.rb
+- lib/ruby_lsp/ruby_lsp_ree/parsing/parsed_rspec_document.rb
 - lib/ruby_lsp/ruby_lsp_ree/ree_formatter.rb
 - lib/ruby_lsp/ruby_lsp_ree/ree_indexing_enhancement.rb
 - lib/ruby_lsp/ruby_lsp_ree/ree_object_finder.rb