module RDoc::RbsSupport

# File lib/rdoc/rbs_support.rb, line 41
def load_signatures(*dirs)
  loader = RBS::EnvironmentLoader.new
  dirs.each { |dir| loader.add(path: Pathname(dir)) }

  env = RBS::Environment.new
  loader.load(env: env)

  signatures = {}

  env.class_decls.each do |type_name, entry|
    class_name = type_name.to_s.delete_prefix('::')

    entry.each_decl do |decl|
      decl.members.each do |member|
        case member
        when RBS::AST::Members::MethodDefinition
          key = member.singleton? ? "#{class_name}::#{member.name}" : "#{class_name}##{member.name}"
          sigs = member.overloads.map { |o| o.method_type.to_s }
          signatures[key] = sigs.join("\n")
        when RBS::AST::Members::AttrReader, RBS::AST::Members::AttrWriter, RBS::AST::Members::AttrAccessor
          key = "#{class_name}.#{member.name}"
          signatures[key] = member.type.to_s
        end
      end
    end
  end

  signatures
end

# File lib/rdoc/rbs_support.rb, line 75
def merge_into_store(store, signatures)
  store.all_classes_and_modules.each do |cm|
    cm.method_list.each do |method|
      next if method.type_signature

      key = method.singleton ? "#{cm.full_name}::#{method.name}" : "#{cm.full_name}##{method.name}"
      if (sig = signatures[key])
        method.type_signature = sig
      end
    end

    cm.attributes.each do |attr|
      next if attr.type_signature

      if (sig = signatures["#{cm.full_name}.#{attr.name}"])
        attr.type_signature = sig
      end
    end
  end
end

# File lib/rdoc/rbs_support.rb, line 108
def signature_to_html(signature, lookup: nil, &resolve_link)
  signature.split("\n").map { |line|
    html = link_type_names_in_line(line, lookup, &resolve_link)
    html.gsub('->', '→')
  }.join("\n")
end

# File lib/rdoc/rbs_support.rb, line 19
def validate_method_type(sig)
  RBS::Parser.parse_method_type(sig, require_eof: true)
  nil
rescue RBS::ParsingError => e
  e.message
end

# File lib/rdoc/rbs_support.rb, line 30
def validate_type(sig)
  RBS::Parser.parse_type(sig, require_eof: true)
  nil
rescue RBS::ParsingError => e
  e.message
end