mlblive/cmd/sportflag.go

/*
Copyright © 2024 filifa

This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free Software
Foundation, either version 3 of the License, or (at your option) any later
version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with
this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cmd

import (
	"strings"

	"scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi"
)

var sportIDs = map[string]statsapi.SportID{
	"mlb":        statsapi.MLB,
	"aaa":        statsapi.TripleA,
	"aa":         statsapi.DoubleA,
	"higha":      statsapi.HighA,
	"a":          statsapi.SingleA,
	"rookie":     statsapi.Rookie,
	"winter":     statsapi.Winter,
	"milb":       statsapi.MILB,
	"indie":      statsapi.Independent,
	"negro":      statsapi.Negro,
	"kbo":        statsapi.Korea,
	"npb":        statsapi.NPB,
	"int":        statsapi.International,
	"int18u":     statsapi.International18U,
	"int16u":     statsapi.International16U,
	"intamateur": statsapi.InternationalAmateur,
	"college":    statsapi.College,
	"hs":         statsapi.HighSchool,
}

type sportFlag string

func (t *sportFlag) String() string {
	return string(*t)
}

func (t *sportFlag) Set(v string) error {
	var err error
	v = strings.ToLower(v)

	_, ok := sportIDs[v]
	if !ok {
		allowed := keys(sportIDs)
		err = flagValueError("sport", allowed)
	} else {
		*t = sportFlag(v)
	}
	return err
}

func (t *sportFlag) Type() string {
	return "sportFlag"
}