package minio

import (
	"context"
	"io"
	"mime"
	"net/url"

	"go-socket/packages/globals"

	"github.com/minio/minio-go/v7"
	"github.com/minio/minio-go/v7/pkg/credentials"
)

var minClient *minio.Client

var canonicalExt = map[string]string{
	"image/jpeg":      ".jpg",
	"image/png":       ".png",
	"image/gif":       ".gif",
	"image/webp":      ".webp",
	"video/mp4":       ".mp4",
	"application/pdf": ".pdf",
}

func extensionFromContentType(ct string) string {
	if ext, ok := canonicalExt[ct]; ok {
		return ext
	}
	exts, err := mime.ExtensionsByType(ct)
	if err != nil || len(exts) == 0 {
		return ".unk"
	}
	return exts[0]
}

func GetKey(hash string, contentType string) string {
	return hash + extensionFromContentType(contentType)
}

func Init() {
	ctx := context.Background()

	var err error
	minClient, err = minio.New("localhost:9000", &minio.Options{
		Creds:  credentials.NewStaticV4("root", "change_to_env", ""),
		Secure: false,
	}) // TODO change in production
	if err != nil {
		panic(err)
	}

	exists, err := minClient.BucketExists(ctx, globals.FileStorageBucketName)
	if err != nil {
		panic(err)
	}

	if !exists {
		err = minClient.MakeBucket(ctx, globals.FileStorageBucketName, minio.MakeBucketOptions{})
		if err != nil {
			return
		}
	}

}

func Upload(ctx context.Context, key string, body io.Reader, size int64, contentType string, metadata map[string]string) error {
	opt := minio.PutObjectOptions{
		ContentType:  contentType,
		PartSize:     globals.FileProcessingPartSize,
		NumThreads:   globals.FileProcessingThreads,
		UserMetadata: metadata,
	}
	opt.SetMatchETagExcept("*")

	_, err := minClient.PutObject(ctx, globals.FileStorageBucketName, key, body, size, opt)
	return err
}

func GetDownloadUrl(ctx context.Context, key string) (*url.URL, error) {
	u, err := minClient.PresignedGetObject(ctx, globals.FileStorageBucketName, key, globals.FileDownloadLinkTtl, nil)
	return u, err
}

func DoesExist(ctx context.Context, key string) bool {
	_, err := minClient.StatObject(ctx, globals.FileStorageBucketName, key, minio.StatObjectOptions{})
	return err == nil
}