JezK

Edit File: locator.rb

module RBS
  class Locator
    attr_reader :decls

def initialize(decls:)
      @decls = decls
    end

def buffer
      decls[0].location&.buffer or raise
    end

def find(line:, column:)
      pos = buffer.loc_to_pos([line, column])

decls.each do |decl|
        array = []
        find_in_decl(pos, decl: decl, array: array) and return array
      end

[]
    end

def find2(line:, column:)
      path = find(line: line, column: column)

return if path.empty?

hd, *tl = path
      if hd.is_a?(Symbol)
        [hd, tl]
      else
        [nil, path]
      end
    end

def find_in_decl(pos, decl:, array:)
      if test_loc(pos, location: decl.location)
        array.unshift(decl)

case decl
        when AST::Declarations::Class
          decl.type_params.each do |param|
            if test_loc(pos, location: param.location)
              array.unshift(param)
              find_in_loc(pos, array: array, location: param.location)
              return true
            end
          end

if super_class = decl.super_class
            if test_loc(pos, location: super_class.location)
              array.unshift(super_class)
              find_in_loc(pos, array: array, location: super_class.location)
              return true
            end
          end

decl.each_decl do |decl_|
            find_in_decl(pos, decl: decl_, array: array) and return true
          end

decl.each_member do |member|
            find_in_member(pos, array: array, member: member) and return true
          end

when AST::Declarations::Module
          decl.type_params.each do |param|
            if test_loc(pos, location: param.location)
              array.unshift(param)
              find_in_loc(pos, array: array, location: param.location)
              return true
            end
          end

decl.self_types.each do |self_type|
            if test_loc(pos, location: self_type.location)
              array.unshift(self_type)
              find_in_loc(pos, array: array, location: self_type.location)
              return true
            end
          end

decl.each_decl do |decl_|
            find_in_decl(pos, decl: decl_, array: array) and return true
          end

decl.each_member do |member|
            find_in_member(pos, array: array, member: member) and return true
          end

when AST::Declarations::Interface
          decl.type_params.each do |param|
            if test_loc(pos, location: param.location)
              array.unshift(param)
              find_in_loc(pos, array: array, location: param.location)
              return true
            end
          end

decl.members.each do |member|
            find_in_member(pos, array: array, member: member) and return true
          end

when AST::Declarations::Constant, AST::Declarations::Global
          find_in_type(pos, array: array, type: decl.type) and return true

when AST::Declarations::Alias
          find_in_type(pos, array: array, type: decl.type) and return true
        end

find_in_loc(pos, location: decl.location, array: array)