filifa
/
mlblive
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

unexport some types

This commit is contained in:
filifa 2024-07-14 19:41:40 -05:00
parent 343a15893d
commit adc5c592f2
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/root.go
View File

@ -27,7 +27,7 @@ import (
"scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi" "scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi"
) )
var Team TeamFlag var team teamFlag
var teamIDs = map[string]statsapi.TeamID{ var teamIDs = map[string]statsapi.TeamID{
"laa": statsapi.LAA, "laa": statsapi.LAA,
"az": statsapi.AZ, "az": statsapi.AZ,
@ -62,7 +62,7 @@ var teamIDs = map[string]statsapi.TeamID{
} }
func getGamePk() (string, error) { func getGamePk() (string, error) {
id := teamIDs[string(Team)] id := teamIDs[string(team)]
sched, err := statsapi.RequestSchedule("1", strconv.Itoa(int(id))) sched, err := statsapi.RequestSchedule("1", strconv.Itoa(int(id)))
if err != nil { if err != nil {
return "", err return "", err
@ -146,11 +146,11 @@ func mlblive(cmd *cobra.Command, args []string) {
var rootCmd = &cobra.Command{ var rootCmd = &cobra.Command{
Use: "mlblive -t [team]", Use: "mlblive -t [team]",
Short: "Output data from Major League Baseball's Stats API", 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 mlblive will establish a websocket connection with ws.statsapi.mlb.com and
output JSON with live updates of a team's game.`, 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. // 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 // Cobra also supports local flags, which will only run
// when this action is called directly. // 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") rootCmd.MarkFlagRequired("team")
} }

12
cmd/teamflag.go
View File

@ -20,24 +20,24 @@ import (
"strings" "strings"
) )
type TeamFlag string type teamFlag string
func (t *TeamFlag) String() string { func (t *teamFlag) String() string {
return string(*t) return string(*t)
} }
func (t *TeamFlag) Set(v string) error { func (t *teamFlag) Set(v string) error {
var err error var err error
v = strings.ToLower(v) v = strings.ToLower(v)
_, ok := teamIDs[v] _, ok := teamIDs[v]
if !ok { if !ok {
err = errors.New("invalid team ID") err = errors.New("invalid team ID")
} else { } else {
*t = TeamFlag(v) *t = teamFlag(v)
} }
return err return err
} }
func (t *TeamFlag) Type() string { func (t *teamFlag) Type() string {
return "team flag" return "teamFlag"
} }