fix for cli adding or removing empty p2p collection.

sourcenetwork · Oct 6, 2023 · 9220ac4 · 9220ac4
1 parent 0d3f6be
commit 9220ac4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/cli/p2p_collection_add.go b/cli/p2p_collection_add.go
@@ -32,7 +32,16 @@ Example: add multiple collections
 		Args: cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
 			store := mustGetStoreContext(cmd)
-			collectionIDs := strings.Split(args[0], ",")
+
+			var collectionIDs []string
+			for _, id := range strings.Split(args[0], ",") {
+				id = strings.TrimSpace(id)
+				if id == "" {
+					continue
+				}
+				collectionIDs = append(collectionIDs, id)
+			}
+
 			return store.AddP2PCollections(cmd.Context(), collectionIDs)
 		},
 	}

diff --git a/cli/p2p_collection_remove.go b/cli/p2p_collection_remove.go
@@ -32,7 +32,16 @@ Example: remove multiple collections
 		Args: cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
 			store := mustGetStoreContext(cmd)
-			collectionIDs := strings.Split(args[0], ",")
+
+			var collectionIDs []string
+			for _, id := range strings.Split(args[0], ",") {
+				id = strings.TrimSpace(id)
+				if id == "" {
+					continue
+				}
+				collectionIDs = append(collectionIDs, id)
+			}
+
 			return store.RemoveP2PCollections(cmd.Context(), collectionIDs)
 		},
 	}