Moving Catalog#write_graph to SimpleGraph, where it belongs.

Signed-off-by: Luke Kanies <luke@madstop.com>

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/puppet/simple_graph.rb b/lib/puppet/simple_graph.rb
index 48f393f77..c926258ca 100644
--- a/lib/puppet/simple_graph.rb
+++ b/lib/puppet/simple_graph.rb
@@ -315,4 +315,16 @@ class Puppet::SimpleGraph
       system( "dot -T#{fmt} #{src} -o #{dot}" )
       dot
     end
+
+    # Produce the graph files if requested.
+    def write_graph(name)
+        return unless Puppet[:graph]
+
+        Puppet.settings.use(:graphing)
+
+        file = File.join(Puppet[:graphdir], "%s.dot" % name.to_s)
+        File.open(file, "w") { |f|
+            f.puts to_dot("name" => name.to_s.capitalize)
+        }
+    end
 end