class Middleman::Extension

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