From 8be2e5de9e36effc5989002d19d0e6c9c83d174b Mon Sep 17 00:00:00 2001 From: filifa Date: Sun, 14 Jul 2024 18:12:24 -0500 Subject: [PATCH] use teamids map --- cmd/teamflag.go | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/cmd/teamflag.go b/cmd/teamflag.go index 34af5b7..60d5432 100644 --- a/cmd/teamflag.go +++ b/cmd/teamflag.go @@ -1,6 +1,10 @@ package cmd -import "errors" +import ( + "errors" + + "scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi" +) type TeamFlag string @@ -9,13 +13,14 @@ func (t *TeamFlag) String() string { } func (t *TeamFlag) Set(v string) error { - switch v { - case "laa", "az", "bal", "bos", "chc", "cin", "cle", "col", "det", "hou", "kc", "lad", "wsh", "nym", "oak", "pit", "sd", "sea", "sf", "stl", "tb", "tex", "tor", "min", "phi", "atl", "cws", "mia", "nyy", "mil": + var err error + _, ok := statsapi.TeamIds[v] + if !ok { + err = errors.New("invalid team ID") + } else { *t = TeamFlag(v) - return nil - default: - return errors.New("invalid team ID") } + return err } func (t *TeamFlag) Type() string {