Contents

class Module
  unless respond_to? :delegate
    def delegate(*methods)
      options = methods.pop
      unless options.is_a?(Hash) && to = options[:to]
        raise ArgumentError, "Delegation needs a target. Supply an options hash with a :to key as the last argument (e.g. delegate :hello, :to => :greeter)."
      end

      if options[:prefix] == true && options[:to].to_s =~ /^[^a-z_]/
        raise ArgumentError, "Can only automatically set the delegation prefix when delegating to a method."
      end

      prefix = options[:prefix] && "#{options[:prefix] == true ? to : options[:prefix]}_" || ''

      file, line = caller.first.split(':', 2)
      line = line.to_i

      methods.each do |method|
        on_nil =
          if options[:allow_nil]
            'return'
          else
            %(raise "#{self}##{prefix}#{method} delegated to #{to}.#{method}, but #{to} is nil: \#{self.inspect}")
          end
        # def customer_name(*args, &block)
        module_eval %{
          def #{prefix}#{method}(*args, &block)
            #{to}.__send__(#{method.inspect}, *args, &block)
          rescue NoMethodError
            if #{to}.nil?
              #{on_nil}
            else
              raise
            end
          end
        }
      end
    end
  end
end

# http://blog.jayfields.com/2008/02/ruby-replace-methodmissing-with-dynamic.html
class DelegateDecorator
  def initialize(subject, options = {})
    options[:only]   ||= []
    options[:except] ||= []
    options[:only].map!(&:to_s)
    options[:except].map!(&:to_s)

    meths = subject.public_methods(false)
    meths = meths & options[:only] unless options[:only].empty?

    meths.each do |meth|
      (class << self; self; end).class_eval do
        unless options[:except].include? meth.to_s
          define_method meth do |*args|
            subject.send meth, *args
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
sugar-high-0.7.3	lib/sugar-high/delegate.rb