steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Using go sdk for ./bricks fs ls DIR_NAME command

This commit is contained in:
Shreyas Goenka 2022-08-31 19:05:51 +02:00
parent 0ae8b60573
commit 6217d20e57
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/fs/ls.go
View File

@ -3,10 +3,9 @@ package fs
import ( import (
"fmt" "fmt"
"github.com/databricks/bricks/project"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"github.com/databricks/databricks-sdk-go/workspaces"
"github.com/databrickslabs/terraform-provider-databricks/storage" "github.com/databricks/databricks-sdk-go/service/dbfs"
) )
// lsCmd represents the ls command // lsCmd represents the ls command
@ -16,11 +15,17 @@ var lsCmd = &cobra.Command{
Long: `Lists files`, Long: `Lists files`,
Args: cobra.ExactArgs(1), Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
api := storage.NewDbfsAPI(cmd.Context(), project.Current.Client()) // Question (shreyas): Where does the client pick up the login creds from ? Context ?
files, err := api.List(args[0], false) // Maybe this client can be added on a higher level ?
// Create issue to add tests for this ? How to write those for cli ?
workspacesClient := workspaces.New()
listStatusResponse, err := workspacesClient.Dbfs.ListStatus(cmd.Context(),
dbfs.ListStatusRequest{Path: args[0]},
)
if err != nil { if err != nil {
panic(err) panic(err)
} }
files := listStatusResponse.Files
// TODO: output formatting: JSON, CSV, tables and default // TODO: output formatting: JSON, CSV, tables and default
for _, v := range files { for _, v := range files {
fmt.Printf("[-] %s (%d, %v)\n", v.Path, v.FileSize, v.IsDir) fmt.Printf("[-] %s (%d, %v)\n", v.Path, v.FileSize, v.IsDir)