Add entities.metadata

internal/inventory/asset.go

@@ -329,9 +329,32 @@ type AssetResourcePolicy struct {
 	Condition  map[string]any `json:"condition,omitempty"`
 }
 
+// EntityMetadata maps metadata required to use Entity Store
+type entityMetadata struct {
+	Category string `json:"category"`
+	Type     string `json:"type"`
+}
+
 // AssetEnricher functional builder function
 type AssetEnricher func(asset *AssetEvent)
 
+func (a *AssetEvent) getEntityMetadata() map[string]entityMetadata {
+	ids := a.Asset.Id
+
+	if len(ids) == 0 {
+		return nil
+	}
+
+	// Picking up only first id, we need to make a decision on if we
+	// have a "primary" id or if we duplicate data
+	return map[string]entityMetadata{
+		ids[0]: {
+			Category: string(a.Asset.Category),
+			Type:     string(a.Asset.Type),
+		},
+	}
+}
+
 func NewAssetEvent(c AssetClassification, ids []string, name string, enrichers ...AssetEnricher) AssetEvent {
 	a := AssetEvent{
 		Asset: Asset{

internal/inventory/inventory.go

@@ -114,6 +114,7 @@ func (a *AssetInventory) publish(assets []AssetEvent) {
 				"iam":               e.IAM,
 				"resource_policies": e.ResourcePolicies,
 				"related.entity":    e.Asset.Id,
+				"entities":          map[string]any{"metadata": e.getEntityMetadata()},
 			},
 		}
 	})

internal/inventory/inventory_test.go

@@ -86,6 +86,14 @@ func TestAssetInventory_Run(t *testing.T) {
 					},
 				},
 				"related.entity": []string{"arn:aws:ec2:us-east::ec2/234567890"},
+				"entities": map[string]any{
+					"metadata": map[string]entityMetadata{
+						"arn:aws:ec2:us-east::ec2/234567890": {
+							Category: string(CategoryInfrastructure),
+							Type:     string(TypeVirtualMachine),
+						},
+					},
+				},
 			},
 		},
 	}