package translators

type Class string

const (
	Infiltrator  Class = "infiltrator"
	LightAssault Class = "light_assault"
	CombatMedic  Class = "combat_medic"
	Engineer     Class = "engineer"
	HeavyAssault Class = "heavy_assault"
	MAX          Class = "max"
)

var (
	LoadoutMap = map[string]Class{
		"1":  Infiltrator,
		"8":  Infiltrator,
		"15": Infiltrator,
		"28": Infiltrator,

		"3":  LightAssault,
		"10": LightAssault,
		"17": LightAssault,
		"29": LightAssault,

		"4":  CombatMedic,
		"11": CombatMedic,
		"18": CombatMedic,
		"30": CombatMedic,

		"5":  Engineer,
		"12": Engineer,
		"19": Engineer,
		"31": Engineer,

		"6":  HeavyAssault,
		"13": HeavyAssault,
		"20": HeavyAssault,
		"32": HeavyAssault,

		"7":  MAX,
		"14": MAX,
		"21": MAX,
		"45": MAX,
	}
)

func ClassFromLoadout(loadoutID string) Class {
	c, ok := LoadoutMap[loadoutID]

	if !ok {
		return "unknown"
	}

	return c
}