expose black magic aura for pre-pull actions

wowsims · Aug 14, 2023 · 3af3eb1 · 3af3eb1
1 parent 3305804
commit 3af3eb1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/sim/common/wotlk/enchant_effects.go b/sim/common/wotlk/enchant_effects.go
@@ -8,6 +8,10 @@ import (
 	"github.com/wowsims/wotlk/sim/core/stats"
 )
 
+func CreateBlackMagicProcAura(character *core.Character) *core.Aura {
+	return character.NewTemporaryStatsAura("Black Magic Proc", core.ActionID{SpellID: 59626}, stats.Stats{stats.MeleeHaste: 250, stats.SpellHaste: 250}, time.Second*10)
+}
+
 func init() {
 	// Keep these in order by item ID.
 
@@ -233,7 +237,7 @@ func init() {
 	core.NewEnchantEffect(3790, func(agent core.Agent) {
 		character := agent.GetCharacter()
 
-		procAura := character.NewTemporaryStatsAura("Black Magic Proc", core.ActionID{SpellID: 59626}, stats.Stats{stats.MeleeHaste: 250, stats.SpellHaste: 250}, time.Second*10)
+		procAura := CreateBlackMagicProcAura(character)
 		icd := core.Cooldown{
 			Timer:    character.NewTimer(),
 			Duration: time.Second * 35,

diff --git a/sim/deathknight/deathknight.go b/sim/deathknight/deathknight.go
@@ -4,6 +4,7 @@ import (
 	"math"
 	"time"
 
+	"github.com/wowsims/wotlk/sim/common/wotlk"
 	"github.com/wowsims/wotlk/sim/core"
 	"github.com/wowsims/wotlk/sim/core/proto"
 	"github.com/wowsims/wotlk/sim/core/stats"
@@ -330,6 +331,9 @@ func (dk *Deathknight) Initialize() {
 			dk.ArmyOfTheDead.Cast(sim, nil)
 		})
 	}
+
+	// allows us to use these auras in the APL pre-pull actions
+	wotlk.CreateBlackMagicProcAura(&dk.Character)
 }
 
 func (dk *Deathknight) registerMindFreeze() {