ruby-lsp-ree 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8eb134e8d6b0ac3bf5321789ed6617ea1347ee096b1e642f953885b398524030
-  data.tar.gz: 460f2cc171155affbafe236535604b116d0af15bbcf14c3925470a8ff3616211
+  metadata.gz: 9c8f904d1cf98e1106296f608f265999c5c606fc83473d3e65a05395dd83ace5
+  data.tar.gz: daba84c44a1bf697f80ec7589ab0a9a436af14aa5d9d8cc6cb12377623de5e8f
 SHA512:
-  metadata.gz: cf49632399253b97f3b3df77ae7767008141900552461de8bca6b7395ac0d48fa4dbc5a353d2245f50147d7d119059b3f4d5a3f55aa41ae4fd11521d58b7930d
-  data.tar.gz: 0f392c77bbe2bdee9e9f613bc856c1d93b9bb9a2c87a93078017388fb253b27325d244ba289b36c24ff7e71e9f72b4f9c11abb663a7a2f675b5a9661aae29554
+  metadata.gz: 318c26fe57dcc12ac7a2209c10abd65b5662034e00eb87fea1a2ea58107ccd955db3b1e22a40c80f0a92e1ff10fef0f27c5a1ed3964b7877daa15dd5c64c61d4
+  data.tar.gz: 13c136bd68c1aa9f505a79b94d6f30b713e4e7f24b22a2157c5ccdadc1300d57e27fd7fb4bc5d75f55e7a441125dccf10d6923048948ec416616d7dd723936b1

data/CHANGELOG.md

@@ -1,3 +1,14 @@
+## [0.1.5] - 2025-02-28
+
+- improved hover format
+- basic support for ree templates
+
+## [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.5)
 
 GEM
   remote: https://rubygems.org/

data/lib/ruby_lsp/ruby_lsp_ree/addon.rb

@@ -5,6 +5,7 @@ require_relative "listeners/hover_listener"
 require_relative "ree_indexing_enhancement"
 require_relative "utils/ree_lsp_utils"
 require_relative "ree_formatter"
+require_relative "ree_template_applicator"
 require_relative "parsing/parsed_document_builder"
 
 module RubyLsp
@@ -13,8 +14,10 @@ module RubyLsp
       def activate(global_state, message_queue)
         @global_state = global_state
         @message_queue = message_queue
+        @template_applicator = RubyLsp::Ree::ReeTemplateApplicator.new
 
         global_state.register_formatter("ree_formatter", RubyLsp::Ree::ReeFormatter.new)
+        register_additional_file_watchers(global_state, message_queue)
       end
 
       def deactivate
@@ -38,6 +41,42 @@ module RubyLsp
         index = @global_state.index
         RubyLsp::Ree::HoverListener.new(response_builder, node_context, index, dispatcher)
       end
