thor-plus 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 65643a66eb29ec071d58e0df7c191b5230d6fc52
-  data.tar.gz: 5cb5b49921fb2a1c29526d9b6891c176ab53e03a
+  metadata.gz: f483251a5a8dc7d64817f01855dd09a1f3073de4
+  data.tar.gz: 49de3b45cc64696aa9e98810508224fafddc2a5d
 SHA512:
-  metadata.gz: 83bd49b2ed6a9c62f805fda556f933b3457c3acd98dd9bb87e7a436644514f5471bc09f4ffca1b925569df9c2015bd3071eddae15ff170bfc1720480f577726f
-  data.tar.gz: fe8320dd9d0447dd8ecede00cd745cafe58c445b1d734af36ca765d644235211fd745e3aefaf0ae77fe5b33597a6ca358d86c3044d85201ec9722812bd0fc1b9
+  metadata.gz: c7fb604ecf73e9d478625b053a8353cc2f2e763d6dd1b6b749e2b84df262713531e7b22dc0d45764c788bea252ef2762afc892ff0fb3f645c31a10b5ffc2f159
+  data.tar.gz: bae1a232716f10d9243bdefa1e35f42bbc8d59d99bfee0883853a2d3955385b3905698c64f621ca7047282ca9e23b502380a022f1500929843e5c6509229d926

data/lib/thor.rb

@@ -27,6 +27,39 @@ class Thor # rubocop:disable ClassLength
     end
     alias_method :default_task, :default_command
 
+    # Adds a callback method that should be executed before a command is executed.
+    #
+    # ==== Parameters
+    # methods<*String|Symbol>:: Callbacks to execute before command runs.
+    def before_command(*methods)
+      methods.each { |m| register_callback(:before, m) }
+    end
+    alias_method :before_task, :before_command
+
+    # Adds a callback method that should be executed after a task is executed.
+    #
+    # ==== Parameters
+    # methods<*String|Symbol>:: Callbacks to execute after command runs.
+    def after_command(*methods)
+      methods.each { |m| register_callback(:after, m) }
+    end
+    alias_method :after_task, :after_command
+
+    # All callbacks defined for this class & superclass.
+    def callbacks
+      @callbacks ||= from_superclass(:callbacks, {})
+    end
+
+    # Register a callback of generic type.
+    #
+    # ==== Parameters
+    # type<Symbol>:: Type of callback to register.
+    # name<Symbol>:: Name of callback command to run.
+    def register_callback(type, name)
+      callbacks[type.to_sym] ||= []
+      callbacks[type.to_sym] << name.to_sym
+    end
+
     # Registers another Thor subclass as a command.
     #
     # ==== Parameters

data/lib/thor/base.rb

@@ -84,6 +84,15 @@ class Thor
       @args = thor_args.remaining
     end
 
+    # Run callbacks by type.
+    #
+    # ==== Parameters
+    # type<Symbol>:: Type of callbacks to run.
+    def run_callbacks(type)
+      callbacks = self.class.callbacks[type.to_sym] || []
+      callbacks.each { |c| __send__(c) }
+    end
+
     class << self
       def included(base) #:nodoc:
         base.extend ClassMethods

data/lib/thor/command.rb

@@ -24,7 +24,10 @@ class Thor
         instance.class.handle_no_command_error(name)
       elsif public_method?(instance)
         arity = instance.method(name).arity
+
+        instance.run_callbacks(:before)
         instance.__send__(name, *args)
+        instance.run_callbacks(:after)
       elsif local_method?(instance, :method_missing)
         instance.__send__(:method_missing, name.to_sym, *args)
       else

data/lib/thor/version.rb

@@ -1,3 +1,3 @@
 class Thor
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: thor-plus
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Yehuda Katz
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-08 00:00:00.000000000 Z
+date: 2016-06-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler