diff --git a/cmd/schedule.go b/cmd/schedule.go index 077ac50..b625698 100644 --- a/cmd/schedule.go +++ b/cmd/schedule.go @@ -25,9 +25,9 @@ import ( "scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi" ) -var team teamFlag var date string -var sportId int + +var team teamFlag var teamIDs = map[string]statsapi.TeamID{ "laa": statsapi.LAA, "az": statsapi.AZ, @@ -61,15 +61,44 @@ var teamIDs = map[string]statsapi.TeamID{ "mil": statsapi.MIL, } +var sport string +var sportIDs = map[string]statsapi.SportID{ + "mlb": statsapi.MLB, + "aaa": statsapi.AAA, + "aa": statsapi.AA, + "higha": statsapi.HighA, + "a": statsapi.A, + "rookie": statsapi.Rookie, + "winter": statsapi.Winter, + "milb": statsapi.MILB, + "indie": statsapi.Independent, + "negro": statsapi.Negro, + "kbo": statsapi.KBO, + "npb": statsapi.NPB, + "int": statsapi.International, + "int18u": statsapi.International18U, + "int16u": statsapi.International16U, + "intamateur": statsapi.InternationalAmateur, + "college": statsapi.College, + "hs": statsapi.HighSchool, +} + func schedule(cmd *cobra.Command, args []string) { - var id string + var teamId string if team == "" { - id = string(team) + teamId = string(team) } else { - id = strconv.Itoa(int(teamIDs[string(team)])) + teamId = strconv.Itoa(int(teamIDs[string(team)])) } - sched, err := statsapi.RequestSchedule(strconv.Itoa(sportId), id, date) + var sportId string + if sport == "" { + sportId = string(sport) + } else { + sportId = strconv.Itoa(int(sportIDs[sport])) + } + + sched, err := statsapi.RequestSchedule(sportId, teamId, date) if err != nil { log.Fatal(err) } @@ -99,5 +128,5 @@ func init() { // is called directly, e.g.: scheduleCmd.Flags().VarP(&team, "team", "t", "team to get schedule 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.Flags().StringVarP(&date, "date", "d", "", "date to get schedule for (YYYY-MM-DD)") - scheduleCmd.Flags().IntVarP(&sportId, "sport", "s", 1, "sport ID get schedule for (default: 1 (MLB))") + scheduleCmd.Flags().StringVarP(&sport, "sport", "s", "mlb", "sport to get schedule for (default: MLB)") }