-- the squid  controller

module (..., package.seeall)

default_action = "general"

general = function( self )

	local info = { status = { value = "stopped" }, version = { value = self.model.get_dansguardian_version() }, srvctrl = { value = srvctrl} };

	local option = { script = ENV["SCRIPT_NAME"],
	                 prefix = self.conf.prefix,
	                 controller = self.conf.controller,
	                 action = self.conf.action,
	                 extra = ""
	               }
	               
	local service = { message="", status="", config="" }
	
	if self.clientdata.srvcmd then
		srvcmd = self.clientdata.srvcmd
		if srvcmd == "start" or srvcmd == "stop" or srvcmd == "restart" then
			service.message = self.model.service_control( srvcmd )
		end
	end
	
	if self.clientdata.cmd then
		if self.clientdata.cmd == "save" then
			local conf = self.clientdata
			local config = { filterip = conf.filterip, filterport = conf.filterport,
			                 proxyip = conf.proxyip, proxyport = conf.proxyport,
			                 accessdeniedaddress = conf.accessdeniedaddress,
			                 naughtynesslimit = conf.naughtynesslimit
			               }
			              
			self.model.update_general_config( config )
		end
	end
	
	service.status = self.model.get_status()
	info.status.value = service.status
	service.config, service.cfgerr = self.model.get_general_config()
	
	return ( cfe ({ option = option, service = service, info = info }) )
end

advanced = function( self )
							
	local info = { status = { value = "stopped" }, version = { value = self.model.get_dansguardian_version() }, srvctrl = { value = srvctrl} };
	
	local option = { script = ENV["SCRIPT_NAME"],
	                 prefix = self.conf.prefix,
	                 controller = self.conf.controller,
	                 action = self.conf.action,
	                 extra = ""
	               }
	               
	local service = { message="", status="", config="" }
	
	if self.clientdata.srvcmd then
		srvcmd = self.clientdata.srvcmd
		if srvcmd == "start" or srvcmd == "stop" or srvcmd == "restart" then
			service.message = self.model.service_control( srvcmd )
		end
	end
	
	service.status = self.model.get_status()
	info.status.value = service.status
	service.config, service.cfgerr = self.model.get_advanced_config()
	
	return ( cfe ({ option = option, service = service, info = info }) )
end

plain = function( self )

	local option = { script = ENV["SCRIPT_NAME"],
	                 prefix = self.conf.prefix,
	                 controller = self.conf.controller,
	                 action = self.conf.action,
	                 extra = ""
	               }
	               
	local service = { message="", status="", config="" }
	
	if self.clientdata.srvcmd then
		srvcmd = self.clientdata.srvcmd
		if srvcmd == "start" or srvcmd == "stop" or srvcmd == "restart" then
			service.message = self.model.service_control( srvcmd )
		end
	end
	
	if self.clientdata.cmd then
		if self.clientdata.cmd == "save" then
			local conf = self.clientdata
			local config = conf.config
			self.model.update_plain_config( config )
		end
	end

	service.status = self.model.get_status()
	service.config, service.cfgerr = self.model.get_plain_config()
	
	return ( cfe ({ option = option, service = service }) )
end

edit = function( self )

	if not self.clientdata.name then
		redirect( self )
	end

	local option = { script = ENV["SCRIPT_NAME"],
	                 prefix = self.conf.prefix,
	                 controller = self.conf.controller,
	                 action = self.conf.action,
	                 extra = ""
	               }
	               
	local service = { message="", status="", config="" }

	if self.clientdata.cmd then
		if self.clientdata.cmd == "save" then 
			self.model.update_edit_config( self.clientdata.name, self.clientdata.config )
		end
	end	
	
	service.status = self.model.get_status()
	service.config, service.cfgerr = self.model.get_edit_config( self.clientdata.name )
	service.name = self.clientdata.name
	
	if service.cfgerr == "Hacker" then
		redirect( self )
	end
	
	return ( cfe ({ option = option, service = service }) )
end

category = function( self )

	local option = { script = ENV["SCRIPT_NAME"],
	                 prefix = self.conf.prefix,
	                 controller = self.conf.controller,
	                 action = self.conf.action,
	                 extra = ""
	               }
	               
	local service = { message="", status="", config="", cfgerr="" }
	
	service.config = self.model.get_categories()
	
	return ( cfe ({ option = option, service = service }) )
end