Move #on method from Command to Performable

ergoserv · Mar 6, 2020 · f46c237 · f46c237
1 parent d42470a
commit f46c237
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/lib/auxiliary_rails/application/command.rb b/lib/auxiliary_rails/application/command.rb
@@ -43,16 +43,6 @@ def i18n_scope
         end
       end
 
-      def on(status, &_block)
-        ensure_execution!
-
-        return self unless status?(status)
-
-        yield(self) if block_given?
-
-        self
-      end
-
       protected
 
       # Shortcut reader for attributes defined by <tt>Dry::Initializer</tt>

diff --git a/lib/auxiliary_rails/concerns/performable.rb b/lib/auxiliary_rails/concerns/performable.rb
@@ -53,6 +53,22 @@ def success?
         status?(:success)
       end
 
+      # Provides ability to execude block of the code depending on
+      # command execution status
+      #
+      # @param status [Symol] Desired command status
+      # @param &_block Code to be executed if status matches
+      # @return [self]
+      def on(status, &_block)
+        ensure_execution!
+
+        return self unless status?(status)
+
+        yield(self) if block_given?
+
+        self
+      end
+
       # Shortcut for <tt>ActiveRecord::Base.transaction</tt>
       def transaction(&block)
         ActiveRecord::Base.transaction(&block) if block_given?