warlock: fix crash when casting DS without haunt talented

Fixes wowsims#3992, wowsims#3999.
Tharre · Nov 9, 2023 · 67ca2fb · 67ca2fb
1 parent f00f661
commit 67ca2fb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/sim/warlock/apl_values.go b/sim/warlock/apl_values.go
@@ -50,9 +50,12 @@ func (value *APLValueWarlockShouldRecastDrainSoul) GetBool(sim *core.Simulation)
 		warlock.CurseOfWeaknessAuras.Get(warlock.CurrentTarget).RemainingDuration(sim),
 	) - warlock.CurseOfAgony.CastTime()
 
-	hauntRefresh := warlock.HauntDebuffAuras.Get(warlock.CurrentTarget).RemainingDuration(sim) -
-		warlock.Haunt.CastTime() -
-		warlock.Haunt.TravelTime()
+	hauntRefresh := 1000 * time.Second
+	if warlock.HauntDebuffAuras != nil {
+		hauntRefresh = warlock.HauntDebuffAuras.Get(warlock.CurrentTarget).RemainingDuration(sim) -
+			warlock.Haunt.CastTime() -
+			warlock.Haunt.TravelTime()
+	}
 
 	timeUntilRefresh := min(uaRefresh, curseRefresh)
 

diff --git a/sim/warlock/drain_soul.go b/sim/warlock/drain_soul.go
@@ -11,7 +11,6 @@ func (warlock *Warlock) registerDrainSoulSpell() {
 
 	calcSoulSiphonMult := func(target *core.Unit) float64 {
 		auras := []*core.Aura{
-			warlock.HauntDebuffAuras.Get(target),
 			warlock.UnstableAffliction.Dot(target).Aura,
 			warlock.Corruption.Dot(target).Aura,
 			warlock.Seed.Dot(target).Aura,
@@ -23,6 +22,9 @@ func (warlock *Warlock) registerDrainSoulSpell() {
 			warlock.ShadowEmbraceDebuffAura(target),
 			// missing: death coil
 		}
+		if warlock.HauntDebuffAuras != nil {
+			auras = append(auras, warlock.HauntDebuffAuras.Get(target))
+		}
 		numActive := 0
 		for _, aura := range auras {
 			if aura.IsActive() {