fix(product_enablement): check for ProductUndefined

fastly · Oct 9, 2023 · 5524e4e · 5524e4e
1 parent af48095
commit 5524e4e
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/commands/products/root.go b/pkg/commands/products/root.go
@@ -79,8 +79,12 @@ func (c *RootCommand) Exec(_ io.Reader, out io.Writer) error {
 	}
 
 	if c.enableProduct != "" {
+		p := identifyProduct(c.enableProduct)
+		if p == fastly.ProductUndefined {
+			return errors.New("unrecognised product")
+		}
 		if _, err := ac.EnableProduct(&fastly.ProductEnablementInput{
-			ProductID: identifyProduct(c.enableProduct),
+			ProductID: p,
 			ServiceID: serviceID,
 		}); err != nil {
 			return fmt.Errorf("failed to enable product '%s': %w", c.enableProduct, err)
@@ -90,8 +94,12 @@ func (c *RootCommand) Exec(_ io.Reader, out io.Writer) error {
 	}
 
 	if c.disableProduct != "" {
+		p := identifyProduct(c.disableProduct)
+		if p == fastly.ProductUndefined {
+			return errors.New("unrecognised product")
+		}
 		if err := ac.DisableProduct(&fastly.ProductEnablementInput{
-			ProductID: identifyProduct(c.disableProduct),
+			ProductID: p,
 			ServiceID: serviceID,
 		}); err != nil {
 			return fmt.Errorf("failed to disable product '%s': %w", c.disableProduct, err)