Module:AttributeData: Difference between revisions

Browse history interactively

← Older edit

@@ Line 1: / Line 1: @@
 local p = {};
-local data = mw.loadJsonData("Data:AttributeData.json")
+local attributes_data = mw.loadJsonData("Data:AttributeData.json")
+local lang_module = require("Module:Lang")
+local category_data = { --category specific parameters
+	Weapon = {
+		unlocalized_name = "CitadelCategoryWeapon",
+		rgb = "213, 144, 63"
+	},
+	Vitality = {
+		unlocalized_name = "CitadelCategoryArmor",
+		rgb = "116, 176, 28"
+	},
+	Spirit = {
+		unlocalized_name = "CitadelCategoryTech",
+		rgb = "194, 136, 240"
+	},
+}
+local brown_icons = { --color these attributes brown instead of grey
+	BaseWeaponDamageIncrease = true,
+	BulletDamage = true,
+	DPS = true,
+	LightMeleeDamage = true,
+	HeavyMeleeDamage = true
+}
+function p.get_category_data()
+	return category_data
+end
+-- returns the color (x) for {{Icon/x}} for a given color using brown_icons
+function p.get_attr_icon_color(attr_key)
+	color_it_brown = brown_icons[attr_key]
+	if (color_it_brown == nil) then
+		return "Grey"
+	else
+		return "Brown"
+	end
+end
 -- returns the table of a specific item
 function get_json_item(name)
-	for i,v in pairs(data) do
+	for i,v in pairs(attributes_data) do
 		if (i == name) then
 			return v
@@ Line 12: / Line 48: @@
 end
-function isElementInArray(array, element)
+-- localize Weapon/Vitality/Spirit from english rather than their keys
-    for _, value in ipairs(array) do
+--{{#invoke:AttributeData|localize_category|Weapon/Vitality/Spirit}}
-        if value == element then
+p.localize_category = function(frame)
-            return true
+	local category_en = frame.args[1]
-        end
+	if (category_en == nil) then return "Weapon/Vitality/Spirit category not provided" end
-    end
-    return false
-end
---{{#invoke:AttributeData|get_attr_category|ATTR_NAME}}--
-p.get_attr_category = function(frame)
-	local attr_name = frame.args[1]
-	for category, attributes in pairs(data) do
+	-- get unlocalized key
-		if isElementInArray(attributes, attr_name) then
+	local unlocalized_key = nil
-			return category
+	for category, cat_data in pairs(category_data) do
+		if category == category_en then
+			unlocalized_key = cat_data["unlocalized_name"]
 		end
 	end
-	return nil
+	if (unlocalized_key == nil) then return "Category should be one of Weapon/Vitality/Spirit" end
+	-- localize it
+	local localized = lang_module.get_string(unlocalized_key)
+	if (localized == nil) then return category_en end --category missing from lang
+	return localized
 end
 return p