2fas-server/internal/common/aws/s3.go

package aws

import (
	"fmt"
	"io"
	"os"
	"path/filepath"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
	"github.com/aws/aws-sdk-go/service/s3/s3manager"

	"github.com/twofas/2fas-server/internal/common/logging"
)

type AwsS3 struct {
	Region          string
	AccessKeyId     string
	AccessSecretKey string
}

func NewAwsS3(region, accessKeyId, secretAccessKey string) *AwsS3 {
	logging.WithFields(logging.Fields{
		"region": region,
	}).Debug("Initialize AWS S3 instance")

	return &AwsS3{
		Region:          region,
		AccessKeyId:     accessKeyId,
		AccessSecretKey: secretAccessKey,
	}
}

func (s *AwsS3) Get(path string) (file *os.File, err error) {
	directory := filepath.Dir(path)
	name := filepath.Base(path)

	sess, err := session.NewSessionWithOptions(session.Options{
		Config: aws.Config{
			Region:      aws.String(s.Region),
			Credentials: credentials.NewStaticCredentials(s.AccessKeyId, s.AccessSecretKey, ""),
		},
	})

	downloader := s3manager.NewDownloader(sess)

	f, err := os.Create(name)
	if err != nil {
		return nil, fmt.Errorf("failed to create file: %w", err)
	}

	_, err = downloader.Download(f, &s3.GetObjectInput{
		Bucket: aws.String(directory),
		Key:    aws.String(name),
	})
	if err != nil {
		return nil, fmt.Errorf("failed to download the object from s3: %w", err)
	}

	return f, nil
}

func (s *AwsS3) Save(path string, data io.Reader) (location string, err error) {
	directory := filepath.Dir(path)
	name := filepath.Base(path)

	sess, err := session.NewSessionWithOptions(session.Options{
		Config: aws.Config{
			Region:      aws.String(s.Region),
			Credentials: credentials.NewStaticCredentials(s.AccessKeyId, s.AccessSecretKey, ""),
		},
	})

	uploader := s3manager.NewUploader(sess)

	result, err := uploader.Upload(&s3manager.UploadInput{
		Bucket: aws.String(directory),
		Key:    aws.String(name),
		Body:   data,
	})

	if err != nil {
		logging.WithFields(logging.Fields{
			"error":    err.Error(),
			"bucket":   directory,
			"filename": name,
		}).Error("Cannot upload file")

		return "", err
	}

	return result.Location, nil
}

func (s *AwsS3) Move(oldPath, newPath string) (location string, err error) {
	sourceDirectory := filepath.Dir(oldPath)
	sourceName := filepath.Base(oldPath)

	sess, err := session.NewSessionWithOptions(session.Options{
		Config: aws.Config{
			Region:      aws.String(s.Region),
			Credentials: credentials.NewStaticCredentials(s.AccessKeyId, s.AccessSecretKey, ""),
		},
	})

	svc := s3.New(sess)

	file, err := s.Get(oldPath)

	if err != nil {
		return "", err
	}

	newLocation, err := s.Save(newPath, file)

	if err != nil {
		return "", err
	}

	_, err = svc.DeleteObject(&s3.DeleteObjectInput{
		Bucket: aws.String(sourceDirectory),
		Key:    aws.String(sourceName)},
	)

	if err != nil {
		logging.WithFields(logging.Fields{
			"error":    err.Error(),
			"bucket":   sourceDirectory,
			"filename": sourceName,
		}).Error("Cannot delete file")

		return newLocation, err
	}

	return newLocation, nil
}