6c4344623b
The main HCL package is more visible this way, and so it's easier than having to pick it out from dozens of other package directories.
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package transform
|
|
|
|
import (
|
|
"github.com/hashicorp/hcl/v2"
|
|
)
|
|
|
|
// A Transformer takes a given body, applies some (possibly no-op)
|
|
// transform to it, and returns the new body.
|
|
//
|
|
// It must _not_ mutate the given body in-place.
|
|
//
|
|
// The transform call cannot fail, but it _can_ return a body that immediately
|
|
// returns diagnostics when its methods are called. NewErrorBody is a utility
|
|
// to help with this.
|
|
type Transformer interface {
|
|
TransformBody(hcl.Body) hcl.Body
|
|
}
|
|
|
|
// TransformerFunc is a function type that implements Transformer.
|
|
type TransformerFunc func(hcl.Body) hcl.Body
|
|
|
|
// TransformBody is an implementation of Transformer.TransformBody.
|
|
func (f TransformerFunc) TransformBody(in hcl.Body) hcl.Body {
|
|
return f(in)
|
|
}
|
|
|
|
type chain []Transformer
|
|
|
|
// Chain takes a slice of transformers and returns a single new
|
|
// Transformer that applies each of the given transformers in sequence.
|
|
func Chain(c []Transformer) Transformer {
|
|
return chain(c)
|
|
}
|
|
|
|
func (c chain) TransformBody(body hcl.Body) hcl.Body {
|
|
for _, t := range c {
|
|
body = t.TransformBody(body)
|
|
}
|
|
return body
|
|
}
|