internal/feed: add severity, severity tests

1 files changed, 47 insertions, 0 deletions

diff --git a/internal/feed/severity.go b/internal/feed/severity.go
new file mode 100644
index 0000000..50969ed
--- /dev/null
+++ b/internal/feed/severity.go
@@ -0,0 +1,47 @@
+package feed
+
+//go:generate stringer -linecomment -type=Severity
+
+import (
+  "encoding/json"
+  "fmt"
+)
+
+type Severity byte
+
+const (
+  SeverityNone Severity = iota // NONE
+  SeverityLow // LOW
+  SeverityMedium // MEDIUM
+  SeverityHigh // HIGH
+  SeverityCritical // CRITICAL
+)
+
+// Unmarshal CVSS severity from JSON.
+func (me *Severity) UnmarshalJSON(b []byte) error {
+  // decode string, check for error
+  var s string
+  if err := json.Unmarshal(b, &s); err != nil {
+    return err
+  }
+
+  // check value
+  switch s {
+  case "NONE":
+    *me = SeverityNone
+  case "LOW":
+    *me = SeverityLow
+  case "MEDIUM":
+    *me = SeverityMedium
+  case "HIGH":
+    *me = SeverityHigh
+  case "CRITICAL":
+    *me = SeverityCritical
+  default:
+    // return error
+    return fmt.Errorf("unknown severity: %s", s)
+  }
+
+  // return success
+  return nil
+}