solargraph 0.21.1 → 0.22.0

data/lib/solargraph/language_server/message/text_document/hover.rb

@@ -38,6 +38,6 @@ module Solargraph::LanguageServer::Message::TextDocument
     def link_documentation path
       uri = "solargraph:/document?query=" + URI.encode(path)
       "[#{path}](#{uri})"
-    end  
+    end
   end
 end

data/lib/solargraph/language_server/message/text_document/references.rb

@@ -0,0 +1,14 @@
+module Solargraph::LanguageServer::Message::TextDocument
+  class References < Base
+    def process
+      locs = host.references_from(uri_to_file(params['textDocument']['uri']), params['position']['line'], params['position']['character'])
+      result = locs.map do |loc|
+        {
+          uri: file_to_uri(loc.filename),
+          range: loc.range.to_hash
+        }
+      end
+      set_result result
+    end
+  end
+end

data/lib/solargraph/language_server/message/text_document/rename.rb

@@ -0,0 +1,17 @@
+module Solargraph::LanguageServer::Message::TextDocument
+  class Rename < Base
+    def process
+      locs = host.references_from(uri_to_file(params['textDocument']['uri']), params['position']['line'], params['position']['character'])
+      changes = {}
+      locs.each do |loc|
+        uri = file_to_uri(loc.filename)
+        changes[uri] ||= []
+        changes[uri].push({
+          range: loc.range.to_hash,
+          newText: params['newName']
+        })
+      end
+      set_result changes: changes
+    end
+  end
+end

data/lib/solargraph/language_server/uri_helpers.rb

@@ -1,6 +1,8 @@
 module Solargraph
   module LanguageServer
     module UriHelpers
+      module_function
+
       # Convert a file URI to a path.
       #
       # @param uri [String]

data/lib/solargraph/library.rb

@@ -141,6 +141,30 @@ module Solargraph
       api_map.signify(fragment)
     end
 
+    def references_from filename, line, column
+      source = read(filename)
+      api_map.virtualize source
+      fragment = source.fragment_at(line, column)
+      pins = api_map.define(fragment)
+      return [] if pins.empty?
+      result = []
+      # @param pin [Solargraph::Pin::Base]
+      pins.uniq.each do |pin|
+        if pin.kind != Solargraph::Pin::NAMESPACE and !pin.location.nil?
+          mn_loc = get_symbol_name_location(pin)
+          result.push mn_loc unless mn_loc.nil?
+        end
+        (workspace.sources + source_hash.values).uniq(&:filename).each do |source|
+          found = source.references(pin.name).select do |loc|
+            referenced = definitions_at(loc.filename, loc.range.ending.line, loc.range.ending.character).first
+            !referenced.nil? and referenced.path == pin.path
+          end
+          result.concat found.sort{|a, b| a.range.start.line <=> b.range.start.line}
+        end
+      end
+      result
+    end
+
     # Get the pin at the specified location or nil if the pin does not exist.
     #
     # @return [Solargraph::Pin::Base]
@@ -270,5 +294,18 @@ module Solargraph
       raise FileNotFoundError, "File not found: #{filename}" unless File.file?(filename)
       Solargraph::Source.load(filename)
     end
+
+    def get_symbol_name_location pin
+      decsrc = read(pin.location.filename)
+      offset = Solargraph::Source::Position.to_offset(decsrc.code, pin.location.range.start)
+      soff = decsrc.code.index(pin.name, offset)
+      eoff = soff + pin.name.length
+      Solargraph::Source::Location.new(
+        pin.location.filename, Solargraph::Source::Range.new(
+          Solargraph::Source::Position.from_offset(decsrc.code, soff),
+          Solargraph::Source::Position.from_offset(decsrc.code, eoff)
+        )
+      )
+    end
   end
 end

data/lib/solargraph/live_map.rb

@@ -9,11 +9,17 @@ module Solargraph
     # @return [Solargraph::ApiMap]
     attr_reader :api_map
 
+    # @param api_map [Solargraph::ApiMap]
     def initialize api_map
       @api_map = api_map
       runners
     end
 
