From 043da1e40db9497fc13cbe3fe758bfcf514f3e19 Mon Sep 17 00:00:00 2001
From: Max Rashes <mrashes2@gmail.com>
Date: Sat, 29 May 2021 13:48:32 -0500
Subject: [PATCH] Implemented requested adjustements

---
 eid_config.lua      |  3 ++-
 main.lua            |  7 +++----
 mod_config_menu.lua | 21 +++++++++++++++++++++
 3 files changed, 26 insertions(+), 5 deletions(-)

diff --git a/eid_config.lua b/eid_config.lua
index 157252e6..15f147ce 100644
--- a/eid_config.lua
+++ b/eid_config.lua
@@ -239,5 +239,6 @@ EID.DefaultConfig = {
 	["ErrorMessage"] = "[Effect not defined]",
 	["TextColor"] = "ColorEIDText",
 	["ErrorColor"] = "ColorEIDError",
-	["EnableEntityDescriptions"] = true
+	["EnableEntityDescriptions"] = true,
+	["ShowQuality"] = true
 }
\ No newline at end of file
diff --git a/main.lua b/main.lua
index 392c5d78..5890ac11 100644
--- a/main.lua
+++ b/main.lua
@@ -245,7 +245,7 @@ function EID:printDescription(desc)
 	--Display Itemname
 	if EID.Config["ShowItemName"] then
 		local curName = desc.Name
-		local quality = EID.itemWeightsLookup[tonumber(desc.ID)]
+		local quality = tonumber(EID.itemConfig:GetCollectible(tonumber(desc.ID)).Quality)
 		if EID.Config["TranslateItemName"] ~= 2 then
 			local curLanguage = EID.Config["Language"]
 			EID.Config["Language"] = "en_us"
@@ -257,12 +257,11 @@ function EID:printDescription(desc)
 				curName = curName.." ("..englishName..")"
 			end
 		end
-		local withQual = curName
 		if EID.Config["ShowQuality"] then
-			withQual = curName.." - "..quality
+			curName = curName.." - {{Quality"..quality.."}}"
 		end
 		EID:renderString(
-			withQual,
+			curName,
 			renderPos + (Vector(offsetX, -3) * EID.Config["Scale"]),
 			textScale,
 			EID:getNameColor()
diff --git a/mod_config_menu.lua b/mod_config_menu.lua
index 575ff170..b0b5ef5d 100644
--- a/mod_config_menu.lua
+++ b/mod_config_menu.lua
@@ -776,6 +776,27 @@ if MCMLoaded then
 			end
 		}
 	)
+	-------Quality ICON---------
+	MCM.AddSetting(
+		"EID",
+		"Visuals",
+		{
+			Type = ModConfigMenu.OptionType.BOOLEAN,
+			CurrentSetting = function()
+				return EID.Config["ShowQuality"]
+			end,
+			Display = function()
+				local onOff = "False"
+				if EID.Config["ShowQuality"] then
+					onOff = "True"
+				end
+				return "Display Qualtiy Info: " .. onOff
+			end,
+			OnChange = function(currentBool)
+				EID.Config["ShowQuality"] = currentBool
+			end
+		}
+	)
 
 	---------------------------------------------------------------------------
 	---------------------------------Visuals-----------------------------------