onyx_online/lottery-cli
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/lottery-cli.git synced 2026-08-17 00:34:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add --kind flag, it allows you to configure the lottery kind directly, skipping the selection prompt

Browse Source 
update the README with more examples.
This commit is contained in:
onyx-and-iris
2026-07-24 13:11:56 +01:00
parent 3f0af6e5ec
commit 1310447664
4 changed files with 65 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
cmd/lottery/main.go
View File

@@ -32,29 +32,40 @@ func versionFromBuild() string {
var cmd = &cobra.Command{
Use: "lottery",
Short: "A CLI for National Lottery games.",
PreRunE: func(cmd *cobra.Command, args []string) error {
kindStr := viper.GetString("kind")
if kindStr != "" {
_, err := lottery.ParseKind(kindStr)
if err != nil {
return err
}
}
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
var selected string
form := huh.NewForm(
huh.NewGroup(
huh.NewSelect[string]().
Title("Pick a lottery.").
Options(
huh.NewOption("Lotto", "lotto"),
huh.NewOption("EuroMillions", "euromillions"),
huh.NewOption("Set For Life", "setforlife"),
huh.NewOption("Thunderball", "thunderball"),
huh.NewOption("Powerball", "powerball"),
).
Value(&selected),
),
)
err := form.Run()
if err != nil {
return err
kindStr := viper.GetString("kind")
if kindStr == "" {
form := huh.NewForm(
huh.NewGroup(
huh.NewSelect[string]().
Title("Pick a lottery.").
Options(
huh.NewOption("Lotto", "lotto"),
huh.NewOption("EuroMillions", "euromillions"),
huh.NewOption("Set For Life", "setforlife"),
huh.NewOption("Thunderball", "thunderball"),
huh.NewOption("Powerball", "powerball"),
).
Value(&kindStr),
),
)
err := form.Run()
if err != nil {
return err
}
}
kind, err := lottery.ParseKind(selected)
kind, err := lottery.ParseKind(kindStr)
if err != nil {
return err
}
@@ -101,6 +112,7 @@ var cmd = &cobra.Command{
}
func init() {
cmd.Flags().StringP("kind", "k", "", "Lottery kind to generate draws for.")
cmd.Flags().IntP("count", "c", 1, "Number of draws to generate.")
cmd.Flags().BoolP("count-prompt", "C", false, "Prompt for the number of draws to generate.")