feat(saas) add vuls tags from env (#1487)

future-architect · Jul 4, 2022 · 9373131 · 9373131
1 parent 999529a
commit 9373131
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/saas/saas.go b/saas/saas.go
@@ -47,6 +47,7 @@ func (w Writer) Write(rs ...models.ScanResult) error {
 	if len(rs) == 0 {
 		return nil
 	}
+	tags := strings.Split(os.Getenv("VULS_TAGS"), ",")
 
 	ipv4s, ipv6s, err := util.IP()
 	if err != nil {
@@ -111,6 +112,13 @@ func (w Writer) Write(rs ...models.ScanResult) error {
 
 	svc := s3.New(sess)
 	for _, r := range rs {
+		if 0 < len(tags) {
+			if r.Optional == nil {
+				r.Optional = map[string]interface{}{}
+			}
+			r.Optional["VULS_TAGS"] = tags
+		}
+
 		b, err := json.Marshal(r)
 		if err != nil {
 			return xerrors.Errorf("Failed to Marshal to JSON: %w", err)