balance

var ItemSetRegaliaOfTheEternalBloosom = core.NewItemSet(core.ItemSet{
	Name:                    "Regalia of the Eternal Blossom",
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(_ core.Agent, setBonusAura *core.Aura) {

			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Pct,
				ClassMask:  druid.DruidSpellStarfallTick,
				FloatValue: 0.2,
			})
		},
		4: func(_ core.Agent, setBonusAura *core.Aura) {

			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:      core.SpellMod_DotNumberOfTicks_Flat,
				ClassMask: druid.DruidSpellMoonfireDoT | druid.DruidSpellSunfireDoT,
				IntValue:  1,
			})
		},
	},
})

var ItemSetRegaliaOfTheHauntedForest = core.NewItemSet(core.ItemSet{
	Name:                    "Regalia of the Haunted Forest",
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(_ core.Agent, setBonusAura *core.Aura) {

			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_BonusCrit_Percent,
				ClassMask:  druid.DruidSpellStarsurge,
				FloatValue: 10,
			})
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {

			druid := agent.(*BalanceDruid)
			aura := druid.NewTemporaryStatsAura(
				"Druid T15 Balance 4P Bonus",
				core.ActionID{SpellID: 138350},
				stats.Stats{stats.CritRating: 1000, stats.MasteryRating: 1000},
				time.Second*15,
			)

			druid.Env.RegisterPreFinalizeEffect(func() {
				druid.NaturesGrace.ApplyOnGain(func(_ *core.Aura, sim *core.Simulation) {
					if setBonusAura.IsActive() {
						aura.Activate(sim)
					}
				})
			})

			setBonusAura.ExposeToAPL(138350)
		},
	},
})

var ItemSetRegaliaOfTheShatteredVale = core.NewItemSet(core.ItemSet{
	Name:                    "Regalia of the Shattered Vale",
	ID:                      1197,
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(agent core.Agent, setBonusAura *core.Aura) {

			moonkin := agent.(*BalanceDruid)
			solarBolt := moonkin.registerT15BoltSpell(144772, core.SpellSchoolNature, SolarEclipse)
			lunarBolt := moonkin.registerT15BoltSpell(144770, core.SpellSchoolArcane, LunarEclipse)

			bothDuringCA := druid.DruidSpellMoonfire | druid.DruidSpellSunfire | druid.DruidSpellStarsurge

			procTriggerSpellMask := druid.DruidSpellMoonfire |
				druid.DruidSpellSunfire |
				druid.DruidSpellStarfall |
				druid.DruidSpellStarfire |
				druid.DruidSpellWrath |
				druid.DruidSpellStarsurge

			setBonusAura.AttachProcTrigger(core.ProcTrigger{
				Callback:           core.CallbackOnCastComplete,
				ClassSpellMask:     procTriggerSpellMask,
				TriggerImmediately: true,

				Handler: func(sim *core.Simulation, spell *core.Spell, _ *core.SpellResult) {
					alignmentActive := moonkin.CelestialAlignment.RelatedSelfBuff.IsActive()

					if spell.SpellSchool.Matches(core.SpellSchoolNature) || (alignmentActive && spell.Matches(bothDuringCA)) {
						solarBolt.Cast(sim, spell.Unit.CurrentTarget)
					}

					if spell.SpellSchool.Matches(core.SpellSchoolArcane) || (alignmentActive && spell.Matches(bothDuringCA)) {
						lunarBolt.Cast(sim, spell.Unit.CurrentTarget)
					}
				},
			})
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {

			moonkin := agent.(*BalanceDruid)
			moonkin.ShootingStarsProcChance += 0.08
		},
	},
})