Adding p4 features in relation to holy shock

proto/paladin.proto

@@ -64,6 +64,7 @@ enum PaladinRune {
 
 	RuneCloakShieldOfRighteousness	= 440658;
 	RuneCloakVindicator				= 440666;
+	RuneCloakShockAndAwe   		    = 462834;
 	RuneCloakRighteousVengeance		= 440672;
 
 	RuneChestDivineStorm			= 407778;

sim/paladin/avenging_wrath.go

@@ -0,0 +1,49 @@
+package paladin
+
+import (
+	"time"
+
+	"github.com/wowsims/sod/sim/core"
+)
+
+func (paladin *Paladin) registerAvengingWrath() {
+	actionID := core.ActionID{SpellID: 407788}
+
+	AvengingWrathAura := paladin.RegisterAura(core.Aura{
+		Label:    "Avenging Wrath",
+		ActionID: actionID,
+		Duration: time.Second * 20,
+		OnGain: func(aura *core.Aura, sim *core.Simulation) {
+			aura.Unit.PseudoStats.DamageDealtMultiplier *= 1.2
+		},
+		OnExpire: func(aura *core.Aura, sim *core.Simulation) {
+			aura.Unit.PseudoStats.DamageDealtMultiplier /= 1.2
+		},
+	})
+	core.RegisterPercentDamageModifierEffect(AvengingWrathAura, 1.2)
+
+	AvengingWrath := paladin.RegisterSpell(core.SpellConfig{
+		ActionID: actionID,
+		Flags:    core.SpellFlagNoOnCastComplete | core.SpellFlagAPL,
+
+		ManaCost: core.ManaCostOptions{
+			BaseCost: 0.08,
+		},
+		Cast: core.CastConfig{
+			CD: core.Cooldown{
+				Timer:    paladin.NewTimer(),
+				Duration: time.Minute*3,
+			},
+
+		},
+		ApplyEffects: func(sim *core.Simulation, _ *core.Unit, _ *core.Spell) {
+			AvengingWrathAura.Activate(sim)
+		},
+	})
+
+	paladin.AddMajorCooldown(core.MajorCooldown{
+		Spell: AvengingWrath,
+		Type:  core.CooldownTypeDPS,
+
+	})
+}

sim/paladin/holy_shock.go

@@ -9,9 +9,18 @@ import (
 )
 
 func (paladin *Paladin) registerHolyShock() {
+
+	hasInfusionOfLight := paladin.hasRune(proto.PaladinRune_RuneWaistInfusionOfLight)
+
+	cdTime := time.Second * 30
+	if hasInfusionOfLight {
+		cdTime = time.Second * 6
+	}
+
+
 	paladin.holyShockCooldown = &core.Cooldown{
 		Timer:    paladin.NewTimer(),
-		Duration: time.Second * 15,
+		Duration: cdTime,
 	}
 
 	if !paladin.Talents.HolyShock {
@@ -30,11 +39,11 @@ func (paladin *Paladin) registerHolyShock() {
 		{level: 56, spellID: 20930, manaCost: 325, minDamage: 365, maxDamage: 395},
 	}
 
-	hasInfusionOfLight := paladin.hasRune(proto.PaladinRune_RuneWaistInfusionOfLight)
-	damageMultiplier := core.TernaryFloat64(hasInfusionOfLight, 1.2, 1.0)
+
+	damageMultiplier := core.TernaryFloat64(hasInfusionOfLight, 1.5, 1.0)
 
-	hasArtOfWar := paladin.hasRune(proto.PaladinRune_RuneFeetTheArtOfWar)
-	manaCostMultiplier := core.TernaryFloat64(hasArtOfWar, 0.2, 1.0)
+	//hasArtOfWar := paladin.hasRune(proto.PaladinRune_RuneFeetTheArtOfWar)
+	manaCostMultiplier := 1.0//core.TernaryFloat64(hasArtOfWar, 0.2, 1.0)
 
 	hasWrath := paladin.hasRune(proto.PaladinRune_RuneHeadWrath)
 
@@ -88,7 +97,7 @@ func (paladin *Paladin) registerHolyShock() {
 				if hasInfusionOfLight && result.Outcome.Matches(core.OutcomeCrit) {
 					paladin.AddMana(sim, rank.manaCost, manaMetrics)
 					paladin.holyShockCooldown.Reset()
-					paladin.exorcismCooldown.Reset()
+					//paladin.exorcismCooldown.Reset()
 				}
 			},
 		})

sim/paladin/paladin.go

@@ -76,6 +76,7 @@ func (paladin *Paladin) Initialize() {
 	paladin.registerDivineFavor()
 	paladin.registerHammerOfWrath()
 	paladin.registerHolyWrath()
+	paladin.registerAvengingWrath()
 }
 
 func (paladin *Paladin) Reset(_ *core.Simulation) {