diff --git a/cmd/root.go b/cmd/root.go index cdc400f..8b74f7a 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -27,7 +27,7 @@ import ( "scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi" ) -var Team TeamFlag +var team teamFlag var teamIDs = map[string]statsapi.TeamID{ "laa": statsapi.LAA, "az": statsapi.AZ, @@ -62,7 +62,7 @@ var teamIDs = map[string]statsapi.TeamID{ } func getGamePk() (string, error) { - id := teamIDs[string(Team)] + id := teamIDs[string(team)] sched, err := statsapi.RequestSchedule("1", strconv.Itoa(int(id))) if err != nil { return "", err @@ -146,11 +146,11 @@ func mlblive(cmd *cobra.Command, args []string) { var rootCmd = &cobra.Command{ Use: "mlblive -t [team]", Short: "Output data from Major League Baseball's Stats API", - Long: `Output data from Major League Baseball's Stats API. + Long: `Output data from Major League Baseball's Stats API. mlblive will establish a websocket connection with ws.statsapi.mlb.com and output JSON with live updates of a team's game.`, - Run: mlblive, + Run: mlblive, } // Execute adds all child commands to the root command and sets flags appropriately. @@ -169,6 +169,6 @@ func init() { // Cobra also supports local flags, which will only run // when this action is called directly. - rootCmd.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)") + rootCmd.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)") rootCmd.MarkFlagRequired("team") } diff --git a/cmd/teamflag.go b/cmd/teamflag.go index 537b96d..5a7888c 100644 --- a/cmd/teamflag.go +++ b/cmd/teamflag.go @@ -20,24 +20,24 @@ import ( "strings" ) -type TeamFlag string +type teamFlag string -func (t *TeamFlag) String() string { +func (t *teamFlag) String() string { return string(*t) } -func (t *TeamFlag) Set(v string) error { +func (t *teamFlag) Set(v string) error { var err error v = strings.ToLower(v) _, ok := teamIDs[v] if !ok { err = errors.New("invalid team ID") } else { - *t = TeamFlag(v) + *t = teamFlag(v) } return err } -func (t *TeamFlag) Type() string { - return "team flag" +func (t *teamFlag) Type() string { + return "teamFlag" }