From a824a9732154f9bee2ce530f753d5747de78aa3f Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Tue, 1 Feb 2022 01:10:24 -0500
Subject: internal/feed: split out datatype, add tests

---
 internal/feed/feed.go | 29 ++---------------------------
 1 file changed, 2 insertions(+), 27 deletions(-)

(limited to 'internal/feed/feed.go')

diff --git a/internal/feed/feed.go b/internal/feed/feed.go
index 1c15a0a..b4d514b 100644
--- a/internal/feed/feed.go
+++ b/internal/feed/feed.go
@@ -9,8 +9,7 @@ import (
 )
 
 const (
-  CveType = iota  // CVE data type
-  MitreFormat     // MITRE data format
+  MitreFormat = iota    // MITRE data format
   DataVersion40   // Version 4.0
 
   OrNodeOp        // OR operator
@@ -42,31 +41,7 @@ const (
   Single          // Single authentication
 )
 
-// TODO: parse cpe, cvss vectors (v3.x, v2)
-
-// Data type for NVD feeds and feed items.
-type DataType int
-
-// Unmarshal DataType from JSON.
-func (me *DataType) 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 "CVE":
-    *me = CveType
-  default:
-    // return error
-    return fmt.Errorf("unknown data type: %s", s)
-  }
-
-  // return success
-  return nil
-}
+// TODO: parse cpe
 
 // Data format for NVD feeds and feed items.
 type DataFormat int
-- 
cgit v1.2.3