1
0
mirror of https://github.com/SAP/jenkins-library.git synced 2024-12-14 11:03:09 +02:00
sap-jenkins-library/pkg/generator/helper/docuHelper.go
2019-11-13 14:34:04 +01:00

468 lines
16 KiB
Go

package helper
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"os"
"sort"
"strings"
"text/template"
"github.com/SAP/jenkins-library/pkg/config"
"github.com/ghodss/yaml"
)
// DocuHelperData is used to transport the needed parameters and functions from the step generator to the docu generation.
type DocuHelperData struct {
IsGenerateDocu bool
DocTemplatePath string
OpenDocTemplateFile func(d string) (io.ReadCloser, error)
DocFileWriter func(f string, d []byte, p os.FileMode) error
}
// ContextDefaultData holds the meta data and the default data for the context default parameter descriptions
type ContextDefaultData struct {
Metadata ContextDefaultMetadata `json:"metadata"`
Parameters []ContextDefaultParameters `json:"params"`
}
// ContextDefaultMetadata holds meta data for the context default parameter descripten (name, description, long description)
type ContextDefaultMetadata struct {
Name string `json:"name"`
Description string `json:"description"`
LongDescription string `json:"longDescription,omitempty"`
}
// ContextDefaultParameters holds the description for the context default parameters
type ContextDefaultParameters struct {
Name string `json:"name"`
Description string `json:"description"`
}
// ReadPipelineContextDefaultData loads step definition in yaml format
func (c *ContextDefaultData) readPipelineContextDefaultData(metadata io.ReadCloser) {
defer metadata.Close()
content, err := ioutil.ReadAll(metadata)
checkError(err)
err = yaml.Unmarshal(content, &c)
checkError(err)
}
// ReadContextDefaultMap maps the default descriptions into a map
func (c *ContextDefaultData) readContextDefaultMap() map[string]string {
var m map[string]string = make(map[string]string)
for _, param := range c.Parameters {
m[param.Name] = param.Description
}
return m
}
func readContextDefaultDescription(contextDefaultPath string) map[string]string {
//read context default description
var ContextDefaultData ContextDefaultData
contextDefaultFile, err := os.Open(contextDefaultPath)
checkError(err)
defer contextDefaultFile.Close()
ContextDefaultData.readPipelineContextDefaultData(contextDefaultFile)
return ContextDefaultData.readContextDefaultMap()
}
// generates the step documentation and replaces the template with the generated documentation
//func generateStepDocumentation(stepData config.StepData, docTemplateFilePath string, docTemplate io.ReadCloser, docFileWriter func(f string, d []byte, p os.FileMode) error) {
func generateStepDocumentation(stepData config.StepData, docuHelperData DocuHelperData) error{
fmt.Printf("Generate docu for: %v\n", stepData.Metadata.Name)
//create the file path for the template and open it.
docTemplateFilePath := fmt.Sprintf("%v%v.md", docuHelperData.DocTemplatePath, stepData.Metadata.Name)
docTemplate, err := docuHelperData.OpenDocTemplateFile(docTemplateFilePath)
defer docTemplate.Close()
//check if there is an error during opening the template (true : skip docu generation for this meta data file)
if err != nil {
return fmt.Errorf("Error occured: %v\n", err)
}
content := readAndAdjustTemplate(docTemplate)
if len(content) <= 0 {
fmt.Printf("Error occured: No content inside of the template\n")
return fmt.Errorf("Error occured: No content inside of the template\n")
}
// binding of functions and placeholder
funcMap := template.FuncMap{
"docGenDescription": docGenDescription,
"docGenStepName": docGenStepName,
"docGenParameters": docGenParameters,
"docGenConfiguration": docGenConfiguration,
}
tmpl, err := template.New("doc").Funcs(funcMap).Parse(content)
checkError(err)
setDefaultStepParameters(&stepData)
// add secrets, context defaults to the step parameters
handleStepParameters(&stepData)
//write executed template data to the previously opened file
var docContent bytes.Buffer
err = tmpl.Execute(&docContent, stepData)
checkError(err)
//overwrite existing file
err = docuHelperData.DocFileWriter(docTemplateFilePath, docContent.Bytes(), 644)
checkError(err)
fmt.Printf("Documentation generation complete for: %v\n", stepData.Metadata.Name)
return nil
}
func setDefaultStepParameters(stepData *config.StepData) {
for k, param := range stepData.Spec.Inputs.Parameters {
if param.Default == nil {
switch param.Type {
case "bool":
param.Default = "false"
}
} else {
switch param.Type {
case "string":
case "bool":
param.Default = fmt.Sprintf("\"%v\"", param.Default)
}
}
stepData.Spec.Inputs.Parameters[k] = param
}
}
func readAndAdjustTemplate(docFile io.ReadCloser) string {
//read template content
content, err := ioutil.ReadAll(docFile)
checkError(err)
contentStr := string(content)
//replace old placeholder with new ones
contentStr = strings.ReplaceAll(contentStr, "${docGenStepName}", "{{docGenStepName .}}")
contentStr = strings.ReplaceAll(contentStr, "${docGenConfiguration}", "{{docGenConfiguration .}}")
contentStr = strings.ReplaceAll(contentStr, "${docGenParameters}", "{{docGenParameters .}}")
contentStr = strings.ReplaceAll(contentStr, "${docGenDescription}", "{{docGenDescription .}}")
contentStr = strings.ReplaceAll(contentStr, "## ${docJenkinsPluginDependencies}", "")
return contentStr
}
// Replaces the docGenDescription placeholder with content from the yaml
func docGenDescription(stepData config.StepData) string {
desc := "Description \n\n"
desc += stepData.Metadata.LongDescription
return desc
}
// Replaces the docGenStepName placeholder with the content from the yaml
func docGenStepName(stepData config.StepData) string {
return stepData.Metadata.Name
}
// Replaces the docGenParameters placeholder with the content from the yaml
func docGenParameters(stepData config.StepData) string {
//create step parameter table
parametersTable := createParametersTable(stepData.Spec.Inputs.Parameters)
//create parameters detail section
parametersDetail := createParametersDetail(stepData.Spec.Inputs.Parameters)
return "Parameters\n\n" + parametersTable + parametersDetail
}
// Replaces the docGenConfiguration placeholder with the content from the yaml
func docGenConfiguration(stepData config.StepData) string {
var conf = "We recommend to define values of step parameters via [config.yml file](../configuration.md). \n\n"
conf += "In following sections of the config.yml the configuration is possible:\n\n"
// create step configuration table
conf += createConfigurationTable(stepData.Spec.Inputs.Parameters)
return conf
}
func createParametersTable(parameters []config.StepParameters) string {
var table = "| name | mandatory | default |\n"
table += "| ---- | --------- | ------- |\n"
m := combineEqualParametersTogether(parameters)
for _, param := range parameters {
if v, ok := m[param.Name]; ok {
table += fmt.Sprintf(" | %v | %v | %v | \n ", param.Name, ifThenElse(param.Mandatory && param.Default == nil, "Yes", "No"), v)
delete(m, param.Name)
}
}
return table
}
func createParametersDetail(parameters []config.StepParameters) string {
var detail = "## Details\n"
var m map[string]bool = make(map[string]bool)
for _, param := range parameters {
if _, ok := m[param.Name]; !ok {
if len(param.Description) > 0 {
detail += fmt.Sprintf(" * ` %v ` : %v \n ", param.Name, param.Description)
m[param.Name] = true
}
}
}
return detail
}
//combines equal parameters and the values
func combineEqualParametersTogether(parameters []config.StepParameters) map[string]string {
var m map[string]string = make(map[string]string)
for _, param := range parameters {
m[param.Name] = fmt.Sprintf("%v", param.Default)
if _, ok := m[param.Name]; ok {
addExistingParameterWithCondition(param, m)
} else {
addNewParameterWithCondition(param, m)
}
}
return m
}
func addExistingParameterWithCondition(param config.StepParameters, m map[string]string) {
if param.Conditions != nil {
for _, con := range param.Conditions {
if con.Params != nil {
for _, p := range con.Params {
m[param.Name] = fmt.Sprintf("%v <br> %v=%v:%v ", m[param.Name], p.Name, p.Value, param.Default)
}
}
}
}
}
func addNewParameterWithCondition(param config.StepParameters, m map[string]string) {
if param.Conditions != nil {
m[param.Name] = ""
for _, con := range param.Conditions {
if con.Params != nil {
for _, p := range con.Params {
m[param.Name] += fmt.Sprintf("%v=%v:%v", p.Name, p.Value, param.Default)
}
}
}
}
}
func createConfigurationTable(parameters []config.StepParameters) string {
var table = "| parameter | general | step/stage |\n"
table += "|-----------|---------|------------|\n"
for _, param := range parameters {
if len(param.Scope) > 0 {
general := contains(param.Scope, "GENERAL")
step := contains(param.Scope, "STEPS")
table += fmt.Sprintf(" | %v | %v | %v | \n ", param.Name, ifThenElse(general, "X", ""), ifThenElse(step, "X", ""))
}
}
return table
}
func handleStepParameters(stepData *config.StepData) {
//add secrets to pstep arameters
appendSecretsToParameters(stepData)
//get the context defaults
context := getDocuContextDefaults(stepData)
if len(context) > 0 {
contextDefaultPath := "pkg/generator/helper/piper-context-defaults.yaml"
mCD := readContextDefaultDescription(contextDefaultPath)
//create StepParemeters items for context defaults
for k, v := range context {
if len(v) > 0 {
stepData.Spec.Inputs.Parameters = append(stepData.Spec.Inputs.Parameters, config.StepParameters{Name: k, Default: v, Mandatory: false, Description: mCD[k]})
}
}
}
//Sort Parameters
sortStepParameters(stepData)
}
func appendSecretsToParameters(stepData *config.StepData) {
secrets := stepData.Spec.Inputs.Secrets
if secrets != nil {
for _, secret := range secrets {
item := config.StepParameters{Name: secret.Name, Type: secret.Type, Description: secret.Description, Mandatory: true}
stepData.Spec.Inputs.Parameters = append(stepData.Spec.Inputs.Parameters, item)
}
}
}
func getDocuContextDefaults(m *config.StepData) map[string]string {
var result map[string]string = make(map[string]string)
//creates the context defaults for containers
if len(m.Spec.Containers) > 0 {
keys := []string{}
resources := map[string][]string{}
for _, container := range m.Spec.Containers {
key := ""
if len(container.Conditions) > 0 {
key = fmt.Sprintf("%v=%v", container.Conditions[0].Params[0].Name, container.Conditions[0].Params[0].Value)
}
if len(container.Command) > 0 {
keys = append(keys, key+"_containerCommand")
}
if m.Metadata.Name == "dockerExecuteOnKubernetes" {
keys = append(keys, key+"_containerName")
}
keys = append(keys, key+"_containerShell")
keys = append(keys, key+"_dockerEnvVars")
keys = append(keys, key+"_dockerImage")
keys = append(keys, key+"_dockerName")
keys = append(keys, key+"_dockerPullImage")
keys = append(keys, key+"_dockerWorkspace")
workingDir := ifThenElse(len(container.WorkingDir) > 0, container.WorkingDir, "\\<empty\\>")
resources[key+"_containerShell"] = append(resources[key+"_containerShell"], container.Shell)
resources[key+"_dockerName"] = append(resources[key+"_dockerName"], container.Name)
//Only for Step: dockerExecuteOnKubernetes
if m.Metadata.Name == "dockerExecuteOnKubernetes" {
resources[key+"_containerName"] = append(resources[key+"_containerName"], container.Name)
}
//ContainerCommand > 0
if len(container.Command) > 0 {
resources[key+"_containerCommand"] = append(resources[key+"_containerCommand"], container.Command[0])
}
//ImagePullPolicy > 0
if len(container.ImagePullPolicy) > 0 {
resources[key+"_dockerPullImage"] = []string{fmt.Sprintf("%v", container.ImagePullPolicy != "Never")}
}
//Different when key is set (Param.Name + Param.Value)
if len(key) > 0 {
resources[key+"_dockerEnvVars"] = append(resources[key+"_dockerEnvVars"], fmt.Sprintf("%v:\\[%v\\]", key, strings.Join(envVarsAsStringSlice(container.EnvVars), "")))
resources[key+"_dockerImage"] = append(resources[key+"_dockerImage"], fmt.Sprintf("%v:%v", key, container.Image))
resources[key+"_dockerWorkspace"] = append(resources[key+"_dockerWorkspace"], fmt.Sprintf("%v:%v", key, workingDir))
} else {
resources[key+"_dockerEnvVars"] = append(resources[key+"_dockerEnvVars"], fmt.Sprintf("%v", strings.Join(envVarsAsStringSlice(container.EnvVars), "")))
resources[key+"_dockerImage"] = append(resources[key+"_dockerImage"], container.Image)
resources[key+"_dockerWorkspace"] = append(resources[key+"_dockerWorkspace"], workingDir)
}
// Ready command not relevant for main runtime container so far
//p[] = container.ReadyCommand
}
for _, key := range keys {
s := strings.Split(key, "_")
if len(strings.Join(resources[key], ", ")) > 1 {
result[s[1]] += fmt.Sprintf("%v <br>", strings.Join(resources[key], ", "))
} else if len(strings.Join(resources[key], ", ")) == 1 {
if _, ok := result[s[1]]; !ok {
result[s[1]] = fmt.Sprintf("%v", strings.Join(resources[key], ", "))
}
}
}
}
//creates the context defaults for sidecars
if len(m.Spec.Sidecars) > 0 {
if len(m.Spec.Sidecars[0].Command) > 0 {
result["sidecarCommand"] += m.Spec.Sidecars[0].Command[0]
}
result["sidecarEnvVars"] = strings.Join(envVarsAsStringSlice(m.Spec.Sidecars[0].EnvVars), "")
result["sidecarImage"] = m.Spec.Sidecars[0].Image
result["sidecarName"] = m.Spec.Sidecars[0].Name
if len(m.Spec.Sidecars[0].ImagePullPolicy) > 0 {
result["sidecarPullImage"] = fmt.Sprintf("%v", m.Spec.Sidecars[0].ImagePullPolicy != "Never")
}
result["sidecarReadyCommand"] = m.Spec.Sidecars[0].ReadyCommand
result["sidecarWorkspace"] = m.Spec.Sidecars[0].WorkingDir
}
// not filled for now since this is not relevant in Kubernetes case
//p["dockerOptions"] = container.
//p["dockerVolumeBind"] = container.
//root["containerPortMappings"] = m.Spec.Sidecars[0].
//root["sidecarOptions"] = m.Spec.Sidecars[0].
//root["sidecarVolumeBind"] = m.Spec.Sidecars[0].
//creates the context defaults for resources
if len(m.Spec.Inputs.Resources) > 0 {
keys := []string{}
resources := map[string][]string{}
//fill the map with the key (condition) and the values (resource.Name) to combine the conditions under the resource.Name
for _, resource := range m.Spec.Inputs.Resources {
if resource.Type == "stash" {
key := ""
if len(resource.Conditions) > 0 {
key = fmt.Sprintf("%v=%v", resource.Conditions[0].Params[0].Name, resource.Conditions[0].Params[0].Value)
}
if resources[key] == nil {
keys = append(keys, key)
resources[key] = []string{}
}
resources[key] = append(resources[key], resource.Name)
}
}
for _, key := range keys {
//more than one key when there are conditions
if len(key) > 0 {
result["stashContent"] += fmt.Sprintf("%v:\\[%v\\] <br>", key, strings.Join(resources[key], ", "))
} else {
//single entry for stash content (no condition)
result["stashContent"] += fmt.Sprintf("\\[%v\\] <br>", strings.Join(resources[key], ", "))
}
}
}
return result
}
func envVarsAsStringSlice(envVars []config.EnvVar) []string {
e := []string{}
c := len(envVars) - 1
for k, v := range envVars {
if k < c {
e = append(e, fmt.Sprintf("%v=%v, <br>", v.Name, ifThenElse(len(v.Value) > 0, v.Value, "\\<empty\\>")))
} else {
e = append(e, fmt.Sprintf("%v=%v", v.Name, ifThenElse(len(v.Value) > 0, v.Value, "\\<empty\\>")))
}
}
return e
}
func sortStepParameters(stepData *config.StepData) {
if stepData.Spec.Inputs.Parameters != nil {
parameters := stepData.Spec.Inputs.Parameters
sort.Slice(parameters[:], func(i, j int) bool {
return parameters[i].Name < parameters[j].Name
})
}
}