Add health verdicts and acceptance tests

audit/cmd/bee/main.go

@@ -27,11 +27,14 @@ func run(args []string, stdout, stderr io.Writer) int {
 
 	if len(args) == 0 {
 		printRootUsage(stderr)
-		return 1
+		return 2
 	}
 
 	switch args[0] {
 	case "help", "--help", "-h":
+		if len(args) > 1 {
+			return runHelp(args[1:], stdout, stderr)
+		}
 		printRootUsage(stdout)
 		return 0
 	case "audit":
@@ -48,7 +51,7 @@ func run(args []string, stdout, stderr io.Writer) int {
 	default:
 		fmt.Fprintf(stderr, "bee: unknown command %q\n\n", args[0])
 		printRootUsage(stderr)
-		return 1
+		return 2
 	}
 }
 
@@ -57,8 +60,29 @@ func printRootUsage(w io.Writer) {
   bee audit   --runtime auto|local|livecd --output stdout|file:<path>
   bee tui     --runtime auto|local|livecd
   bee export  --target <device>
-  bee sat nvidia
-  bee version`)
+  bee sat nvidia|memory|storage
+  bee version
+  bee help [command]`)
+}
+
+func runHelp(args []string, stdout, stderr io.Writer) int {
+	switch args[0] {
+	case "audit":
+		return runAudit([]string{"--help"}, stdout, stdout)
+	case "tui":
+		return runTUI([]string{"--help"}, stdout, stdout)
+	case "export":
+		return runExport([]string{"--help"}, stdout, stdout)
+	case "sat":
+		return runSAT([]string{"--help"}, stdout, stderr)
+	case "version":
+		fmt.Fprintln(stdout, "usage: bee version")
+		return 0
+	default:
+		fmt.Fprintf(stderr, "bee help: unknown command %q\n\n", args[0])
+		printRootUsage(stderr)
+		return 2
+	}
 }
 
 func runAudit(args []string, stdout, stderr io.Writer) int {
@@ -72,6 +96,13 @@ func runAudit(args []string, stdout, stderr io.Writer) int {
 		fs.PrintDefaults()
 	}
 	if err := fs.Parse(args); err != nil {
+		if err == flag.ErrHelp {
+			return 0
+		}
+		return 2
+	}
+	if fs.NArg() != 0 {
+		fs.Usage()
 		return 2
 	}
 	if *showVersion {
@@ -107,6 +138,13 @@ func runTUI(args []string, stdout, stderr io.Writer) int {
 		fs.PrintDefaults()
 	}
 	if err := fs.Parse(args); err != nil {
+		if err == flag.ErrHelp {
+			return 0
+		}
+		return 2
+	}
+	if fs.NArg() != 0 {
+		fs.Usage()
 		return 2
 	}
 
@@ -137,6 +175,13 @@ func runExport(args []string, stdout, stderr io.Writer) int {
 		fs.PrintDefaults()
 	}
 	if err := fs.Parse(args); err != nil {
+		if err == flag.ErrHelp {
+			return 0
+		}
+		return 2
+	}
+	if fs.NArg() != 0 {
+		fs.Usage()
 		return 2
 	}
 	if strings.TrimSpace(*targetDevice) == "" {
@@ -169,21 +214,44 @@ func runExport(args []string, stdout, stderr io.Writer) int {
 }
 
 func runSAT(args []string, stdout, stderr io.Writer) int {
-	if len(args) == 0 || args[0] == "help" || args[0] == "--help" || args[0] == "-h" {
-		fmt.Fprintln(stderr, "usage: bee sat nvidia")
+	if len(args) == 0 {
+		fmt.Fprintln(stderr, "usage: bee sat nvidia|memory|storage")
 		return 2
 	}
-	if args[0] != "nvidia" {
+	if args[0] == "help" || args[0] == "--help" || args[0] == "-h" {
+		fmt.Fprintln(stdout, "usage: bee sat nvidia|memory|storage")
+		return 0
+	}
+	if args[0] != "nvidia" && args[0] != "memory" && args[0] != "storage" {
 		fmt.Fprintf(stderr, "bee sat: unknown target %q\n", args[0])
-		fmt.Fprintln(stderr, "usage: bee sat nvidia")
+		fmt.Fprintln(stderr, "usage: bee sat nvidia|memory|storage")
+		return 2
+	}
+	if len(args) > 1 {
+		fmt.Fprintln(stderr, "usage: bee sat nvidia|memory|storage")
 		return 2
 	}
 	application := app.New(platform.New())
-	archive, err := application.RunNvidiaAcceptancePack("")
+	var (
+		archive string
+		err     error
+		label   string
+	)
+	switch args[0] {
+	case "nvidia":
+		label = "nvidia"
+		archive, err = application.RunNvidiaAcceptancePack("")
+	case "memory":
+		label = "memory"
+		archive, err = application.RunMemoryAcceptancePack("")
+	case "storage":
+		label = "storage"
+		archive, err = application.RunStorageAcceptancePack("")
+	}
 	if err != nil {
-		slog.Error("run nvidia sat", "err", err)
+		slog.Error("run sat", "target", label, "err", err)
 		return 1
 	}
-	slog.Info("nvidia sat archive written", "path", archive)
+	slog.Info("sat archive written", "target", label, "path", archive)
 	return 0
 }