unitZone={}
unitZone.version = "2.0.1"
unitZone.verbose = false 
unitZone.ups = 1 
unitZone.requiredLibs = {
	"dcsCommon", -- always
	"cfxZones", -- Zones, of course 
}
--[[--
	Version History 
	2.0.0 - matchAll option (internal, automatic look for "*" in names)
		  - lookFor defaults to "*"
		  - OOP dmlZones
		  - uzDirect correctly initialized at start 
		  - synonyms uzDirect#, uzDirectInv#
		  - uzDirectInv better support
		  - unitZone now used to define the coalition, coalition DEPRECATED
		  - filter synonym 
		  - direct#, directInv# synonyms 
	2.0.1 - code hardening 
--]]--

unitZone.unitZones = {}

function unitZone.addUnitZone(theZone)
	table.insert(unitZone.unitZones, theZone)
end

function unitZone.getUnitZoneByName(aName) 
	for idx, aZone in pairs(unitZone.unitZones) do 
		if aName == aZone.name then return aZone end 
	end
	if unitZone.verbose then 
		trigger.action.outText("+++uZne: no unitZone with name <" .. aName ..">", 30)
	end 
	
	return nil 
end

function unitZone.string2cat(filterString)

	if not filterString then return 2 end -- default ground 
	filterString = filterString:lower()
	filterString = dcsCommon.trim(filterString)

	local catNum = tonumber(filterString)
	if catNum then 
		if catNum < 0 then catNum = 0 end 
		if catNum > 4 then catNum = 4 end 
		return catNum 
	end
	
	catNum = 2 -- ground default 
	if dcsCommon.stringStartsWith(filterString, "grou") then catNum = 2 end 
	if dcsCommon.stringStartsWith(filterString, "air") then catNum = 0 end
	if dcsCommon.stringStartsWith(filterString, "hel") then catNum = 1 end
	if dcsCommon.stringStartsWith(filterString, "shi") then catNum = 3 end
	if dcsCommon.stringStartsWith(filterString, "trai") then catNum = 4 end

	return catNum
end

