FranklinWH-Toolbox/service/utils/minio/minio.go

package minio

import (
	"context"
	"fmt"
	"fwh-toolbox-server/config"
	"io"
	"log"
	"net/url"
	"strings"
	"time"

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

var Client *minio.Client
var BucketName string

func InitClient(cfg *config.Config) {
	var err error
	Client, err = minio.New(cfg.MinIoEndpoint, &minio.Options{
		Creds:  credentials.NewStaticV4(cfg.MinioAccessKey, cfg.MinioSecretKey, ""),
		Secure: strings.HasPrefix(strings.ToLower(cfg.MinIoEndpoint), "https://"),
	})
	if err != nil {
		log.Fatalln("❌ Failed to init MinIO client")
		panic(err)
	}
	BucketName = cfg.MinioBucketName

	ctx := context.Background()
	bucketInfo, err := Client.ListBuckets(ctx)
	if err != nil {
		log.Fatalln("❌ List Buckets failed")
		panic(err)
	}
	log.Println("Exists Bucket:")
	for _, info := range bucketInfo {
		log.Printf("%s\n", info.Name)
	}
	exists, err := Client.BucketExists(ctx, BucketName)
	if err != nil {
		log.Fatalf("❌ Bucket check failed: %s\n", BucketName)
		panic(err)
	}
	if !exists {
		err = Client.MakeBucket(ctx, BucketName, minio.MakeBucketOptions{})
		if err != nil {
			log.Fatalln("❌ Failed to create bucket")
			panic(err)
		}
		log.Println("✅ Created bucket:", BucketName)
	} else {
		log.Println("✅ Using existing bucket:", BucketName)
	}
}

func UploadFile(localPath, objectName string) error {
	ctx := context.Background()
	info, err := Client.FPutObject(ctx, BucketName, objectName, localPath, minio.PutObjectOptions{})
	if err != nil {
		return err
	}
	zap.L().Info(fmt.Sprintf("✅ Uploaded %s (%d bytes)", info.Key, info.Size))
	return nil
}

func DownloadFile(objectName, localPath string) error {
	ctx := context.Background()
	return Client.FGetObject(ctx, BucketName, objectName, localPath, minio.GetObjectOptions{})
}

func ListFiles() ([]string, error) {
	ctx := context.Background()
	var result []string
	for obj := range Client.ListObjects(ctx, BucketName, minio.ListObjectsOptions{Recursive: true}) {
		if obj.Err != nil {
			return nil, obj.Err
		}
		result = append(result, obj.Key)
	}
	return result, nil
}

func DeleteFile(objectName string) error {
	ctx := context.Background()
	return Client.RemoveObject(ctx, BucketName, objectName, minio.RemoveObjectOptions{})
}

func GetObject(objectName string) (*minio.Object, error) {
	ctx := context.Background()
	return Client.GetObject(ctx, BucketName, objectName, minio.GetObjectOptions{})
}

// 生成临时下载链接
func GetPresignedURL(objectName string, expire time.Duration) (*url.URL, error) {
	ctx := context.Background()
	return Client.PresignedGetObject(ctx, BucketName, objectName, expire, nil)
}

// 从流上传（用于HTTP上传）
func UploadFromReader(file io.Reader, size int64, objectName, contentType string) error {
	ctx := context.Background()
	info, err := Client.PutObject(ctx, BucketName, objectName, file, size, minio.PutObjectOptions{
		ContentType: contentType,
	})
	if err != nil {
		return err
	}
	zap.L().Info(fmt.Sprintf("✅ Uploaded %s (%d bytes)", info.Key, info.Size))
	return nil
}