solargraph 0.21.0 → 0.22.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 04dd49e264774d545fbdf1a7f52d0b7227ff3f11
-  data.tar.gz: cb617a8800b2eec8688c46fc89cf4ee79445dd50
+SHA256:
+  metadata.gz: c647321bb57cb5ba859e1762032855324aa967e78a562c0e259508e5c24c22ee
+  data.tar.gz: 057c1a4298f9c4574c0773425cce496216ab60e22165871487d6d77e13733745
 SHA512:
-  metadata.gz: db20a5e3053060171bf9077c1ce572de8428193b4d562c7d939c9a7003074de64f41ffe879100b7f7a31bb42f625a32e69ad7755afc68b8c31d2cc64d15be353
-  data.tar.gz: bb9a311819e91501145c05fc2045178a791933fb6f9598e6265f68efe301664cbb6bfa50170928989d1fd8543a726cd36e0079b2f650c2ded36f9ad3a66c92c6
+  metadata.gz: ae8cb69f92e2f3ad78959bd8767b78dda5a7d4baa861fa59680be191b7c92944f0df15ff12ccad2ed4a93231afcc55db53bd40cf9b42473387491f4d168f1cef
+  data.tar.gz: 8d02f2a10b1a71579a0b48e9dae4268689cc0ec642c137b1adedf6fc99749ae5a1b04f46b4d9b30448d3316a71b65dfdf79b5066c0af499e2181fcc5a107201f

data/lib/solargraph/api_map/completion.rb

@@ -1,5 +1,8 @@
 module Solargraph
   class ApiMap
+    # The result of a completion request containing the pins that describe
+    # completion options and the range to be replaced.
+    #
     class Completion
       # @return [Array<Solargraph::Pin::Base>]
       attr_reader :pins

data/lib/solargraph/api_map/probe.rb

@@ -1,12 +1,7 @@
 module Solargraph
   class ApiMap
     class Probe
-      class VirtualPin
-        attr_reader :return_type
-        def initialize return_type
-          @return_type = return_type
-        end
-      end
+      include TypeMethods
 
       # @return [Solargraph::ApiMap]
       attr_reader :api_map
@@ -24,8 +19,8 @@ module Solargraph
         return infer_word_pins(base, context_pin, locals) if rest.nil?
         pins = infer_word_pins(base, context_pin, locals).map do |pin|
           next pin unless pin.return_type.nil?
-          type = resolve_pin_type(pin)
-          VirtualPin.new(type)
+          type = resolve_pin_type(pin, locals - [pin])
+          Pin::ProxyMethod.new(type)
         end
         return [] if pins.empty?
         rest = rest.split('.')
@@ -53,7 +48,7 @@ module Solargraph
       def infer_signature_type signature, context_pin, locals
         pins = infer_signature_pins(signature, context_pin, locals)
         pins.each do |pin|
-          type = resolve_pin_type(pin)
+          type = resolve_pin_type(pin, locals - [pin])
           return qualify(type, pin.named_context) unless type.nil?
         end
         nil
@@ -70,15 +65,25 @@ module Solargraph
         namespace, scope = extract_namespace_and_scope_from_pin(context_pin)
         return api_map.pins.select{|pin| word_matches_context?(word, namespace, scope, pin)} if variable_name?(word)
         result = []
-        result.concat api_map.get_path_suggestions(word)
+        cparts = namespace.split('::')
+        until cparts.empty?
+          break unless result.empty?
+          cur = cparts.join('::')
+          result.concat api_map.get_path_suggestions("#{cur}::#{word}")
+          cparts.pop
+        end
+        result.concat api_map.get_path_suggestions(word) if result.empty?
         result.concat api_map.get_methods(namespace, scope: scope, visibility: [:public, :private, :protected]).select{|pin| pin.name == word} unless word.include?('::')
-        result.concat api_map.get_constants('', namespace).select{|pin| pin.name == word}
+        result.concat api_map.get_constants('', namespace).select{|pin| pin.name == word} if result.empty?
         result
       end
 
       # Method name search is external by default
+      # @param method_name [String]
+      # @param context_pin [Solargraph::Pin::Base]
       def infer_method_name_pins method_name, context_pin, internal = false
