mage

var ItemSetFirehawkRobesOfConflagration = core.NewItemSet(core.ItemSet{
	Name: "Firehawk Robes of Conflagration",
	Bonuses: map[int32]core.ApplySetBonus{

		2: func(agent core.Agent, setBonusAura *core.Aura) {
			mage := agent.(MageAgent).GetMage()

			setBonusAura.AttachProcTrigger(core.ProcTrigger{
				Name:           "Item - Mage T12 2P Bonus",
				Callback:       core.CallbackOnCastComplete,
				ClassSpellMask: MageSpellArcaneBlast | MageSpellFireball | MageSpellFrostfireBolt | MageSpellFrostbolt,
				ProcChance:     0.20,
				ICD:            time.Second * 45,
				Handler: func(sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
					mage.t12MirrorImage.EnableWithTimeout(sim, mage.t12MirrorImage, time.Second*15)
				},
			})
		},

		4: func(agent core.Agent, setBonusAura *core.Aura) {
			mage := agent.(MageAgent).GetMage()

			setBonusAura.ApplyOnGain(func(_ *core.Aura, _ *core.Simulation) {
				mage.brainFreezeProcChance += .15
				mage.baseHotStreakProcChance += 0.30
			})

			setBonusAura.ApplyOnExpire(func(_ *core.Aura, _ *core.Simulation) {
				mage.brainFreezeProcChance -= .15
				mage.baseHotStreakProcChance -= .30
			})

			setBonusAura.ExposeToAPL(99064)

			mage.T12_4pc = setBonusAura
		},
	},
})

var ItemSetFirelordsVestments = core.NewItemSet(core.ItemSet{
	Name: "Firelord's Vestments",
	Bonuses: map[int32]core.ApplySetBonus{
		2: func(_ core.Agent, setBonusAura *core.Aura) {

			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_BonusCrit_Percent,
				ClassMask:  MageSpellArcaneMissilesTick | MageSpellIceLance | MageSpellPyroblast | MageSpellPyroblastDot,
				FloatValue: 5,
			})
		},
		4: func(_ core.Agent, setBonusAura *core.Aura) {

			setBonusAura.AttachSpellMod(core.SpellModConfig{
				Kind:       core.SpellMod_CastTime_Pct,
				ClassMask:  MageSpellArcaneBlast | MageSpellFireball | MageSpellFrostfireBolt | MageSpellFrostbolt,
				FloatValue: -0.1,
			})
		},
	},
})

var ItemSetTimeLordsRegalia = core.NewItemSet(core.ItemSet{
	Name: "Time Lord's Regalia",
	Bonuses: map[int32]core.ApplySetBonus{

		2: func(agent core.Agent, setBonusAura *core.Aura) {
			character := agent.GetCharacter()
			mage := agent.(MageAgent).GetMage()

			mage.t13ProcAura = core.MakeStackingAura(character, core.StackingStatAura{
				Aura: core.Aura{
					Label:     "Stolen Time",
					ActionID:  core.ActionID{SpellID: 105785},
					Duration:  time.Second * 30,
					MaxStacks: 10,
				},
				BonusPerStack: stats.Stats{stats.HasteRating: 50},
			})

			newStolenTimeTrigger := func(procChance float64, spellMask int64) *core.Aura {
				return setBonusAura.MakeDependentProcTriggerAura(&character.Unit, core.ProcTrigger{
					Name:           fmt.Sprintf("Stolen Time Trigger %f", procChance),
					ActionID:       core.ActionID{ItemID: 105788},
					Callback:       core.CallbackOnSpellHitDealt,
					ClassSpellMask: spellMask,
					ProcChance:     procChance,
					Outcome:        core.OutcomeLanded,
					Handler: func(sim *core.Simulation, spell *core.Spell, result *core.SpellResult) {
						mage.t13ProcAura.Activate(sim)
						mage.t13ProcAura.AddStack(sim)
					},
				})
			}

			newStolenTimeTrigger(1, MageSpellArcaneBlast)
			newStolenTimeTrigger(0.5, MageSpellFireball|MageSpellPyroblast|MageSpellFrostfireBolt|MageSpellFrostbolt)
		},

		4: func(agent core.Agent, setBonusAura *core.Aura) {

			mage := agent.(MageAgent).GetMage()
			mage.T13_4pc = setBonusAura
		},
	},
})

var TalentTreeSizes = [3]int{21, 21, 19}