fahd
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
)
|
|
|
|
var S3Client S3
|
|
|
|
type S3 interface {
|
|
GetPresignedURL(ctx context.Context, key, contentType string, expiry time.Duration) (string, error)
|
|
DoesFileExist(ctx context.Context, key string) (bool, error)
|
|
}
|
|
|
|
type S3Concrete struct {
|
|
S3Client *s3.Client
|
|
Bucket string
|
|
}
|
|
|
|
func (svc S3Concrete) GetPresignedURL(ctx context.Context, key, contentType string, expiry time.Duration) (string, error) {
|
|
presignClient := s3.NewPresignClient(svc.S3Client)
|
|
|
|
request, err := presignClient.PresignPutObject(ctx, &s3.PutObjectInput{
|
|
Bucket: &svc.Bucket,
|
|
Key: &key,
|
|
ContentType: &contentType,
|
|
}, s3.WithPresignExpires(expiry))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return request.URL, nil
|
|
}
|
|
|
|
func (svc S3Concrete) DoesFileExist(ctx context.Context, key string) (bool, error) {
|
|
_, error := svc.S3Client.HeadObject(ctx, &s3.HeadObjectInput{
|
|
Bucket: &svc.Bucket,
|
|
Key: &key,
|
|
})
|
|
|
|
if error != nil {
|
|
return false, error
|
|
}
|
|
|
|
return true, nil
|
|
}
|