From 0d5b5644a9a7f3366ca00a3f058e9e0eb72a924f Mon Sep 17 00:00:00 2001
From: Sumsebrum <kyle@xicon.de>
Date: Thu, 29 Jul 2021 14:19:13 +0200
Subject: [PATCH] testmode DR only with enabled DRs

---
 Modules/Diminishings.lua | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/Modules/Diminishings.lua b/Modules/Diminishings.lua
index db48ad6..0541109 100644
--- a/Modules/Diminishings.lua
+++ b/Modules/Diminishings.lua
@@ -257,17 +257,17 @@ end
 
 function Diminishings:Test(unit)
     if Gladdy.db.drEnabled then
-        local spells = { 33786, 118, 8643, 8983 }
-        for i = 1, 4 do
-            if i == 1 then
-                self:AuraFade(unit, spells[i])
-            elseif i == 2 then
-                self:AuraFade(unit, spells[i])
-                self:AuraFade(unit, spells[i])
-            else
-                self:AuraFade(unit, spells[i])
-                self:AuraFade(unit, spells[i])
-                self:AuraFade(unit, spells[i])
+        local limit = {}
+        for spellID,category in pairs(DRData:GetSpells()) do
+            if Gladdy.db.drCategories[category].enabled then
+                if not limit[category] then
+                    limit[category] = { count = 1, limit = math.random(1,3) }
+                else
+                    limit[category].count = limit[category].count + 1
+                end
+                if limit[category].count <= limit[category].limit then
+                    self:AuraFade(unit, spellID)
+                end
             end
         end
     end