warlock

var ItemSetMaleficRaiment = core.NewItemSet(core.ItemSet{
	Name: "Shadowflame Regalia",
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(agent core.Agent, setBonusAura *core.Aura) {
			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_CastTime_Pct,
				ClassMask:  WarlockSpellChaosBolt | WarlockSpellHandOfGuldan | WarlockSpellHaunt,
				FloatValue: -0.1,
			})
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {
			warlock := agent.(WarlockAgent).GetWarlock()

			dmgMod := warlock.AddDynamicMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Flat,
				ClassMask:  WarlockSpellFelFlame,
				FloatValue: 3.0,
			})

			aura := warlock.RegisterAura(core.Aura{
				Label:     "Fel Spark",
				ActionID:  core.ActionID{SpellID: 89937},
				Duration:  15 * time.Second,
				MaxStacks: 2,
				OnGain: func(aura *core.Aura, sim *core.Simulation) {
					dmgMod.Activate()
				},
				OnExpire: func(aura *core.Aura, sim *core.Simulation) {
					dmgMod.Deactivate()
				},
				OnSpellHitDealt: func(aura *core.Aura, sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					if spell.Matches(WarlockSpellFelFlame) && result.Landed() {
						aura.RemoveStack(sim)
					}
				},
			})

			setBonusAura.AttachProcTrigger(core.ProcTrigger{
				Name:           "Item - Warlock T11 4P Bonus",
				ActionID:       core.ActionID{SpellID: 89935},
				ClassSpellMask: WarlockSpellImmolateDot | WarlockSpellUnstableAffliction,
				Callback:       core.CallbackOnPeriodicDamageDealt,
				ProcChance:     0.02,
				Handler: func(sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					aura.Activate(sim)
					aura.SetStacks(sim, 2)
				},
			})
		},
	},
})

