exp/change flags

cmd/mk/main.go

@@ -1,43 +1,43 @@
 package main
 
 import (
-	"flag"
 	"fmt"
 	"os"
 	"path/filepath"
 	"strings"
 
 	"github.com/devkcud/mk/internal/help"
 	"github.com/devkcud/mk/internal/utils"
+	flag "github.com/spf13/pflag"
 )
 
 var Version string
 
 func main() {
 	flag.Usage = func() { help.ShowHelp(Version) }
 
-	showHelp := flag.Bool("help", false, "Show help menu")
-	showVersion := flag.Bool("version", false, "Show version")
-	quiet := flag.Bool("quiet", false, "Disable output")
+	flagHelp := flag.BoolP("help", "h", false, "Show help menu")
+	flagVersion := flag.BoolP("version", "V", false, "Show version")
+	flagQuiet := flag.BoolP("quiet", "q", false, "Disable output")
 
 	flag.Parse()
 
-	if *showVersion {
+	if *flagQuiet {
+		// os.Stdin = nil
+		os.Stdout = nil
+		os.Stderr = nil
+	}
+
+	if *flagVersion {
 		fmt.Printf("mk %s\n", Version)
 		return
 	}
 
-	if *showHelp || flag.NArg() == 0 {
+	if *flagHelp || flag.NArg() == 0 {
 		flag.Usage()
 		return
 	}
 
-	if *quiet {
-		// os.Stdin = nil
-		os.Stdout = nil
-		os.Stderr = nil
-	}
-
 	var dirstack []string
 
 	for _, name := range flag.Args() {

go.mod

@@ -1,3 +1,5 @@
 module github.com/devkcud/mk
 
 go 1.21
+
+require github.com/spf13/pflag v1.0.5 // indirect

go.sum

@@ -0,0 +1,2 @@
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=

internal/help/help.go

@@ -1,8 +1,9 @@
 package help
 
 import (
-	"flag"
 	"fmt"
+
+	"github.com/spf13/pflag"
 )
 
 var examples []string = []string{"src/main.js", "tests/", "lib/ myclass.go otherclass.go andanotherclass.go", "src/main.cpp Makefile"}
@@ -12,7 +13,7 @@ func ShowHelp(version string) {
 
 	fmt.Printf("Usage: mk [-flags] [..] [folder/] [files...]\n\nFlags:\n")
 
-	flag.PrintDefaults()
+	pflag.PrintDefaults()
 
 	fmt.Println("\nExamples:")