main.go

package main

import (
	"fmt"
	"os"
	"strings"

	"github.com/Leixb/jutge/commands"

	"github.com/alecthomas/kong"
	"github.com/posener/complete"
	"github.com/willabides/kongplete"
)

type Globals struct {
	WorkDir     string      `type:"path" help:"Directory to save jutge files" default:"~/Documents/jutge" env:"JUTGE_WORK_DIR"`
	Concurrency uint        `help:"Maximum concurrent routines" default:"5"`
	Regex       string      `help:"Regular expression used to validate and find problem codes in filenames" default:"[PGQX]\d{5}_(ca|en|es|fr|de)"`
	Username    string      `help:"Username" env:"JUTGE_USERNAME"`
	Password    string      `help:"Password" env:"JUTGE_PASSWORD"`
	Version     VersionFlag `help:"Print version and exit"`
	URL         string      `help:"URL to jutge.org" default:"https://jutge.org"`
}

type CLI struct {
	Globals

	Download DownloadCmd `cmd:"" help:"Download problems from jutge.org" aliases:"dl,down"`
	Test     TestCmd     `cmd:"" help:"Test your solutions"`
	Upload   UploadCmd   `cmd:"" help:"Upload your solutions" aliases:"up"`
	Check    CheckCmd    `cmd:"" help:"Check the status of your solutions" aliases:"chk"`
	Database DatabaseCmd `cmd:"" help:"Manage your database" aliases:"db"`
	New      NewCmd      `cmd:"" help:"Create a new file"`

	InstallCompletions kongplete.InstallCompletions `cmd:"" help:"install shell completions"`
}

type VersionFlag string

func (v VersionFlag) Decode(ctx *kong.DecodeContext) error { return nil }
func (v VersionFlag) IsBool() bool                         { return true }
func (v VersionFlag) BeforeApply(app *kong.Kong, vars kong.Vars) error {
	fmt.Println(vars["version"])
	app.Exit(0)
	return nil
}

func main() {
	cli := CLI{
		Globals: Globals{
			Version: VersionFlag("0.3.1"),
		},
	}

	parser := kong.Must(&cli,
		kong.Name("jutge"),
		kong.Description("Jutge is a command line tool to download and test problems from jutge.org"),
		kong.UsageOnError(),
		// kong.ConfigureHelp(kong.HelpOptions{
		// 	Compact: true,
		// }),
		kong.Vars{
			"version":   "0.3.1",
			"compilers": strings.Join(commands.GetCompilers(), ","),
		})

	kongplete.Complete(parser,
		kongplete.WithPredictor("file", complete.PredictFiles("*")),
	)

	ctx, err := parser.Parse(os.Args[1:])
	parser.FatalIfErrorf(err)

	err = ctx.Run(ctx, &cli.Globals)
	ctx.FatalIfErrorf(err)
}