Ignoreing case for name and rank of unit when comparing

StarWarsDev · Feb 26, 2020 · 70db21d · 70db21d
1 parent 37d5203
commit 70db21d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/data/sw-legion-data.go b/internal/data/sw-legion-data.go
@@ -3,6 +3,7 @@ package data
 import (
 	"fmt"
 	"github.com/StarWarsDev/go-legion-data"
+	"strings"
 	"time"
 )
 
@@ -32,7 +33,8 @@ func ExtUnit(name, rank string) (*legiondata.Unit, error) {
 	}
 
 	for _, unit := range data.Units {
-		if unit.Name == name && unit.Rank == rank {
+		if strings.ToLower(unit.Name) == strings.ToLower(name) &&
+			strings.ToLower(unit.Rank) == strings.ToLower(rank) {
 			return &unit, nil
 		}
 	}