mirror of https://github.com/databricks/cli.git
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
|
|
"github.com/databricks/cli/libs/dyn"
|
|
)
|
|
|
|
type rewriteSyncPaths struct{}
|
|
|
|
func RewriteSyncPaths() bundle.Mutator {
|
|
return &rewriteSyncPaths{}
|
|
}
|
|
|
|
func (m *rewriteSyncPaths) Name() string {
|
|
return "RewriteSyncPaths"
|
|
}
|
|
|
|
// makeRelativeTo returns a dyn.MapFunc that joins the relative path
|
|
// of the file it was defined in w.r.t. the bundle root path, with
|
|
// the contents of the string node.
|
|
//
|
|
// For example:
|
|
// - The bundle root is /foo
|
|
// - The configuration file that defines the string node is at /foo/bar/baz.yml
|
|
// - The string node contains "somefile.*"
|
|
//
|
|
// Then the resulting value will be "bar/somefile.*".
|
|
func (m *rewriteSyncPaths) makeRelativeTo(root string) dyn.MapFunc {
|
|
return func(_ dyn.Path, v dyn.Value) (dyn.Value, error) {
|
|
dir := filepath.Dir(v.Location().File)
|
|
rel, err := filepath.Rel(root, dir)
|
|
if err != nil {
|
|
return dyn.NilValue, err
|
|
}
|
|
|
|
return dyn.NewValue(filepath.Join(rel, v.MustString()), v.Location()), nil
|
|
}
|
|
}
|
|
|
|
func (m *rewriteSyncPaths) Apply(ctx context.Context, b *bundle.Bundle) error {
|
|
return b.Config.Mutate(func(v dyn.Value) (dyn.Value, error) {
|
|
return dyn.Map(v, "sync", func(_ dyn.Path, v dyn.Value) (nv dyn.Value, err error) {
|
|
v, err = dyn.Map(v, "include", dyn.Foreach(m.makeRelativeTo(b.Config.Path)))
|
|
if err != nil {
|
|
return dyn.NilValue, err
|
|
}
|
|
v, err = dyn.Map(v, "exclude", dyn.Foreach(m.makeRelativeTo(b.Config.Path)))
|
|
if err != nil {
|
|
return dyn.NilValue, err
|
|
}
|
|
return v, nil
|
|
})
|
|
})
|
|
}
|