class Solargraph::Pin::BaseVariable

def == other
  return false unless super
  assignment == other.assignment
end

def completion_item_kind
  Solargraph::LanguageServer::CompletionItemKinds::VARIABLE
end

def desc
  "#{to_rbs} = #{assignment&.type.inspect}"
end

def generate_complex_type
  tag = docstring.tag(:type)
  return ComplexType.try_parse(*tag.types) unless tag.nil? || tag.types.nil? || tag.types.empty?
  ComplexType.new
end

def initialize assignment: nil, **splat
  super(**splat)
  @assignment = assignment
  # @type [nil, ::Array(Parser::AST::Node, Integer)]

  @mass_assignment = nil
end

def nil_assignment?
  return_type.nil?
end

def probe api_map
  unless @assignment.nil?
    types = return_types_from_node(@assignment, api_map)
    return ComplexType.new(types.uniq) unless types.empty?
  end
  unless @mass_assignment.nil?
    mass_node, index = @mass_assignment
    types = return_types_from_node(mass_node, api_map)
    types.map! do |type|
      if type.tuple?
        type.all_params[index]
      elsif ['::Array', '::Set', '::Enumerable'].include?(type.rooted_name)
        type.all_params.first
      end
    end.compact!
    return ComplexType.new(types.uniq) unless types.empty?
  end
  ComplexType::UNDEFINED
end

def return_type
  @return_type ||= generate_complex_type
end

def return_types_from_node(parent_node, api_map)
  types = []
  value_position_nodes_only(parent_node).each do |node|
    # Nil nodes may not have a location

    if node.nil? || node.type == :NIL || node.type == :nil
      types.push ComplexType::NIL
    else
      rng = Range.from_node(node)
      next if rng.nil?
      pos = rng.ending
      clip = api_map.clip_at(location.filename, pos)
      # Use the return node for inference. The clip might infer from the

      # first node in a method call instead of the entire call.

      chain = Parser.chain(node, nil, nil)
      result = chain.infer(api_map, closure, clip.locals).self_to_type(closure.context)
      types.push result unless result.undefined?
    end
  end
  types
end

def symbol_kind
  Solargraph::LanguageServer::SymbolKinds::VARIABLE
end

def try_merge! pin
  return false unless super
  @assignment = pin.assignment
  @return_type = pin.return_type
  true
end

def variable?
  true
end