mirror of https://github.com/databricks/cli.git
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package fs
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/databricks/cli/cmd/root"
|
|
"github.com/databricks/cli/libs/filer"
|
|
)
|
|
|
|
func filerForPath(ctx context.Context, fullPath string) (filer.Filer, string, error) {
|
|
// Split path at : to detect any file schemes
|
|
parts := strings.SplitN(fullPath, ":", 2)
|
|
|
|
// If no scheme is specified, then local path
|
|
if len(parts) < 2 {
|
|
f, err := filer.NewLocalClient("")
|
|
return f, fullPath, err
|
|
}
|
|
|
|
// On windows systems, paths start with a drive letter. If the scheme
|
|
// is a single letter and the OS is windows, then we conclude the path
|
|
// is meant to be a local path.
|
|
if runtime.GOOS == "windows" && len(parts[0]) == 1 {
|
|
f, err := filer.NewLocalClient("")
|
|
return f, fullPath, err
|
|
}
|
|
|
|
if parts[0] != "dbfs" {
|
|
return nil, "", fmt.Errorf("invalid scheme: %s", parts[0])
|
|
}
|
|
|
|
path := parts[1]
|
|
w := root.WorkspaceClient(ctx)
|
|
|
|
// If the specified path has the "Volumes" prefix, use the Files API.
|
|
if strings.HasPrefix(path, "/Volumes/") {
|
|
f, err := filer.NewFilesClient(w, "/")
|
|
return f, path, err
|
|
}
|
|
|
|
// The file is a dbfs file, and uses the DBFS APIs
|
|
f, err := filer.NewDbfsClient(w, "/")
|
|
return f, path, err
|
|
}
|
|
|
|
func isDbfsPath(path string) bool {
|
|
return strings.HasPrefix(path, "dbfs:/")
|
|
}
|