adding the check for CVSS3 feature flag

Checkmarx · Nov 2, 2023 · b6061e6 · b6061e6
1 parent 1ccc35e
commit b6061e6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/internal/commands/predicates.go b/internal/commands/predicates.go
@@ -142,7 +142,10 @@ func runTriageUpdate(resultsPredicatesWrapper wrappers.ResultsPredicatesWrapper)
 		state, _ := cmd.Flags().GetString(params.StateFlag)
 		comment, _ := cmd.Flags().GetString(params.CommentFlag)
 		scanType, _ := cmd.Flags().GetString(params.ScanTypeFlag)
-
+		// check if the current tenant has critical severity available
+		if !wrappers.FeatureFlags[wrappers.CVSSV3Enabled] && severity == "CRITICAL" {
+			return errors.Errorf("%s", "Critical severity is not available for your tenant")
+		}
 		predicate := &wrappers.PredicateRequest{
 			SimilarityID: similarityID,
 			ProjectID:    projectID,

diff --git a/internal/wrappers/feature-flags.go b/internal/wrappers/feature-flags.go
@@ -6,6 +6,7 @@ import (
 
 const tenantIDClaimKey = "tenant_id"
 const PackageEnforcementEnabled = "PACKAGE_ENFORCEMENT_ENABLED"
+const CVSSV3Enabled = "CVSS_V3_ENABLED"
 
 var FeatureFlagsBaseMap = []CommandFlags{
 	{
@@ -17,6 +18,15 @@ var FeatureFlagsBaseMap = []CommandFlags{
 			},
 		},
 	},
+	{
+		CommandName: "cx triage update",
+		FeatureFlags: []FlagBase{
+			{
+				Name:    CVSSV3Enabled,
+				Default: false,
+			},
+		},
+	},
 }
 
 var FeatureFlags = map[string]bool{}