package storage

import (
	"io"

	"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/aws/aws-sdk-go/service/s3/s3manager/s3manageriface"
)

type Storage interface {
	Store(path string, content io.Reader) error
}

type S3 struct {
	bucket string
	svc    s3manageriface.UploaderAPI
}

func (s *S3) Store(path string, content io.Reader) error {
	_, err := s.svc.Upload(&s3manager.UploadInput{
		Bucket: aws.String(s.bucket),
		Key:    aws.String(path),
		Body:   content,
	})
	return err
}

func New(bucket string) (Storage, error) {
	sess, err := session.NewSession(aws.NewConfig())
	if err != nil {
		return nil, err
	}
	uploader := s3manager.NewUploader(sess, func(u *s3manager.Uploader) {
		u.PartSize = 5 * 1024 * 1024
	})
	return &S3{bucket: bucket, svc: uploader}, nil
}