Contents

require 'webrick'

module YARD
  module Server
    # The main adapter to initialize a WEBrick server.
    class WebrickAdapter < Adapter
      # Initializes a WEBrick server. If {Adapter#server_options} contains a
      # +:daemonize+ key set to true, the server will be daemonized.
      def start
        server_options[:ServerType] = WEBrick::Daemon if server_options[:daemonize]
        server = WEBrick::HTTPServer.new(server_options)
        server.mount('/', WebrickServlet, self)
        trap("INT") { server.shutdown }
        server.start
      end
    end

    # The main WEBrick servlet implementation, accepting only GET requests.
    class WebrickServlet < WEBrick::HTTPServlet::AbstractServlet
      attr_accessor :adapter
      
      def initialize(server, adapter)
        super
        self.adapter = adapter
      end
      
      def do_GET(request, response)
        status, headers, body = *adapter.router.call(request)
        response.status = status
        response.body = body.is_a?(Array) ? body[0] : body
        headers.each do |key, value|
          response[key] = value
        end
      end
    end
  end
end

# @private
class WEBrick::HTTPRequest
  def xhr?; (self['X-Requested-With'] || "").downcase == 'xmlhttprequest' end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
yard-0.6.8	lib/yard/server/webrick_adapter.rb
yard-0.6.7	lib/yard/server/webrick_adapter.rb
yard-0.6.6	lib/yard/server/webrick_adapter.rb
yard-0.6.5	lib/yard/server/webrick_adapter.rb
yard-0.6.4	lib/yard/server/webrick_adapter.rb