Editing Module:AttributeData

local p = {};
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(attributes_data) do
		if (i == name) then
			return v
		end
	end
	return nil
end

-- localize Weapon/Vitality/Spirit from english rather than their keys
--{{#invoke:AttributeData|localize_category|Weapon/Vitality/Spirit}}
p.localize_category = function(frame)
	local category_en = frame.args[1]
	if (category_en == nil) then return "Weapon/Vitality/Spirit category not provided" end
	
	-- get unlocalized key
	local unlocalized_key = nil
	for category, cat_data in pairs(category_data) do
		if category == category_en then
			unlocalized_key = cat_data["unlocalized_name"]
		end
	end
	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