64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package s3
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"fiskerinc.com/modules/utils/envtool"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/s3/s3manager"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var instanceS3 Interface
|
|
var bucket string = envtool.GetEnv("OTA_UPDATE_BUCKET", "fisker-upload-dev")
|
|
var downloadurl = envtool.GetEnv("OTA_UPDATE_DOWNLOAD_URL", "https://upload-dev.fiskerdps.com")
|
|
|
|
// S3 AWS S3 bucket
|
|
type S3 struct {
|
|
}
|
|
|
|
// Interface interface for S3
|
|
type Interface interface {
|
|
PutBucket(key string, reader io.Reader, contentType string) (string, error)
|
|
}
|
|
|
|
// PutBucket put file into S3 bucket
|
|
func (s *S3) PutBucket(key string, reader io.Reader, contentType string) (string, error) {
|
|
if len(key) > 255 {
|
|
return "", errors.New("S3 key too long")
|
|
}
|
|
|
|
sess := session.Must(session.NewSession())
|
|
uploader := s3manager.NewUploader(sess)
|
|
_, err := uploader.Upload(&s3manager.UploadInput{
|
|
Bucket: aws.String(bucket),
|
|
Key: aws.String(key),
|
|
Body: reader,
|
|
ACL: aws.String("public-read"),
|
|
ContentType: aws.String(contentType),
|
|
})
|
|
if err != nil {
|
|
return "", errors.WithStack(err)
|
|
}
|
|
|
|
return fmt.Sprintf("%s/%s", downloadurl, key), nil
|
|
}
|
|
|
|
// GetS3 returns S3 struct
|
|
func GetS3() Interface {
|
|
if instanceS3 != nil {
|
|
return instanceS3
|
|
}
|
|
|
|
return &S3{}
|
|
}
|
|
|
|
// SetS3Instance set S3 instance to return for mocking
|
|
func SetS3Instance(value Interface) {
|
|
instanceS3 = value
|
|
}
|