From 0ac209dda95931ad56a12729c861c2a77df26cce Mon Sep 17 00:00:00 2001 From: filifa Date: Fri, 19 Jul 2024 22:14:09 -0500 Subject: [PATCH] make team flag optional --- cmd/schedule.go | 11 ++++++++--- cmd/teamflag.go | 6 ++++++ 2 files changed, 14 insertions(+), 3 deletions(-) diff --git a/cmd/schedule.go b/cmd/schedule.go index a2a3eb3..f47b174 100644 --- a/cmd/schedule.go +++ b/cmd/schedule.go @@ -60,8 +60,14 @@ var teamIDs = map[string]statsapi.TeamID{ } func schedule(cmd *cobra.Command, args []string) { - id := teamIDs[string(team)] - sched, err := statsapi.RequestSchedule("1", strconv.Itoa(int(id))) + var id string + if team == "" { + id = string(team) + } else { + id = strconv.Itoa(int(teamIDs[string(team)])) + } + + sched, err := statsapi.RequestSchedule("1", id) if err != nil { log.Fatal(err) } @@ -89,5 +95,4 @@ func init() { // Cobra supports local flags which will only run when this command // is called directly, e.g.: scheduleCmd.Flags().VarP(&team, "team", "t", "team to get updates for (atl, az, bal, bos, chc, cin, cle, col, cws, det, hou, kc, laa, lad, mia, mil, min, nym, nyy, oak, phi, pit, sd, sea, sf, stl, tb, tex, tor, wsh)") - scheduleCmd.MarkFlagRequired("team") } diff --git a/cmd/teamflag.go b/cmd/teamflag.go index 5a7888c..88b810d 100644 --- a/cmd/teamflag.go +++ b/cmd/teamflag.go @@ -29,6 +29,12 @@ func (t *teamFlag) String() string { func (t *teamFlag) Set(v string) error { var err error v = strings.ToLower(v) + + if v == "" { + *t = teamFlag(v) + return nil + } + _, ok := teamIDs[v] if !ok { err = errors.New("invalid team ID")