cli flags added

Author: typomedia

go.mod

@@ -4,4 +4,7 @@ go 1.19
 
 require git.shangtai.net/staffan/go-ico v1.0.0-beta.2
 
-require golang.org/x/image v0.0.0-20211028202545-6944b10bf410
+require (
+	github.com/spf13/pflag v1.0.5
+	golang.org/x/image v0.0.0-20211028202545-6944b10bf410
+)

main.go

@@ -11,19 +11,29 @@ import (
 	"time"
 
 	"git.shangtai.net/staffan/go-ico"
+	"github.com/spf13/pflag"
 	"golang.org/x/image/draw"
 )
 
 func main() {
+	var (
+		input  string
+		output string
+	)
 
-	start := time.Now()
+	pflag.StringVarP(&output, "out", "o", "", "output ico file")
+	pflag.Parse()
 
-	if len(os.Args) < 2 {
+	if len(pflag.Args()) < 1 {
 		os.Exit(1)
 	}
-	path := os.Args[1]
 
-	file, err := os.ReadFile(path)
+	input = pflag.Arg(0)
+	fmt.Println("Input:", input)
+
+	start := time.Now()
+
+	file, err := os.ReadFile(input)
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -47,8 +57,13 @@ func main() {
 			Y: (size - img.Bounds().Dy()) / 2,
 		}), img, image.Point{}, draw.Over)
 
-	basename := name(path)
-	icoFile, err := os.Create(basename + ".ico")
+	basename := name(input) + ".ico"
+	if output != "" {
+		basename = output
+	}
+	fmt.Println("Output:", basename)
+
+	icoFile, err := os.Create(basename)
 	if err != nil {
 		log.Fatal(err)
 	}