-        namespace, scope = extract_namespace_and_scope(context_pin.return_type)
+        relname, scope = extract_namespace_and_scope(context_pin.return_type)
+        namespace = api_map.qualify(relname, context_pin.namespace)
         visibility = [:public]
         visibility.push :protected, :private if internal
         result = api_map.get_methods(namespace, scope: scope, visibility: visibility).select{|pin| pin.name == method_name}
@@ -110,35 +115,6 @@ module Solargraph
         type.sub(/<#{rns}>/, "<#{res}>")
       end
 
-      # Extract a namespace from a type.
-      #
-      # @example
-      #   extract_namespace('String') => 'String'
-      #   extract_namespace('Class<String>') => 'String'
-      #
-      # @return [String]
-      def extract_namespace type
-        extract_namespace_and_scope(type)[0]
-      end
-
-      # Extract a namespace and a scope (:instance or :class) from a type.
-      #
-      # @example
-      #   extract_namespace('String')            #=> ['String', :instance]
-      #   extract_namespace('Class<String>')     #=> ['String', :class]
-      #   extract_namespace('Module<Enumerable') #=> ['Enumberable', :class]
-      #
-      # @return [Array] The namespace (String) and scope (Symbol).
-      def extract_namespace_and_scope type
-        scope = :instance
-        result = type.to_s.gsub(/<.*$/, '')
-        if (result == 'Class' or result == 'Module') and type.include?('<')
-          result = type.match(/<([a-z0-9:_]*)/i)[1]
-          scope = :class
-        end
-        [result, scope]
-      end
-
       # Extract a namespace and a scope from a pin. For now, the pin must
       # be either a namespace, a method, or a block.
       #
@@ -172,11 +148,11 @@ module Solargraph
         pin
       end
 
-      def resolve_pin_type pin
+      def resolve_pin_type pin, locals
         pin.return_type
         return pin.return_type unless pin.return_type.nil?
         return resolve_block_parameter(pin) if pin.kind == Pin::BLOCK_PARAMETER
-        return resolve_variable(pin) if pin.variable?
+        return resolve_variable(pin, locals) if pin.variable?
         nil
       end
 
@@ -205,10 +181,10 @@ module Solargraph
         nil
       end
 
-      def resolve_variable(pin)
+      def resolve_variable(pin, locals)
         return nil if pin.nil_assignment?
         # @todo Do we need the locals here?
-        return infer_signature_type(pin.signature, pin.context, [])
+        return infer_signature_type(pin.signature, pin.context, locals)
       end
 
       def get_subtypes type

data/lib/solargraph/api_map/store.rb

@@ -43,8 +43,8 @@ module Solargraph
         }
       end
 
-      def get_attrs fqns
-        namespace_pins(fqns).select{ |pin| pin.kind == Pin::ATTRIBUTE }
+      def get_attrs fqns, scope
+        namespace_pins(fqns).select{ |pin| pin.kind == Pin::ATTRIBUTE and pin.scope == scope }
       end
 
       def get_superclass fqns
@@ -80,6 +80,10 @@ module Solargraph
         namespace_pins(fqns).select{|pin| pin.kind == Pin::INSTANCE_VARIABLE and pin.scope == scope}
       end
 
+      def get_class_variables(fqns)
+        namespace_pins(fqns).select{|pin| pin.kind == Pin::CLASS_VARIABLE}
+      end
+
       def get_symbols
         symbols.uniq(&:name)
       end

data/lib/solargraph/api_map/type_methods.rb

@@ -0,0 +1,40 @@
+module Solargraph
+  class ApiMap
+    # Module functions for processing YARD types.
+    #
+    module TypeMethods
+      module_function
+
+      # Extract a namespace from a type.
+      #
+      # @example
+      #   extract_namespace('String') => 'String'
+      #   extract_namespace('Class<String>') => 'String'
+      #
+      # @param type [String]
+      # @return [String]
+      def extract_namespace type
+        extract_namespace_and_scope(type)[0]
+      end
+
+      # Extract a namespace and a scope (:instance or :class) from a type.
+      #
+      # @example
+      #   extract_namespace('String')            #=> ['String', :instance]
+      #   extract_namespace('Class<String>')     #=> ['String', :class]
+      #   extract_namespace('Module<Enumerable') #=> ['Enumberable', :class]
+      #
+      # @param type [String]
+      # @return [Array] The namespace (String) and scope (Symbol).
+      def extract_namespace_and_scope type
+        scope = :instance
+        result = type.to_s.gsub(/<.*$/, '')
+        if (result == 'Class' or result == 'Module') and type.include?('<')
+          result = type.match(/<([a-z0-9:_]*)/i)[1]
+          scope = :class
+        end
+        [result, scope]
+      end
+    end
+  end
+end

