Add callbacks

1 files changed, 28 insertions, 0 deletions

diff --git a/lib/action_mailroom/mailbox/callbacks.rb b/lib/action_mailroom/mailbox/callbacks.rb
new file mode 100644
index 0000000000..ae5a461d8f
--- /dev/null
+++ b/lib/action_mailroom/mailbox/callbacks.rb
@@ -0,0 +1,28 @@
+require "active_support/callbacks"
+
+module ActionMailroom
+  class Mailbox
+    module Callbacks
+      extend  ActiveSupport::Concern
+      include ActiveSupport::Callbacks
+
+      included do
+        define_callbacks :process
+      end
+
+      module ClassMethods
+        def before_processing(*methods, &block)
+          set_callback(:process, :before, *methods, &block)
+        end
+
+        def after_processing(*methods, &block)
+          set_callback(:process, :after, *methods, &block)
+        end
+
+        def around_processing(*methods, &block)
+          set_callback(:process, :around, *methods, &block)
+        end
+      end
+    end
+  end
+end