+    def get_path_pin path
+      cache.get_path_pin(path)
+    end
+
+    # @return [Array<Solargraph::Pin::Base>]
     def get_methods(namespace, root = '', scope = 'instance', with_private = false)
       fqns = api_map.find_fully_qualified_namespace(namespace, root)
       params = {
@@ -26,7 +32,7 @@ module Solargraph
       runners.each do |p|
         next if did_runtime and p.runtime?
         p.get_methods(namespace: namespace, root: root, scope: scope, with_private: with_private).each do |m|
-          result.push Suggestion.new(m['name'], kind: Suggestion::METHOD, docstring: YARD::Docstring.new('(defined at runtime)'), path: "#{fqns}.#{m['name']}", arguments: m['parameters'])
+          result.push Solargraph::Pin::Method.new(nil, namespace, m['name'], YARD::Docstring.new('(defined at runtime)'), scope.to_sym, nil, [])
         end
         did_runtime = true if p.runtime?
       end
@@ -34,7 +40,7 @@ module Solargraph
       result
     end
 
-    # @return [Array<Solargraph::Suggestion>]
+    # @return [Array<Solargraph::Pin::Base>]
     def get_constants(namespace, root = '')
       cached = cache.get_constants(namespace, root)
       return cached unless cached.nil?

data/lib/solargraph/live_map/cache.rb

@@ -4,6 +4,7 @@ module Solargraph
       def initialize
         @method_cache = {}
         @constant_cache = {}
+        @path_cache = {}
       end
 
       def get_methods options
@@ -12,6 +13,7 @@ module Solargraph
 
       def set_methods options, values
         @method_cache[options] = values
+        values.each { |pin| @path_cache[pin.path] = pin }
       end
 
       def get_constants namespace, root
@@ -22,9 +24,14 @@ module Solargraph
         @constant_cache[[namespace, root]] = values
       end
 
+      def get_path_pin path
+        @path_cache[path]
+      end
+
       def clear
         @method_cache.clear
         @constant_cache.clear
+        @path_cache.clear
       end
     end
   end

data/lib/solargraph/pin.rb

@@ -21,6 +21,7 @@ module Solargraph
     autoload :Documenting,      'solargraph/pin/documenting'
     autoload :Block,            'solargraph/pin/block'
     autoload :Localized,        'solargraph/pin/localized'
+    autoload :ProxyMethod,      'solargraph/pin/proxy_method'
 
     ATTRIBUTE = 1
     CLASS_VARIABLE = 2

data/lib/solargraph/pin/attribute.rb

@@ -4,10 +4,14 @@ module Solargraph
       # @return [Symbol] :reader or :writer
       attr_reader :access
 
-      def initialize location, namespace, name, docstring, access
+      # @return [Symbol] :class or :instance
+      attr_reader :scope
+
+      def initialize location, namespace, name, docstring, access, scope
         super(location, namespace, name, docstring)
         @access = access
         @docstring = docstring
+        @scope = scope
       end
 
       def kind

data/lib/solargraph/pin/namespace.rb

@@ -1,8 +1,6 @@
 module Solargraph
   module Pin
     class Namespace < Pin::Base
-      include Solargraph::NodeMethods
-
       attr_reader :visibility
 
       attr_reader :type

data/lib/solargraph/pin/proxy_method.rb

@@ -0,0 +1,31 @@
+module Solargraph
+  module Pin
+    # ProxyMethod serves as a quick, disposable shortcut for providing context
+    # to type inference methods. ApiMap::Probe can treat it as an anonymous
+    # method while analyzing signatures.
+    #
+    class ProxyMethod
+      # @return [String]
+      attr_reader :return_type
+
+      def initialize return_type
+        @return_type = return_type
+      end
+
+      # @return [String]
+      def namespace
+        @namespace ||= ApiMap::TypeMethods.extract_namespace(@return_type)
+      end
+
+      # @return [Integer]
+      def kind
+        Pin::METHOD
+      end
+
+      # @return [Symbol]
+      def scope
+        :instance
+      end
+    end
+  end
+end

data/lib/solargraph/plugin/process.rb

@@ -33,7 +33,7 @@ module Solargraph
           rescue JSON::ParserError => e
             STDOUT.puts respond_err "Error parsing input: #{e.message}"
           rescue Exception => e
-            STDOUT.puts respond_err "Error processing input: #{e.message}"
+            STDOUT.puts respond_err "Error processing input: #{e.message}\n#{e.backtrace}"
           end
           STDOUT.flush
         end

data/lib/solargraph/plugin/runtime.rb

@@ -61,7 +61,6 @@ module Solargraph
 
       def load_environment
         return if api_map.nil?
-        STDERR.puts "Required paths given to Runtime: #{api_map.required}"
         send_require api_map.required
         @current_required = api_map.required.clone
       end

data/lib/solargraph/shell.rb

@@ -69,7 +69,6 @@ module Solargraph
       matches = []
       if options[:extensions]
         Gem::Specification.each do |g|
-          puts g.name
           if g.name.match(/^solargraph\-[A-Za-z0-9_\-]*?\-ext/)
             require g.name
             matches.push g.name

data/lib/solargraph/source.rb

@@ -11,6 +11,7 @@ module Solargraph
     autoload :Updater,       'solargraph/source/updater'
     autoload :Change,        'solargraph/source/change'
     autoload :Mapper,        'solargraph/source/mapper'
+    autoload :NodeMethods,   'solargraph/source/node_methods'
 
     # @return [String]
     attr_reader :code
@@ -40,6 +41,7 @@ module Solargraph
     # @return [Time]
     attr_reader :stime
 
+    # @return [Array<Solargraph::Pin::Base>]
     attr_reader :pins
 
     attr_reader :requires
@@ -132,6 +134,22 @@ module Solargraph
       symbol_pins
     end
 
+    # @return [Array<Source::Location>]
+    def references name
+      inner_node_references(name, node).map do |n|
+        offset = Position.to_offset(code, get_node_start_position(n))
+        soff = code.index(name, offset)
+        eoff = soff + name.length
+        Location.new(
+          filename,
+          Solargraph::Source::Range.new(
+            Position.from_offset(code, soff),
+            Position.from_offset(code, eoff)
+          )
+        )
+      end
+    end
+
     def locate_named_path_pin line, character
       _locate_pin line, character, Pin::NAMESPACE, Pin::METHOD
     end
@@ -157,7 +175,8 @@ module Solargraph
 
     # Get the nearest node that contains the specified index.
     #
-    # @param index [Integer]
+    # @param line [Integer]
+    # @param column [Integer]
     # @return [AST::Node]
     def node_at(line, column)
       tree_at(line, column).first
@@ -246,6 +265,17 @@ module Solargraph
 
     private
 
+    def inner_node_references name, top
+      result = []
+      if top.kind_of?(AST::Node)
+        if (top.type == :const and top.children[1].to_s == name) or (top.type == :send and top.children[1].to_s == name)
+          result.push top
+        end
+        top.children.each { |c| result.concat inner_node_references(name, c) }
+      end
+      result
+    end
+
     def parse
       node, comments = inner_parse(@fixed, filename)
       @node = node

data/lib/solargraph/source/fragment.rb

@@ -3,12 +3,11 @@ module Solargraph
     class Fragment
       include NodeMethods
 
-      attr_reader :tree
-
       attr_reader :line
 
       attr_reader :column
 
+      # @return [Solargraph::Source]
       attr_reader :source
 
       # @param source [Solargraph::Source]
@@ -46,17 +45,6 @@ module Solargraph
         @argument ||= !signature_position.nil?
       end
 
-      def chained?
-        if @chained.nil?
-          @chained = false
-          @tree.each do |n|
-            @chained = true if n.type == :send
-            break
-          end
-        end
-        @chained
-      end
-
       # @return [Fragment]
       def recipient
         return nil if signature_position.nil?
@@ -319,9 +307,18 @@ module Solargraph
           @base_literal = 'Integer'
         # @todo Smelly exceptional case for array literals
         elsif signature.start_with?('.[]')
-          index += 3
-          signature = signature[index+3..-1].to_s
+          index += 2
+          signature = signature[3..-1].to_s
           @base_literal = 'Array'
+        elsif signature.start_with?('.')
+          pos = Position.from_offset(source.code, index)
+          node = source.node_at(pos.line, pos.character)
+          lit = infer_literal_node_type(node)
+          unless lit.nil?
+            signature = signature[1..-1].to_s
+            index += 1
+            @base_literal = lit
+          end
         end
         [index + 1, signature]
       end

data/lib/solargraph/source/location.rb

@@ -1,7 +1,10 @@
 module Solargraph
   class Source
     class Location
+      # @return [String]
       attr_reader :filename
+
+      # @return [Solargraph::Source::Range]
       attr_reader :range
 
       def initialize filename, range

data/lib/solargraph/source/mapper.rb

@@ -106,9 +106,17 @@ module Solargraph
                     u = c.updated(:ivasgn, c.children + ora.children[1..-1], nil)
                     @docstring_hash[u.loc] = docstring_for(ora)
                     pins.push Solargraph::Pin::InstanceVariable.new(get_node_location(u), fqn || '', c.children[0].to_s, docstring_for(u), resolve_node_signature(u.children[1]), infer_literal_node_type(u.children[1]), context)
+                    if visibility == :module_function and context.kind == Pin::METHOD
+                      other = pins.select{|pin| pin.path == "#{context.namespace}.#{context.name}"}.first
+                      pins.push Solargraph::Pin::InstanceVariable.new(get_node_location(u), fqn || '', c.children[0].to_s, docstring_for(u), resolve_node_signature(u.children[1]), infer_literal_node_type(u.children[1]), other) unless other.nil?
+                    end
                   end
                 else
                   pins.push Solargraph::Pin::InstanceVariable.new(get_node_location(c), fqn || '',c.children[0].to_s, docstring_for(c), resolve_node_signature(c.children[1]), infer_literal_node_type(c.children[1]), context)
+                  if visibility == :module_function and context.kind == Pin::METHOD
+                    other = pins.select{|pin| pin.path == "#{context.namespace}.#{context.name}"}.first
+                    pins.push Solargraph::Pin::InstanceVariable.new(get_node_location(c), fqn || '',c.children[0].to_s, docstring_for(c), resolve_node_signature(c.children[1]), infer_literal_node_type(c.children[1]), other)
+                  end
                 end
               elsif c.type == :cvasgn
                 here = get_node_start_position(c)
@@ -212,8 +220,14 @@ module Solargraph
                     ref = pins.select{|p| p.namespace == (fqn || '') and p.name == cn}.first
                     unless ref.nil?
                       pins.delete ref
-                      pins.push Solargraph::Pin::Method.new(ref.location, ref.namespace, ref.name, ref.docstring, :class, :public, ref.parameters)
-                      pins.push Solargraph::Pin::Method.new(ref.location, ref.namespace, ref.name, ref.docstring, :instance, :private, ref.parameters)
+                      mm = Solargraph::Pin::Method.new(ref.location, ref.namespace, ref.name, ref.docstring, :class, :public, ref.parameters)
+                      cm = Solargraph::Pin::Method.new(ref.location, ref.namespace, ref.name, ref.docstring, :instance, :private, ref.parameters)
+                      pins.push mm, cm
+                      pins.select{|pin| pin.kind == Pin::INSTANCE_VARIABLE and pin.context == ref}.each do |ivar|
+                        pins.delete ivar
+                        pins.push Solargraph::Pin::InstanceVariable.new(ivar.location, ivar.namespace, ivar.name, ivar.docstring, ivar.signature, ivar.instance_variable_get(:@literal), mm)
+                        pins.push Solargraph::Pin::InstanceVariable.new(ivar.location, ivar.namespace, ivar.name, ivar.docstring, ivar.signature, ivar.instance_variable_get(:@literal), cm)
+                      end
                     end
                   end
                 elsif c.children[2].type == :def
@@ -253,10 +267,10 @@ module Solargraph
               elsif c.type == :send and [:attr_reader, :attr_writer, :attr_accessor].include?(c.children[1])
                 c.children[2..-1].each do |a|
                   if c.children[1] == :attr_reader or c.children[1] == :attr_accessor
-                    pins.push Solargraph::Pin::Attribute.new(get_node_location(c), fqn || '', "#{a.children[0]}", docstring_for(c), :reader) #AttrPin.new(c)
+                    pins.push Solargraph::Pin::Attribute.new(get_node_location(c), fqn || '', "#{a.children[0]}", docstring_for(c), :reader, scope)
                   end
                   if c.children[1] == :attr_writer or c.children[1] == :attr_accessor
-                    pins.push Solargraph::Pin::Attribute.new(get_node_location(c), fqn || '', "#{a.children[0]}=", docstring_for(c), :writer) #AttrPin.new(c)
+                    pins.push Solargraph::Pin::Attribute.new(get_node_location(c), fqn || '', "#{a.children[0]}=", docstring_for(c), :writer, scope)
                   end
                 end
               elsif c.type == :sclass and c.children[0].type == :self
@@ -330,10 +344,6 @@ module Solargraph
         nil
       end
 
-      def get_node_start_position(node)
-        Position.new(node.loc.line - 1, node.loc.column)
-      end
-
       def get_node_location(node)
         if node.nil?
           st = Position.new(0, 0)
@@ -392,10 +402,10 @@ module Solargraph
               t = (d.tag.types.nil? || d.tag.types.empty?) ? nil : d.tag.types.flatten.join('')
               if t.nil? or t.include?('r')
                 # location, namespace, name, docstring, access
-                pins.push Solargraph::Pin::Attribute.new(get_node_location(k.node), namespace_for(k.node).path, d.tag.name, docstring, :reader)
+                pins.push Solargraph::Pin::Attribute.new(get_node_location(k.node), namespace_for(k.node).path, d.tag.name, docstring, :reader, :instance)
               end
               if t.nil? or t.include?('w')
-                pins.push Solargraph::Pin::Attribute.new(get_node_location(k.node), namespace_for(k.node).path, "#{d.tag.name}=", docstring, :reader)
+                pins.push Solargraph::Pin::Attribute.new(get_node_location(k.node), namespace_for(k.node).path, "#{d.tag.name}=", docstring, :writer, :instance)
               end
             elsif d.tag.tag_name == 'method'
               gen_src = Source.new("def #{d.tag.name};end", filename)