# * George Moschovitis  <gm@navel.gr>
# (c) 2004-2005 Navel, all rights reserved.
# $Id$

require 'webrick'

require 'glue/flexob'

require 'nitro/context'
require 'nitro/dispatcher'

module WEBrick
	
	class HTTPResponse
		def header=(header)
			@header = header
		end
		
		def cookies=(cookies)
			@cookies = cookies
		end
	end

end

module N
	
# Helper methods for the WebrickAdapter.

class Webrick
	
	class << self
		attr_accessor :server

		def start(conf)
			conf = Flexob.new(conf) unless conf.is_a?(Flexob)
			
			# patch for OSX
			
			Socket.do_not_reverse_lookup = true
	 
			if :none == conf.log and RUBY_PLATFORM !~ /mswin32/
				accesslog = WEBrick::BasicLog::new('/dev/null')
				refererlog = WEBrick::BasicLog::new('/dev/null')	
			elsif (conf.accesslog || conf.refererlog)
				accesslog = WEBrick::BasicLog::new(conf.accesslog || 'log/access.log')
				refererlog = WEBrick::BasicLog::new(conf.refererlog || 'log/referer.log')	
			else
				accesslog = STDERR
				refererlog = STDERR
			end
	
			@server = WEBrick::HTTPServer.new(
				:BindAddress => conf.host, 
				:Port => conf.port, 
				:DocumentRoot => conf.dispatcher.root,
				:AccessLog => [
					[accesslog, WEBrick::AccessLog::COMMON_LOG_FORMAT],
					[refererlog, WEBrick::AccessLog::REFERER_LOG_FORMAT]
				]
			)
			
			trap('INT') { @server.shutdown }

			@server.mount('/', WebrickAdapter, conf)
			@server.start
		end
	end

end

# A Webrick Adapter for Nitro.

class WebrickAdapter < WEBrick::HTTPServlet::AbstractServlet

	# REQUEST_MUTEX = Mutex.new
		
  def initialize(server, conf)
		@conf = conf

		# Handles static resources. Useful when running 
		# a stand-alone webrick server.
		
    @file_handler = WEBrick::HTTPServlet::FileHandler.new(
				server, 
				conf.dispatcher.root, conf.webrick_options || {})
  end
	
	def handle(req, res)
		path = req.request_uri.path

    if path =~ /\./
			@file_handler.do_GET(req, res)
		else
			# gmosx, TODO: move this into a filter.
			Og.db.get_connection if defined?(Og)

			# REQUEST_MUTEX.lock

			context = Context.new(@conf)
			
			context.params = req.query 
			context.headers = {}
			req.header.each { |h, v| context.headers[h.upcase] = v.first }
			context.headers.update(req.meta_vars)
			context.cookies = {}
			req.cookies.each { |c| context.cookies[c.name] = c.value }
		
			context.render(path)

			res.status = context.status
			res.header = context.response_headers || {}
			res.cookies = context.response_cookies || {}
			res.body = context.out
		end
	ensure
		# REQUEST_MUTEX.unlock if REQUEST_MUTEX.locked?
		Og.db.put_connection if defined?(Og)
	end

	alias do_GET handle 
	alias do_POST handle

end

end