lib/pyro/server.rb in pyro-0.8.1 vs lib/pyro/server.rb in pyro-0.8.2

@@ -2,22 +2,16 @@
 require 'sinatra/base'
 require 'erb'
 
 class PyroServer < Sinatra::Base
   include Pyro
-  
+
   set :port, 5678
   set :public_folder, './builds/development'
   set :fast_build, false
-  
-  #helpers do
-  #  define_method(:script)     { |*args| Pyro.script     public_folder, args }
-  #  define_method(:stylesheet) { |*args| Pyro.stylesheet args }
-  #  define_method(:template)   { |*args| Pyro.template   args }
-  #end
-  
+
   get '/?' do
-    Pyro.build('development', settings.fast_build)
+    Pyro.build('development', settings.fast_build, settings.src_folder)
     settings.fast_build = settings.fast
-    File.read('./builds/development/index.html')
+    File.read("#{settings.src_folder}/builds/development/index.html")
   end
 end