Skip to content

Commit

Permalink
lakectl local: implement list
Browse files Browse the repository at this point in the history
  • Loading branch information
N-o-Z committed Aug 2, 2023
1 parent 305d1ec commit b82d4f4
Show file tree
Hide file tree
Showing 10 changed files with 451 additions and 63 deletions.
59 changes: 0 additions & 59 deletions cmd/lakectl/cmd/index.go

This file was deleted.

11 changes: 8 additions & 3 deletions cmd/lakectl/cmd/local_init.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ import (

"github.com/spf13/cobra"
"github.com/treeverse/lakefs/pkg/git"
"github.com/treeverse/lakefs/pkg/local"
)

const (
Expand Down Expand Up @@ -36,18 +37,22 @@ var localInitCmd = &cobra.Command{
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
DieErr(err)
}
if IndexExists(localPath) && !force {
exists, err := local.IndexExists(localPath)
if err != nil {
DieErr(err)
}
if exists && !force {
Die(fmt.Sprintf("directory '%s' already linked to a lakefs path, run command with --force to overwrite", localPath), 1)
}

// dereference
head := resolveCommitOrDie(cmd.Context(), getClient(), remote.Repository, remote.Ref)
err = WriteIndex(localPath, remote, head)
err = local.WriteIndex(localPath, remote, head)
if err != nil {
DieErr(err)
}

ignoreFile, err := git.Ignore(localPath, []string{localPath, IndexFileName}, []string{IndexFileName}, IgnoreMarker)
ignoreFile, err := git.Ignore(localPath, []string{localPath, local.IndexFileName}, []string{local.IndexFileName}, local.IgnoreMarker)
if err != nil && !errors.Is(err, git.ErrNotARepository) {
DieErr(err)
} else if err == nil {
Expand Down
76 changes: 76 additions & 0 deletions cmd/lakectl/cmd/local_list.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
package cmd

import (
"path/filepath"

"github.com/cockroachdb/errors"
"github.com/jedib0t/go-pretty/v6/table"
"github.com/spf13/cobra"
"github.com/treeverse/lakefs/pkg/git"
"github.com/treeverse/lakefs/pkg/local"
)

const (
localListMinArgs = 0
localListMaxArgs = 1

indicesListTemplate = `{{.IndicesListTable | table -}}`
)

var localListCmd = &cobra.Command{
Use: "list [directory]",
Short: "find and list directories that are synced with lakeFS",
Args: cobra.RangeArgs(localListMinArgs, localListMaxArgs),
Run: func(cmd *cobra.Command, args []string) {
dir := "."
if len(args) > 0 {
dir = args[0]
}
abs, err := filepath.Abs(dir)
if err != nil {
DieErr(err)
}
gitRoot, err := git.GetRepositoryPath(abs)
if err == nil {
abs = gitRoot
} else if !(errors.Is(err, git.ErrNotARepository) || errors.Is(err, git.ErrNoGit)) { // allow support in environments with no git
DieErr(err)
}

dirs, err := local.FindIndices(abs)
if err != nil {
DieErr(err)
}

var rows [][]interface{}
for _, d := range dirs {
idx, err := local.ReadIndex(d)
if err != nil {
DieErr(err)
}
remote, err := idx.GetCurrentURI()
if err != nil {
DieErr(err)
}
rows = append(rows, table.Row{d, remote, idx.AtHead})
}
data := struct {
IndicesListTable *Table
}{
IndicesListTable: &Table{
Headers: []interface{}{
"Directory",
"Remote URI",
"Last synced HEAD",
},
Rows: rows,
},
}
Write(indicesListTemplate, data)
},
}

//nolint:gochecknoinits
func init() {
localCmd.AddCommand(localListCmd)
}
17 changes: 17 additions & 0 deletions docs/reference/cli.md
Original file line number Diff line number Diff line change
Expand Up @@ -2616,6 +2616,23 @@ lakectl local init <path uri> [directory] [flags]



### lakectl local list

find and list directories that are synced with lakeFS

```
lakectl local list [directory] [flags]
```

#### Options
{:.no_toc}

```
-h, --help help for list
```



### lakectl log

Show log of commits
Expand Down
43 changes: 42 additions & 1 deletion pkg/fileutil/io.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,16 @@
package fileutil

import "os"
import (
"errors"
"fmt"
"io/fs"
"os"
"path/filepath"
)

const (
DefaultDirectoryMask = 0o755
)

// IsDir Returns true if p is a directory, otherwise false
func IsDir(p string) (bool, error) {
Expand All @@ -10,3 +20,34 @@ func IsDir(p string) (bool, error) {
}
return stat.IsDir(), nil
}

func FindInParents(path, filename string) (string, error) {
var lookup string
fullPath, err := filepath.Abs(path)
if err != nil {
return "", err
}
for fullPath != string(filepath.Separator) && fullPath != filepath.VolumeName(fullPath) {
info, err := os.Stat(fullPath)
if errors.Is(err, fs.ErrNotExist) {
return "", fmt.Errorf("%s: %w", fullPath, fs.ErrNotExist)
} else if err != nil {
return "", err
}
if !info.IsDir() {
// find filename here
lookup = filepath.Join(filepath.Dir(fullPath), filename)
} else {
lookup = filepath.Join(fullPath, filename)
}
_, err = os.Stat(lookup)
if os.IsNotExist(err) {
fullPath = filepath.Dir(fullPath)
continue
} else if err != nil {
return "", err
}
return lookup, nil
}
return "", nil
}
85 changes: 85 additions & 0 deletions pkg/fileutil/io_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
package fileutil_test

import (
"os"
"path/filepath"
"testing"

"github.com/stretchr/testify/require"
"github.com/treeverse/lakefs/pkg/fileutil"
)

func TestFindInParents(t *testing.T) {
root := t.TempDir()
dirTree := filepath.Join(root, "foo", "bar", "baz", "taz")
require.NoError(t, os.MkdirAll(dirTree, fileutil.DefaultDirectoryMask))
t.Run("does not exist", func(t *testing.T) {
found, err := fileutil.FindInParents(root, ".doesnotexist21348329043289")
if err != nil {
t.Fatal(err)
}
if found != "" {
t.Errorf("expected found to be empty, got %v", found)
}
})

tests := []struct {
name string
deep string
filename string
filepath string
find bool
}{
{
name: "find_at_leaf",
deep: filepath.Join(root, "foo", "bar", "baz"),
filename: "some_file0",
filepath: filepath.Join(root, "foo", "bar", "baz", "some_file0"),
find: true,
},
{
name: "find_at_root",
deep: filepath.Join(root, "foo", "bar", "baz"),
filename: "some_file1",
filepath: filepath.Join(root, "some_file1"),
find: true,
},
{
name: "find_at_subpath",
deep: filepath.Join(root, "foo", "bar", "baz", "taz"),
filename: "some_file2",
filepath: filepath.Join(root, "foo", "some_file2"),
find: true,
},
{
name: "not_found_above",
deep: filepath.Join(root, "foo", "bar", "baz"),
filename: "some_file3",
filepath: filepath.Join(root, "foo", "bar", "baz", "taz", "some_file3"),
find: false,
},
{
name: "doesnt_exist",
deep: filepath.Join(root, "foo", "bar", "baz"),
filename: ".doesnotexist21348329043289",
filepath: filepath.Join(root, "foo", "bar", "some_file4"),
find: false,
},
}

for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
f, err := os.Create(tt.filepath)
require.NoError(t, err)
require.NoError(t, f.Close())

found, err := fileutil.FindInParents(tt.deep, tt.filename)
require.NoError(t, err)
if tt.find {
require.Equal(t, tt.filepath, found)
} else {
require.Equal(t, "", found)
}
})
}
}
1 change: 1 addition & 0 deletions pkg/git/errors.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,4 +7,5 @@ import (
var (
ErrGitError = errors.New("git error")
ErrNotARepository = errors.New("not a git repository")
ErrNoGit = errors.New("no git support")
)
4 changes: 4 additions & 0 deletions pkg/git/git.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,10 @@ const (
)

func git(dir string, args ...string) (string, error) {
_, err := exec.LookPath("git") // assume git is in path, otherwise consider as not having git support
if err != nil {
return "", ErrNoGit
}
cmd := exec.Command("git", args...)
cmd.Dir = dir
out, err := cmd.CombinedOutput()
Expand Down
Loading

0 comments on commit b82d4f4

Please sign in to comment.