function unitZone.createUnitZone(theZone)
	-- start val - a range
	theZone.lookFor = theZone:getStringFromZoneProperty("lookFor", "*") -- default to match all  
	if theZone.lookFor == "*" then 
		theZone.matchAll = true 
		if theZone.verbose or unitZone.verbose then 
			trigger.action.outText("+++uZne: zone <" .. theZone.name .. "> set up to matche all names", 30)
		end
	elseif dcsCommon.stringEndsWith(theZone.lookFor, "*") then 
		theZone.lookForBeginsWith = true
		theZone.matchAll = false
		theZone.lookFor = dcsCommon.removeEnding(theZone.lookFor, "*") 
	end
	
	theZone.matching = theZone:getStringFromZoneProperty("matching", "group") -- group, player [, name, type]
	theZone.matching = dcsCommon.trim(theZone.matching:lower())
	if theZone.matching == "groups" then theZone.matching = "group" end -- some simplification 
	if theZone.matching == "players" then theZone.matching = "player" end -- some simplification 

	-- coalition 
	theZone.uzCoalition = theZone:getCoalitionFromZoneProperty("unitZone", 0) -- now with main attribute
	-- DEPRECATED 2023 SEPT: provided for legacy compatibility
	if theZone:hasProperty("coalition") then 
		theZone.uzCoalition = theZone:getCoalitionFromZoneProperty("coalition", 0) -- 0 = all
	elseif theZone:hasProperty("uzCoalition") then 
		theZone.uzCoalition = theZone:getCoalitionFromZoneProperty("uzCoalition", 0)
	end
	
	-- DML Method 
	theZone.uzMethod = theZone:getStringFromZoneProperty("method", "inc")
	if theZone:hasProperty("uzMethod") then 
		theZone.uzMethod = theZone:getStringFromZoneProperty("uzMethod", "inc")
	end

	if theZone:hasProperty("enterZone!") then 
		theZone.enterZone = theZone:getStringFromZoneProperty("enterZone!", "*<none>")
	end 
	if theZone:hasProperty("exitZone!") then 
		theZone.exitZone = theZone:getStringFromZoneProperty("exitZone!", "*<none>")
	end 
	
	if theZone:hasProperty("changeZone!") then 
		theZone.changeZone = theZone:getStringFromZoneProperty("changeZone!", "*<none>")
	end 
	
	if theZone:hasProperty("filterFor") then 
		local filterString = theZone:getStringFromZoneProperty( "filterFor", "1") -- ground 
		theZone.filterFor = unitZone.string2cat(filterString)
		if unitZone.verbose or theZone.verbose then 
			trigger.action.outText("+++uZne: filtering " .. theZone.filterFor .. " in " .. theZone.name, 30)
		end 
	elseif theZone:hasProperty("filter") then 
		local filterString = theZone:getStringFromZoneProperty( "filter", "1") -- ground 
		theZone.filterFor = unitZone.string2cat(filterString)
		if unitZone.verbose or theZone.verbose then 
			trigger.action.outText("+++uZne: filtering " .. theZone.filterFor .. " in " .. theZone.name, 30)
		end
	end	
	
	-- uzDirect
	if theZone:hasProperty("uzDirect") then
		theZone.uzDirect = theZone:getStringFromZoneProperty("uzDirect", "*<none>")
	elseif
		theZone:hasProperty("uzDirect#") then
		theZone.uzDirect = theZone:getStringFromZoneProperty("uzDirect#", "*<none>")
	elseif
		theZone:hasProperty("direct#") then
		theZone.uzDirect = theZone:getStringFromZoneProperty("direct#", "*<none>")
	end 
	if theZone:hasProperty("uzDirectInv") then
		theZone.uzDirectInv = theZone:getStringFromZoneProperty("uzDirectInv", "*<none>")
	elseif theZone:hasProperty("uzDirectInv#") then
		theZone.uzDirectInv = theZone:getStringFromZoneProperty("uzDirectInv#", "*<none>")
	elseif theZone:hasProperty("directInv#") then
		theZone.uzDirectInv = theZone:getStringFromZoneProperty("directInv#", "*<none>")
	end 
	
	-- on/off flags
	theZone.uzPaused = false -- we are turned on 
	if theZone:hasProperty("uzOn?") then 
		theZone.triggerOnFlag = theZone:getStringFromZoneProperty("uzOn?", "*<none1>")
		theZone.lastTriggerOnValue = theZone:getFlagValue(theZone.triggerOnFlag)
	end 
	
	if theZone:hasProperty("uzOff?") then 
		theZone.triggerOffFlag = theZone:getStringFromZoneProperty("uzOff?", "*<none2>")
		theZone.lastTriggerOffValue = theZone:getFlagValue(theZone.triggerOffFlag)
	end 
	
	theZone.uzTriggerMethod = theZone:getStringFromZoneProperty("triggerMethod", "change")
	if theZone:hasProperty("uzTriggerMethod") then 
		theZone.uzTriggerMethod = theZone:getStringFromZoneProperty("uzTriggerMethod", "change")
	end 
	
	-- now get initial zone status ?
	theZone.lastStatus = unitZone.checkZoneStatus(theZone)
	if theZone.uzDirect then 
		if newState then 
			theZone:setFlagValue(theZone.uzDirect, 1)
		else 
			theZone:setFlagValue(theZone.uzDirect, 0)
		end
	end
	if theZone.uzDirectInv then 
		if newState then 
			theZone:setFlagValue(theZone.uzDirectInv, 0)
		else 
			theZone:setFlagValue(theZone.uzDirectInv, 1)
		end
	end
	
	if unitZone.verbose or theZone.verbose then 
		trigger.action.outText("+++uZne: processsed unit zone <" .. theZone.name .. "> with status = (" .. dcsCommon.bool2Text(theZone.lastStatus) .. ")", 30)
	end	
