diff --git a/pkg/document/template.go b/pkg/document/template.go index 5bebe67..8b7e73b 100644 --- a/pkg/document/template.go +++ b/pkg/document/template.go @@ -1,13 +1,14 @@ package document import ( - "github.com/norwoodj/helm-docs/pkg/util" "io/ioutil" "os" "path" "strings" "text/template" + "github.com/norwoodj/helm-docs/pkg/util" + "github.com/Masterminds/sprig" log "github.com/sirupsen/logrus" @@ -91,6 +92,15 @@ func getAppVersionTemplate() string { return appVersionBuilder.String() } +func getBadgesTemplates() string { + badgeBuilder := strings.Builder{} + badgeBuilder.WriteString(`{{ define "chart.badgesSection" }}`) + badgeBuilder.WriteString(`{{ template "chart.versionBadge" . }}{{ template "chart.typeBadge" . }}{{ template "chart.appVersionBadge" . }}`) + badgeBuilder.WriteString("{{ end }}") + + return badgeBuilder.String() +} + func getDescriptionTemplate() string { descriptionBuilder := strings.Builder{} descriptionBuilder.WriteString(`{{ define "chart.description" }}`) @@ -271,6 +281,7 @@ func getDocumentationTemplates(chartDirectory string, chartSearchRoot string, te getHeaderTemplate(), getDeprecatedTemplate(), getAppVersionTemplate(), + getBadgesTemplates(), getDescriptionTemplate(), getVersionTemplates(), getTypeTemplate(),