mirror of https://github.com/databricks/cli.git
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"slices"
|
|
|
|
"github.com/databricks/cli/bundle/config/mutator/paths"
|
|
|
|
"github.com/databricks/cli/libs/dyn"
|
|
)
|
|
|
|
func (t *translateContext) applyJobTranslations(ctx context.Context, v dyn.Value) (dyn.Value, error) {
|
|
var err error
|
|
|
|
fallback, err := gatherFallbackPaths(v, "jobs")
|
|
if err != nil {
|
|
return dyn.InvalidValue, err
|
|
}
|
|
|
|
// Do not translate job task paths if using Git source
|
|
var ignore []string
|
|
for key, job := range t.b.Config.Resources.Jobs {
|
|
if job.GitSource != nil {
|
|
ignore = append(ignore, key)
|
|
}
|
|
}
|
|
|
|
return paths.VisitJobPaths(v, func(p dyn.Path, kind paths.PathKind, v dyn.Value) (dyn.Value, error) {
|
|
key := p[2].Key()
|
|
|
|
// Skip path translation if the job is using git source.
|
|
if slices.Contains(ignore, key) {
|
|
return v, nil
|
|
}
|
|
|
|
dir, err := v.Location().Directory()
|
|
if err != nil {
|
|
return dyn.InvalidValue, fmt.Errorf("unable to determine directory for job %s: %w", key, err)
|
|
}
|
|
|
|
mode, err := getJobTranslateMode(kind)
|
|
if err != nil {
|
|
return dyn.InvalidValue, err
|
|
}
|
|
|
|
opts := translateOptions{
|
|
Mode: mode,
|
|
}
|
|
|
|
// Try to rewrite the path relative to the directory of the configuration file where the value was defined.
|
|
nv, err := t.rewriteValue(ctx, p, v, dir, opts)
|
|
if err == nil {
|
|
return nv, nil
|
|
}
|
|
|
|
// If we failed to rewrite the path, try to rewrite it relative to the fallback directory.
|
|
// We only do this for jobs and pipelines because of the comment in [gatherFallbackPaths].
|
|
if fallback[key] != "" {
|
|
nv, nerr := t.rewriteValue(ctx, p, v, fallback[key], opts)
|
|
if nerr == nil {
|
|
// TODO: Emit a warning that this path should be rewritten.
|
|
return nv, nil
|
|
}
|
|
}
|
|
|
|
return dyn.InvalidValue, err
|
|
})
|
|
}
|
|
|
|
func getJobTranslateMode(kind paths.PathKind) (TranslateMode, error) {
|
|
switch kind {
|
|
case paths.PathKindLibrary:
|
|
return TranslateModeLocalRelative, nil
|
|
case paths.PathKindNotebook:
|
|
return TranslateModeNotebook, nil
|
|
case paths.PathKindWorkspaceFile:
|
|
return TranslateModeFile, nil
|
|
case paths.PathKindDirectory:
|
|
return TranslateModeDirectory, nil
|
|
case paths.PathKindWithPrefix:
|
|
return TranslateModeLocalRelativeWithPrefix, nil
|
|
}
|
|
|
|
return TranslateMode(0), fmt.Errorf("unsupported path kind: %d", kind)
|
|
}
|