86 lines
2.3 KiB
Go
86 lines
2.3 KiB
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
|
|
}
|