lib/lita/registry.rb in lita-4.0.4 vs lib/lita/registry.rb in lita-4.1.0

@@ -45,12 +45,12 @@
       #   Adds an adapter to the registry under the provided key.
       #   @param key [String, Symbol] The key that identifies the adapter.
       #   @yield The body of the adapter class.
       #   @return [void]
       #   @since 4.0.0
-      def register_adapter(key, adapter = nil)
-        adapter = PluginBuilder.new(key, &proc).build_adapter if block_given?
+      def register_adapter(key, adapter = nil, &block)
+        adapter = PluginBuilder.new(key, &block).build_adapter if block
 
         unless adapter.is_a?(Class)
           raise ArgumentError, I18n.t("lita.core.register_adapter.block_or_class_required")
         end
 
@@ -65,12 +65,12 @@
       #   Adds a handler to the registry.
       #   @param key [String] The namespace of the handler.
       #   @yield The body of the handler class.
       #   @return [void]
       #   @since 4.0.0
-      def register_handler(handler_or_key)
-        if block_given?
-          handler = PluginBuilder.new(handler_or_key, &proc).build_handler
+      def register_handler(handler_or_key, &block)
+        if block
+          handler = PluginBuilder.new(handler_or_key, &block).build_handler
         else
           handler = handler_or_key
 
           unless handler.is_a?(Class)
             raise ArgumentError, I18n.t("lita.core.register_handler.block_or_class_required")