package tests

import (
	"testing"

	nbattle "github.com/chompy/nbattle_go"
	"github.com/chompy/nbattle_go/event"
)

type TestEffect struct {
	hpStat *nbattle.StatDef
}

func (e *TestEffect) OnAdd(ctx *nbattle.EffectCtx) {
	stat, _ := ctx.Target.GetStat(e.hpStat)
	stat.SetBase(29)
}

func (e *TestEffect) OnRemove(ctx *nbattle.EffectCtx) {
	stat, _ := ctx.Target.GetStat(e.hpStat)
	stat.SetBase(31)
}

func (e *TestEffect) OnEvent(ctx *nbattle.EffectCtx, evt event.Event) {
	if evt.Type() == event.TickEvent {
		stat, _ := ctx.Target.GetStat(e.hpStat)
		stat.AddBase(-1)
	}
}

func TestCombatantEffect(t *testing.T) {

	ctx := nbattle.New()

	statDefHP := ctx.NewStatDef("hp", 0, 99)
	effectDefTest := ctx.NewEffectDef("test", func() nbattle.Effect {
		return &TestEffect{hpStat: statDefHP}
	})

	cmbt := ctx.NewCombatant()
	hpStat, err := cmbt.GetStat(statDefHP)
	if err != nil {
		t.Fatal(err)
	}
	hpStat.SetBase(15)

	srcCmbt := ctx.NewCombatant()

	cmbt.SetEffect(effectDefTest, 1, srcCmbt)

	hpStat, err = cmbt.GetStat(statDefHP)
	if err != nil {
		t.Fatal(err)
	}
	if hpStat.GetValue() != 29 {
		t.Fatal("expected effect to set combatant hp to 29 on add")
	}

	ctx.Tick()

	hpStat, err = cmbt.GetStat(statDefHP)
	if err != nil {
		t.Fatal(err)
	}
	if hpStat.GetValue() != 28 {
		t.Fatal("expected effect reduce combatant hp by 1 on tick")
	}

	ctx.Tick()

	hpStat, err = cmbt.GetStat(statDefHP)
	if err != nil {
		t.Fatal(err)
	}
	if hpStat.GetValue() != 27 {
		t.Fatal("expected effect reduce combatant hp by 1 on tick")
	}

	cmbt.SetEffect(effectDefTest, 0, nil)

	hpStat, err = cmbt.GetStat(statDefHP)
	if err != nil {
		t.Fatal(err)
	}
	if hpStat.GetValue() != 31 {
		t.Fatal("expected effect to set combatant hp to 31 on removal")
	}

}