main.go

package main

import (
	"fmt"
	"os"

	"github.com/fezcode/atlas.bench/pkg/bench"
	"github.com/fezcode/atlas.bench/pkg/ui"

	"github.com/spf13/pflag"
)

var Version = "dev"

func printHelp() {
	fmt.Println("Atlas Bench — high-precision benchmarking TUI for CLI commands.")
	fmt.Println()
	fmt.Println("Usage:")
	fmt.Println("  atlas.bench [options] \"command1\" \"command2\" ...")
	fmt.Println()
	fmt.Println("Options:")
	fmt.Println("  -r, --runs int         Timed iterations per command (default 10)")
	fmt.Println("  -w, --warmup int       Unmeasured warm-up runs (default 3)")
	fmt.Println("  -t, --timeout dur      Per-run timeout (e.g. 5s, 100ms; 0 = none)")
	fmt.Println("  -s, --shell string     Shell to use (default auto)")
	fmt.Println("  -v, --version          Show version")
	fmt.Println("  -h, --help             Show this help")
	fmt.Println()
	fmt.Println("Inside the UI:")
	fmt.Println("  ↑↓/jk navigate    ↵ toggle detail    r restart    q quit")
}

func main() {
	if len(os.Args) > 1 {
		switch os.Args[1] {
		case "-v", "--version":
			fmt.Printf("atlas.bench v%s\n", Version)
			return
		case "-h", "--help", "help":
			printHelp()
			return
		}
	}

	runs := pflag.IntP("runs", "r", 10, "Timed iterations per command")
	warmup := pflag.IntP("warmup", "w", 3, "Unmeasured warm-up runs")
	timeout := pflag.DurationP("timeout", "t", 0, "Per-run timeout")
	shell := pflag.StringP("shell", "s", "", "Shell to use")
	pflag.BoolP("version", "v", false, "Show version")
	pflag.Usage = printHelp
	pflag.Parse()

	commands := pflag.Args()
	if len(commands) == 0 {
		printHelp()
		os.Exit(1)
	}

	cfg := bench.Config{
		Commands: commands,
		Runs:     *runs,
		Warmup:   *warmup,
		Timeout:  *timeout,
		Shell:    *shell,
	}

	if err := ui.Start(Version, cfg); err != nil {
		fmt.Printf("Error starting UI: %v\n", err)
		os.Exit(1)
	}
}