From c20a289c065f1308e17e22ae17b59d214786c799 Mon Sep 17 00:00:00 2001
From: filifa <filifa.breath652@8alias.com>
Date: Sun, 28 Jul 2024 17:35:56 -0500
Subject: [PATCH] use sport abbrevs instead of ids

---
 cmd/schedule.go | 43 ++++++++++++++++++++++++++++++++++++-------
 1 file changed, 36 insertions(+), 7 deletions(-)

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)")
 }