Merge pull request #43 from orangekame3/add-datetime

[feat]: Add datetime

Author: orangekame3

README.md

@@ -250,6 +250,7 @@ Usage:
   stree [bucket/prefix] [flags]
 
 Flags:
+  -D, --date-time             Print the last modified time of each file.
   -e, --endpoint-url string   AWS endpoint URL to use (useful for local testing with LocalStack)
   -f, --full-path             Print the full path prefix for each file.
   -h, --help                  help for stree
@@ -262,7 +263,7 @@ Flags:
   -p, --profile string        AWS profile to use (default "default")
   -r, --region string         AWS region to use (overrides the region specified in the profile)
   -s, --size                  Print the size of each file in bytes along with the name.
-  -v, --version               version for stree
+
 
 ```
 

cmd/root.go

@@ -50,6 +50,7 @@ var (
 	fileName    string
 	size 	  bool
 	humanReadable bool
+	dateTime   bool
 )
 
 var rootCmd = &cobra.Command{
@@ -80,7 +81,7 @@ var rootCmd = &cobra.Command{
 		if level > 0 {
 			maxDepth = &level
 		}
-		keys, err := pkg.FetchS3ObjectKeys(s3Svc, bucket, prefix, maxDepth,size, humanReadable)
+		keys, err := pkg.FetchS3ObjectKeys(s3Svc, bucket, prefix, maxDepth,size, humanReadable,dateTime)
 		if err != nil {
 			log.Fatalf("failed to fetch S3 object keys: %v", err)
 			return
@@ -139,6 +140,7 @@ func init() {
 	rootCmd.Flags().StringVarP(&fileName, "output", "o", "", "Send output to filename.")
 	rootCmd.Flags().BoolVarP(&size, "size", "s", false, "Print the size of each file in bytes along with the name.")
 	rootCmd.Flags().BoolVarP(&humanReadable, "human-readable", "H", false, "Print the size of each file but in a more human readable way, e.g. appending a size letter for kilobytes (K), megabytes (M), gigabytes (G), terabytes (T), petabytes (P) and exabytes(E).")
+	rootCmd.Flags().BoolVarP(&dateTime, "date-time", "D", false, "Print the last modified time of each file.")
 }
 
 func extractBucketAndPrefix(input string) (string, string, error) {

pkg/s3.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/config"
@@ -86,7 +87,7 @@ func formatBytes(b int64) string {
 }
 
 // FetchS3ObjectKeys returns a slice of keys for all objects in the specified bucket and prefix
-func FetchS3ObjectKeys(s3Client *s3.Client, bucket string, prefix string, maxDepth *int,size, humanReadable bool) ([][]string, error) {
+func FetchS3ObjectKeys(s3Client *s3.Client, bucket string, prefix string, maxDepth *int,size, humanReadable bool,dateTime bool) ([][]string, error) {
 	var delimiter *string
 	if maxDepth != nil {
 		delimiter = aws.String("/")
@@ -124,10 +125,22 @@ func FetchS3ObjectKeys(s3Client *s3.Client, bucket string, prefix string, maxDep
 
 			for _, obj := range page.Contents {
 				key := strings.Split(*obj.Key, "/")
+			
+				meta := []string{}
 				if humanReadable {
-					key[len(key)-1] = fmt.Sprintf("[%7s] %s", formatBytes(*obj.Size), key[len(key)-1])
+					meta = append(meta, formatBytes(*obj.Size))
 				} else if size {
-					key[len(key)-1] = fmt.Sprintf("[%7d] %s", *obj.Size, key[len(key)-1])
+					meta = append(meta, fmt.Sprintf("%d", *obj.Size))
+				}
+				if dateTime {
+					t := *obj.LastModified
+					layout := "Jan 2 15:04"
+					formatted := t.In(time.Local).Format(layout)
+					fmt.Println(formatted)
+					meta = append(meta, formatted)
+				}
+				if len(meta) > 0 {
+					key[len(key)-1] = fmt.Sprintf("[%7s] %s", strings.Join(meta, " "), key[len(key)-1])
 				}
 				keys = append(keys, key)
 			}