Allow block to be passed through from Middleman.server()

This commit is contained in:
David W. Keith 2013-06-03 11:48:16 -07:00
parent cc418c7a2d
commit f2f17c33b3

View file

@ -136,7 +136,7 @@ module Middleman
def server(&block)
@@servercounter ||= 0
@@servercounter += 1
const_set("MiddlemanApplication#{@@servercounter}", Class.new(Middleman::Application))
const_set("MiddlemanApplication#{@@servercounter}", Class.new(Middleman::Application, &block))
end
end
@ -148,7 +148,7 @@ module Middleman
#
# @return [Class]
def server(&block)
::Middleman::Application.server
::Middleman::Application.server(&block)
end
end