[Command Tags] Make startup and shutdown calls to all modules

1 files changed, 4 insertions, 3 deletions

diff --git a/app/lib/command_tag/processor.rb b/app/lib/command_tag/processor.rb
index cb447f851..414c08638 100644
--- a/app/lib/command_tag/processor.rb
+++ b/app/lib/command_tag/processor.rb
@@ -49,7 +49,7 @@ class CommandTag::Processor
 
     reset_status_caches
 
-    initialize_handlers!
+    all_handlers!(:startup)
 
     parse_statements
 
@@ -79,13 +79,14 @@ class CommandTag::Processor
     end
 
     execute_statements(:at_end)
+    all_handlers!(:shutdown)
     reset_status_caches
   end
 
   private
 
-  def initialize_handlers!
-    self.class.instance_methods.grep(/\Ainitialize_\w+!\z/).each do |name|
+  def all_handlers!(affix)
+    self.class.instance_methods.grep(/\Ahandle_\w+_#{affix}\z/).each do |name|
       public_send(name)
     end
   end