class Tapioca::Gem::Listeners::SourceLocation

def add_source_location_comment(node, file, line)
  return unless file && line
  path = Pathname.new(file)
  return unless File.exist?(path)
  # On native extensions, the source location may point to a shared object (.so, .bundle) file, which we cannot
  # use for jump to definition. Only add source comments on Ruby files
  return unless path.extname == ".rb"
  realpath = path.realpath.to_s
  gem = Gemfile::GemSpec.spec_lookup_by_file_path[realpath]
  return if gem.nil?
  path = gem.relative_path_for(path)
  version = gem.version
  # we can clear the gem version if the gem is the same one we are processing
  version = "" if gem == @pipeline.gem
  uri = URI::Source.build(
    gem_name: gem.name,
    gem_version: version,
    path: path.to_s,
    line_number: line.to_s,
  )
  node.comments << RBI::Comment.new("") if node.comments.any?
  node.comments << RBI::Comment.new(uri.to_s)
rescue URI::InvalidComponentError, URI::InvalidURIError
  # Bad uri
end

def on_const(event)
  file, line = Object.const_source_location(event.symbol)
  add_source_location_comment(event.node, file, line)
end

def on_method(event)
  file, line = event.method.source_location
  add_source_location_comment(event.node, file, line)
end

def on_scope(event)
  # Instead of using `const_source_location`, which always reports the first place where a constant is defined,
  # we filter the locations tracked by ConstantDefinition. This allows us to provide the correct location for
  # constants that are defined by multiple gems.
  locations = Runtime::Trackers::ConstantDefinition.locations_for(event.constant)
  location = locations.find do |loc|
    Pathname.new(loc.path).realpath.to_s.include?(@pipeline.gem.full_gem_path)
  end
  # The location may still be nil in some situations, like constant aliases (e.g.: MyAlias = OtherConst). These
  # are quite difficult to attribute a correct location, given that the source location points to the original
  # constants and not the alias
  add_source_location_comment(event.node, location.path, location.lineno) unless location.nil?
end