Ensure output format is supported at artifact_get.go (#1988)

mindersec · Dec 20, 2023 · 969ec04 · 969ec04
1 parent eacc684
commit 969ec04
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/cli/app/artifact/artifact_get.go b/cmd/cli/app/artifact/artifact_get.go
@@ -56,6 +56,11 @@ func getCommand(ctx context.Context, cmd *cobra.Command, conn *grpc.ClientConn)
 		return cli.MessageAndError(fmt.Sprintf("Provider %s is not supported yet", provider), fmt.Errorf("invalid argument"))
 	}
 
+	// Ensure the output format is supported
+	if !app.IsOutputFormatSupported(format) {
+		return cli.MessageAndError(fmt.Sprintf("Output format %s not supported", format), fmt.Errorf("invalid argument"))
+	}
+
 	// check artifact by name
 	art, err := client.GetArtifactById(ctx, &minderv1.GetArtifactByIdRequest{
 		Context:        &minderv1.Context{Provider: &provider, Project: &project},