var ItemSetRegaliaOfTheHornedNightmare = core.NewItemSet(core.ItemSet{
	Name:                    "Regalia of the Horned Nightmare",
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(agent core.Agent, setBonusAura *core.Aura) {
			warlock := agent.(WarlockAgent).GetWarlock()
			var buff *core.Aura
			switch warlock.Spec {
			case proto.Spec_SpecAfflictionWarlock:
				buff = warlock.RegisterAura(core.Aura{
					ActionID: core.ActionID{SpellID: 145082},
					Label:    "Regalia of the Horned Nightmare - Affli - 2pc",
					Duration: time.Second * 10,
				}).AttachSpellMod(core.SpellModConfig{
					Kind:       core.SpellMod_DamageDone_Pct,
					FloatValue: 0.15,
					ClassMask:  WarlockSpellDrainSoul | WarlockSpellMaleficGrasp,
				})

				warlock.T16_2pc_buff = buff
				setBonusAura.OnSpellHitDealt = func(aura *core.Aura, sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					if spell.Matches(WarlockSpellUnstableAffliction) && result.DidCrit() && sim.Proc(0.5, "T16 - 2pc") {
						buff.Activate(sim)
						return
					}
				}
			case proto.Spec_SpecDemonologyWarlock:

				buffAction := core.ActionID{SpellID: 145075}
				applyBuffAura := func(unit *core.Unit) {
					unit.RegisterAura(core.Aura{
						ActionID: buffAction,
						Label:    "Regalia of the Horned Nightmare - Demo - 2pc",
						Duration: time.Second * 10,
					}).AttachMultiplicativePseudoStatBuff(&unit.PseudoStats.DamageDealtMultiplier, 1.2)
				}

				applyBuffAura(&warlock.Unit)
				for _, pet := range warlock.Pets {
					if pet.IsGuardian() {
						continue
					}

					applyBuffAura(&pet.Unit)
				}

				setBonusAura.OnSpellHitDealt = func(aura *core.Aura, sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					if spell.Matches(WarlockSpellSoulFire) && sim.Proc(0.2, "T16 - 2pc") {
						warlock.GetAuraByID(buffAction).Activate(sim)
						for _, pet := range warlock.Pets {
							if pet.IsGuardian() {
								continue
							}

							if !pet.IsActive() {
								continue
							}

							pet.GetAuraByID(buffAction).Activate(sim)
						}
					}
				}
			case proto.Spec_SpecDestructionWarlock:
				buff = warlock.RegisterAura(core.Aura{
					ActionID: core.ActionID{SpellID: 145075},
					Label:    "Regalia of the Horned Nightmare - Destro - 2pc",
					Duration: time.Second * 10,
				}).AttachSpellMod(core.SpellModConfig{
					Kind:       core.SpellMod_BonusCrit_Percent,
					FloatValue: 0.1,
					ClassMask:  WarlockSpellImmolate | WarlockSpellImmolateDot | WarlockSpellIncinerate | WarlockSpellFaBIncinerate,
				})

				setBonusAura.OnSpellHitDealt = func(aura *core.Aura, sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					if spell.Matches(WarlockSpellConflagrate|WarlockSpellFaBConflagrate) && result.DidCrit() && sim.Proc(0.2, "T16 - 2pc") {
						buff.Activate(sim)
						return
					}
				}
			default:
				return
			}
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {
			warlock := agent.(WarlockAgent).GetWarlock()
			switch agent.GetCharacter().Spec {
			case proto.Spec_SpecAfflictionWarlock:
				warlock.OnSpellRegistered(func(spell *core.Spell) {
					if !spell.Matches(WarlockSpellHaunt) {
						return
					}

					for _, target := range warlock.Env.Encounter.AllTargets {
						dot := spell.Dot(&target.Unit)
						if dot == nil {
							break
						}
						dot.ApplyOnExpire(func(_ *core.Aura, sim *core.Simulation) {
							if sim.Proc(0.1, "T16 4p") {
								warlock.GetSecondaryResourceBar().Gain(sim, 1, spell.ActionID)
							}
						})
					}

				})
			case proto.Spec_SpecDemonologyWarlock:
				setBonusAura.AttachProcTrigger(core.ProcTrigger{
					Callback:       core.CallbackOnCastComplete,
					ClassSpellMask: WarlockSpellShadowBolt | WarlockSpellTouchOfChaos,
					ProcChance:     0.08,
					Handler: func(sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {

					},
				})
			case proto.Spec_SpecDestructionWarlock:
				buff := warlock.RegisterAura(core.Aura{
					ActionID: core.ActionID{SpellID: 145164},
					Label:    "Regalia of the Horned Nightmare - Demo - 4pc",
					Duration: time.Second * 5,
					Icd: &core.Cooldown{
						Timer:    warlock.NewTimer(),
						Duration: time.Second * 10,
					},
				}).AttachStatBuff(stats.CritRating, core.CritRatingPerCritPercent*15)

				warlock.GetSecondaryResourceBar().RegisterOnGain(func(
					sim *core.Simulation,
					_, realGain float64,
					actionID core.ActionID,
				) {
					if realGain == 0 || buff.Icd.IsReady(sim) {
						return
					}

					old := warlock.GetSecondaryResourceBar().Value() - realGain
					if int(old/10) == int(warlock.GetSecondaryResourceBar().Value()/10) {
						return
					}

					buff.Activate(sim)
				})
			}
		},
	},
})

var ItemSetRegaliaOfTheThousandfeldHells = core.NewItemSet(core.ItemSet{
	Name:                    "Regalia of the Thousandfold Hells",
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(agent core.Agent, setBonusAura *core.Aura) {
			warlock := agent.(WarlockAgent).GetWarlock()
			warlock.T15_2pc = agent.GetCharacter().RegisterAura(core.Aura{
				Label:    "Regalia of the Thousandfold Hells - 2P Buff",
				Duration: time.Second * 20,
			}).AttachSpellMod(core.SpellModConfig{
				Kind:      core.SpellMod_DotNumberOfTicks_Flat,
				IntValue:  2,
				ClassMask: WarlockSpellHaunt,
			})

			agent.GetCharacter().OnSpellRegistered(func(spell *core.Spell) {
				if spell.Matches(WarlockDarkSoulSpell) {
					spell.RelatedSelfBuff.ApplyOnGain(func(aura *core.Aura, sim *core.Simulation) {
						warlock.T15_2pc.Activate(sim)
					})
				}
			})
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {
			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Pct,
				FloatValue: 0.05,
				ClassMask:  WarlockSpellMaleficGrasp | WarlockSpellDrainSoul,
			})

			warlock := agent.(WarlockAgent).GetWarlock()
			warlock.T15_4pc = setBonusAura
		},
	},
})

var ItemSetShaSkinRegalia = core.NewItemSet(core.ItemSet{
	Name:                    "Sha-Skin Regalia",
	DisabledInChallengeMode: true,
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(agent core.Agent, setBonusAura *core.Aura) {
			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Pct,
				FloatValue: 0.1,
				ClassMask:  WarlockSpellCorruption,
			}).AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Pct,
				FloatValue: 0.05,
				ClassMask:  WarlockSpellIncinerate | WarlockSpellFaBIncinerate,
			}).AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_DamageDone_Pct,
				FloatValue: 0.02,
				ClassMask:  WarlockSpellShadowBolt | WarlockSpellDemonicSlash | WarlockSpellTouchOfChaos,
			})
		},
		4: func(agent core.Agent, setBonusAura *core.Aura) {
			buff := agent.GetCharacter().RegisterAura(core.Aura{
				Label:    "Sha-Skin Regalia - 4P Buff",
				ActionID: core.ActionID{SpellID: 148463},
				Duration: time.Second * 20,
			}).AttachMultiplicativePseudoStatBuff(&agent.GetCharacter().PseudoStats.DamageDealtMultiplier, 1.10)

			agent.GetCharacter().OnSpellRegistered(func(spell *core.Spell) {
				if spell.Matches(WarlockDarkSoulSpell) {
					spell.RelatedSelfBuff.ApplyOnGain(func(aura *core.Aura, sim *core.Simulation) {
						buff.Activate(sim)
					})
				}
			})
		},
	},
})