Add loadout warning message and support for loadout import prefix

Signed-off-by: Tomas Slusny <slusnucky@gmail.com>

Author: deathbeam

src/Classes/ImportTab.lua

@@ -315,7 +315,20 @@ You can get this from your web browser's cookies while logged into the Path of E
 				self.build.viewMode = "TREE"
 			end)
 		elseif self.controls.importCodeMode.selIndex == 3 then
-			self.build:ImportLoadouts(self.importCodeXML)
+			local controls = { }
+			t_insert(controls, new("LabelControl", nil, {0, 20, 0, 16}, colorCodes.WARNING.."Warning:^7 Importing many loadouts into the same build"))
+			t_insert(controls, new("LabelControl", nil, {0, 36, 0, 16}, "may cause performance issues. Use with caution."))
+			t_insert(controls, new("LabelControl", nil, {0, 64, 0, 16}, "^7Prefix for imported loadouts (optional):"))
+			controls.prefix = new("EditControl", nil, {0, 84, 350, 20}, "", nil, nil, 50)
+			controls.import = new("ButtonControl", nil, {-45, 114, 80, 20}, "Import", function()
+				local prefix = controls.prefix.buf
+				main:ClosePopup()
+				self.build:ImportLoadouts(self.importCodeXML, prefix ~= "" and prefix or nil)
+			end)
+			t_insert(controls, new("ButtonControl", nil, {45, 114, 80, 20}, "Cancel", function()
+				main:ClosePopup()
+			end))
+			main:OpenPopup(380, 144, "Import Loadouts", controls, "import")
 		else
 			self.build:Shutdown()
 			self.build:Init(false, "Imported build", self.importCodeXML, false, self.importCodeSite and self.controls.importCodeIn.buf or nil)

src/Classes/ItemsTab.lua

@@ -958,7 +958,7 @@ holding Shift will put it in the second.]])
 	self.lastSlot = self.slots[baseSlots[#baseSlots]]
 end)
 
-function ItemsTabClass:Load(xml, dbFileName, appendItems)
+function ItemsTabClass:Load(xml, dbFileName, appendItems, prefix)
 	if not appendItems then
 		self.activeItemSetId = 0
 		self.itemSets = { }
@@ -1037,7 +1037,11 @@ function ItemsTabClass:Load(xml, dbFileName, appendItems)
 			end
 		elseif node.elem == "ItemSet" then
 			local itemSet = self:NewItemSet(not appendItems and tonumber(node.attrib.id) or nil)
-			itemSet.title = node.attrib.title
+			local title = node.attrib.title
+			if appendItems and prefix then
+				title = prefix .. (title or "")
+			end
+			itemSet.title = title
 			itemSet.useSecondWeaponSet = node.attrib.useSecondWeaponSet == "true"
 			for _, child in ipairs(node) do
 				if child.elem == "Slot" then

src/Classes/PassiveSpec.lua

@@ -83,8 +83,12 @@ function PassiveSpecClass:Init(treeVersion, convert)
 	self.hashOverrides = { }
 end
 
-function PassiveSpecClass:Load(xml, dbFileName)
-	self.title = xml.attrib.title
+function PassiveSpecClass:Load(xml, dbFileName, prefix)
+	local title = xml.attrib.title
+	if prefix then
+		title = prefix .. (title or "")
+	end
+	self.title = title
 	local url
 	for _, node in pairs(xml) do
 		if type(node) == "table" then

src/Classes/SkillsTab.lua

@@ -374,7 +374,7 @@ function SkillsTabClass:LoadSkill(node, skillSetId)
 	t_insert(self.skillSets[skillSetId].socketGroupList, socketGroup)
 end
 
-function SkillsTabClass:Load(xml, fileName, appendSkills)
+function SkillsTabClass:Load(xml, fileName, appendSkills, prefix)
 	if not appendSkills then
 		self.activeSkillSetId = 0
 		self.skillSets = { }
@@ -416,7 +416,11 @@ function SkillsTabClass:Load(xml, fileName, appendSkills)
 
 		if node.elem == "SkillSet" then
 			local skillSet = self:NewSkillSet(not appendSkills and tonumber(node.attrib.id) or nil)
-			skillSet.title = node.attrib.title
+			local title = node.attrib.title
+			if appendSkills and prefix then
+				title = prefix .. (title or "")
+			end
+			skillSet.title = title
 			t_insert(self.skillSetOrderList, skillSet.id)
 			for _, subNode in ipairs(node) do
 				self:LoadSkill(subNode, skillSet.id)

src/Classes/TreeTab.lua

@@ -444,7 +444,7 @@ function TreeTabClass:GetSpecList()
 	return newSpecList
 end
 
-function TreeTabClass:Load(xml, dbFileName, appendSpecs, itemIdMap)
+function TreeTabClass:Load(xml, dbFileName, appendSpecs, prefix, itemIdMap)
 	if not appendSpecs then
 		self.specList = { }
 	end
@@ -482,7 +482,7 @@ function TreeTabClass:Load(xml, dbFileName, appendSpecs, itemIdMap)
 				end
 
 				local newSpec = new("PassiveSpec", self.build, node.attrib.treeVersion or defaultTreeVersion)
-				newSpec:Load(node, dbFileName)
+				newSpec:Load(node, dbFileName, prefix)
 				t_insert(self.specList, newSpec)
 			end
 		end

src/Modules/Build.lua

@@ -1881,7 +1881,7 @@ function buildMode:LoadDBFile()
 	return self:LoadDB(xmlText, self.dbFileName)
 end
 
-function buildMode:ImportLoadouts(xmlText)
+function buildMode:ImportLoadouts(xmlText, prefix)
 	-- Parse the XML
 	local dbXML = self:ParseXML(xmlText, self.dbFileName)
 	if not dbXML then
@@ -1930,19 +1930,20 @@ function buildMode:ImportLoadouts(xmlText)
 
 	local itemIdMap = { }
 	if itemsSection then
-		itemIdMap = self.itemsTab:Load(itemsSection, self.dbFileName, true) or { }
+		itemIdMap = self.itemsTab:Load(itemsSection, self.dbFileName, true, prefix) or { }
 	end
 	if skillsSection then
-		self.skillsTab:Load(skillsSection, self.dbFileName, true)
+		self.skillsTab:Load(skillsSection, self.dbFileName, true, prefix)
 	end
 	if treeSection then
-		self.treeTab:Load(treeSection, self.dbFileName, true, itemIdMap)
+		self.treeTab:Load(treeSection, self.dbFileName, true, prefix, itemIdMap)
 		self.treeTab:PostLoad()
 	end
 
 	-- Mark build as modified
 	self.modFlag = true
 	self.buildFlag = true
+	self:SyncLoadouts()
 
 	-- Show success message
 	local message = "Successfully imported:\n"