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(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(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 }) }) }