Add count argument to source:local instruction

Author: enindu

commands/source/exceptions.go

@@ -19,4 +19,5 @@ import "errors"
 var (
 	errInvalidWidth  error = errors.New("the width is invalid (integer, min 1)")
 	errInvalidHeight error = errors.New("the height is invalid (integer, min 1)")
+	errInvalidCount  error = errors.New("the count is invalid (integer, min 1)")
 )

commands/source/help.go

@@ -26,7 +26,7 @@ func Help(a []string) {
 	
 Available subcommands and arguments:
 
-	local [width] [height]                  # Get a random, static color image.
+	local [width] [height] [count]          # Get random, static color images.
 	picsum [width] [height]                 # Get a random image from "picsum.photos".
 	pexels [key] [width] [height] [keyword] # Get a random image from "pexels.com".
 	help                                    # View help message.

commands/source/local.go

@@ -28,7 +28,7 @@ import (
 
 func Local(a []string) {
 	// Validate arguments and get parameters
-	if len(a) != 2 {
+	if len(a) != 3 {
 		Help(nil)
 		return
 	}
@@ -55,45 +55,59 @@ func Local(a []string) {
 		return
 	}
 
-	// Create graphic
-	rectangle := image.Rect(0, 0, width, height)
-	graphic := image.NewRGBA(rectangle)
-
-	fill := color.RGBA{
-		R: uint8(rand.Int()),
-		G: uint8(rand.Int()),
-		B: uint8(rand.Int()),
-		A: uint8(rand.Int()),
-	}
-
-	for x := range width {
-		for y := range height {
-			graphic.SetRGBA(x, y, fill)
-		}
-	}
-
-	// Create file
-	directory, err := os.UserHomeDir()
+	count, err := strconv.Atoi(a[2])
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
 		return
 	}
 
-	path := filepath.Join(directory, "local.jpg")
-
-	file, err := os.Create(path)
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
+	if count < 1 {
+		fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(errInvalidCount.Error()))
 		return
 	}
 
-	// Encode graphic into the file
-	err = jpeg.Encode(file, graphic, nil)
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
-		return
-	}
+	for i := range count {
+		// Create graphic
+		rectangle := image.Rect(0, 0, width, height)
+		graphic := image.NewRGBA(rectangle)
 
-	// Print message
-	fmt.Fprintf(os.Stdout, "Image saved in \"%s\"\n", path)
+		fill := color.RGBA{
+			R: uint8(rand.Intn(256)),
+			G: uint8(rand.Intn(256)),
+			B: uint8(rand.Intn(256)),
+			A: uint8(rand.Intn(256)),
+		}
+
+		for x := range width {
+			for y := range height {
+				graphic.SetRGBA(x, y, fill)
+			}
+		}
+
+		// Create file
+		directory, err := os.UserHomeDir()
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
+			return
+		}
+
+		name := fmt.Sprintf("local-%d.jpg", i+1)
+		path := filepath.Join(directory, name)
+
+		file, err := os.Create(path)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
+			return
+		}
+
+		// Encode graphic into the file
+		err = jpeg.Encode(file, graphic, nil)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "%s\n", strings.ToLower(err.Error()))
+			return
+		}
+
+		// Print message
+		fmt.Fprintf(os.Stdout, "Image saved in \"%s\"\n", path)
+	}
 }

version.go

@@ -20,7 +20,7 @@ import (
 )
 
 func version() {
-	message := `genimg v1.0.1
+	message := `genimg v1.1.0
 Copyright (C) 2025 Enindu Alahapperuma
 Licensed under GNU GPL 3.0`