bucket.go

package database

import (
	"context"
	"fmt"
	"log"
	"mime/multipart"
	"os"
	"time"

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

func InitBucket() *minio.Client {
	endpoint := os.Getenv("BUCKET_ENDPOINT")
	accessKeyID := os.Getenv("BUCKET_ACCESS_KEY")
	secretAccessKey := os.Getenv("BUCKET_SECRET_KEY")
	useSSL := false

	minioClient, err := minio.New(endpoint, &minio.Options{
		Creds:  credentials.NewStaticV4(accessKeyID, secretAccessKey, ""),
		Secure: useSSL,
	})
	if err != nil {
		log.Fatalln("Failed to initialize minio client:", err)
	}
	log.Println("Successfully connected to Minio")
	return minioClient
}

func UploadFile(client *minio.Client, file *multipart.FileHeader) (string, error) {
	ctx := context.Background()
	bucketName := "user-profile"
	publicDomain := os.Getenv("BUCKET_PUBLIC_DOMAIN")
	if publicDomain == "" {
		publicDomain = "http://localhost:8334"
	}

	src, err := file.Open()
	if err != nil {
		return "", err
	}
	defer func(src multipart.File) {
		err := src.Close()
		if err != nil {

		}
	}(src)

	exists, err := client.BucketExists(ctx, bucketName)
	if err != nil {
		return "", err
	}
	if !exists {
		err := client.MakeBucket(ctx, bucketName, minio.MakeBucketOptions{})
		if err != nil {
			return "", fmt.Errorf("failed to create bucket %s: %v", bucketName, err)
		}
		log.Println("Successfully created bucket:", bucketName)
	}

	objectName := fmt.Sprintf("file_%d_%s", time.Now().Unix(), file.Filename)
	contentType := file.Header.Get("Content-Type")

	info, err := client.PutObject(ctx, bucketName, objectName, src, file.Size, minio.PutObjectOptions{
		ContentType: contentType,
	})
	if err != nil {
		return "", err
	}
	log.Println("Successfully uploaded file:", info)
	fileURL := fmt.Sprintf("%s/%s/%s", publicDomain, bucketName, objectName)
	return fileURL, nil
}