end
--
-- process zone 
--
function unitZone.collectGroups(theZone)
	local collector = {} -- players: units, groups: groups
	if theZone.matching == "player" then 
		-- collect all players matching coalition
		if theZone.uzCoalition == 1 or theZone.uzCoalition == 0 then 
			local allPlayers = coalition.getPlayers(1)
			for idx, pUnit in pairs(allPlayers) do 
				table.insert(collector, pUnit)
			end
		end 
		if theZone.uzCoalition == 2 or theZone.uzCoalition == 0 then 
			local allPlayers = coalition.getPlayers(2)
			for idx, pUnit in pairs(allPlayers) do 
				table.insert(collector, pUnit)
			end
		end
	elseif theZone.matching == "group" then 
		if theZone.uzCoalition == 1 or theZone.uzCoalition == 0 then 
			local allGroups = coalition.getGroups(1, theZone.filterFor)
			for idx, aGroup in pairs(allGroups) do 
				table.insert(collector, aGroup)
			end
		end 
		if theZone.uzCoalition == 2 or theZone.uzCoalition == 0 then 
			local allGroups = coalition.getGroups(2, theZone.filterFor)
			for idx, aGroup in pairs(allGroups) do 
				table.insert(collector, aGroup)
			end
		end
	else 
		trigger.action.outText("+++uZne: unknown matching: " .. theZone.matching, 30)
		return {}
	end
	
	return collector
end

function unitZone.checkZoneStatus(theZone)
	-- returns true (at least one unit found in zone)
	-- or false (no unit found in zone)
	-- collect all groups to inspect 
	local theGroups = unitZone.collectGroups(theZone)
	local lookFor = theZone.lookFor
	-- now see if the groups match name and then check inside status for each 
	local playerCheck = theZone.matching == "player"
	if playerCheck then 
		-- we check the names for players only 
		-- collector holds units for players, not groups 
		for idx, pUnit in pairs(theGroups) do
			if Unit.isExist(pUnit) then 
				local puName = pUnit:getName()
				local hasMatch = theZone.matchAll 
				if not hasMatch then 
					if theZone.lookForBeginsWith then 
						hasMatch = dcsCommon.stringStartsWith(puName, lookFor)
					else 
						hasMatch = puName == lookFor 
					end
				end 
				if hasMatch then 
					if cfxZones.unitInZone(pUnit, theZone) then 
						return true
					end
				end
			end 
		end 
        
	else 
		-- we perform group check. 
		for idx, aGroup in pairs(theGroups) do 
			if Group.isExist(aGroup) then 
				local gName=aGroup:getName()
				local hasMatch = theZone.matchAll
				if not hasMatch then 
					if theZone.lookForBeginsWith then 
						hasMatch = dcsCommon.stringStartsWith(gName, lookFor)
					else 
						hasMatch = gName == lookFor 
					end
				end 
				if hasMatch and aGroup:isExist() then 
					-- check all living units in zone 
					local gUnits = aGroup:getUnits()
					for idy, aUnit in pairs (gUnits) do
						if cfxZones.unitInZone(aUnit, theZone) then 
							return true
						end
					end
				end
			end
		end
	end
	return false 
end
--
-- update 
--
function unitZone.bangState(theZone, newState)
	
	if theZone.changeZone then 
		theZone:pollFlag(theZone.changeZone, theZone.uzMethod)
	end 
	if newState then 
		if theZone.enterZone then 
			theZone:pollFlag(theZone.enterZone, theZone.uzMethod)
			if unitZone.verbose then 
				trigger.action.outText("+++uZone: banging enter! with <" .. theZone.uzMethod .. "> on <" .. theZone.enterZone .. "> for " .. theZone.name, 30)
			end
		end
	else 
		if theZone.exitZone then 
			theZone:pollFlag(theZone.exitZone, theZone.uzMethod)
			if unitZone.verbose then 
				trigger.action.outText("+++uZone: banging exit! with <" .. theZone.uzMethod .. "> on <" .. theZone.exitZone .. "> for " .. theZone.name, 30)
			end
		end 
	end
end

