lib/duck_puncher.rb in duck_puncher-2.4.0 vs lib/duck_puncher.rb in duck_puncher-2.5.0

@@ -1,43 +1,72 @@
 require 'pathname'
 require 'fileutils'
+require 'delegate'
 require 'logger'
 require 'duck_puncher/version'
 
 module DuckPuncher
   autoload :JSONStorage, 'duck_puncher/json_storage'
   autoload :GemInstaller, 'duck_puncher/gem_installer'
   autoload :Duck, 'duck_puncher/duck'
   autoload :Ducks, 'duck_puncher/ducks'
 
-  def self.punch!(*names)
-    names.each do |name|
-      if duck = Ducks[name]
+  class << self
+    attr_accessor :log
+
+    def delegate_class(name)
+      @delegations ||= {}
+      @delegations[name] ||= Ducks[name].dup.delegated
+    end
+
+    # @description Extends functionality to a copy of the specified class
+    def punch(*names)
+      singular = names.size == 1
+      punched_ducks = names.map do |name|
+        duck = Ducks[name]
+        duck_class = Class.new(duck.klass)
+        if duck.punch duck_class
+          duck_class
+        else
+          log.error %Q(Failed to punch #{name}!)
+        end
+      end
+      punched_ducks.compact!
+      punched_ducks = punched_ducks.first if singular
+      punched_ducks
+    end
+
+    def punch!(*names)
+      names.each do |name|
+        duck = Ducks[name]
         if duck.punched?
           log.info %Q(Already punched #{name})
         else
-          log.warn %Q(Punching the #{name} ducky)
+          log.warn %Q(Punching #{name} ducky)
           unless duck.punch
             log.error %Q(Failed to punch #{name}!)
           end
         end
-      else
-        log.info %Q(Couldn't find "#{name}" in my list of Ducks! I know about: #{Ducks.list.map(&:name).map(&:to_s)})
       end
+      nil
     end
-    nil
-  end
 
-  def self.punch_all!
-    log.warn 'Punching all ducks! Watch out!'
-    Ducks.list.each &:punch
+    def punch_all!
+      log.warn 'Punching all ducks! Watch out!'
+      Ducks.list.each &:punch
+    end
   end
 
-  class << self
-    attr_accessor :log
-  end
-
+  # @description Default logger
+  # @example Silence logging
+  #
+  #   `DuckPuncher.log.level = Logger::ERROR`
+  #
   self.log = Logger.new(STDOUT).tap do |config|
     config.level = Logger::INFO
     config.formatter = proc { |*args| "#{args.first}: #{args.last.to_s}\n" }
   end
+end
+
+def punch(name, val)
+  DuckPuncher.delegate_class(name).new val
 end