mirror of https://github.com/databricks/cli.git
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package template
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
// Executes the template by applying config on it. Returns the materialized config
|
|
// as a string
|
|
func executeTemplate(config map[string]any, templateDefination string) (string, error) {
|
|
// configure template with helper functions
|
|
tmpl, err := template.New("").Funcs(HelperFuncs).Parse(templateDefination)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// execute template
|
|
result := strings.Builder{}
|
|
err = tmpl.Execute(&result, config)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return result.String(), nil
|
|
}
|
|
|
|
func generateFile(config map[string]any, pathTemplate, contentTemplate string) error {
|
|
// compute file content
|
|
fileContent, err := executeTemplate(config, contentTemplate)
|
|
if errors.Is(err, errSkipThisFile) {
|
|
// skip this file
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// compute the path for this file
|
|
path, err := executeTemplate(config, pathTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// create any intermediate directories required. Directories are lazily generated
|
|
// only when they are required for a file.
|
|
err = os.MkdirAll(filepath.Dir(path), 0755)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// write content to file
|
|
return os.WriteFile(path, []byte(fileContent), 0644)
|
|
}
|
|
|
|
func walkFileTree(config map[string]any, templatePath, instancePath string) error {
|
|
entries, err := os.ReadDir(templatePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
// compute directory name
|
|
dirName, err := executeTemplate(config, entry.Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// recusively generate files and directories inside inside our newly generated
|
|
// directory from the template defination
|
|
err = walkFileTree(config, filepath.Join(templatePath, entry.Name()), filepath.Join(instancePath, dirName))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
// case: materialize a template file with it's contents
|
|
b, err := os.ReadFile(filepath.Join(templatePath, entry.Name()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
contentTemplate := string(b)
|
|
fileNameTemplate := entry.Name()
|
|
err = generateFile(config, filepath.Join(instancePath, fileNameTemplate), contentTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|