function unitZone.update()
	-- call me in a second to poll triggers
	timer.scheduleFunction(unitZone.update, {}, timer.getTime() + 1/unitZone.ups)
		
	for idx, aZone in pairs(unitZone.unitZones) do
		-- check if we need to pause/unpause 
		if aZone.triggerOnFlag and 
		   aZone:testZoneFlag( aZone.triggerOnFlag, aZone.uzTriggerMethod, "lastTriggerOnValue") then 
			if unitZone.verbose or aZone.verbose then 
				trigger.action.outText("+++uZone: turning " .. aZone.name .. " on", 30)
			end 
			aZone.uzPaused = false 
		end
		
		if aZone.triggerOffFlag and 
		   aZone:testZoneFlag(aZone.triggerOffFlag, aZone.uzTriggerMethod, "lastTriggerOffValue") then 
			if unitZone.verbose or aZone.verbose then 
				trigger.action.outText("+++uZone: turning " .. aZone.name .. " OFF", 30)
			end 
			aZone.uzPaused = true 
		end
		
		-- scan all zones 
		if not aZone.uzPaused then 
			local newState = unitZone.checkZoneStatus(aZone) -- returns true if at least one unit in zone 

			if newState ~= aZone.lastStatus then 
				-- bang on change! 
				unitZone.bangState(aZone, newState)
				aZone.lastStatus = newState 
			end
			
			-- output direct state suite
			if aZone.uzDirect then 
				if aZone.verbose or unitZone.verbose then 
					trigger.action.outText("+++uZone: <" .. aZone.name .. "> setting uzDirect <" .. aZone.uzDirect .. "> to ".. dcsCommon.bool2Num(newState), 30)
				end
				if newState then 
					aZone:setFlagValue(aZone.uzDirect, 1)
				else 
					aZone:setFlagValue(aZone.uzDirect, 0)
				end
			end
			if aZone.uzDirectInv then 
				local invState = not newState
				if aZone.verbose or unitZone.verbose then 
					trigger.action.outText("+++uZone: <" .. aZone.name .. "> setting INVuzDirect <" .. aZone.uzDirectInv .. "> to ".. dcsCommon.bool2Num(invState), 30)
				end
				if newState then 
					aZone:setFlagValue(aZone.uzDirectInv, 0)
				else 
					aZone:setFlagValue(aZone.uzDirectInv, 1)
				end
			end
		end 
	end
end
--
-- Config & Start
--
function unitZone.readConfigZone()
	unitZone.name = "unitZoneConfig"
	local theZone = cfxZones.getZoneByName("unitZoneConfig") 
	if not theZone then 
		theZone = cfxZones.createSimpleZone("unitZoneConfig")
	end 
	unitZone.verbose = cfxZones.getBoolFromZoneProperty(theZone, "verbose", false)
	unitZone.ups = cfxZones.getNumberFromZoneProperty(theZone, "ups", 1)
	if unitZone.verbose then 
		trigger.action.outText("+++uZne: read config", 30)
	end 
end

function unitZone.start()
	-- lib check
	if not dcsCommon.libCheck then 
		trigger.action.outText("cfx Unit Zone requires dcsCommon", 30)
		return false 
	end 
	if not dcsCommon.libCheck("cfx Unit Zone", unitZone.requiredLibs) then
		return false 
	end
	-- read config 
	unitZone.readConfigZone()
	-- process cloner Zones 
	local attrZones = cfxZones.getZonesWithAttributeNamed("unitZone")
	for k, aZone in pairs(attrZones) do 
		unitZone.createUnitZone(aZone) -- process attributes
		unitZone.addUnitZone(aZone) -- add to list
	end
	-- start update 
	unitZone.update()
	trigger.action.outText("cfx Unit Zone v" .. unitZone.version .. " started.", 30)
	return true 
end

-- let's go!
if not unitZone.start() then 
	trigger.action.outText("cfx Unit Zone aborted: missing libraries", 30)
	unitZone = nil 
end

--ToDo: add 'neutral' support and add 'both' option 
--ToDo: add API