-- GridStatus.lua

--{{{ Libraries

local RL = AceLibrary("RosterLib-2.0")

--{{{ GridStatus

GridStatus = Grid:NewModule("GridStatus", "AceModuleCore-2.0")
GridStatus:SetModuleMixins("AceDebug-2.0", "AceEvent-2.0")

--{{{ Module prototype

GridStatus.modulePrototype.core = GridStatus
GridStatus.modulePrototype.debugFrame = GridStatus.debugFrame

function GridStatus.modulePrototype:OnInitialize()
	self.debugging = Grid.debugging
	self.debugFrame = Grid.debugFrame
end

function GridStatus.modulePrototype:InitializeOptions()
	local module = self
	if not self.options then
		self.options = {
			type = "group",
			name = (self.menuName or self.name),
			desc = "Options for ".. self.name,
			args = {},
		}

	end

	if self.extraOptions then
		for name, option in pairs(self.extraOptions) do
			self.options.args[name] = option
		end
	end
end

function GridStatus.modulePrototype:RegisterStatus(status, desc, options)
	local module = self

	GridStatus:RegisterStatus(status, desc, (self.name or true))

	if not self.options then
		self:InitializeOptions()
	end

	if not self.options.args[status] then
		self.options.args[status] = {
			type = "group",
			name = desc,
			desc = "Send ".. desc,
			args = {
				["enable"] = {
					type = "toggle",
					name = "Enable",
					desc = "Enable ".. desc,
					get = function ()
						      return module.db.profile[status].enable
					      end,
					set = function (v)
						      module.db.profile[status].enable = v
					      end,
				},
				["color"] = {
					type = "color",
					name = "Color",
					desc = "Color for ".. desc,
					hasAlpha = true,
					get = function ()
						      local color = module.db.profile[status].color
						      return color.r, color.g, color.b, color.a
					      end,
					set = function (r, g, b, a)
						      local color = module.db.profile[status].color
						      color.r = r
						      color.g = g
						      color.b = b
						      color.a = a or 1
					      end,
				},
				["priority"] = {
					type = "range",
					name = "Priority",
					desc = "Priority for ".. desc,
					max = 99,
					min = 0,
					step = 1,
					get = function ()
						      return module.db.profile[status].priority
					      end,
					set = function (v)
						      module.db.profile[status].priority = v
					      end,
				},
				["range"] = {
					type = 'toggle',
					name = "Range filter",
					desc = "Range filter for".. desc,
					get = function() return module.db.profile[status].range end,
					set = function()
						module.db.profile[status].range = not module.db.profile[status].range
					end,
				}
			},
		}

		if options then
			for name, option in pairs(options) do
				if not option then
					self.options.args[status].args[name] = nil
				else
					self.options.args[status].args[name] = option
				end
			end
		end
	end
end

--}}}

--{{{ AceDB defaults

GridStatus.defaultDB = {
	debug = true,
	range = true,
}

--}}}
--{{{ AceOptions table

GridStatus.options = {
	type = "group",
	name = "Status",
	desc = "Options for GridStatus.",
	args = {
	},
}

--}}}

function GridStatus:OnInitialize()
	self.super.OnInitialize(self)
	self.registry = Compost:Acquire()
	self.cache = Compost:Acquire()
end

--{{{ Status registry

function GridStatus:RegisterStatus(status, moduleName)
	if not self.registry then
		self.registry = Compost:Acquire()
	end

	if not self.registry[status] then
		self:Debug("Registered", status, "for", moduleName)
		self.registry[status] = (moduleName or true)
		self:TriggerEvent("Grid_StatusRegistered", status, moduleName)
	else
		-- error if status is already registered?
		self:Debug("RegisterStatus:", status, "is already registered.")
	end
end

function GridStatus:UnregisterStatus(status, moduleName)
	local name

	if self:IsStatusRegistered(status) then
		self:Debug("Unregistered", status, "for", moduleName)
		-- need to remove from cache
		for name in pairs(self.cache) do
			self:SendStatusLost(name, status)
		end

		-- now we can remove from registry
		self.registry[status] = nil
		self:TriggerEvent("Grid_StatusUnregistered", status)
	end
end

function GridStatus:IsStatusRegistered(status)
	return (self.registry and
		self.registry[status] and
		true)
end

function GridStatus:RegisteredStatusIterator()
	local status
	local gsreg = self.registry
	return function ()
		status = next(gsreg, status)
		return status, gsreg[status]
	end
end

--}}}
--{{{ Caching status functions

function GridStatus:SendStatusGained(name, status, priority, range, color, text,  value, maxValue, texture)
	local u = RL:GetUnitObjectFromName(name)
	local cached

	-- ignore unit if it is not in the roster
	if not u then
		return
	end

	if color and not type(color) == "table" then
		self:Debug("color is not a table for", status)
	end

	if range and type(range) ~= "number" then
		self:Debug("Range is not a number for", status)
	end

	-- create cache for unit if needed
	if not self.cache[name] then
		self.cache[name] = Compost:Acquire()
	end

	if not self.cache[name][status] then
		self.cache[name][status] = Compost:Acquire()
	end

	cached = self.cache[name][status]

	-- if no changes were made, return rather than triggering an event
	if cached and
		cached.priority == priority and
		cached.range == range and
		cached.color == color and
		cached.text == text and
		cached.value == value and
		cached.maxValue == maxValue and
		cached.texture == texture then

		return
	end

	-- update cache
	cached.priority = priority
	cached.range = range
	cached.color = color
	cached.text = (text or "")
	cached.value = value
	cached.maxValue = maxValue
	cached.texture = texture

	self:TriggerEvent("Grid_StatusGained", name, status,
			  priority, range, color, text, value, maxValue,
			  texture)

end

function GridStatus:SendStatusLost(name, status)

	-- if status isn't cached, don't send status lost event
	if (not self.cache[name]) or (not self.cache[name][status]) then
		return
	end

	Compost:Reclaim(self.cache[name][status])
	self.cache[name][status] = nil

	self:TriggerEvent("Grid_StatusLost", name, status)
end

function GridStatus:GetCachedStatus(name, status)
	return (self.cache[name] and self.cache[name][status])
end

--}}}
--}}}