api/internal/mediaStorage/service.go

package mediaStorage

import (
	"context"
	"fmt"
	"github.com/minio/minio-go/v7"
	log "github.com/sirupsen/logrus"
	"io"
	"net/url"
	"strings"
	"time"
)

type Service struct {
	client     *minio.Client
	endpoint   string
	secureMode bool
}

func newService(client *minio.Client, endpoint string, secureMode bool) *Service {
	return &Service{
		client:     client,
		endpoint:   endpoint,
		secureMode: secureMode,
	}
}

func (s *Service) CheckBucketExists(bucketName string) (bool, error) {
	ctx := context.Background()
	exists, err := s.client.BucketExists(ctx, bucketName)
	if err != nil {
		log.WithError(err).Fatal("Media storage | Failed to check bucket existence")
		return exists, err
	}
	log.Infof("Media storage | Bucket %s exists", bucketName)
	return exists, nil
}

func (s *Service) Upload(ctx context.Context, bucket, object string, reader io.Reader, size int64) error {
	_, err := s.client.PutObject(ctx, bucket, object, reader, size, minio.PutObjectOptions{ContentType: "image/jpeg"})
	return err
}

func (s *Service) GetPublicLink(ctx context.Context, bucket, object string) (string, string, error) {
	stat, err := s.client.StatObject(ctx, bucket, object, minio.StatObjectOptions{})
	if err != nil {
		log.WithFields(log.Fields{
			"error": err,
			"key":   bucket + "/" + object,
		}).Error("Media storage | Failed to get public link")
		return "", "", err
	}

	var scheme string
	if s.secureMode {
		scheme = "https"
	} else {
		scheme = "http"
	}

	link := fmt.Sprintf("%s://%s/%s/%s", scheme, strings.TrimRight(s.endpoint, "/"), bucket, object)
	log.WithFields(log.Fields{"link": link}).Debug("Media storage | Get public link")

	return link, stat.ETag, nil
}

func (s *Service) GetPresignedLink(ctx context.Context, bucket, object string, expires time.Duration, params url.Values) (string, error) {
	presigned, err := s.client.PresignedGetObject(ctx, bucket, object, expires, params)
	if err != nil {
		return "", err
	}

	return presigned.String(), nil
}

func (s *Service) Delete(ctx context.Context, bucket, object string) error {
	return s.client.RemoveObject(ctx, bucket, object, minio.RemoveObjectOptions{})
}

func (s *Service) GetObjectEtag(ctx context.Context, bucketName, object string) (string, error) {
	info, err := s.client.StatObject(ctx, bucketName, object, minio.StatObjectOptions{})
	if err != nil {
		return "", err
	}
	return info.ETag, nil
}