class Gem::RDoc

def self.generation_hook installer, specs
  start = Time.now
  types = installer.document
  generate_rdoc = types.include? 'rdoc'
  generate_ri   = types.include? 'ri'
  specs.each do |spec|
    new(spec, generate_rdoc, generate_ri).generate
  end
  return unless generate_rdoc or generate_ri
  duration = (Time.now - start).to_i
  names    = specs.map(&:name).join ', '
  say "Done installing documentation for #{names} after #{duration} seconds"
end

def self.load_rdoc
  return if @rdoc_version
  require 'rdoc/rdoc'
  @rdoc_version = if ::RDoc.const_defined? :VERSION then
                    Gem::Version.new ::RDoc::VERSION
                  else
                    Gem::Version.new '1.0.1'
                  end
rescue LoadError => e
  raise Gem::DocumentError, "RDoc is not installed: #{e}"
end

def delete_legacy_args args
  args.delete '--inline-source'
  args.delete '--promiscuous'
  args.delete '-p'
  args.delete '--one-file'
end

def document generator, options, destination
  generator_name = generator
  options = options.dup
  options.exclude ||= [] # TODO maybe move to RDoc::Options#finish
  options.setup_generator generator
  options.op_dir = destination
  options.finish
  generator = options.generator.new @rdoc.store, options
  @rdoc.options = options
  @rdoc.generator = generator
  say "Installing #{generator_name} documentation for #{@spec.full_name}"
  FileUtils.mkdir_p options.op_dir
  Dir.chdir options.op_dir do
    begin
      @rdoc.class.current = @rdoc
      @rdoc.generator.generate @file_info
    ensure
      @rdoc.class.current = nil
    end
  end
end

def generate
  return unless @generate_ri or @generate_rdoc
  setup
  options = nil
  if Gem::Requirement.new('< 3').satisfied_by? self.class.rdoc_version then
    generate_legacy
    return
  end
  ::RDoc::TopLevel.reset # TODO ::RDoc::RDoc.reset
  ::RDoc::Parser::C.reset
  args = @spec.rdoc_options
  args.concat @spec.source_paths
  args.concat @spec.extra_rdoc_files
  case config_args = Gem.configuration[:rdoc]
  when String then
    args = args.concat config_args.split
  when Array then
    args = args.concat config_args
  end
  delete_legacy_args args
  Dir.chdir @spec.full_gem_path do
    options = ::RDoc::Options.new
    options.default_title = "#{@spec.full_name} Documentation"
    options.parse args
  end
  options.quiet = !Gem.configuration.really_verbose
  @rdoc = new_rdoc
  @rdoc.options = options
  say "Parsing documentation for #{@spec.full_name}"
  Dir.chdir @spec.full_gem_path do
    @file_info = @rdoc.parse_files options.files
  end
  document 'ri',       options, @ri_dir if
    @generate_ri   and (@force or not File.exist? @ri_dir)
  document 'darkfish', options, @rdoc_dir if
    @generate_rdoc and (@force or not File.exist? @rdoc_dir)
end

def generate_legacy
  if @generate_rdoc then
    FileUtils.rm_rf @rdoc_dir
    say "Installing RDoc documentation for #{@spec.full_name}"
    legacy_rdoc '--op', @rdoc_dir
  end
  if @generate_ri then
    FileUtils.rm_rf @ri_dir
    say "Installing ri documentation for #{@spec.full_name}"
    legacy_rdoc '--ri', '--op', @ri_dir
  end
end

def initialize spec, generate_rdoc = true, generate_ri = true
  @doc_dir   = spec.doc_dir
  @file_info = nil
  @force     = false
  @rdoc      = nil
  @spec      = spec
  @generate_rdoc = generate_rdoc
  @generate_ri   = generate_ri
  @rdoc_dir = spec.doc_dir 'rdoc'
  @ri_dir   = spec.doc_dir 'ri'
end

def legacy_rdoc *args
  args << @spec.rdoc_options
  args << '--quiet'
  args << @spec.require_paths.clone
  args << @spec.extra_rdoc_files
  args << '--title' << "#{@spec.full_name} Documentation"
  args = args.flatten.map do |arg| arg.to_s end
  delete_legacy_args args if
    Gem::Requirement.new('>= 2.4.0') =~ self.class.rdoc_version
  r = new_rdoc
  verbose { "rdoc #{args.join ' '}" }
  Dir.chdir @spec.full_gem_path do
    begin
      r.document args
    rescue Errno::EACCES => e
      dirname = File.dirname e.message.split("-")[1].strip
      raise Gem::FilePermissionError, dirname
    rescue Interrupt => e
      raise e
    rescue Exception => ex
      alert_error "While generating documentation for #{@spec.full_name}"
      ui.errs.puts "... MESSAGE:   #{ex}"
      ui.errs.puts "... RDOC args: #{args.join(' ')}"
      ui.backtrace ex
      ui.errs.puts "(continuing with the rest of the installation)"
    end
  end
end

def new_rdoc # :nodoc:
  ::RDoc::RDoc.new
end

def rdoc_installed?
  File.exist? @rdoc_dir
end

def remove
  base_dir = @spec.base_dir
  raise Gem::FilePermissionError, base_dir unless File.writable? base_dir
  FileUtils.rm_rf @rdoc_dir
  FileUtils.rm_rf @ri_dir
end

def ri_installed?
  File.exist? @ri_dir
end

def setup
  self.class.load_rdoc
  raise Gem::FilePermissionError, @doc_dir if
    File.exist?(@doc_dir) and not File.writable?(@doc_dir)
  FileUtils.mkdir_p @doc_dir unless File.exist? @doc_dir
end