class Middleman::Extension

def bind_before_build
  ext = self
  if ext.respond_to?(:before_build)
    @klass.before_build do |builder|
      if ext.method(:before_build).arity == 1
        ext.before_build(builder)
      else
        ext.before_build
      end
    end
  end
end