From f640f6fa6ae1757d4059578895e813fac7ee1f8a Mon Sep 17 00:00:00 2001
From: Sumsebrum <kyle@xicon.de>
Date: Mon, 17 May 2021 23:22:07 +0200
Subject: [PATCH] TotemPlates add option to show friendly/enemy icons

---
 Modules/TotemPlates.lua | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/Modules/TotemPlates.lua b/Modules/TotemPlates.lua
index ae4dc2c..91d17bb 100644
--- a/Modules/TotemPlates.lua
+++ b/Modules/TotemPlates.lua
@@ -1,5 +1,5 @@
 local select, pairs, string_lower, tremove, tinsert, format, string_gsub, ipairs = select, pairs, string.lower, tremove, tinsert, format, string.gsub, ipairs
-local UnitExists, UnitIsUnit, UnitName = UnitExists, UnitIsUnit, UnitName
+local UnitExists, UnitIsUnit, UnitName, UnitIsEnemy = UnitExists, UnitIsUnit, UnitName, UnitIsEnemy
 local C_NamePlate = C_NamePlate
 local Gladdy = LibStub("Gladdy")
 local L = Gladdy.L
@@ -222,6 +222,8 @@ end
 
 local TotemPlates = Gladdy:NewModule("TotemPlates", nil, {
     npTotems = true,
+    npTotemsShowFriendly = true,
+    npTotemsShowEnemy = true,
     npTotemPlatesBorderStyle = "Interface\\AddOns\\Gladdy\\Images\\Border_rounded_blp",
     npTotemPlatesSize = 40,
     npTotemPlatesWidthFactor = 1,
@@ -303,6 +305,13 @@ end
 
 function TotemPlates:NAME_PLATE_UNIT_ADDED(...)
     local unitID = ...
+    local isEnemy = UnitIsEnemy("player", unitID)
+    if not Gladdy.db.npTotemsShowEnemy and isEnemy then
+        return
+    end
+    if not Gladdy.db.npTotemsShowFriendly and not isEnemy then
+        return
+    end
     local nameplateName = UnitName(unitID)
     local totemName = string_gsub(nameplateName, "^%s+", "") --trim
     totemName = string_gsub(totemName, "%s+$", "") --trim
@@ -414,6 +423,21 @@ function TotemPlates:GetOptions()
             name = L["Totem icons on/off"],
             desc = L["Turns totem icons instead of nameplates on or off. (Requires reload)"],
             order = 3,
+            width = 0.9,
+        }),
+        npTotemsShowFriendly = Gladdy:option({
+            type = "toggle",
+            name = L["Show friendly"],
+            desc = L["Turns totem icons instead of nameplates on or off. (Requires reload)"],
+            order = 4,
+            width = 0.65,
+        }),
+        npTotemsShowEnemy = Gladdy:option({
+            type = "toggle",
+            name = L["Show enemy"],
+            desc = L["Turns totem icons instead of nameplates on or off. (Requires reload)"],
+            order = 5,
+            width = 0.6,
         }),
         group = {
             type = "group",