Adding pre- and post-hooks, as requested in #233.

git-svn-id: https://reductivelabs.com/svn/puppet/trunk@1541 980ebf18-57e1-0310-9a29-db15c13687c0

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/puppet/util/classgen.rb b/lib/puppet/util/classgen.rb
index d7622cf93..9c7e571f6 100644
--- a/lib/puppet/util/classgen.rb
+++ b/lib/puppet/util/classgen.rb
@@ -65,6 +65,10 @@ module Puppet::Util::ClassGen
             end
         end
 
+        if klass.respond_to? :preinit
+            klass.preinit
+        end
+
         block ||= options[:block]
 
         # Evaluate the passed block if there is one.  This should usually
@@ -73,6 +77,10 @@ module Puppet::Util::ClassGen
             klass.class_eval(&block)
         end
 
+        if klass.respond_to? :postinit
+            klass.postinit
+        end
+
         # If we were told to stick it in a hash, then do so
         if hash = options[:hash]
             if hash.include? name and ! options[:overwrite]