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
	domain   string
	endpoint string
}

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

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) Get(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
	}

	link := presigned.String()
	if s.domain != "" {
		link = strings.Replace(link, fmt.Sprintf("http://%s", s.endpoint), s.domain, 1)
	}

	return link, 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
}