summaryrefslogtreecommitdiffstats
path: root/src/k8splugin/internal/rb/profile_yaml.go
blob: ba55fba9750351af082d34976325178ad5987de0 (plain)
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
/*
 * Copyright 2018 Intel Corporation, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package rb

import (
	"io/ioutil"
	"log"
	"path/filepath"

	"github.com/ghodss/yaml"
	pkgerrors "github.com/pkg/errors"
)

/*
#Sample Yaml format for profile manifest.yaml
---
version: v1
type:
  values: "values_override.yaml"
  configresource:
    - filepath: config.yaml
      chartpath: chart/config/resources/config.yaml
    - filepath: config2.yaml
      chartpath: chart/config/resources/config2.yaml
*/

type overrideFiles struct {
	FilePath  string `yaml:"filepath"`
	ChartPath string `yaml:"chartpath"`
}

type supportedOverrides struct {
	ConfigResource []overrideFiles `yaml:"configresource"`
	Values         string          `yaml:"values"`
}

type profileOverride struct {
	Version string             `yaml:"version"`
	Type    supportedOverrides `yaml:"type"`
}

type ProfileYamlClient struct {
	path     string
	override profileOverride
}

func (p ProfileYamlClient) Print() {
	log.Println(p.override)
}

//GetValues returns a path to the override values.yam
//that was part of the profile
func (p ProfileYamlClient) GetValues() string {
	return filepath.Join(p.path, p.override.Type.Values)
}

//CopyConfigurationOverrides copies the various files that are
//provided as overrides to their corresponding locations within
//the destination chart.
func (p ProfileYamlClient) CopyConfigurationOverrides(chartPath string) error {

	//Iterate over each configresource and copy that file into
	//the respective path in the chart.
	for _, v := range p.override.Type.ConfigResource {
		data, err := ioutil.ReadFile(filepath.Join(p.path, v.FilePath))
		if err != nil {
			return pkgerrors.Wrap(err, "Reading configuration file")
		}
		err = ioutil.WriteFile(filepath.Join(chartPath, v.ChartPath), data, 0644)
		if err != nil {
			return pkgerrors.Wrap(err, "Writing configuration file into chartpath")
		}
	}

	return nil
}

//ProcessProfileYaml parses the manifest.yaml file that is part of the profile
//package and creates the appropriate structures out of it.
func ProcessProfileYaml(fpath string, manifestFileName string) (ProfileYamlClient, error) {

	p := filepath.Join(fpath, manifestFileName)
	data, err := ioutil.ReadFile(p)
	if err != nil {
		return ProfileYamlClient{}, pkgerrors.Wrap(err, "Reading manifest file")
	}

	out := profileOverride{}
	err = yaml.Unmarshal(data, &out)
	if err != nil {
		return ProfileYamlClient{}, pkgerrors.Wrap(err, "Marshaling manifest yaml file")
	}

	return ProfileYamlClient{path: fpath, override: out}, nil
}