Merge pull request #119 from trickest/feat/files-list

Add files list command

README.md

@@ -267,6 +267,19 @@ trickest files delete --file delete_me.txt
 | --file               | string  | /       | File or files (comma-separated)                                     |
 
 
+#### List files
+Use the **list** command with the optional **--query** flag to list or search for files
+
+```
+trickest files list
+```
+
+| Flag                 | Type    | Default  | Description                                                         |
+|----------------------|---------|----------|---------------------------------------------------------------------|
+| --query              | string  | /        | Filter listed files using the specified search query                |
+| --json               | boolean | false    | Display output in JSON format                                       |
+
+
 ## Tools command
 Manage [private tools](https://trickest.com/docs/tutorials/private-tools/private-tools-library/)
 

cmd/files/files.go

@@ -26,9 +26,6 @@ var FilesCmd = &cobra.Command{
 }
 
 func init() {
-	FilesCmd.PersistentFlags().StringVar(&Files, "file", "", "File or files (comma-separated)")
-	FilesCmd.MarkPersistentFlagRequired("file")
-
 	FilesCmd.SetHelpFunc(func(command *cobra.Command, strings []string) {
 		_ = FilesCmd.Flags().MarkHidden("workflow")
 		_ = FilesCmd.Flags().MarkHidden("project")
@@ -40,16 +37,29 @@ func init() {
 }
 
 func getMetadata(searchQuery string) ([]types.File, error) {
-	resp := request.Trickest.Get().DoF("file/?search=%s&vault=%s", searchQuery, util.GetVault())
-	if resp == nil || resp.Status() != http.StatusOK {
-		return nil, fmt.Errorf("unexpected response status code: %d", resp.Status())
-	}
-	var metadata types.Files
+	pageSize := 100
+
+	page := 1
+	var allFiles []types.File
+	for {
+		resp := request.Trickest.Get().DoF("file/?search=%s&vault=%s&page_size=%d&page=%d", searchQuery, util.GetVault(), pageSize, page)
+		if resp == nil || resp.Status() != http.StatusOK {
+			return nil, fmt.Errorf("unexpected response status code: %d", resp.Status())
+		}
+
+		var metadata types.Files
+		err := json.Unmarshal(resp.Body(), &metadata)
+		if err != nil {
+			return nil, fmt.Errorf("couldn't unmarshal file IDs response: %s", err)
+		}
+
+		allFiles = append(allFiles, metadata.Results...)
 
-	err := json.Unmarshal(resp.Body(), &metadata)
-	if err != nil {
-		return nil, fmt.Errorf("couldn't unmarshal file IDs response: %s", err)
+		if metadata.Next == "" {
+			break
+		}
+		page++
 	}
 
-	return metadata.Results, nil
+	return allFiles, nil
 }

cmd/files/filesCreate.go

@@ -36,6 +36,9 @@ var filesCreateCmd = &cobra.Command{
 
 func init() {
 	FilesCmd.AddCommand(filesCreateCmd)
+
+	filesCreateCmd.Flags().StringVar(&Files, "file", "", "File or files (comma-separated)")
+	filesCreateCmd.MarkFlagRequired("file")
 }
 
 func createFile(filePath string) error {

cmd/files/filesDelete.go

@@ -29,6 +29,9 @@ var filesDeleteCmd = &cobra.Command{
 
 func init() {
 	FilesCmd.AddCommand(filesDeleteCmd)
+
+	filesDeleteCmd.Flags().StringVar(&Files, "file", "", "File or files (comma-separated)")
+	filesDeleteCmd.MarkFlagRequired("file")
 }
 
 func deleteFile(fileName string) error {

cmd/files/filesGet.go

@@ -37,8 +37,9 @@ var filesGetCmd = &cobra.Command{
 func init() {
 	FilesCmd.AddCommand(filesGetCmd)
 
+	filesGetCmd.Flags().StringVar(&Files, "file", "", "File or files (comma-separated)")
+	filesGetCmd.MarkFlagRequired("file")
 	filesGetCmd.Flags().StringVar(&outputDir, "output-dir", ".", "Path to directory which should be used to store files")
-
 	filesGetCmd.Flags().BoolVar(&partialNameMatch, "partial-name-match", false, "Get all files with a partial name match")
 }
 

cmd/files/filesList.go

@@ -0,0 +1,62 @@
+package files
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/trickest/trickest-cli/types"
+	"github.com/xlab/treeprint"
+)
+
+var (
+	searchQuery string
+	jsonOutput  bool
+)
+
+// filesListCmd represents the filesGet command
+var filesListCmd = &cobra.Command{
+	Use:   "list",
+	Short: "List files in the Trickest file storage",
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+		files, err := getMetadata(searchQuery)
+		if err != nil {
+			fmt.Printf("Error: %s\n", err)
+		} else {
+			printFiles(files, jsonOutput)
+		}
+	},
+}
+
+func init() {
+	FilesCmd.AddCommand(filesListCmd)
+
+	filesListCmd.Flags().StringVar(&searchQuery, "query", "", "Filter listed files using the specified search query")
+	filesListCmd.Flags().BoolVar(&jsonOutput, "json", false, "Display output in JSON format")
+}
+
+func printFiles(files []types.File, jsonOutput bool) {
+	var output string
+
+	if jsonOutput {
+		data, err := json.Marshal(files)
+		if err != nil {
+			fmt.Println("Error marshalling response data")
+			return
+		}
+		output = string(data)
+	} else {
+		tree := treeprint.New()
+		tree.SetValue("Files")
+		for _, file := range files {
+			fileSubBranch := tree.AddBranch("\U0001f4c4 " + file.Name)                             //📄
+			fileSubBranch.AddNode("\U0001f522 " + file.PrettySize)                                 //🔢
+			fileSubBranch.AddNode("\U0001f4c5 " + file.ModifiedDate.Format("2006-01-02 15:04:05")) //📅
+		}
+
+		output = tree.String()
+	}
+
+	fmt.Println(output)
+}

types/files.go

@@ -16,9 +16,6 @@ type Files struct {
 type File struct {
 	ID           string    `json:"id"`
 	Name         string    `json:"name"`
-	Vault        string    `json:"vault"`
-	TweID        string    `json:"twe_id"`
-	ArtifactID   string    `json:"artifact_id"`
 	Size         int       `json:"size"`
 	PrettySize   string    `json:"pretty_size"`
 	ModifiedDate time.Time `json:"modified_date"`