Contents

module DuckPuncher
  class Duck
    attr_accessor :target, :mod, :options

    # @param target [String,Class] Class or module to punch
    # @param mod [String,Module] The module that defines the extensions
    # @param [Hash] options to modify the duck #punch method behavior
    # @option options :before [Proc] A hook that is called with the target class before +punch+
    # @option options :after [Proc] A hook that is called with the target class after +punch+
    def initialize(target, mod, options = {})
      @options = options
      @target = DuckPuncher.lookup_constant(target)
      @mod = DuckPuncher.lookup_constant(mod)
    end

    # @param [Hash] opts to modify punch
    # @option options [Class] :target Specifies the class to be punched
    # @option options [Array,Symbol] :only Specifies the methods to extend onto the current object
    # @option options [Symbol,String] :method Specifies if the methods should be included or prepended (:include)
    # @return [Class] The class that was just punched
    def punch(opts = {})
      opts = options.merge(opts)
      targets = Array(opts[:target] || self.target)
      targets.each do |target|
        options[:before].call(target) if options[:before]
        punches = Array(opts[:only] || Ducks::Module.instance_method(:local_methods).bind(mod).call)
        DuckPuncher.logger.info %Q(#{target}#{" <-- #{mod.name}#{punches}" if punches.any?})
        extender = Usable::ModExtender.new(mod, only: options.delete(:only), method: options.delete(:method))
        extender.call target
        extender = nil
        options[:after].call(target) if options[:after]
      end
      targets
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
duck_puncher-4.4.1	lib/duck_puncher/duck.rb
duck_puncher-4.4.0	lib/duck_puncher/duck.rb