mirror of https://github.com/databricks/cli.git
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package bundle
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/env"
|
|
"github.com/databricks/cli/folders"
|
|
)
|
|
|
|
// getRootEnv returns the value of the bundle root environment variable
|
|
// if it set and is a directory. If the environment variable is set but
|
|
// is not a directory, it returns an error. If the environment variable is
|
|
// not set, it returns an empty string.
|
|
func getRootEnv(ctx context.Context) (string, error) {
|
|
path, ok := env.Root(ctx)
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
stat, err := os.Stat(path)
|
|
if err == nil && !stat.IsDir() {
|
|
err = fmt.Errorf("not a directory")
|
|
}
|
|
if err != nil {
|
|
return "", fmt.Errorf(`invalid bundle root %s="%s": %w`, env.RootVariable, path, err)
|
|
}
|
|
return path, nil
|
|
}
|
|
|
|
// getRootWithTraversal returns the bundle root by traversing the filesystem
|
|
// from the working directory to the root looking for a configuration file.
|
|
func getRootWithTraversal() (string, error) {
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
for _, file := range config.FileNames {
|
|
path, err := folders.FindDirWithLeaf(wd, file)
|
|
if err == nil {
|
|
return path, nil
|
|
}
|
|
}
|
|
|
|
return "", fmt.Errorf(`unable to locate bundle root: %s not found`, config.FileNames[0])
|
|
}
|
|
|
|
// mustGetRoot returns a bundle root or an error if one cannot be found.
|
|
func mustGetRoot(ctx context.Context) (string, error) {
|
|
path, err := getRootEnv(ctx)
|
|
if path != "" || err != nil {
|
|
return path, err
|
|
}
|
|
return getRootWithTraversal()
|
|
}
|
|
|
|
// tryGetRoot returns a bundle root or an empty string if one cannot be found.
|
|
func tryGetRoot(ctx context.Context) (string, error) {
|
|
// Note: an invalid value in the environment variable is still an error.
|
|
path, err := getRootEnv(ctx)
|
|
if path != "" || err != nil {
|
|
return path, err
|
|
}
|
|
// Note: traversal failing means the bundle root cannot be found.
|
|
path, _ = getRootWithTraversal()
|
|
return path, nil
|
|
}
|