Moved testimonials to S3

lib/s3.go

@@ -11,13 +11,16 @@ import (
 	"github.com/aws/aws-sdk-go/service/s3"
 )
 
+// Generates a presigned URL for the given key. Returns the pre-signed URL.
 func GeneratePublicURL(key string) string {
+	// Get the S3 bucket name
 	bucket := os.Getenv("BUCKET_NAME")
 	if bucket == "" {
-		fmt.Println("No S3 bucket specified, skipping upload.")
+		fmt.Println("No S3 bucket specified.")
 		return ""
 	}
 
+	// Create the S3 session
 	endpoint := os.Getenv("AWS_ENDPOINT_URL_S3")
 	accessKeyID := os.Getenv("AWS_ACCESS_KEY_ID")
 	secretAccessKey := os.Getenv("AWS_SECRET_ACCESS_KEY")
@@ -38,6 +41,7 @@ func GeneratePublicURL(key string) string {
 
 	svc := s3.New(sess)
 
+	// Generate the presigned URL
 	req, _ := svc.GetObjectRequest(&s3.GetObjectInput{
 		Bucket: &bucket,
 		Key:    &key,
@@ -50,3 +54,72 @@ func GeneratePublicURL(key string) string {
 
 	return urlStr
 }
+
+// Generates a list of presigned URLs for all items in the given directory. Returns an array of pre-signed URLs.
+func GeneratePublicURLsFromDirectory(directory string) []string {
+	// Get the S3 bucket name
+	bucket := os.Getenv("BUCKET_NAME")
+	if bucket == "" {
+		fmt.Println("No S3 bucket specified.")
+		return []string{}
+	}
+
+	// Create the S3 session
+	endpoint := os.Getenv("AWS_ENDPOINT_URL_S3")
+	accessKeyID := os.Getenv("AWS_ACCESS_KEY_ID")
+	secretAccessKey := os.Getenv("AWS_SECRET_ACCESS_KEY")
+	region := os.Getenv("AWS_REGION")
+
+	sess, err := session.NewSession(&aws.Config{
+		Region: &region,
+		Credentials: credentials.NewStaticCredentials(
+			accessKeyID,
+			secretAccessKey,
+			"",
+		),
+		Endpoint: aws.String(endpoint),
+	})
+	if err != nil {
+		return []string{}
+	}
+
+	svc := s3.New(sess)
+
+	// Create the input for the ListObjectsV2 call
+	input := &s3.ListObjectsV2Input{
+		Bucket: aws.String(bucket),
+		Prefix: aws.String(directory),
+	}
+
+	// Get the list of items in the directory
+	result, err := svc.ListObjectsV2(input)
+	if err != nil {
+		return []string{}
+	}
+
+	//Remove the items in results that are directories
+	for i := 0; i < len(result.Contents); i++ {
+		if *result.Contents[i].Key == directory {
+			result.Contents = append(result.Contents[:i], result.Contents[i+1:]...)
+			i--
+		}
+	}
+
+	// Generate the presigned URLs
+	urls := []string{}
+	for _, item := range result.Contents {
+		req, _ := svc.GetObjectRequest(&s3.GetObjectInput{
+			Bucket: aws.String(bucket),
+			Key:    item.Key,
+		})
+		urlStr, err := req.Presign(15 * time.Minute)
+
+		if err != nil {
+			continue
+		}
+
+		urls = append(urls, urlStr)
+	}
+
+	return urls
+}