chore: switch to map[string]bool to simplify code

satisfactorymodding · Oct 20, 2023 · 357f674 · 357f674
1 parent dbfdd4f
commit 357f674
Showing 1 changed file with 8 additions and 15 deletions.
diff --git a/gql/resolver_mods.go b/gql/resolver_mods.go
@@ -23,20 +23,13 @@ import (
 	"github.com/satisfactorymodding/smr-api/util/converter"
 )
 
-var DisallowedModReferences = map[string]struct{}{
-	"satisfactory":          {},
-	"factorygame":           {},
-	"sml":                   {},
-	"satisfactorymodloader": {},
-	"examplemod":            {},
-	"docmod":                {},
-}
-
-func ModReferenceBanned(modReference string) bool {
-	if _, found := DisallowedModReferences[strings.ToLower(modReference)]; found {
-		return true
-	}
-	return false
+var DisallowedModReferences = map[string]bool{
+	"satisfactory":          true,
+	"factorygame":           true,
+	"sml":                   true,
+	"satisfactorymodloader": true,
+	"examplemod":            true,
+	"docmod":                true,
 }
 
 func (r *mutationResolver) CreateMod(ctx context.Context, mod generated.NewMod) (*generated.Mod, error) {
@@ -48,7 +41,7 @@ func (r *mutationResolver) CreateMod(ctx context.Context, mod generated.NewMod)
 		return nil, errors.Wrap(err, "validation failed")
 	}
 
-	if ModReferenceBanned(mod.ModReference) {
+	if DisallowedModReferences[strings.ToLower(mod.ModReference)] {
 		return nil, errors.New("using this mod reference is not allowed")
 	}