data/lib/solargraph/api_map.rb

@@ -9,15 +9,25 @@ module Solargraph
     autoload :Completion,   'solargraph/api_map/completion'
     autoload :Probe,        'solargraph/api_map/probe'
     autoload :Store,        'solargraph/api_map/store'
+    autoload :TypeMethods,  'solargraph/api_map/type_methods'
 
     include Solargraph::ApiMap::SourceToYard
     include CoreFills
+    include TypeMethods
 
     # The workspace to analyze and process.
     #
     # @return [Solargraph::Workspace]
     attr_reader :workspace
 
+    # @return [ApiMap::Store]
+    attr_reader :store
+
+    # Get a LiveMap associated with the current workspace.
+    #
+    # @return [Solargraph::LiveMap]
+    attr_reader :live_map
+
     # @param workspace [Solargraph::Workspace]
     def initialize workspace = Solargraph::Workspace.new(nil)
       @workspace = workspace
@@ -26,7 +36,7 @@ module Solargraph
       @yard_stale = true
       # process_maps
       @sources = workspace.sources
-      yard_map
+      refresh_store_and_maps
     end
 
     # Create an ApiMap with a workspace in the specified directory.
@@ -36,11 +46,6 @@ module Solargraph
       self.new(Solargraph::Workspace.new(directory))
     end
 
-    # @return [ApiMap::Store]
-    def store
-      @store ||= ApiMap::Store.new(@sources)
-    end
-
     def pins
       store.pins
     end
@@ -66,19 +71,12 @@ module Solargraph
     # @return [Solargraph::YardMap]
     def yard_map
       # refresh
-      if @yard_map.nil? || @yard_map.required.to_set != required.to_set
+      if @yard_map.required.to_set != required.to_set
         @yard_map = Solargraph::YardMap.new(required: required, workspace: workspace)
       end
       @yard_map
     end
 
-    # Get a LiveMap associated with the current workspace.
-    #
-    # @return [Solargraph::LiveMap]
-    def live_map
-      @live_map ||= Solargraph::LiveMap.new(self)
-    end
-
     # Declare a virtual source that will be included in the map regardless of
     # whether it's in the workspace.
     #
@@ -229,7 +227,7 @@ module Solargraph
     # @param namespace [String] A fully qualified namespace
     # @return [Array<Solargraph::Pin::ClassVariable>]
     def get_class_variable_pins(namespace)
-      prefer_non_nil_variables(@cvar_pins[namespace] || [])
+      prefer_non_nil_variables(store.get_class_variables(namespace))
     end
 
     # @return [Array<Solargraph::Pin::Base>]
@@ -260,6 +258,11 @@ module Solargraph
       else
         result.concat inner_get_methods(fqns, scope, visibility, deep, skip)
       end
+      live = live_map.get_methods(fqns, '', scope.to_s, visibility.include?(:private))
+      unless live.empty?
+        exist = result.map(&:name)
+        result.concat live.reject{|p| exist.include?(p.name)}
+      end
       result
     end
 
@@ -293,7 +296,7 @@ module Solargraph
             if fragment.base.empty?
               result.concat get_methods(pin.path)
             else
-              type = probe.infer_signature_type(fragment.base, pin, fragment.locals)
+              type = probe.infer_signature_type("#{pin.path}.new.#{fragment.base}", pin, fragment.locals)
               unless type.nil?
                 namespace, scope = extract_namespace_and_scope(type)
                 result.concat get_methods(namespace, scope: scope)
@@ -310,7 +313,8 @@ module Solargraph
           end
         end
       end
-      filtered = result.uniq(&:identifier).select{|s| s.kind != Pin::METHOD or s.name.match(/^[a-z0-9_]*(\!|\?|=)?$/i)}.sort_by.with_index{ |x, idx| [x.name, idx] }
+      frag_start = fragment.word.to_s.downcase
+      filtered = result.uniq(&:identifier).select{|s| s.name.downcase.start_with?(frag_start) and (s.kind != Pin::METHOD or s.name.match(/^[a-z0-9_]*(\!|\?|=)?$/i))}.sort_by.with_index{ |x, idx| [x.name, idx] }
       Completion.new(filtered, fragment.whole_word_range)
     end
 
