mirror of https://github.com/k3s-io/k3s
48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
// +build !no_stage
|
|
|
|
package deploy
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func Stage(dataDir string, templateVars map[string]string, skips map[string]bool) error {
|
|
staging:
|
|
for _, name := range AssetNames() {
|
|
nameNoExtension := strings.TrimSuffix(name, filepath.Ext(name))
|
|
if skips[name] || skips[nameNoExtension] {
|
|
continue staging
|
|
}
|
|
namePath := strings.Split(name, string(os.PathSeparator))
|
|
for i := 1; i < len(namePath); i++ {
|
|
subPath := filepath.Join(namePath[0:i]...)
|
|
if skips[subPath] {
|
|
continue staging
|
|
}
|
|
}
|
|
|
|
content, err := Asset(name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for k, v := range templateVars {
|
|
content = bytes.Replace(content, []byte(k), []byte(v), -1)
|
|
}
|
|
p := filepath.Join(dataDir, name)
|
|
os.MkdirAll(filepath.Dir(p), 0700)
|
|
logrus.Info("Writing manifest: ", p)
|
|
if err := ioutil.WriteFile(p, content, 0600); err != nil {
|
|
return errors.Wrapf(err, "failed to write to %s", name)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|