lib/yoda/model/node_signature.rb in yoda-language-server-0.6.1 vs lib/yoda/model/node_signature.rb in yoda-language-server-0.6.2

@@ -1,9 +1,14 @@
 module Yoda
   module Model
     class NodeSignature
-      attr_reader :node, :trace
+      # @return [::Parser::AST::Node]
+      attr_reader :node
+
+      # @return [Typing::Traces::Base]
+      attr_reader :trace
+
       # @param node  [::Parser::AST::Node]
       # @param trace [Typing::Traces::Base]
       def initialize(node, trace)
         @node = node
         @trace = trace
@@ -14,9 +19,19 @@
         Parsing::Range.of_ast_location(node.location)
       end
 
       # @return [Array<Descriptions::Base>]
       def descriptions
+        [top_description, *type_descriptions]
+      end
+
+      # @return [Descriptions::NodeDescription]
+      def top_description
+        Descriptions::NodeDescription.new(node, trace)
+      end
+
+      # @return [Array<Descriptions::Base>]
+      def type_descriptions
         case trace
         when Typing::Traces::Send
           trace.functions.map { |function| Descriptions::FunctionDescription.new(function) }.take(1)
         else
           trace.values.map { |value| Descriptions::ValueDescription.new(value) }