feat: Shift more severity & ecosystem functionality into place

underdog-tech · Aug 30, 2023 · e57e2b2 · e57e2b2
1 parent a0ba10a
commit e57e2b2
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 10 deletions.
diff --git a/config/config.go b/config/config.go
@@ -10,16 +10,6 @@ import (
 	"github.com/spf13/viper"
 )
 
-type SeverityConfig struct {
-	Label       string
-	Slack_emoji string
-}
-
-type EcosystemConfig struct {
-	Label       string
-	Slack_emoji string
-}
-
 type TeamConfig struct {
 	Name          string
 	Github_slug   string

diff --git a/config/ecosystems.go b/config/ecosystems.go
@@ -18,3 +18,18 @@ const (
 	FindingEcosystemRust   FindingEcosystemType = "rust"
 	FindingEcosystemSwift  FindingEcosystemType = "swift"
 )
+
+func GetConsoleEcosystemIcons() map[FindingEcosystemType]string {
+	return map[FindingEcosystemType]string{
+		FindingEcosystemGo:     "🦦",
+		FindingEcosystemJava:   "🪶 ",
+		FindingEcosystemJS:     "⬢ ",
+		FindingEcosystemPython: "🐍",
+		FindingEcosystemRuby:   "♦️ ",
+	}
+}
+
+type EcosystemConfig struct {
+	Label       string
+	Slack_emoji string
+}
diff --git a/config/severities.go b/config/severities.go
@@ -45,3 +45,19 @@ func GetSeverityReportOrder() []FindingSeverityType {
 		FindingSeverityUndefined,
 	}
 }
+
+func GetConsoleSeverityColors() map[FindingSeverityType]string {
+	return map[FindingSeverityType]string{
+		FindingSeverityCritical:  "#B21515",
+		FindingSeverityHigh:      "#D26C00",
+		FindingSeverityModerate:  "#FBD100",
+		FindingSeverityLow:       "#233EB5",
+		FindingSeverityInfo:      "#56B8F5",
+		FindingSeverityUndefined: "#CFD0D1",
+	}
+}
+
+type SeverityConfig struct {
+	Label       string
+	Slack_emoji string
+}