feat: distribution optional

blackstork-io · Dec 15, 2024 · a762cf7 · a762cf7
1 parent 4b112e2
commit a762cf7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/internal/misp/client/misp_models.go b/internal/misp/client/misp_models.go
@@ -56,7 +56,7 @@ type AddEventReportRequest struct {
 	EventId        string  `json:"event_id"`
 	Name           string  `json:"name"`
 	Content        string  `json:"content"`
-	Distribution   string  `json:"distribution"`
+	Distribution   *string `json:"distribution"`
 	SharingGroupId *string `json:"sharing_group_id"`
 	Timestamp      *string `json:"timestamp"`
 	Deleted        bool    `json:"deleted"`

diff --git a/internal/misp/publish_misp_event_reports.go b/internal/misp/publish_misp_event_reports.go
@@ -41,9 +41,8 @@ func makeMispEventReportsPublisher(loader ClientLoaderFn) *plugin.Publisher {
 					Constraints: constraint.RequiredMeaningful,
 				},
 				{
-					Name:        "distribution",
-					Type:        cty.String,
-					Constraints: constraint.RequiredMeaningful,
+					Name: "distribution",
+					Type: cty.String,
 					OneOf: []cty.Value{
 						cty.StringVal("0"),
 						cty.StringVal("1"),
@@ -123,13 +122,17 @@ func publishEventReport(loader ClientLoaderFn) plugin.PublishFunc {
 
 		timestamp := fmt.Sprintf("%d", time.Now().Unix())
 		report := client.AddEventReportRequest{
-			Uuid:         uuid.New().String(),
-			EventId:      params.Args.GetAttrVal("event_id").AsString(),
-			Name:         params.Args.GetAttrVal("name").AsString(),
-			Content:      buff.String(),
-			Distribution: params.Args.GetAttrVal("distribution").AsString(),
-			Timestamp:    &timestamp,
-			Deleted:      false,
+			Uuid:      uuid.New().String(),
+			EventId:   params.Args.GetAttrVal("event_id").AsString(),
+			Name:      params.Args.GetAttrVal("name").AsString(),
+			Content:   buff.String(),
+			Timestamp: &timestamp,
+			Deleted:   false,
+		}
+		distribution := params.Args.GetAttrVal("distribution")
+		if !distribution.IsNull() {
+			distributionStr := distribution.AsString()
+			report.Distribution = &distributionStr
 		}
 		sharingGroupId := params.Args.GetAttrVal("sharing_group_id")
 		if !sharingGroupId.IsNull() {