95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"encoding/json"
|
||
|
|
"text/template"
|
||
|
|
//"github.com/hashicorp/hcl2/gohcl"
|
||
|
|
"github.com/hashicorp/hcl/v2"
|
||
|
|
"github.com/hashicorp/hcl/v2/hclsyntax"
|
||
|
|
//"github.com/zclconf/go-cty/cty"
|
||
|
|
//"github.com/zclconf/go-cty/cty/gocty"
|
||
|
|
ctyjson "github.com/zclconf/go-cty/cty/json"
|
||
|
|
)
|
||
|
|
|
||
|
|
func checkErr(e error) {
|
||
|
|
if e != nil {
|
||
|
|
panic(e)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func processGoTemplate(file string, config []byte) {
|
||
|
|
|
||
|
|
// The JSON configuration
|
||
|
|
var confData map[string]interface{}
|
||
|
|
err := json.Unmarshal(config, &confData)
|
||
|
|
checkErr(err)
|
||
|
|
|
||
|
|
// Read the template
|
||
|
|
data, err := os.ReadFile(file)
|
||
|
|
checkErr(err)
|
||
|
|
|
||
|
|
|
||
|
|
tpl, err := template.New("conf").Parse(string(data))
|
||
|
|
checkErr(err)
|
||
|
|
|
||
|
|
checkErr(tpl.Execute(os.Stdout,config))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func processHCLTemplate(file string, config []byte) {
|
||
|
|
|
||
|
|
fct, err:= os.ReadFile(file)
|
||
|
|
checkErr(err)
|
||
|
|
|
||
|
|
expr, diags := hclsyntax.ParseTemplate(fct, file, hcl.Pos{Line:1, Column: 1})
|
||
|
|
if diags.HasErrors() {
|
||
|
|
panic(diags.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
ctyType, err := ctyjson.ImpliedType(config)
|
||
|
|
checkErr(err)
|
||
|
|
varsVal, err := ctyjson.Unmarshal(config,ctyType)
|
||
|
|
checkErr(err)
|
||
|
|
|
||
|
|
ctx := &hcl.EvalContext{
|
||
|
|
Variables: varsVal.AsValueMap(),
|
||
|
|
}
|
||
|
|
|
||
|
|
for n := range ctx.Variables {
|
||
|
|
if !hclsyntax.ValidIdentifier(n) {
|
||
|
|
panic(fmt.Errorf("invalid template variable name %q: must start with a letter, followed by zero or more letters, digits, and underscores", n))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, traversal := range expr.Variables() {
|
||
|
|
root := traversal.RootName()
|
||
|
|
if _, ok := ctx.Variables[root]; !ok {
|
||
|
|
panic(fmt.Errorf("vars map does not contain key %q, referenced at %s", root, traversal[0].SourceRange()))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
val, diags := expr.Value(ctx)
|
||
|
|
if diags.HasErrors() {
|
||
|
|
panic(diags.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("%s",val.AsString())
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
// The template to process
|
||
|
|
templateType := os.Args[1]
|
||
|
|
templateFile := os.Args[2]
|
||
|
|
config := []byte(os.Args[3])
|
||
|
|
|
||
|
|
|
||
|
|
if templateType == "go" {
|
||
|
|
processGoTemplate(templateFile, config)
|
||
|
|
} else if templateType == "hcl" {
|
||
|
|
processHCLTemplate(templateFile, config)
|
||
|
|
} else {
|
||
|
|
panic(fmt.Errorf("Unsupported template type"))
|
||
|
|
}
|
||
|
|
}
|