Add auto-filter

Threagile · Nov 7, 2024 · edf140a · edf140a
1 parent 83880b3
commit edf140a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pkg/report/excel.go b/pkg/report/excel.go
@@ -181,6 +181,8 @@ func WriteRisksExcelToFile(parsedModel *types.Model, filename string, config rep
 						}
 					}
 
+					cellWidth += 5 // add some extra width for auto filter
+
 					if cellWidth > largestWidth {
 						largestWidth = cellWidth
 					}
@@ -235,6 +237,17 @@ func WriteRisksExcelToFile(parsedModel *types.Model, filename string, config rep
 
 	excel.SetActiveSheet(sheetIndex)
 
+	lastColumn, err := excelize.ColumnNumberToName(len(columns))
+	if err != nil {
+		return fmt.Errorf("failed to get last column name: %w", err)
+	}
+	lastRow := len(riskItems) + 1
+	filterRange := fmt.Sprintf("A1:%s%d", lastColumn, lastRow)
+	err = excel.AutoFilter(sheetName, filterRange, []excelize.AutoFilterOptions{})
+	if err != nil {
+		return fmt.Errorf("failed to add autofilter: %w", err)
+	}
+
 	// save file
 	saveAsError := excel.SaveAs(filename)
 	if saveAsError != nil {