module Puppet::Util::RDoc

# File lib/puppet/util/rdoc.rb, line 49
def manifestdoc(files)
  Puppet[:ignoreimport] = true
  files.select { |f| FileTest.file?(f) }.each do |f|
    parser = Puppet::Parser::Parser.new(Puppet::Node::Environment.new(Puppet[:environment]))
    parser.file = f
    ast = parser.parse
    output(f, ast)
  end
end

# File lib/puppet/util/rdoc.rb, line 61
def output(file, ast)
  astobj = []
  ast.instantiate('').each do |resource_type|
    astobj << resource_type if resource_type.file == file
  end

  astobj.sort! {|a,b| a.line <=> b.line }.each do |k|
    output_astnode_doc(k)
  end
end

# File lib/puppet/util/rdoc.rb, line 72
def output_astnode_doc(ast)
  puts ast.doc if !ast.doc.nil? and !ast.doc.empty?
  if Puppet.settings[:document_all]
    # scan each underlying resources to produce documentation
    code = ast.code.children if ast.code.is_a?(Puppet::Parser::AST::ASTArray)
    code ||= ast.code
    output_resource_doc(code) unless code.nil?
  end
end

# File lib/puppet/util/rdoc.rb, line 82
def output_resource_doc(code)
  code.sort { |a,b| a.line <=> b.line }.each do |stmt|
    output_resource_doc(stmt.children) if stmt.is_a?(Puppet::Parser::AST::ASTArray)

    if stmt.is_a?(Puppet::Parser::AST::Resource)
      puts stmt.doc if !stmt.doc.nil? and !stmt.doc.empty?
    end
  end
end

# File lib/puppet/util/rdoc.rb, line 7
def rdoc(outputdir, files, charset = nil)
  unless Puppet.features.rdoc1?
    raise "the version of RDoc included in Ruby #{::RUBY_VERSION} is not supported"
  end

  begin
    Puppet[:ignoreimport] = true

    # then rdoc
    require 'rdoc/rdoc'
    require 'rdoc/options'

    # load our parser
    require 'puppet/util/rdoc/parser'

    r = RDoc::RDoc.new

    RDoc::RDoc::GENERATORS["puppet"] = RDoc::RDoc::Generator.new(
      "puppet/util/rdoc/generators/puppet_generator.rb",
      :PuppetGenerator,
      "puppet"
    )

    # specify our own format & where to output
    options = [ "--fmt", "puppet",
                "--quiet",
                "--exclude", "/modules/[^/]*/files/.*$",
                "--exclude", "/modules/[^/]*/templates/.*$",
                "--op", outputdir ]

    options << "--force-update" if Options::OptionList.options.any? { |o| o[0] == "--force-update" }
    options += [ "--charset", charset] if charset
    options += files

    # launch the documentation process
    r.document(options)
  rescue RDoc::RDocError => e
    raise Puppet::ParseError.new("RDoc error #{e}")
  end
end