Module:HeroData/nav: Difference between revisions

Browse history interactively

← Older edit

@@ Line 1: / Line 1: @@
 local p = {}
 local heroes_data = mw.loadJsonData("Data:HeroData.json")
+local hero_release_module = require('Module:HeroData/release')
 local lang_module = require('Module:Lang')
--- for [[Template:Hero Navbox]], i.e. {{Hero Card Nav|Hero=Abrams}} {{Hero Card Nav|Hero=Bebop}}
+-- for [[Template:Deadlock_Wiki/Heroes]] on home page, i.e. [[Template:Hero card2]]
 function p.get_hero_nav_cards(frame)
-	local current_or_upcoming = frame.args[1]
+	local in_herolabs = frame.args['in_herolabs'] --true or false
-	if current_or_upcoming ~= 'Current' and current_or_upcoming ~= 'Upcoming' then
+	local text_size = frame.args['text_size']
-		return 'First parameter must be Current or Upcoming'
+	local card_size = frame.args['card_size']
+	local sticker_size = frame.args['sticker_size']
+	local herolabs_text_size = frame.args['herolabs_text_size']
+	local debug_mode = frame.args['debug_mode']
+	local lang_code = frame.args['lang_code']
+	-- Turn debug_mode from string to bool, default to false
+	if debug_mode == nil then
+		debug_mode = false
+	elseif debug_mode == 'true' then
+		 debug_mode = true
+	elseif debug_mode == 'false' then
+		debug_mode = false
+	end
+	if in_herolabs ~= 'true' and in_herolabs ~= 'false' then
+		return 'in_herolabs must be either true or false'
 	end
-	local ret = ''
+	local subpage = ''
+	if lang_code ~= nil and lang_code ~= 'en' then subpage = '/' .. lang_code end
+	--Iterate heroes
+	local heroes = {}
 	for hero_key, hero_data in pairs(heroes_data) do
-		local card
+		if hero_data["Name"] ~= nil then
+			-- if herolabs is requested, only needs to have herolabs=true
-		if hero_data["Name"] ~= nil and hero_data["IsDisabled"] == false then
+			-- otherwise, needs to be not disabled, in development, nd herolabs is nil or false
-			if (hero_data["InDevelopment"] == false and current_or_upcoming == 'Current' or hero_data["InDevelopment"] == true and current_or_upcoming == 'Upcoming') then
+			local this_hero_status = hero_release_module.get_hero_release_status(hero_key)
-				card = '{{Hero Card Nav|Hero=' .. lang_module.get_string(hero_key) .. '}}'
-				ret = ret .. card .. ' '
+			if this_in_herolabs == nil then this_in_herolabs = false end
+			if ((in_herolabs == 'true' and this_hero_status == 'HeroLabs') or
+				(in_herolabs == 'false' and this_hero_status == 'Released')) then
+				local is_recommended = hero_data['IsRecommended']
+				if is_recommended == nil then
+					is_recommended = false
+				end
+				local sub_page_str = ''
+				local hero_str = '|name=' .. lang_module.get_string(hero_key, nil, 'en') .. '|link=' .. lang_module.get_string(hero_key, 'en') .. '|subpage=' .. subpage .. '|recommended=' .. tostring(is_recommended)
+				table.insert(heroes, hero_str)
 			end
 		end
 	end
-	return frame:preprocess(ret)
+	--Order list alphabetically
+	table.sort(heroes) --O(nlogn)
+	--Add each item to output
+	local ret = ''
+	for index, hero_name in ipairs(heroes) do
+		ret = ret .. '{{Hero card2' .. hero_name
+		ret = ret .. '|text_size=' .. text_size
+		ret = ret .. '|herolabs=' .. in_herolabs
+		ret = ret .. '|card_size=' .. card_size
+		ret = ret .. '|sticker_size=' .. sticker_size
+		ret = ret .. '|herolabs_text_size=' .. herolabs_text_size
+		ret = ret .. '}} '
+	end
+	--Remove the last space
+	ret = string.sub(ret, 1, -(string.len(' '))-1)
+	if debug_mode then
+		return ret
+	else -- evaluate wikitext if not in debug mode
+		return frame:preprocess(ret)
+	end
 end
 return p