-
Notifications
You must be signed in to change notification settings - Fork 297
/
csv.go
92 lines (80 loc) · 2.63 KB
/
csv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package utils
import (
"encoding/csv"
"encoding/json"
"os"
"path/filepath"
"strconv"
"testing"
"github.com/stretchr/testify/require"
)
// CSVToJSON - converts CSV to JSON Structure
func CSVToJSON(t *testing.T, filename string) []byte {
cwd, _ := os.Getwd()
filePath := filepath.Join("output", filename)
fullPath := filepath.Join(cwd, filePath)
csvFile, err := os.Open(fullPath)
require.NoError(t, err, "Error reading file: %s", fullPath)
reader := csv.NewReader(csvFile)
reader.FieldsPerRecord = -1
csvData, err := reader.ReadAll()
require.NoError(t, err, "Error reading CSV file: %s", fullPath)
err = csvFile.Close()
require.NoError(t, err, "Error when closing file: %s", fullPath)
var csvStruct csvSchema
var csvItems []csvSchema
for _, row := range csvData[1:] {
line, lineErr := strconv.Atoi(row[14])
require.NoError(t, lineErr, "Error when converting CSV: %s", fullPath)
searchLine, searchErr := strconv.Atoi(row[17])
require.NoError(t, searchErr, "Error when converting CSV: %s", fullPath)
csvStruct.QueryName = row[0]
csvStruct.QueryID = row[1]
csvStruct.QueryURI = row[2]
csvStruct.Severity = row[3]
csvStruct.Platform = row[4]
csvStruct.CloudProvider = row[5]
csvStruct.Category = row[6]
csvStruct.DescriptionID = row[7]
csvStruct.Description = row[8]
csvStruct.CISDescriptionIDFormatted = row[9]
csvStruct.CISDescriptionTitle = row[10]
csvStruct.CISDescriptionTextFormatted = row[11]
csvStruct.FileName = row[12]
csvStruct.SimilarityID = row[13]
csvStruct.Line = line
csvStruct.IssueType = row[15]
csvStruct.SearchKey = row[16]
csvStruct.SearchLine = searchLine
csvStruct.SearchValue = row[18]
csvStruct.ExpectedValue = row[19]
csvStruct.ActualValue = row[20]
csvItems = append(csvItems, csvStruct)
}
jsondata, err := json.Marshal(csvItems)
require.NoError(t, err, "Error marshaling file: %s", fullPath)
return jsondata
}
type csvSchema struct {
QueryName string
QueryID string
QueryURI string
Severity string
Platform string
CloudProvider string
Category string
DescriptionID string
Description string
CISDescriptionIDFormatted string
CISDescriptionTitle string
CISDescriptionTextFormatted string
FileName string
SimilarityID string
Line int
IssueType string
SearchKey string
SearchLine int
SearchValue string
ExpectedValue string
ActualValue string
}