-- acf model for squid
-- Copyright(c) 2007 A. Brodmann - Licensed under terms of GPL2
module (..., package.seeall)

dansguardiancfg = "/etc/dansguardian/dansguardian.conf"

get_status = function()

	local retval = "stopped"
	
	local ptr = io.popen( "/bin/pidof dansguardian" )
	local pid = ptr:read( "*a" )
	ptr:close()
	if pid ~= nil then
		if #pid > 1 then
			retval = "running"
		end
	end
	
	return retval
end

service_control = function( control )

	local retval = ""
	
	local ptr = io.popen( "/etc/init.d/dansguardian " .. control, "r" )
	if ptr ~= nil then
		local retmsg = ptr:read( "*a" )
		ptr:close()
		if retmsg ~= nil then
			retval = retmsg
		else
			retval = "service_control(): Failed to read output from initscript!\n"
		end
	else
		retval = "service_control(): Failed to start/stop/restart service!\n"
	end

	return retval
end

get_general_config = function()

	local retval = {}
	local error = ""
	
	retval = { filterip = { label="Filter IP", type="text", value="" },
	           filterport = { label="Filter Port", type="text", value="" },
	           proxyip = { label="Proxy IP", type="text", value="" },
	           proxyport = { label="Proxy Port", type="text", value="" },
	           accessdeniedaddress = { label="AccessDeniedAddress", type="text", value="" },
	           naughtynesslimit = { label="NaughtynessLimit", type="text", value="" }
	         }
	         
	local fptr = io.open( dansguardiancfg, "r" )
	if fptr ~= nil then
		local line = fptr:read( "*l" )
		while line ~= nil do
			if string.sub( line, 1, 1 ) ~= "#" then
				if string.sub( line, 1, 8 ) == "filterip" then
					retval.filterip.value = get_cfg_value( line )
				elseif string.sub( line, 1, 10 ) == "filterport" then
					retval.filterport.value = get_cfg_value( line )
				elseif string.sub( line, 1, 7 ) == "proxyip" then
					retval.proxyip.value = get_cfg_value( line )
				elseif string.sub( line, 1, 9 ) == "proxyport" then
					retval.proxyport.value = get_cfg_value( line )
				elseif string.sub( line, 1, 19 ) == "accessdeniedaddress" then
					retval.accessdeniedaddress.value = get_cfg_value( line )
				end
			end
			line = fptr:read( "*l" )   -- read one config file
		end
		fptr:close()
	else
		error = "Failed to open " .. dansguardiancfg .. " file!"
	end
	           
	return retval, error
end

get_plain_config = function()

	local retval = ""
	local error = ""
	
	local fptr = io.open( dansguardiancfg, "r" )
	if fptr ~= nil then
		retval = fptr:read( "*a" )
		fptr:close()
		if retval == nil then
			retval = ""
			error = "Failed to read " .. dansguardiancfg .. " file!"
		end
	else
		error = "Failed to open " .. dansguardiancfg .. " file!"
	end
	
	return retval, error
end

update_general_config = function( config )

	local retval = ""
	local tmpfilename = os.tmpname()
	local tmpfile = -1
	local cfgptr = -1
	local line = ""

	tmpfile = io.open( tmpfilename, "wb+" )
	if tmpfile == nil then
		return "Failed to create temporary config file!"
	end	
	
	cfgptr = io.open( dansguardiancfg, "r" )
	if cfgptr == nil then
		tmpfile:close()
		os.remove( tmpfilename )
		return "Failed to open " .. dansguardiancfg .. "!"
	end
	
	line = cfgptr:read( "*l" )
	while line ~= nil do
		if string.sub( line, 1, 8 ) == "filterip" then
			tmpfile:write( "filterip = " .. config.filterip .. "\n" )
		elseif string.sub( line, 1, 10 ) == "filterport" then
			tmpfile:write( "filterport = " .. config.filterport .. "\n" )
		elseif string.sub( line, 1, 7 ) == "proxyip" then
			tmpfile:write( "proxyip = " .. config.proxyip .. "\n" )
		elseif string.sub( line, 1, 9 ) == "proxyport" then
			tmpfile:write( "proxyport = " .. config.proxyport .. "\n" )
		elseif string.sub( line, 1, 19 ) == "accessdeniedaddress" then
			tmpfile:write( "accessdeniedaddress = " .. config.accessdeniedaddress .. "\n" )
		else
			tmpfile:write( line .. "\n" )
		end
		line = cfgptr:read( "*l" )
	end
	
	tmpfile:close()
	cfgptr:close()
	
	os.rename( tmpfilename, dansguardiancfg )
	
	return retval
end

update_plain_config = function( config )

	local retval = ""
	local cfgptr = -1
	local error = ""
	
	cfgptr = io.open( dansguardiancfg, "wb+" )
	if cfgptr ~= nil then
		cfgptr:write( config )
		cfgptr:close()
	else
		retval = "Failed to open " .. dansguardiancfg .. " file!"
	end
	
	return retval
end

get_cfg_value = function( str )

	local retval = ""
	local pos = 1
	local found = false
	local found2 = false
	
	while not found and pos < #str -1 do
		if string.sub( str, pos, pos ) == "=" then
		  found = true
		end
		pos = pos + 1
	end
	
	if found then
		pos = pos - 1
		while not found2 and pos < #str -1 do
			if string.sub( str, pos+1, pos+1 ) ~= " " then
				found2 = true
			end
			pos = pos + 1
		end
	end
	
	if found2 then
		retval = string.sub( str, pos )
	end
	
	return retval
end