module Solargraph::ApiMap::SourceToYard

# File lib/solargraph/api_map/source_to_yard.rb, line 13
def code_object_at path, klass = YARD::CodeObjects::Base
  obj = code_object_map[path]
  obj if obj&.is_a?(klass)
end

# File lib/solargraph/api_map/source_to_yard.rb, line 19
def code_object_paths
  code_object_map.keys
end

# File lib/solargraph/api_map/source_to_yard.rb, line 25
def rake_yard store
  YARD::Registry.clear
  code_object_map.clear
  store.namespace_pins.each do |pin|
    next if pin.path.nil? || pin.path.empty?
    if pin.code_object
      code_object_map[pin.path] ||= pin.code_object
      next
    end
    if pin.type == :class
      # @param obj [YARD::CodeObjects::RootObject]
      code_object_map[pin.path] ||= YARD::CodeObjects::ClassObject.new(root_code_object, pin.path) { |obj|
        next if pin.location.nil? || pin.location.filename.nil?
        obj.add_file(pin.location.filename, pin.location.range.start.line, !pin.comments.empty?)
      }
    else
      # @param obj [YARD::CodeObjects::RootObject]
      code_object_map[pin.path] ||= YARD::CodeObjects::ModuleObject.new(root_code_object, pin.path) { |obj|
        next if pin.location.nil? || pin.location.filename.nil?
        obj.add_file(pin.location.filename, pin.location.range.start.line, !pin.comments.empty?)
      }
    end
    code_object_map[pin.path].docstring = pin.docstring
    store.get_includes(pin.path).each do |ref|
      include_object = code_object_at(pin.path, YARD::CodeObjects::ClassObject)
      unless include_object.nil? || include_object.nil?
        include_object.instance_mixins.push code_object_map[ref.type.to_s]
      end
    end
    store.get_extends(pin.path).each do |ref|
      extend_object = code_object_at(pin.path, YARD::CodeObjects::ClassObject)
      next unless extend_object
      code_object = code_object_map[ref.type.to_s]
      next unless code_object
      extend_object.class_mixins.push code_object
      # @todo add spec showing why this next line is necessary
      extend_object.instance_mixins.push code_object
    end
  end
  store.method_pins.each do |pin|
    if pin.code_object
      code_object_map[pin.path] ||= pin.code_object
      next
    end

    # @param obj [YARD::CodeObjects::RootObject]
    code_object_map[pin.path] ||= YARD::CodeObjects::MethodObject.new(code_object_at(pin.namespace, YARD::CodeObjects::NamespaceObject), pin.name, pin.scope) { |obj|
      next if pin.location.nil? || pin.location.filename.nil?
      obj.add_file pin.location.filename, pin.location.range.start.line
    }
    method_object = code_object_at(pin.path, YARD::CodeObjects::MethodObject)
    method_object.docstring = pin.docstring
    method_object.visibility = pin.visibility || :public
    method_object.parameters = pin.parameters.map do |p|
      [p.full_name, p.asgn_code]
    end
  end
end

# File lib/solargraph/api_map/source_to_yard.rb, line 87
def code_object_map
  @code_object_map ||= {}
end

# File lib/solargraph/api_map/source_to_yard.rb, line 92
def root_code_object
  @root_code_object ||= YARD::CodeObjects::RootObject.new(nil, 'root')
end