mirror of https://github.com/databricks/cli.git
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
|
package mutator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"path"
|
||
|
|
|
||
|
|
"github.com/databricks/bricks/bundle"
|
||
|
|
)
|
||
|
|
|
||
|
|
type defaultArtifactPath struct{}
|
||
|
|
|
||
|
|
// DefaultArtifactPath configures the artifact path if it isn't already set.
|
||
|
|
func DefaultArtifactPath() bundle.Mutator {
|
||
|
|
return &defaultArtifactPath{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *defaultArtifactPath) Name() string {
|
||
|
|
return "DefaultArtifactPath"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *defaultArtifactPath) Apply(ctx context.Context, b *bundle.Bundle) ([]bundle.Mutator, error) {
|
||
|
|
if b.Config.Workspace.ArtifactPath.IsSet() {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
me := b.Config.Workspace.CurrentUser
|
||
|
|
if me == nil {
|
||
|
|
return nil, fmt.Errorf("cannot configured default artifact path if current user isn't set")
|
||
|
|
}
|
||
|
|
|
||
|
|
// We assume we deal with notebooks only for the time being.
|
||
|
|
// When we need to upload wheel files or other non-notebook files,
|
||
|
|
// the workspace must support "Files in Workspace".
|
||
|
|
// If it doesn't, we need to resort to storing artifacts on DBFS.
|
||
|
|
home := fmt.Sprintf("/Users/%s", me.UserName)
|
||
|
|
root := path.Join(home, ".bundle", b.Config.Bundle.Name, b.Config.Bundle.Environment, "artifacts")
|
||
|
|
b.Config.Workspace.ArtifactPath.Workspace = root
|
||
|
|
return nil, nil
|
||
|
|
}
|