Allow multiple status in pman ls --f (#36)

theredditbandit · May 18, 2024 · 325f38d · 325f38d
1 parent d36532d
commit 325f38d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/cmd/ls.go b/cmd/ls.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/spf13/cobra"
 
@@ -25,7 +26,7 @@ var lsCmd = &cobra.Command{
 		}
 		if filterFlag != "" {
 			fmt.Println("Filtering by status : ", filterFlag)
-			data = utils.FilterByStatus(data, filterFlag)
+			data = utils.FilterByStatuses(data, strings.Split(filterFlag, ","))
 		}
 		if oldUI {
 			return ui.RenderTable(data)
@@ -36,6 +37,6 @@ var lsCmd = &cobra.Command{
 
 func init() {
 	rootCmd.AddCommand(lsCmd)
-	lsCmd.Flags().String("f", "", "Filter projects by status. Usage : pman ls --f <status>")
+	lsCmd.Flags().String("f", "", "Filter projects by status. Usage : pman ls --f <status1[,status2]>")
 	lsCmd.Flags().Bool("c", false, "list projects using the colorful table. Usage : pman ls --c")
 }
diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
@@ -26,11 +26,13 @@ func TitleCase(s string) string {
 	return c.String(s)
 }
 
-func FilterByStatus(data map[string]string, status string) map[string]string {
+func FilterByStatuses(data map[string]string, status []string) map[string]string {
 	filteredData := make(map[string]string)
 	for k, v := range data {
-		if v == status {
-			filteredData[k] = v
+		for _, s := range status {
+			if v == s {
+				filteredData[k] = v
+			}
 		}
 	}
 	return filteredData