+
+      def register_additional_file_watchers(global_state, message_queue)
+        # Clients are not required to implement this capability
+        return unless global_state.supports_watching_files
+        
+        return unless @template_applicator.template_dir_exists?
+
+        message_queue << Request.new(
+          id: "ruby-lsp-ree-file-create-watcher",
+          method: "client/registerCapability",
+          params: Interface::RegistrationParams.new(
+            registrations: [
+              Interface::Registration.new(
+                id: "workspace/didCreateWatchedFilesRee",
+                method: "workspace/didChangeWatchedFiles",
+                register_options: Interface::DidChangeWatchedFilesRegistrationOptions.new(
+                  watchers: [
+                    Interface::FileSystemWatcher.new(
+                      glob_pattern: "**/*.rb",
+                      kind: Constant::WatchKind::CREATE,
+                    ),
+                  ],
+                ),
+              ),
+            ],
+          ),
+        )
+      end
+
+      def workspace_did_change_watched_files(changes)
+        $stderr.puts("workspace_did_change_watched_files #{changes.inspect}")
+
+        changes.each do |change_item|
+          @template_applicator.apply(change_item)
+        end
+      end
     end
   end
 end

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
+        \`\`\`ruby
+        #{node.name.to_s}#{get_detail_string(ree_object)}
+        \`\`\`
+        ---
+        ree type: :#{@finder.object_type(ree_object)} package: #{package_name_from_uri(ree_object.uri)}
+
+        [#{path_from_package_folder(ree_object.uri)}](#{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/ree_object_finder.rb

@@ -71,21 +71,10 @@ module RubyLsp
       end
 
       def object_type(ree_object)
-        # TODO rewrite to use string split
-        case ree_object.comments.lines[1]&.chomp
-        when DAO_TYPE_STRING
-          :dao
-        when BEAN_TYPE_STRING
-          :bean
-        when ENUM_TYPE_STRING
-          :enum
-        when MAPPER_TYPE_STRING
-          :mapper
-        when AGGREGATE_TYPE_STRING
-          :aggregate
-        else
-          nil
-        end
+        type_str = ree_object.comments.lines[1]&.chomp
+        return unless type_str
+          
+        type_str.split(' ').last[1..-1].to_sym
       end
     end
   end

data/lib/ruby_lsp/ruby_lsp_ree/ree_template_applicator.rb

@@ -0,0 +1,100 @@
+module RubyLsp
+  module Ree
+    class ReeTemplateApplicator
+      include RubyLsp::Ree::ReeLspUtils
+
+      TEMPLATES_FOLDER = '.vscode-ree/templates'
+      DEFAULT_TEMPLATE_FILENAME = 'default.rb'
+      RSPEC_TEMPLATE_PATH = 'spec_template.rb'
+
+      def initialize
+        return unless template_dir_exists?
+
+        @template_types = Dir
+          .entries(TEMPLATES_FOLDER)
+          .select{ |entry| 
+            File.directory? File.join(TEMPLATES_FOLDER,entry) and !(entry =='.' || entry == '..')
+          }
+      end
+
+      def template_dir_exists?
+        File.exist?(TEMPLATES_FOLDER)
+      end
+
+      def apply(change_item)
+        uri = change_item[:uri]
+        path = URI.parse(uri).path
+
+        file_content = File.read(path)
+        return if file_content.size > 0
+
+        if path.end_with?('_spec.rb')
+          template_str = fetch_rspec_template
+          template_info = fetch_rspec_template_info(uri)
+        else
+          template_type = get_template_type_from_uri(uri)
+          return unless template_type
+  
+          template_str = fetch_template(template_type)
+          template_info = fetch_template_info(uri)
+        end
+        
+        template_content = replace_placeholders(template_str, template_info)
+
+        File.write(path, template_content)
+      end
+
+      def get_template_type_from_uri(uri)
+        uri_parts = File.dirname(uri).split('/')
+
+        uri_parts.reverse.detect{ @template_types.include?(_1) }
+      end
+
+      def fetch_template(template_type)
+        File.read(File.join(TEMPLATES_FOLDER, template_type, DEFAULT_TEMPLATE_FILENAME))
+      end
+
+      def fetch_rspec_template
+        File.read(File.join(TEMPLATES_FOLDER, RSPEC_TEMPLATE_PATH))
+      end
+
+      def fetch_template_info(uri)
+        object_name = File.basename(uri, '.rb')
+        object_class = object_name.split('_').collect(&:capitalize).join
+        package_name = package_name_from_uri(uri)
+        package_class = package_name.split('_').collect(&:capitalize).join
+        
+        {
+          'PACKAGE_MODULE' => package_class,
+          'PACKAGE_NAME' => package_name,
+          'OBJECT_CLASS' => object_class,
+          'OBJECT_NAME' => object_name,
+        }
+      end
+
+      def fetch_rspec_template_info(uri)
+        object_name = File.basename(uri, '.rb')
+        object_class = object_name.split('_').collect(&:capitalize).join
+        package_name = package_name_from_spec_uri(uri)
+        package_class = package_name.split('_').collect(&:capitalize).join
+        file_path = spec_relative_file_path_from_uri(URI(uri))&.delete_suffix('_spec')
+        
+        {
+          'RELATIVE_FILE_PATH' => file_path,
+          'MODULE_NAME' => package_class,
+          'CLASS_NAME' => object_class,
+          'OBJECT_NAME' => object_name, 
+          'PACKAGE_NAME' => package_name,
+        }
+      end
+      
+      def replace_placeholders(template_str, template_info)
+        template_info.each do |k,v|
+          template_str.gsub!(k, v)
+        end
+
+        template_str
+      end
+    end
+  end
+end

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

@@ -17,8 +17,17 @@ module RubyLsp
         uri_parts[package_folder_index + 1]
       end
 
+      def package_name_from_spec_uri(uri)
+        uri_parts = uri.to_s.split('/')
+        
+        spec_folder_index = uri_parts.find_index('spec')
+        return unless spec_folder_index
+
+        uri_parts[spec_folder_index + 1]
+      end
+
       def path_from_package_folder(uri)
-        uri_parts = uri.to_s.chomp(File.extname(uri)).split('/')
+        uri_parts = uri.to_s.chomp(File.extname(uri.to_s)).split('/')
 
         package_folder_index = uri_parts.index('package')
         return unless package_folder_index
@@ -26,6 +35,14 @@ module RubyLsp
         uri_parts.drop(package_folder_index+1).join('/')
       end
 
+      def spec_relative_file_path_from_uri(uri)
+        uri_parts = uri.path.split('/')
+        spec_folder_index = uri_parts.index('spec')
+        return unless spec_folder_index
+
+        uri_parts[spec_folder_index+1..-1].join('/').chomp('.rb')
+      end
+
       def get_ree_type(ree_object)
         type_comment = ree_object.comments.to_s.lines[1]
         return unless type_comment
@@ -49,6 +66,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.5"
   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.5
 platform: ruby
 authors:
 - Ruslan Gatiyatov
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2025-02-21 00:00:00.000000000 Z
+date: 2025-02-28 00:00:00.000000000 Z
 dependencies: []
 description: A Ruby LSP addon that adds extra editor functionality for Ree applications
 email:
@@ -27,15 +27,19 @@ 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
+- lib/ruby_lsp/ruby_lsp_ree/ree_template_applicator.rb
 - lib/ruby_lsp/ruby_lsp_ree/utils/ree_lsp_utils.rb
 - lib/ruby_lsp_ree.rb
 - lib/ruby_lsp_ree/version.rb