@@ -318,7 +322,11 @@ module Solargraph
     # @return [Array<Solargraph::Pin::Base>]
     def define fragment
       return [] if fragment.string? or fragment.comment?
-      probe.infer_signature_pins fragment.whole_signature, fragment.named_path, fragment.locals
+      if fragment.base_literal?
+        probe.infer_signature_pins fragment.whole_signature, Pin::ProxyMethod.new(fragment.base_literal), fragment.locals
+      else
+        probe.infer_signature_pins fragment.whole_signature, fragment.named_path, fragment.locals
+      end
     end
 
     # Infer a return type from a fragment. This method will attempt to resolve
@@ -342,9 +350,13 @@ module Solargraph
     def signify fragment
       return [] unless fragment.argument?
       return [] if fragment.recipient.whole_signature.nil? or fragment.recipient.whole_signature.empty?
-      probe.infer_signature_pins(
-        fragment.recipient.whole_signature, fragment.named_path, fragment.locals
-      ).select{ |pin| pin.kind == Pin::METHOD }
+      result = []
+      if fragment.recipient.base_literal?
+        result.concat probe.infer_signature_pins(fragment.recipient.whole_signature, Pin::ProxyMethod.new(fragment.recipient.base_literal), fragment.locals)
+      else
+        result.concat probe.infer_signature_pins(fragment.recipient.whole_signature, fragment.named_path, fragment.locals)
+      end
+      result.select{ |pin| pin.kind == Pin::METHOD }
     end
 
     # Get an array of all suggestions that match the specified path.
@@ -356,6 +368,10 @@ module Solargraph
       result = []
       result.concat store.get_path_pins(path)
       result.concat yard_map.objects(path)
+      if result.empty?
+        lp = live_map.get_path_pin(path)
+        result.push lp unless lp.nil?
+      end
       result
     end
 
@@ -423,6 +439,12 @@ module Solargraph
 
     private
 
+    def refresh_store_and_maps
+      @store = ApiMap::Store.new(@sources)
+      @live_map = Solargraph::LiveMap.new(self)
+      @yard_map = Solargraph::YardMap.new(required: required, workspace: workspace)
+    end
+
     def process_virtual
       unless @virtual_source.nil?
         map_source @virtual_source
@@ -448,9 +470,7 @@ module Solargraph
       return [] if skip.include?(reqstr)
       skip.push reqstr
       result = []
-      if scope == :instance
-        result.concat store.get_attrs(fqns)
-      end
+      result.concat store.get_attrs(fqns, scope)
       result.concat store.get_methods(fqns, scope: scope, visibility: visibility)
       if deep
         sc = store.get_superclass(fqns)
@@ -475,7 +495,7 @@ module Solargraph
           result.concat yard_map.get_methods(fqns, '', visibility: visibility)
           type = get_namespace_type(fqns)
           result.concat inner_get_methods('Class', :instance, fqns == '' ? [:public] : visibility, deep, skip) if type == :class
-          result.concat inner_get_methods('Module', :instance, fqns == '' ? [:public] : visibility, deep, skip)
+          result.concat inner_get_methods('Module', :instance, fqns == '' ? [:public] : visibility, deep, skip) if type == :module
         end
       end
       result
@@ -568,15 +588,5 @@ module Solargraph
       return yard_map.get_namespace_type(fqns) if pin.nil?
       pin.type
     end
-
-    def extract_namespace_and_scope type
-      scope = :instance
-      result = type.to_s.gsub(/<.*$/, '')
-      if (result == 'Class' or result == 'Module') and type.include?('<')
-        result = type.match(/<([a-z0-9:_]*)/i)[1]
-        scope = :class
-      end
-      [result, scope]
-    end
   end
 end

data/lib/solargraph/diagnostics/require_not_found.rb

@@ -1,5 +1,10 @@
 module Solargraph
   module Diagnostics
+    # RequireNotFound reports required paths that could not be resolved to
+    # either a file in the workspace or a gem.
+    #
+    # @todo Some stdlib paths can result in false positives.
+    #
     class RequireNotFound < Base
       def diagnose source, api_map
         result = []

