ws and pruner done!!!

translators/loadouts.go

@@ -12,40 +12,40 @@ const (
 )
 
 var (
-	LoadoutMap = map[string]Class{
-		"1":  Infiltrator,
-		"8":  Infiltrator,
-		"15": Infiltrator,
-		"28": Infiltrator,
+	LoadoutMap = map[uint16]Class{
+		1:  Infiltrator,
+		8:  Infiltrator,
+		15: Infiltrator,
+		28: Infiltrator,
 
-		"3":  LightAssault,
-		"10": LightAssault,
-		"17": LightAssault,
-		"29": LightAssault,
+		3:  LightAssault,
+		10: LightAssault,
+		17: LightAssault,
+		29: LightAssault,
 
-		"4":  CombatMedic,
-		"11": CombatMedic,
-		"18": CombatMedic,
-		"30": CombatMedic,
+		4:  CombatMedic,
+		11: CombatMedic,
+		18: CombatMedic,
+		30: CombatMedic,
 
-		"5":  Engineer,
-		"12": Engineer,
-		"19": Engineer,
-		"31": Engineer,
+		5:  Engineer,
+		12: Engineer,
+		19: Engineer,
+		31: Engineer,
 
-		"6":  HeavyAssault,
-		"13": HeavyAssault,
-		"20": HeavyAssault,
-		"32": HeavyAssault,
+		6:  HeavyAssault,
+		13: HeavyAssault,
+		20: HeavyAssault,
+		32: HeavyAssault,
 
-		"7":  MAX,
-		"14": MAX,
-		"21": MAX,
-		"45": MAX,
+		7:  MAX,
+		14: MAX,
+		21: MAX,
+		45: MAX,
 	}
 )
 
-func ClassFromLoadout(loadoutID string) Class {
+func ClassFromLoadout(loadoutID uint16) Class {
 	c, ok := LoadoutMap[loadoutID]
 
 	if !ok {

translators/loadouts_test.go

@@ -3,11 +3,11 @@ package translators_test
 import (
 	"testing"
 
-	"github.com/genudine/saerro-go/pkg/translators"
+	"github.com/genudine/saerro-go/translators"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestLoadouts(t *testing.T) {
-	assert.Equal(t, translators.ClassFromLoadout("1"), translators.Infiltrator)
-	assert.Equal(t, translators.ClassFromLoadout("0"), translators.Class("unknown"))
+	assert.Equal(t, translators.ClassFromLoadout(1), translators.Infiltrator)
+	assert.Equal(t, translators.ClassFromLoadout(0), translators.Class("unknown"))
 }

translators/vehicles_test.go

@@ -3,7 +3,7 @@ package translators_test
 import (
 	"testing"
 
-	"github.com/genudine/saerro-go/pkg/translators"
+	"github.com/genudine/saerro-go/translators"
 	"github.com/stretchr/testify/assert"
 )