feat: add autocomplete func

Author: zezaeoh

internal/cmd/delete.go

@@ -32,4 +32,16 @@ var deleteCmd = &cobra.Command{
 		}
 		log.Infof("Delete: %s", name)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		stg, err := storage.GetStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		return stg.GetMatched(toComplete), cobra.ShellCompDirectiveNoFileComp
+	},
 }

internal/cmd/get.go

@@ -35,4 +35,16 @@ var getCmd = &cobra.Command{
 
 		fmt.Println(data)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		stg, err := storage.GetStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		return stg.GetMatched(toComplete), cobra.ShellCompDirectiveNoFileComp
+	},
 }

internal/cmd/storage/set.go

@@ -43,4 +43,16 @@ var setCmd = &cobra.Command{
 		}
 		log.Infof("Storage Configured: %s", name)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		cfg, err := storage.GetConfig()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		return cfg.GetMatchedStorage(toComplete), cobra.ShellCompDirectiveNoFileComp
+	},
 }

internal/storage/config.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"os"
 	"path/filepath"
+	"strings"
 )
 
 const (
@@ -136,3 +137,13 @@ func (cfg *config) AddStorage(n, k string, s map[string]interface{}) {
 		cfg.CurrentStorage = n
 	}
 }
+
+func (cfg *config) GetMatchedStorage(toMatched string) []string {
+	var matched []string
+	for _, s := range cfg.Storages {
+		if strings.HasPrefix(s.Name, toMatched) {
+			matched = append(matched, s.Name)
+		}
+	}
+	return matched
+}

internal/storage/github/storage.go

@@ -146,6 +146,38 @@ func (s *Storage) List() (tree.Node, error) {
 	return gitTreeToNode(t)
 }
 
+func (s *Storage) GetMatched(toMatched string) []string {
+	var matched []string
+
+	ctx := context.Background()
+	client := s.getClient()
+
+	if err := s.checkRepoExist(); err != nil {
+		return matched
+	}
+
+	b, _, err := client.Repositories.GetBranch(ctx, s.spec.owner, s.spec.repo, s.spec.Branch, true)
+	if err != nil {
+		return matched
+	}
+
+	t, _, err := client.Git.GetTree(ctx, s.spec.owner, s.spec.repo, b.GetCommit().GetSHA(), true)
+	if err != nil {
+		return matched
+	}
+
+	for _, e := range t.Entries {
+		if e.GetType() == "blob" && strings.HasSuffix(e.GetPath(), gboxSuffix) {
+			name := e.GetPath()
+			name = name[:len(name)-len(gboxSuffix)]
+			if strings.HasPrefix(name, toMatched) {
+				matched = append(matched, name)
+			}
+		}
+	}
+	return matched
+}
+
 func (s *Storage) checkRepoExist() error {
 	ctx := context.Background()
 	client := s.getClient()

internal/storage/storage.go

@@ -12,6 +12,7 @@ type Storage interface {
 	Set(name, data string) error
 	Delete(name string) error
 	List() (tree.Node, error)
+	GetMatched(toMatched string) []string
 }
 
 var (