Method: RubyTransform::Registry#register

Defined in:

lib/ruby_transform/registry.rb

#register(name, transform = nil, &block) ⇒ Object

# File 'lib/ruby_transform/registry.rb', line 20

def register(name, transform = nil, &block)
  if block
    self[name.to_s] = RubyTransform::Transformers::Custom.new(&block)
  elsif transform.is_a?(Array)
    self[name.to_s] = RubyTransform::Transformers::Composite.new(transform.map {|t|
      if t.is_a?(Symbol) or t.is_a?(String)
        self[t] or raise ArgumentError, "#{t} is not a registered transform name."
      else
        t
      end
    })
  elsif transform.is_a?(RubyTransform::Transformer)
    self[name.to_s] = transform
  else
    raise ArgumentError, "Please provide a transform or custom block processor."
  end
end