mirror of https://github.com/databricks/cli.git
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package fs
|
|
|
|
import (
|
|
"io/fs"
|
|
"path"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/databricks/cli/cmd/root"
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type jsonDirEntry struct {
|
|
Name string `json:"name"`
|
|
IsDir bool `json:"is_directory"`
|
|
Size int64 `json:"size"`
|
|
ModTime time.Time `json:"last_modified"`
|
|
}
|
|
|
|
func toJsonDirEntry(f fs.DirEntry, baseDir string, isAbsolute bool) (*jsonDirEntry, error) {
|
|
info, err := f.Info()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
name := f.Name()
|
|
if isAbsolute {
|
|
name = path.Join(baseDir, name)
|
|
}
|
|
|
|
return &jsonDirEntry{
|
|
Name: name,
|
|
IsDir: f.IsDir(),
|
|
Size: info.Size(),
|
|
ModTime: info.ModTime(),
|
|
}, nil
|
|
}
|
|
|
|
func newLsCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "ls DIR_PATH",
|
|
Short: "Lists files.",
|
|
Long: `Lists files in DBFS and UC Volumes.`,
|
|
Args: root.ExactArgs(1),
|
|
PreRunE: root.MustWorkspaceClient,
|
|
}
|
|
|
|
var long bool
|
|
var absolute bool
|
|
cmd.Flags().BoolVarP(&long, "long", "l", false, "Displays full information including size, file type and modification time since Epoch in milliseconds.")
|
|
cmd.Flags().BoolVar(&absolute, "absolute", false, "Displays absolute paths.")
|
|
|
|
cmd.RunE = func(cmd *cobra.Command, args []string) error {
|
|
ctx := cmd.Context()
|
|
|
|
f, path, err := filerForPath(ctx, args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
entries, err := f.ReadDir(ctx, path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
jsonDirEntries := make([]jsonDirEntry, len(entries))
|
|
for i, entry := range entries {
|
|
jsonDirEntry, err := toJsonDirEntry(entry, args[0], absolute)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
jsonDirEntries[i] = *jsonDirEntry
|
|
}
|
|
sort.Slice(jsonDirEntries, func(i, j int) bool {
|
|
return jsonDirEntries[i].Name < jsonDirEntries[j].Name
|
|
})
|
|
|
|
// Use template for long mode if the flag is set
|
|
if long {
|
|
return cmdio.RenderWithTemplate(ctx, jsonDirEntries, "", cmdio.Heredoc(`
|
|
{{range .}}{{if .IsDir}}DIRECTORY {{else}}FILE {{end}}{{.Size}} {{.ModTime|pretty_date}} {{.Name}}
|
|
{{end}}
|
|
`))
|
|
}
|
|
return cmdio.RenderWithTemplate(ctx, jsonDirEntries, "", cmdio.Heredoc(`
|
|
{{range .}}{{.Name}}
|
|
{{end}}
|
|
`))
|
|
}
|
|
|
|
return cmd
|
|
}
|