data/lib/solargraph/diagnostics/rubocop.rb

@@ -3,6 +3,8 @@ require 'shellwords'
 
 module Solargraph
   module Diagnostics
+    # This reporter provides linting through RuboCop.
+    #
     class Rubocop < Base
       # The rubocop command
       #
@@ -13,6 +15,8 @@ module Solargraph
         @command = command
       end
 
+      # @param source [Solargraph::Source]
+      # @param api_map [Solargraph::ApiMap]
       # @return [Array<Hash>]
       def diagnose source, api_map
         begin
@@ -20,7 +24,12 @@ module Solargraph
           filename = source.filename
           raise DiagnosticsError, 'No command specified' if command.nil? or command.empty?
           cmd = "#{Shellwords.escape(command)} -f j -s #{Shellwords.escape(filename)}"
+          unless api_map.workspace.nil? or api_map.workspace.directory.nil?
+            rc = File.join(api_map.workspace.directory, '.rubocop.yml')
+            cmd += " -c #{Shellwords.escape(rc)}" if File.file?(rc)
+          end
           o, e, s = Open3.capture3(cmd, stdin_data: text)
+          STDERR.puts e unless e.empty?
           raise DiagnosticsError, "Command '#{command}' is not available (gem exception)" if e.include?('Gem::Exception')
           raise DiagnosticsError, "RuboCop returned empty data" if o.empty?
           make_array JSON.parse(o)

data/lib/solargraph/diagnostics.rb

@@ -1,10 +1,15 @@
 module Solargraph
+  # The Diagnostics library provides reporters for analyzing problems in code
+  # and providing the results to language server clients.
+  #
   module Diagnostics
     autoload :Base, 'solargraph/diagnostics/base'
     autoload :Severities, 'solargraph/diagnostics/severities'
     autoload :Rubocop, 'solargraph/diagnostics/rubocop'
     autoload :RequireNotFound, 'solargraph/diagnostics/require_not_found'
 
+    # Reporters identified by name for activation in .solargraph.yml files.
+    #
     REPORTERS = {
       'rubocop' => Rubocop,
       'require_not_found' => RequireNotFound

data/lib/solargraph/language_server/host.rb

@@ -323,6 +323,14 @@ module Solargraph
         result
       end
 
+      def references_from filename, line, column
+        result = nil
+        @change_semaphore.synchronize do
+          result = library.references_from(filename, line, column)
+        end
+        result
+      end
+
       def query_symbols query
         result = nil
         @change_semaphore.synchronize { result = library.query_symbols(query) }
@@ -446,7 +454,7 @@ module Solargraph
         Thread.new do
           diagnoser = Diagnostics::Rubocop.new
           until stopped?
-            sleep 1
+            sleep 0.1
             if !options['diagnostics']
               @change_semaphore.synchronize { @diagnostics_queue.clear }
               next

data/lib/solargraph/language_server/message/base.rb

@@ -37,14 +37,7 @@ module Solargraph
           }
         end
 
