accept an argument

Author: tgvashworth

cli-files/go-ls/cmd/root.go

@@ -11,10 +11,20 @@ var rootCmd = &cobra.Command{
 	Use:   "go-ls",
 	Short: "go-ls is a re-implementation of the ls command",
 	RunE: func(cmd *cobra.Command, args []string) error {
-		files, err := os.ReadDir(".")
+		// By default, ls the current directory
+		dir := "."
+		// If an argument was passed, use the first as our directory
+		if len(args) > 0 {
+			dir = args[0]
+		}
+
+		// Read this directory to get a list of files
+		// https://pkg.go.dev/os#ReadDir
+		files, err := os.ReadDir(dir)
 		if err != nil {
 			return err
 		}
+		// Iterate through each file in the directory, printing the file name
 		for _, file := range files {
 			fmt.Println(file.Name())
 		}