local proto = argv[1] or 'all'

local service_name
if proto == 'all' then
	service_name = 'blf'
else
	service_name = proto
end

require "resources.functions.config"
require "resources.functions.split"
require "resources.functions.trim";

local log = require "resources.functions.log"[service_name]
local presence_in = require "resources.functions.presence_in"
local Database = require "resources.functions.database"
local BasicEventService = require "resources.functions.basic_event_service"

local find_voicemail do

	local find_voicemail_sql = [[select t1.voicemail_message_uuid
	from v_voicemail_messages t1
	inner join v_domains t2 on t1.domain_uuid = t2.domain_uuid
	inner join v_voicemails t3 on t1.voicemail_uuid = t3.voicemail_uuid
	where t2.domain_name = :domain_name and t3.voicemail_id = :extension and t1.message_status != 'saved']]
	
	function find_voicemail(user)
		local ext, domain_name = split_first(user, '@', true)
		log.notice("ext: " .. ext);
		log.notice("domain_name: " .. domain_name);
		if not domain_name then return end
		local dbh = Database.new('system')
		if not dbh then return end
		local voicemail = dbh:first_row(find_voicemail_sql, {domain_name = domain_name, extension = ext})
		dbh:release()
		return voicemail
	end
	
end

local find_call_flow do

local find_call_flow_sql = [[select t1.call_flow_uuid, t1.call_flow_status
from v_call_flows t1 inner join v_domains t2 on t1.domain_uuid = t2.domain_uuid
where t2.domain_name = :domain_name and (t1.call_flow_feature_code = :feature_code
or t1.call_flow_feature_code = :short_feature_code)
]]

function find_call_flow(user)
	local ext, domain_name = split_first(user, '@', true)
	local _, short = split_first(ext, '+', true)
	if not domain_name then return end
	local dbh = Database.new('system')
	if not dbh then return end
	local row = dbh:first_row(find_call_flow_sql, {
		domain_name = domain_name, feature_code = ext, short_feature_code = short
	})
	dbh:release()
	if not row then return end
	return row.call_flow_uuid, row.call_flow_status
end

end

local find_dnd do

local find_dnd_sql = [[select t1.do_not_disturb
from v_extensions t1 inner join v_domains t2 on t1.domain_uuid = t2.domain_uuid
where t2.domain_name = :domain_name and (t1.extension = :extension or t1.number_alias=:extension)]]

find_dnd = function(user)
	local ext, domain_name = split_first(user, '@', true)
	if not domain_name then return end
	local dbh = Database.new('system')
	if not dbh then return end
	local dnd = dbh:first_value(find_dnd_sql, {domain_name = domain_name, extension = ext})
	dbh:release()
	return dnd
end

end

local find_call_forward do

local find_call_forward_sql = [[select t1.forward_all_destination, t1.forward_all_enabled
from v_extensions t1 inner join v_domains t2 on t1.domain_uuid = t2.domain_uuid
where t2.domain_name = :domain_name and (t1.extension = :extension or t1.number_alias=:extension)]]

find_call_forward = function(user)
	local ext, domain_name, number = split_first(user, '@', true)
	if not domain_name then return end
	ext, number = split_first(ext, '/', true)
	local dbh = Database.new('system')
	if not dbh then return end
	local row = dbh:first_row(find_call_forward_sql, {domain_name = domain_name, extension = ext})
	dbh:release()
	if not (row and row.forward_all_enabled) then return end
	if row.forward_all_enabled ~= 'true' then return 'false' end
	if number then
		return number == row.forward_all_destination and 'true' or 'false',
			row.forward_all_destination
	end
	return 'true', row.forward_all_destination
end

end

local find_agent_status do

local find_agent_uuid_sql = [[select t1.call_center_agent_uuid
from v_call_center_agents t1 inner join v_domains t2 on t1.domain_uuid = t2.domain_uuid
where t2.domain_name = :domain_name and t1.agent_name = :agent_name
]]

function find_agent_status(user)
	local agent_name, domain_name = split_first(user, '@', true)
	local _, short = split_first(agent_name, '+', true)
	if not domain_name then return end
	local dbh = Database.new('system')
	if not dbh then return end
	local row = dbh:first_row(find_agent_uuid_sql, {
		domain_name = domain_name, agent_name = agent_name
	})
	dbh:release()
	if not row then return end
	if row.call_center_agent_uuid then
		local cmd = "callcenter_config agent get status "..row.call_center_agent_uuid.."";
		freeswitch.consoleLog("notice", "[user status][login] "..cmd.."\n");
		user_status = trim(api:executeString(cmd));
	end
	return row.call_center_agent_uuid, user_status
