v1.0.0

cmd/main.go

@@ -1,25 +1,65 @@
-// main package contains the cli functionality
 package main
 
 import (
+	"flag"
 	"fmt"
+	"io"
 	"os"
 
 	"github.com/ivuorinen/go-test-sarif-action/internal"
 )
 
-func main() {
-	if len(os.Args) < 3 {
-		fmt.Println("Usage: go-test-sarif <input.json> <output.sarif>")
-		os.Exit(1)
-	}
+var (
+	version = "dev"
+	commit  = "none"
+	date    = "unknown"
+	builtBy = "unknown"
+)
 
-	inputFile := os.Args[1]
-	outputFile := os.Args[2]
-
-	err := internal.ConvertToSARIF(inputFile, outputFile)
-	if err != nil {
-		fmt.Println("Error:", err)
-		os.Exit(1)
-	}
+func printVersion(w io.Writer) {
+	_, _ = fmt.Fprintf(w, "go-test-sarif %s\n", version)
+	_, _ = fmt.Fprintf(w, "  commit: %s\n", commit)
+	_, _ = fmt.Fprintf(w, "  built at: %s\n", date)
+	_, _ = fmt.Fprintf(w, "  built by: %s\n", builtBy)
+}
+
+func printUsage(w io.Writer) {
+	_, _ = fmt.Fprintln(w, "Usage: go-test-sarif <input.json> <output.sarif>")
+	_, _ = fmt.Fprintln(w, "       go-test-sarif --version")
+}
+
+func run(args []string, stdout, stderr io.Writer) int {
+	fs := flag.NewFlagSet("go-test-sarif", flag.ContinueOnError)
+	fs.SetOutput(stderr)
+
+	var versionFlag bool
+	fs.BoolVar(&versionFlag, "version", false, "Display version information")
+	fs.BoolVar(&versionFlag, "v", false, "Display version information (short)")
+	if err := fs.Parse(args[1:]); err != nil {
+		return 1
+	}
+
+	if versionFlag {
+		printVersion(stdout)
+		return 0
+	}
+
+	if fs.NArg() < 2 {
+		printUsage(stderr)
+		return 1
+	}
+
+	inputFile := fs.Arg(0)
+	outputFile := fs.Arg(1)
+
+	if err := internal.ConvertToSARIF(inputFile, outputFile); err != nil {
+		_, _ = fmt.Fprintf(stderr, "Error: %v\n", err)
+		return 1
+	}
+
+	return 0
+}
+
+func main() {
+	os.Exit(run(os.Args, os.Stdout, os.Stderr))
 }