1 files changed, 85 insertions, 0 deletions
diff --git a/lib/puppet/util/rdoc.rb b/lib/puppet/util/rdoc.rb
new file mode 100644
index 000000000..b33e67c71
--- /dev/null
+++ b/lib/puppet/util/rdoc.rb
@@ -0,0 +1,85 @@
+
+module Puppet::Util::RDoc
+
+    module_function
+
+    # launch a rdoc documenation process
+    # with the files/dir passed in +files+
+    def rdoc(outputdir, files)
+        begin
+            Puppet[:ignoreimport] = true
+
+            # then rdoc
+            require 'rdoc/rdoc'
+
+            # 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".intern,
+                                                                       "puppet")
+            # specify our own format & where to output
+            options = [ "--fmt", "puppet",
+                        "--quiet",
+                        "--op", outputdir ]
+
+            options += files
+
+            # launch the documentation process
+            r.document(options)
+        rescue RDoc::RDocError => e
+            raise Puppet::ParseError.new("RDoc error %s" % e)
+        end
+    end
+
+    # launch a output to console manifest doc
+    def manifestdoc(files)
+        Puppet[:ignoreimport] = true
+        files.select { |f| FileTest.file?(f) }.each do |f|
+            parser = Puppet::Parser::Parser.new(:environment => Puppet[:environment])
+            parser.file = f
+            ast = parser.parse
+            output(f, ast)
+        end
+    end
+
+    # Ouputs to the console the documentation
+    # of a manifest
+    def output(file, ast)
+        astobj = []
+        ast[:nodes].each do |name, k|
+            astobj << k if k.file == file
+        end
+        ast[:classes].each do |name, k|
+            astobj << k if k.file == file
+        end
+        ast[:definitions].each do |name, k|
+            astobj << k if k.file == file
+        end
+        astobj.sort! {|a,b| a.line <=> b.line }.each do |k|
+            output_astnode_doc(k)
+        end
+    end
+
+    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
+
+    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
+
+end
+\ No newline at end of file