feat: service and metadata fields add to CSV export (#1360)

Add to metadata and service column to export csv
tailwarden · Mar 4, 2024 · 06b21f8 · 06b21f8
1 parent 23cc666
commit 06b21f8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/handlers/csv_handler.go b/handlers/csv_handler.go
@@ -245,7 +245,7 @@ func respondWithCSVDownload(resources []models.Resource, c *gin.Context) {
 	fw := bufio.NewWriter(file)
 	csvWriter := csv.NewWriter(fw)
 
-	header := []string{"id", "provider", "account", "name", "region", "tags", "cost"}
+	header := []string{"id", "provider", "account", "name", "service", "region", "tags", "cost", "metadata"}
 	if err := csvWriter.Write(header); err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": "could not write csv"})
 		return
@@ -256,9 +256,13 @@ func respondWithCSVDownload(resources []models.Resource, c *gin.Context) {
 		if err != nil {
 			log.Fatalf("Could not marshal tags")
 		}
+		metadata, err := json.Marshal(record.Metadata)
+		if err != nil {
+			log.Fatalf("Could not marshal metadata")
+		}
 
 		row := []string{
-			record.ResourceId, record.Provider, record.Account, record.Name, record.Region, string(tags), fmt.Sprintf("%2.f", record.Cost),
+			record.ResourceId, record.Provider, record.Account, record.Name, record.Service, record.Region, string(tags), fmt.Sprintf("%2.f", record.Cost), string(metadata),
 		}
 		if err := csvWriter.Write(row); err != nil {
 			c.JSON(http.StatusInternalServerError, gin.H{"error": "could not write csv"})