Editing Module:HeroData/headcount

local p = {}
local heroes_data = mw.loadJsonData("Data:HeroData.json")

-- {{Heroes count}} access point
function p.get_headcount(frame)
	local hero_type = frame.args[1]
	
	local iter = 0
	for _, v in pairs(heroes_data) do
		if hero_type == 'Released' then
			if (v["InDevelopment"] == false and v["IsDisabled"] == false and v["Name"] ~= nil) then
				iter = iter + 1	
			end
		elseif hero_type == 'HeroLabs' then
			if (v["InHeroLabs"] == true and v["Name"] ~= nil) then
				iter = iter + 1	
			end
		else
			return "get_headcount()'s hero_type parameter must be either 'Released' or 'HeroLabs'"
		end
	end
	return iter
end

return p