/
model.go
85 lines (74 loc) · 2.4 KB
/
model.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
package document
import (
"fmt"
"sort"
"strconv"
log "github.com/sirupsen/logrus"
"github.com/norwoodj/helm-docs/pkg/helm"
"github.com/spf13/viper"
"gopkg.in/yaml.v3"
)
type valueRow struct {
Key string
Type string
AutoDefault string
Default string
AutoDescription string
Description string
Column int
LineNumber int
}
type chartTemplateData struct {
helm.ChartDocumentationInfo
HelmDocsVersion string
Values []valueRow
}
func getChartTemplateData(chartDocumentationInfo helm.ChartDocumentationInfo, helmDocsVersion string) (chartTemplateData, error) {
// handle empty values file case
if chartDocumentationInfo.ChartValues.Kind == 0 {
return chartTemplateData{
ChartDocumentationInfo: chartDocumentationInfo,
HelmDocsVersion: helmDocsVersion,
Values: make([]valueRow, 0),
}, nil
}
if chartDocumentationInfo.ChartValues.Kind != yaml.DocumentNode {
return chartTemplateData{}, fmt.Errorf("invalid node kind supplied: %d", chartDocumentationInfo.ChartValues.Kind)
}
if chartDocumentationInfo.ChartValues.Content[0].Kind != yaml.MappingNode {
return chartTemplateData{}, fmt.Errorf("values file must resolve to a map, not %s", strconv.Itoa(int(chartDocumentationInfo.ChartValues.Kind)))
}
valuesTableRows, err := createValueRowsFromField(
"",
nil,
chartDocumentationInfo.ChartValues.Content[0],
chartDocumentationInfo.ChartValuesDescriptions,
true,
)
sortOrder := viper.GetString("sort-values-order")
if sortOrder == FileSortOrder {
sort.Slice(valuesTableRows, func(i, j int) bool {
if valuesTableRows[i].LineNumber == valuesTableRows[j].LineNumber {
return valuesTableRows[i].Column < valuesTableRows[j].Column
}
return valuesTableRows[i].LineNumber < valuesTableRows[i].LineNumber
})
} else if sortOrder == AlphaNumSortOrder {
sort.Slice(valuesTableRows, func(i, j int) bool {
return valuesTableRows[i].Key < valuesTableRows[j].Key
})
} else {
log.Warnf("Invalid sort order provided %s, defaulting to %s", sortOrder, AlphaNumSortOrder)
sort.Slice(valuesTableRows, func(i, j int) bool {
return valuesTableRows[i].Key < valuesTableRows[j].Key
})
}
if err != nil {
return chartTemplateData{}, err
}
return chartTemplateData{
ChartDocumentationInfo: chartDocumentationInfo,
HelmDocsVersion: helmDocsVersion,
Values: valuesTableRows,
}, nil
}