-        def response
-          response = {}
-          response[:result] = result
-          response[:error] = error unless error.nil?
-          response
-        end
-
-        def send
+        def send_response
           unless id.nil? or host.cancel?(id)
             response = {
               jsonrpc: "2.0",

data/lib/solargraph/language_server/message/extended/check_gem_version.rb

@@ -11,7 +11,7 @@ module Solargraph
         class CheckGemVersion < Base
           def process
             o, s = Open3.capture2("gem search solargraph")
-            match = o.match(/solargraph \([0-9\.]*?\)/)
+            match = o.match(/solargraph \(([\d]*\.[\d]*\.[\d]*)\)/)
             # @todo Error if no match or status code != 0
             available = Gem::Version.new(match[1])
             current = Gem::Version.new(Solargraph::VERSION)
@@ -30,6 +30,10 @@ module Solargraph
             elsif params['verbose']
               host.show_message "The Solargraph gem is up to date (version #{Solargraph::VERSION})."
             end
+            set_result({
+              installed: current,
+              available: available
+            })
           end
         end
       end

data/lib/solargraph/language_server/message/extended/download_core.rb

@@ -0,0 +1,25 @@
+require 'open3'
+
+module Solargraph
+  module LanguageServer
+    module Message
+      module Extended
+        # Update core Ruby documentation.
+        #
+        class DownloadCore < Base
+          def process
+            cmd = "solargraph download-core"
+            o, s = Open3.capture2(cmd)
+            if s != 0
+              host.show_message "An error occurred while downloading documentation.", LanguageServer::MessageTypes::ERROR
+            else
+              ver = o.match(/[\d]*\.[\d]*\.[\d]*/)[0]
+              host.show_message "Downloaded documentation for Ruby #{ver}.", LanguageServer::MessageTypes::INFO
+              # @todo YardMap should be refreshed
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/solargraph/language_server/message/extended.rb

@@ -11,6 +11,7 @@ module Solargraph
         autoload :Search, 'solargraph/language_server/message/extended/search'
         autoload :CheckGemVersion, 'solargraph/language_server/message/extended/check_gem_version'
         autoload :DocumentGems, 'solargraph/language_server/message/extended/document_gems'
+        autoload :DownloadCore, 'solargraph/language_server/message/extended/download_core'
       end
     end
   end

data/lib/solargraph/language_server/message/initialize.rb

@@ -11,12 +11,14 @@ module Solargraph
               definitionProvider: true,
               documentSymbolProvider: true,
               workspaceSymbolProvider: true,
+              renameProvider: true,
               workspace: {
                 workspaceFolders: {
                   supported: true,
                   changeNotifications: true
                 }
-              }
+              },
+              referencesProvider: true
             }
           }
           result[:capabilities].merge! static_completion unless dynamic_completion?

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

@@ -11,7 +11,7 @@ module Solargraph::LanguageServer::Message::TextDocument
       line = params['position']['line']
       col = params['position']['character']
       suggestions = host.definitions_at(filename, line, col)
-      locations = suggestions.map do |pin|
+      locations = suggestions.reject{|pin| pin.location.nil?}.map do |pin|
         unless pin.location.nil?
           {
             uri: file_to_uri(pin.location.filename),

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/message/text_document.rb

@@ -2,19 +2,21 @@ module Solargraph
   module LanguageServer
     module Message
       module TextDocument
-        autoload :Base, 'solargraph/language_server/message/text_document/base'
-        autoload :Completion, 'solargraph/language_server/message/text_document/completion'
-        autoload :DidOpen, 'solargraph/language_server/message/text_document/did_open'
-        autoload :DidChange, 'solargraph/language_server/message/text_document/did_change'
-        autoload :DidClose, 'solargraph/language_server/message/text_document/did_close'
-        autoload :DidSave, 'solargraph/language_server/message/text_document/did_save'
-        autoload :Hover, 'solargraph/language_server/message/text_document/hover'
-        autoload :SignatureHelp, 'solargraph/language_server/message/text_document/signature_help'
+        autoload :Base,             'solargraph/language_server/message/text_document/base'
+        autoload :Completion,       'solargraph/language_server/message/text_document/completion'
+        autoload :DidOpen,          'solargraph/language_server/message/text_document/did_open'
+        autoload :DidChange,        'solargraph/language_server/message/text_document/did_change'
+        autoload :DidClose,         'solargraph/language_server/message/text_document/did_close'
+        autoload :DidSave,          'solargraph/language_server/message/text_document/did_save'
+        autoload :Hover,            'solargraph/language_server/message/text_document/hover'
+        autoload :SignatureHelp,    'solargraph/language_server/message/text_document/signature_help'
         autoload :DiagnosticsQueue, 'solargraph/language_server/message/text_document/diagnostics_queue'
         autoload :OnTypeFormatting, 'solargraph/language_server/message/text_document/on_type_formatting'
-        autoload :Definition, 'solargraph/language_server/message/text_document/definition'
-        autoload :DocumentSymbol, 'solargraph/language_server/message/text_document/document_symbol'
-        autoload :Formatting, 'solargraph/language_server/message/text_document/formatting'
+        autoload :Definition,       'solargraph/language_server/message/text_document/definition'
+        autoload :DocumentSymbol,   'solargraph/language_server/message/text_document/document_symbol'
+        autoload :Formatting,       'solargraph/language_server/message/text_document/formatting'
+        autoload :References,       'solargraph/language_server/message/text_document/references'
+        autoload :Rename,           'solargraph/language_server/message/text_document/rename'
       end
     end
   end