feat: add CSV format flag

Bearer · Aug 25, 2023 · 861079a · 861079a
1 parent 9941cda
commit 861079a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pkg/flag/report_flags.go b/pkg/flag/report_flags.go
@@ -13,6 +13,7 @@ var (
 	FormatJSON       = "json"
 	FormatYAML       = "yaml"
 	FormatHTML       = "html"
+	FormatCSV        = "csv"
 	FormatEmpty      = ""
 
 	ReportPrivacy   = "privacy"
@@ -26,7 +27,7 @@ var (
 )
 
 var ErrInvalidFormatSecurity = errors.New("invalid format argument for security report; supported values: json, yaml, sarif, gitlab-sast, rdjson, html")
-var ErrInvalidFormatPrivacy = errors.New("invalid format argument for privacy report; supported values: json, yaml html")
+var ErrInvalidFormatPrivacy = errors.New("invalid format argument for privacy report; supported values: csv, json, yaml, html")
 var ErrInvalidFormatDefault = errors.New("invalid format argument; supported values: json, yaml")
 var ErrInvalidReport = errors.New("invalid report argument; supported values: security, privacy")
 var ErrInvalidSeverity = errors.New("invalid severity argument; supported values: critical, high, medium, low, warning")
@@ -134,6 +135,10 @@ func (f *ReportFlagGroup) ToOptions() (ReportOptions, error) {
 		if report != ReportPrivacy && report != ReportSecurity {
 			return ReportOptions{}, invalidFormat
 		}
+	case FormatCSV:
+		if report != ReportPrivacy {
+			return ReportOptions{}, invalidFormat
+		}
 	case FormatSarif, FormatGitLabSast, FormatReviewDog:
 		if report != ReportSecurity {
 			return ReportOptions{}, invalidFormat

diff --git a/pkg/report/output/privacy/formatter.go b/pkg/report/output/privacy/formatter.go
@@ -24,7 +24,7 @@ func NewFormatter(reportData *outputtypes.ReportData, config settings.Config) *F
 
 func (f Formatter) Format(format string) (output *string, err error) {
 	switch format {
-	case flag.FormatEmpty:
+	case flag.FormatEmpty, flag.FormatCSV:
 		stringBuilder, err := BuildCsvString(f.ReportData, f.Config)
 		if err != nil {
 			return output, err