class Solargraph::ApiMap::Index

# File lib/solargraph/api_map/index.rb, line 9
def initialize pins = []
  catalog pins
end

# File lib/solargraph/api_map/index.rb, line 64
def extend_references
  # @param h [String]
  # @param k [Array<String>]
  @extend_references ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 57
def include_reference_pins
  # @param h [String]
  # @param k [Array<Pin::Reference::Include>]
  @include_reference_pins ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 50
def include_references
  # @param h [String]
  # @param k [Array<String>]
  @include_references ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 86
def merge pins
  deep_clone.catalog pins
end

# File lib/solargraph/api_map/index.rb, line 19
def namespace_hash
  # @param h [String]
  # @param k [Array<Pin::Namespace>]
  @namespace_hash ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 33
def path_pin_hash
  # @param h [String]
  # @param k [Array<Pin::Base>]
  @path_pin_hash ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 26
def pin_class_hash
  # @param h [String]
  # @param k [Array<Pin::Base>]
  @pin_class_hash ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 14
def pins
  @pins ||= []
end

# File lib/solargraph/api_map/index.rb, line 42
def pins_by_class klass
  # @type [Set<Solargraph::Pin::Base>]
  s = Set.new
  # @sg-ignore need to support destructured args in blocks
  @pin_select_cache[klass] ||= pin_class_hash.each_with_object(s) { |(key, o), n| n.merge(o) if key <= klass }
end

# File lib/solargraph/api_map/index.rb, line 71
def prepend_references
  # @param h [String]
  # @param k [Array<String>]
  @prepend_references ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 78
def superclass_references
  # @param h [String]
  # @param k [Array<String>]
  @superclass_references ||= Hash.new { |h, k| h[k] = [] }
end

# File lib/solargraph/api_map/index.rb, line 113
def catalog new_pins
  # @type [Hash{Class<generic<T>> => Set<generic<T>>}]
  @pin_select_cache = {}
  pins.concat new_pins
  set = new_pins.to_set
  # @param k [String]
  # @param v [Set<Pin::Base>]
  set.classify(&:class)
    .map { |k, v| pin_class_hash[k].concat v.to_a }
  # @param k [String]
  # @param v [Set<Pin::Namespace>]
  set.classify(&:namespace)
    .map { |k, v| namespace_hash[k].concat v.to_a }
  # @param k [String]
  # @param v [Set<Pin::Base>]
  set.classify(&:path)
    .map { |k, v| path_pin_hash[k].concat v.to_a }
  @namespaces = path_pin_hash.keys.compact.to_set
  map_references Pin::Reference::Include, include_references
  map_references Pin::Reference::Prepend, prepend_references
  map_references Pin::Reference::Extend, extend_references
  map_references Pin::Reference::Superclass, superclass_references
  map_overrides
  self
end

# File lib/solargraph/api_map/index.rb, line 96
def deep_clone
  Index.allocate.tap do |copy|
    copy.pin_select_cache = {}
    copy.pins = pins.clone
    %i[
      namespace_hash pin_class_hash path_pin_hash include_references extend_references prepend_references
      superclass_references
    ].each do |sym|
      copy.send("#{sym}=", send(sym).clone)
      copy.send(sym)&.transform_values!(&:clone)
    end
  end
end

# File lib/solargraph/api_map/index.rb, line 152
def map_overrides
  # @param ovr [Pin::Reference::Override]
  pins_by_class(Pin::Reference::Override).each do |ovr|
    logger.debug { "ApiMap::Index#map_overrides: Looking at override #{ovr} for #{ovr.name}" }
    pins = path_pin_hash[ovr.name]
    logger.debug { "ApiMap::Index#map_overrides: pins for path=#{ovr.name}: #{pins}" }
    pins.each do |pin|
      new_pin = if pin.path.end_with?('#initialize')
                  path_pin_hash[pin.path.sub(/#initialize/, '.new')].first
                end
      (ovr.tags.map(&:tag_name) + ovr.delete).uniq.each do |tag|
        pin.docstring.delete_tags tag
        new_pin.docstring.delete_tags tag if new_pin
      end
      ovr.tags.each do |tag|
        pin.docstring.add_tag(tag)
        redefine_return_type pin, tag
        if new_pin
          new_pin.docstring.add_tag(tag)
          redefine_return_type new_pin, tag
        end
      end
    end
  end
end

# File lib/solargraph/api_map/index.rb, line 144
def map_references klass, hash
  # @param pin [generic<T>]
  pins_by_class(klass).each do |pin|
    hash[pin.namespace].push pin
  end
end

# File lib/solargraph/api_map/index.rb, line 181
def redefine_return_type pin, tag
  # @todo can this be made to not mutate existing pins and use
  #   proxy() / proxy_with_signatures() instead?
  return unless pin && tag.tag_name == 'return'
  pin.instance_variable_set(:@return_type, ComplexType.try_parse(tag.type))
  pin.signatures.each do |sig|
    sig.instance_variable_set(:@return_type, ComplexType.try_parse(tag.type))
  end
  pin.reset_generated!
end