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
93

package k8s

import (
	"fmt"
	"heroku/pkg/api/log"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/client-go/kubernetes/scheme"
	"regexp"
	"strings"
)

func ParseK8sYaml(fileR []byte) []runtime.Object {

	acceptedK8sTypes := regexp.MustCompile(`(Role|ClusterRole|RoleBinding|ClusterRoleBinding|ServiceAccount|Deployment|StatefulSet|Service|Ingress|HorizontalPodAutoscaler)`)
	fileAsString := string(fileR[:])
	sepYamlfiles := strings.Split(fileAsString, "---")
	retVal := make([]runtime.Object, 0, len(sepYamlfiles))

	for _, f := range sepYamlfiles {
		if f == "\n" || f == "" {
			// ignore empty cases
			continue
		}

		checkList := strings.Split(f, "#")
		if len(checkList) > 10 {
			// ignore annotation resource
			log.Warn(fmt.Sprintf("ignore annotation resource: %s", f[:10]))

			continue
		}

		decode := scheme.Codecs.UniversalDeserializer().Decode
		obj, groupVersionKind, err := decode([]byte(f), nil, nil)

		if err != nil {
			log.Warn(fmt.Sprintf("Error while decoding YAML object. Err was: %s", err))
			continue
		}
		log.Debug(fmt.Sprintf("Helm-Manitest:--:%s", groupVersionKind))

		if !acceptedK8sTypes.MatchString(groupVersionKind.Kind) {
			log.Info(fmt.Sprintf("The custom-roles configMap contained K8s object types which are not supported! Skipping object with type: %s", groupVersionKind.Kind))
		} else {
			retVal = append(retVal, obj)
		}

	}
	return retVal
}

func stringToFile(outDir string, aut model.Aut) (configFile string, err error) {
	filename := filepath.Join(outDir, "config")

	var data = []byte(aut.K8sConf)
	err = ioutil.WriteFile(filename, data, 0666)
	if err != nil {
		return "", errors.New(fmt.Sprintf("k8s-config load to tmp Error : %s", err.Error()))
	}
	return filename, nil
}

func Show(helmRelease string, aut model.Aut) (instances []runtime.Object, err error) {
	tmp, err := ioutil.TempDir("", "curator-")
	if err != nil {
		return nil, errors.Wrapf(err, "Error while preparing temp Dir")
	}

	defer os.RemoveAll(tmp) // clean up

	configPath, err := stringToFile(tmp, aut)
	if err != nil {
		return nil, err
	}

	log.Info("Started Helm-show:")

	args := []string{
		"--kubeconfig", configPath,
		"get",
		"manifest",
		helmRelease,
		"-n", aut.Namespace,
	}

	stdout, err := utils.ExecCMD("helm", args)
	if err != nil {
		return nil, err
	}

	instances = k8s.ParseK8sYaml(stdout)
	return instances, err
}