lib/emittance/brokerage.rb in emittance-0.0.2 vs lib/emittance/brokerage.rb in emittance-0.0.3

@@ -4,30 +4,53 @@
   ##
   # The clearinghouse for brokers. Registers brokers, and decides which broker to use when sent an event. First point of
   # contact for event propagation.
   #
   class Brokerage
+    @enabled = true
+
     class << self
+      # @param event [Emittance::Event] the event object
+      # @param broker_id [Symbol] a symbol that can be used to identify a broker by
       def send_event(event, broker_id)
         broker = registry.fetch(broker_id)
         broker.process_event event
       end
 
+      # @param broker [Emittance::Broker] the broker you would like to register
       def register_broker(broker)
         registry.register broker
       end
 
       def registry
         Emittance::Brokerage::Registry
       end
+
+      def enable!
+        @enabled = true
+      end
+
+      def disable!
+        @enabled = false
+      end
+
+      def enabled?
+        @enabled
+      end
+
+      private
+
+      attr_accessor :enabled
     end
 
     # @private
     module Registry
       @brokers = {}
 
       class << self
+        include Emittance::Helpers::StringHelpers
+
         attr_reader :brokers
 
         def register(broker)
           broker_sym = generate_broker_sym(broker)
           brokers[broker_sym] = broker
@@ -40,16 +63,9 @@
         private
 
         def generate_broker_sym(broker)
           camel_case = broker.name.split('::').last
           snake_case(camel_case).to_sym
-        end
-
-        def snake_case(str)
-          str.gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
-            .gsub(/([a-z\d])([A-Z])/, '\1_\2')
-            .tr('-', '_')
-            .downcase
         end
       end
     end
   end
 end