end

end

local protocols = {}

protocols.voicemail = function(event)
	local from, to = event:getHeader('from'), event:getHeader('to')
	local expires = tonumber(event:getHeader('expires'))
	if expires and expires > 0 then
		local proto, user = split_first(to, '+', true)
		local voicemail_status = find_voicemail(user)
		if voicemail_status then
			log.noticef("Find VOICEMAIL: %s status: %s", to, tostring(voicemail_status))
			presence_in.turn_lamp(true, to)
		else
			log.warningf("Can not find VOICEMAIL: %s", to)
			presence_in.turn_lamp(false, to)
		end
	else
		log.noticef("%s UNSUBSCRIBE from %s", from, to)
	end
end

protocols.flow = function(event)
	local from, to = event:getHeader('from'), event:getHeader('to')
	local expires = tonumber(event:getHeader('expires'))
	if expires and expires > 0 then
		local call_flow_uuid, call_flow_status = find_call_flow(to)
		if call_flow_uuid then
			log.noticef("Find call flow: %s status: %s", to, tostring(call_flow_status))
			presence_in.turn_lamp(call_flow_status == "false", to, call_flow_uuid)
		else
			log.warningf("Can not find call flow: %s", to)
		end
	else
		log.noticef("%s UNSUBSCRIBE from %s", from, to)
	end
end

protocols.dnd = function(event)
	local from, to = event:getHeader('from'), event:getHeader('to')
	local expires = tonumber(event:getHeader('expires'))
	if expires and expires > 0 then
		local proto, user = split_first(to, '+', true)
		user = user or proto
		local dnd_status = find_dnd(user)
		if dnd_status then
			log.noticef("Find DND: %s status: %s", to, tostring(dnd_status))
			presence_in.turn_lamp(dnd_status == "true", to)
		else
			log.warningf("Can not find DND: %s", to)
		end
	else
		log.noticef("%s UNSUBSCRIBE from %s", from, to)
	end
end

protocols.forward = function(event)
	local from, to = event:getHeader('from'), event:getHeader('to')
	local expires = tonumber(event:getHeader('expires'))
	if expires and expires > 0 then
		local proto, user = split_first(to, '+', true)
		user = user or proto
		local status, number = find_call_forward(user)
		if status then
			if status == 'true' then
				log.noticef("CF: %s to number %s", to, tostring(number))
			else
				log.noticef("CF: %s disabled", to)
			end
			presence_in.turn_lamp(status == "true", to)
		else
			log.warningf("Can not find CF: %s", to)
		end
	else
		log.noticef("%s UNSUBSCRIBE from %s", from, to)
	end
end

protocols.agent = function(event)
	local from, to = event:getHeader('from'), event:getHeader('to')
	local expires = tonumber(event:getHeader('expires'))
	if expires and expires > 0 then
		local proto, user = split_first(to, '+', true)
		user = user or proto
		local call_center_agent_uuid, agent_status = find_agent_status(user)
		if agent_status then
			log.noticef("Find agent: %s status: %s", user, tostring(agent_status))
			presence_in.turn_lamp(agent_status == "Available", to)
		else
			log.warningf("Can not find agent status: %s", to)
		end
	else
		log.noticef("%s UNSUBSCRIBE from %s", from, to)
	end
end

if proto ~= 'all' then
	for name in pairs(protocols) do
		if proto ~= name then
			protocols[name] = nil
		end
	end
end

if not next(protocols) then
	log.errorf('Unknown subscribe protocol: %s', proto)
	return
end

for name in pairs(protocols) do
	log.noticef('add subscribe protocol: %s', name)
end

local service = BasicEventService.new(log, service_name)

-- FS receive SUBSCRIBE to BLF from device
service:bind("PRESENCE_PROBE", function(self, name, event)
	local proto = event:getHeader('proto')
	local handler = proto and protocols[proto]
	if not handler then return end
	return handler(event)
end)

log.notice("start")

service:run()

log.notice("stop")
