#!/usr/bin/env ruby

require 'uri'
require 'pathname'

require 'mongrel2' unless defined?( Mongrel2 )
require 'mongrel2/config' unless defined?( Mongrel2::Config )
require 'mongrel2/constants'


# Mongrel2 Server configuration class
class Mongrel2::Config::Server < Mongrel2::Config( :server )
	include Mongrel2::Constants

	### As of Mongrel2/1.8.0:
	# CREATE TABLE server (id INTEGER PRIMARY KEY,
	#     uuid TEXT,
	#     access_log TEXT,
	#     error_log TEXT,
	#     chroot TEXT DEFAULT '/var/www',
	#     pid_file TEXT,
	#     default_host TEXT,
	#     name TEXT DEFAULT '',
	#     bind_addr TEXT DEFAULT "0.0.0.0",
	#     port INTEGER,
	#     use_ssl INTEGER default 0);

	##
	# Return the dataset for looking up a server by its UUID.
	# :singleton-method: by_uuid
	# :call-seq:
	#    by_uuid( uuid )
	dataset_module do
		def by_uuid( uuid )
			filter(:uuid => uuid).limit(1)
		end
	end


	##
	# The hosts[rdoc-ref:Mongrel2::Config::Host] that belong to this server.
	one_to_many :hosts

	##
	# The filters[rdoc-ref:Mongrel2::Config::Filter] that will be loaded by this server.
	one_to_many :filters


	##
	# :method: uuid
	# Get the server identifier, which is typically a UUID, but can
	# in reality be any string of alphanumeric characters and dashes.

	##
	# :method: uuid=( newuuid )
	# Set the server identifier.

	##
	# :method: access_log
	# Get the path to the server's access log as a String.

	### Get the path to the server's access log as a Pathname
	def access_log_path
		path = self.access_log or return nil
		return Pathname( path )
	end


	##
	# :method: error_log
	# Get the path tot he server's error log as a String.

	### Get the path to the server's error log as a Pathname
	def error_log_path
		path = self.error_log or return nil
		return Pathname( path )
	end


	##
	# :method: chroot
	# Get the name of the directory Mongrel2 will chroot to if run as root
	# as a String.

	### Return a Pathname for the server's chroot directory.
	def chroot_path
		path = self.chroot or return nil
		return Pathname( path )
	end


	##
	# :method: pid_file
	# Get the path to the server's PID file as a String.

	### The path to the server's PID file as a Pathname.
	def pid_file_path
		path = self.pid_file or return nil
		return Pathname( path )
	end


	##
	# :method: default_host
	# Get the name of the default virtualhost for the server. If none
	# of the hosts' names (or matching pattern) matches the request's Host:
	# header, the default_host will be used.

	##
	# :method: name
	# The huamn-readable name of the server.

	##
	# :method: bind_addr
	# The address to bind to on startup.

	##
	# :method: port
	# The port to listen on.


	### Returns +true+ if the server uses SSL.
	def use_ssl?
		return self.use_ssl.nonzero?
	end


	### If +enabled+, the server will use SSL.
	def use_ssl=( enabled )
		if !enabled || enabled == 0
			super( 0 )
		else
			super( 1 )
		end
	end


	### Return the URI for its control socket.
	def control_socket_uri
		# Find the control socket relative to the server's chroot
		csock_uri = Mongrel2::Config.settings[:control_port] || DEFAULT_CONTROL_SOCKET
		self.log.debug "Chrooted control socket uri is: %p" % [ csock_uri ]

		scheme, sock_path = csock_uri.split( '://', 2 )
		self.log.debug "  chrooted socket path is: %p" % [ sock_path ]

		csock_path = self.chroot_path + sock_path
		if csock_path.socket?
			self.log.debug "  socket path is relative to the chroot: %p" % [ csock_path ]
		else
			csock_path = Pathname.pwd + sock_path
			raise "Unable to find the socket path %p" % [ csock_uri ] unless csock_path.socket?
			self.log.debug "  socket path is relative to the PWD: %p" % [ csock_path ]
		end

		csock_uri = "%s://%s" % [ scheme, csock_path ]

		self.log.debug "  control socket URI is: %p" % [ csock_uri ]
		return csock_uri
	end


	### Return the Mongrel2::Control object for the server's control socket.
	def control_socket
		return Mongrel2::Control.new( self.control_socket_uri )
	end


	### Sequel validation callback: add errors if the record is invalid.
	def validate
		self.validates_presence [ :access_log, :error_log, :pid_file, :default_host, :port ],
			message: 'is missing or nil'
	end


	### Stringification method -- return a human-readable description of the server.
	def to_s
		return "%s {%s} %s:%d" % [ self.name, self.uuid, self.bind_addr, self.port ]
	end



	### DSL methods for the Server context besides those automatically-generated from its
	### columns.
	module DSLMethods

		### Add a Mongrel2::Config::Host to the Server object with the given +hostname+. If a
		### +block+ is specified, it can be used to further configure the Host.
		def host( name, &block )
			self.target.save( :validate => false )

			self.log.debug "Host [%s] (block: %p)" % [ name, block ]
			adapter = Mongrel2::Config::DSL::Adapter.new( Mongrel2::Config::Host, name: name )
			adapter.target.matching = name
			adapter.instance_eval( &block ) if block
			self.target.add_host( adapter.target )
		end


		### Add a Mongrel2::Config::Filter to the Server object with the specified
		### +path+ (name) and +settings+ hash.
		def filter( path, settings={} )
			self.target.save( :validate => false )

			self.log.debug "Filter [%s]: %p" % [ path, settings ]
			self.target.add_filter( name: path, settings: settings )
		end

	end # module DSLMethods

end # class Mongrel2::Config::Server