From 9c17b97cd0f83be3fff9fa4e87fd1d29052ea616 Mon Sep 17 00:00:00 2001 From: Paul Duncan Date: Fri, 4 Feb 2022 00:35:31 -0500 Subject: rename to github.com/pablotron/cvez, remove internal libs --- feed/datatype.go | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 feed/datatype.go (limited to 'feed/datatype.go') diff --git a/feed/datatype.go b/feed/datatype.go new file mode 100644 index 0000000..6eaa145 --- /dev/null +++ b/feed/datatype.go @@ -0,0 +1,36 @@ +package feed + +//go:generate stringer -linecomment -type=DataType + +import ( + "encoding/json" + "fmt" +) + +// Data type for NVD feeds and feed items. +type DataType byte + +const ( + CveType DataType = iota // CVE +) + +// 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 +} -- cgit v1.2.3