Moving classfile-writing to the Catalog

This work was done by the Agent class before,
but it's really related to the catalog, so that's
where it is now.

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

1 files changed, 11 insertions, 0 deletions

diff --git a/lib/puppet/resource/catalog.rb b/lib/puppet/resource/catalog.rb
index 0692b5ddd..8f013e7db 100644
--- a/lib/puppet/resource/catalog.rb
+++ b/lib/puppet/resource/catalog.rb
@@ -393,6 +393,17 @@ class Puppet::Resource::Catalog < Puppet::SimpleGraph
         to_catalog :to_resource
     end
 
+    # Store the classes in the classfile.
+    def write_class_file
+        begin
+            ::File.open(Puppet[:classfile], "w") do |f|
+                f.puts classes.join("\n")
+            end
+        rescue => detail
+            Puppet.err "Could not create class file %s: %s" % [Puppet[:classfile], detail]
+        end
+    end
+
     # Produce the graph files if requested.
     def write_graph(name)
         # We only want to graph the main host catalog.