require 'webrick' module Nanoc::Extra # Nanoc::Extra::AutoCompiler is a web server that will automatically compile # pages as they are requested. It also serves static files such as # stylesheets and images. class AutoCompiler # Error that is raised when the autocompiler is started if the specified # handler cannot be found. class UnknownHandlerError < Nanoc::Error ; end HANDLER_NAMES = [ :thin, :mongrel, :webrick, :ebb, :cgi, :fastcgi, :lsws, :scgi ] ERROR_404 = < 404 File Not Found

404 File Not Found

The file you requested, <%=h path %>, was not found on this server.

END ERROR_500 = < 500 Server Error

500 Server Error

An error occurred while compiling the page you requested, <%=h path %>.

If you think this is a bug in nanoc, please do report it—thanks!

Message:

<%=h message %>

Page compilation stack:

Page <%= item.page.path %> (rep <%= item.name %>)
Layout <%= item.path %>

Backtrace:

<%= line %>

END # Creates a new autocompiler for the given site. def initialize(site, include_outdated=false) # Set site @site = site # Set options @include_outdated = include_outdated # Create mutex to prevent parallel requests @mutex = Mutex.new end # Starts the server on the given port. # # +port+:: The port the autocompiler web server should be started on. Can # be nil; in this case the server will be started on port 3000. # # +handler_name+:: A symbol containing the name of the handler to use. See # HANDLER_NAMES for a list of supported handlers. Can be # set to nil; in this case the best handler will be # picked. def start(port, handler_name) require 'mime/types' require 'rack' # Determine handler if handler_name.nil? handler = preferred_handler else handler = handler_named(handler_name.to_sym) raise UnknownHandlerError.new(handler_name) if handler.nil? end # Build Rack app app = lambda { |env| handle_request(env['PATH_INFO']) } # Run Rack app port ||= 3000 handler.run(app, :Port => port, :port => port) do |server| trap(:INT) { server.stop } end end private def preferred_handler return @preferred_handler unless @preferred_handler.nil? HANDLER_NAMES.each do |handler_name| # Get handler @preferred_handler = handler_named(handler_name) # Make sure we have one break unless @preferred_handler.nil? end @preferred_handler end def handler_named(handler_name) # Build list of handlers @handlers ||= { :cgi => { :proc => lambda { Rack::Handler::CGI } }, :fastcgi => { # FIXME buggy :proc => lambda { Rack::Handler::FastCGI } }, :lsws => { # FIXME test :proc => lambda { Rack::Handler::LSWS } }, :mongrel => { :proc => lambda { Rack::Handler::Mongrel } }, :scgi => { # FIXME buggy :proc => lambda { Rack::Handler::SCGI } }, :webrick => { :proc => lambda { Rack::Handler::WEBrick } }, :thin => { :proc => lambda { Rack::Handler::Thin }, :requires => [ 'thin' ] }, :ebb => { :proc => lambda { Rack::Handler::Ebb }, :requires => [ 'ebb' ] } } begin # Lookup handler handler = @handlers[handler_name] # Load requirements (handler[:requires] || []).each { |r| require r } # Get handler class handler[:proc].call rescue NameError, LoadError nil end end def handle_request(path) @mutex.synchronize do # Reload site data @site.load_data(true) # Get page or file page_reps = @site.pages.map { |p| p.reps }.flatten page_rep = page_reps.find { |p| p.web_path == path.cleaned_path } file_path = @site.config[:output_dir] + path if page_rep.nil? # Serve file if File.file?(file_path) serve_file(file_path) else serve_404(path) end else # Serve page rep serve_page_rep(page_rep) end end end def h(s) ERB::Util.html_escape(s) end def mime_type_of(path, fallback) mime_type = MIME::Types.of(path).first mime_type = mime_type.nil? ? fallback : mime_type.simplified end def serve_404(path) # Build response [ 404, { 'Content-Type' => 'text/html' }, [ ERB.new(ERROR_404).result(binding) ] ] end def serve_500(path, exception) # Build message case exception when Nanoc::Errors::UnknownLayoutError message = "Unknown layout: #{exception.message}" when Nanoc::Errors::UnknownFilterError message = "Unknown filter: #{exception.message}" when Nanoc::Errors::CannotDetermineFilterError message = "Cannot determine filter for layout: #{exception.message}" when Nanoc::Errors::RecursiveCompilationError message = "Recursive call to page content. Page stack:" when Nanoc::Errors::NoLongerSupportedError message = "No longer supported: #{exception.message}" else message = "Unknown error: #{exception.message}" end # Build response [ 500, { 'Content-Type' => 'text/html' }, [ ERB.new(ERROR_500).result(binding) ] ] end def serve_file(path) # Build response [ 200, { 'Content-Type' => mime_type_of(path, 'application/octet-stream') }, [ File.read(path) ] ] end def serve_page_rep(page_rep) # Recompile page rep begin @site.compiler.run([ page_rep.page ], :even_when_not_outdated => @include_outdated) rescue Exception => exception return serve_500(page_rep.web_path, exception) end # Build response [ 200, { 'Content-Type' => mime_type_of(page_rep.disk_path, 'text/html') }, [ page_rep.content(:post) ] ] end end end