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, 2 insertions, 19 deletions

diff --git a/lib/puppet/agent.rb b/lib/puppet/agent.rb
index f0ee101bf..5cb6f0019 100644
--- a/lib/puppet/agent.rb
+++ b/lib/puppet/agent.rb
@@ -138,6 +138,8 @@ class Puppet::Agent
         return nil unless result
 
         result.retrieval_duration = duration
+        result.host_config = true
+        result.write_class_file
         return result
     end
 
@@ -188,25 +190,6 @@ class Puppet::Agent
         lockfile.locked?
     end
 
-    # Store the classes in the classfile, but only if we're not local.
-    def setclasses(ary)
-        if @local
-            return
-        end
-        unless ary and ary.length > 0
-            Puppet.info "No classes to store"
-            return
-        end
-        begin
-            ::File.open(Puppet[:classfile], "w") { |f|
-                f.puts ary.join("\n")
-            }
-        rescue => detail
-            Puppet.err "Could not create class file %s: %s" %
-                [Puppet[:classfile], detail]
-        end
-    end
-
     private
 
     def self.timeout