Add generation hooks.

jnicklas · Oct 7, 2008 · 4346909 · 4346909
1 parent 5a76b09
commit 4346909
Show file tree

Hide file tree

Showing 2 changed files with 80 additions and 66 deletions.
diff --git a/lib/templater/cli/generator.rb b/lib/templater/cli/generator.rb
@@ -73,6 +73,11 @@ def run(arguments)
           puts "Generating with #{@generator_name} generator:" 
         end
         step_through_templates
+
+        # Run hooks
+        @generator.after_run
+        @generator.after_generation unless @options[:delete]
+        @generator.after_deletion   if     @options[:delete]
       end
 
       def step_through_templates
@@ -165,9 +170,6 @@ def output_diff_line(diff)
           say "#{diff.action}  #{diff.element.chomp}"
         end
       end
-
     end
-
   end
-
-end
+end