Add browser

aws/s3.go

@@ -0,0 +1,48 @@
+package aws
+
+import (
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/s3"
+)
+
+type Image struct {
+	Url      string
+	Download string
+	Name     string
+	S3Path   string
+}
+
+type BucketList struct {
+	Images     []Image
+	StartAfter string
+}
+
+func (s *Service) ListBucketContents(continuationToken *string) (*BucketList, error) {
+	v2, err := s.s3.ListObjectsV2(&s3.ListObjectsV2Input{
+		Bucket:            aws.String(s.bucket),
+		MaxKeys:           aws.Int64(20),
+		ContinuationToken: continuationToken,
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	bucketList := BucketList{
+		Images: make([]Image, 0),
+	}
+
+	if v2.NextContinuationToken != nil {
+		bucketList.StartAfter = *v2.NextContinuationToken
+	}
+
+	for _, item := range v2.Contents {
+		image := Image{
+			Name:   *item.Key,
+			S3Path: "s3://" + s.bucket + "/" + *item.Key,
+		}
+		bucketList.Images = append(bucketList.Images, image)
+	}
+
+	return &bucketList, nil
+}