mirror of https://github.com/databricks/cli.git
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/dyn"
|
|
)
|
|
|
|
type prependWorkspacePrefix struct{}
|
|
|
|
// PrependWorkspacePrefix prepends the workspace root path to all paths in the bundle.
|
|
func PrependWorkspacePrefix() bundle.Mutator {
|
|
return &prependWorkspacePrefix{}
|
|
}
|
|
|
|
func (m *prependWorkspacePrefix) Name() string {
|
|
return "PrependWorkspacePrefix"
|
|
}
|
|
|
|
var skipPrefixes = []string{
|
|
"/Workspace/",
|
|
"/Volumes/",
|
|
}
|
|
|
|
func (m *prependWorkspacePrefix) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
patterns := []dyn.Pattern{
|
|
dyn.NewPattern(dyn.Key("workspace"), dyn.Key("root_path")),
|
|
dyn.NewPattern(dyn.Key("workspace"), dyn.Key("file_path")),
|
|
dyn.NewPattern(dyn.Key("workspace"), dyn.Key("artifact_path")),
|
|
dyn.NewPattern(dyn.Key("workspace"), dyn.Key("state_path")),
|
|
}
|
|
|
|
err := b.Config.Mutate(func(v dyn.Value) (dyn.Value, error) {
|
|
var err error
|
|
for _, pattern := range patterns {
|
|
v, err = dyn.MapByPattern(v, pattern, func(p dyn.Path, pv dyn.Value) (dyn.Value, error) {
|
|
path, ok := pv.AsString()
|
|
if !ok {
|
|
return dyn.InvalidValue, fmt.Errorf("expected string, got %s", v.Kind())
|
|
}
|
|
|
|
for _, prefix := range skipPrefixes {
|
|
if strings.HasPrefix(path, prefix) {
|
|
return pv, nil
|
|
}
|
|
}
|
|
|
|
return dyn.NewValue(fmt.Sprintf("/Workspace%s", path), v.Locations()), nil
|
|
})
|
|
|
|
if err != nil {
|
|
return dyn.InvalidValue, err
|
|
}
|
|
}
|
|
return v, nil
|
|
})
|
|
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
return nil
|
|
}
|