common/minio/minio.go

package minio

import (
	"context"
	"fmt"
	"gitea.com/red-future/common/utils"
	"github.com/gogf/gf/v2/os/glog"
	"github.com/gogf/gf/v2/util/gconv"
	"net/http"
	"path/filepath"
	"time"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/google/uuid"
	"github.com/minio/minio-go/v7"
	"github.com/minio/minio-go/v7/pkg/credentials"
)

// IoConfig 映射 YAML 中的 minio 配置节点
type IoConfig struct {
	FilePrefix string `yaml:"filePrefix"` // 文件前缀
	Endpoint   string `yaml:"endpoint"`   // MinIO API 地址
	AccessKey  string `yaml:"accessKey"`  // AK
	SecretKey  string `yaml:"secretKey"`  // SK
	Secure     bool   `yaml:"secure"`     // 是否启用 SSL
	Region     string `yaml:"region"`     // 区域
}

// 全局 MinIO 客户端（初始化一次，避免重复创建）
var minioClient *minio.Client
var minioCfg IoConfig

// initMinIO 初始化 MinIO 客户端。
func init() {
	ctx := context.Background()
	if !g.Cfg().MustGet(ctx, "minio").IsEmpty() {
		// 加载 MinIO 配置（可从配置文件/环境变量读取，这里硬编码示例）
		minioCfg = IoConfig{
			FilePrefix: g.Cfg().MustGet(ctx, "filePrefix").String(),
			Endpoint:   g.Cfg().MustGet(ctx, "minio.endpoint").String(),
			AccessKey:  g.Cfg().MustGet(ctx, "minio.accessKey").String(),
			SecretKey:  g.Cfg().MustGet(ctx, "minio.secretKey").String(),
			Secure:     g.Cfg().MustGet(ctx, "minio.secure").Bool(),
			Region:     g.Cfg().MustGet(ctx, "minio.region").String(),
		}
		// 创建 MinIO 客户端
		var err error
		if minioClient, err = minio.New(minioCfg.Endpoint, &minio.Options{
			Creds:  credentials.NewStaticV4(minioCfg.AccessKey, minioCfg.SecretKey, ""),
			Secure: minioCfg.Secure,
			Region: minioCfg.Region,
		}); err != nil {
			glog.Errorf(ctx, "初始化 MinIO 客户端失败: %v", err)
		}
	}
}

func UploadFile(ctx context.Context, fileHeader *ghttp.UploadFile) (imagesUrl string, err error) {
	return uploadFile(ctx, getBucketName(ctx), fileHeader)
}

func uploadFile(ctx context.Context, bucketName string, fileHeader *ghttp.UploadFile) (imagesUrl string, err error) {
	// 检查/创建桶
	exists, err := minioClient.BucketExists(ctx, bucketName)
	if err != nil {
		glog.Errorf(ctx, "检查桶是否存在失败: %v", err)
		return
	}
	if !exists {
		if err = minioClient.MakeBucket(ctx, bucketName, minio.MakeBucketOptions{Region: minioCfg.Region}); err != nil {
			glog.Errorf(ctx, "创建桶失败: %v", err)
			return
		}
		glog.Infof(ctx, "成功创建 MinIO 桶: %s", bucketName)
	}
	// 打开文件，获取 io.Reader（*os.File 实现了 io.Reader）
	file, err := fileHeader.Open()
	if err != nil {
		glog.Errorf(ctx, "打开文件失败: %v", err)
		return
	}
	defer file.Close() // 必须关闭，避免文件句柄泄露
	// 获取文件类型
	buffer := make([]byte, 512)
	_, err = file.Read(buffer)
	if err != nil {
		glog.Errorf(ctx, "读取文件头失败: %v", err)
		return
	}
	contentType := http.DetectContentType(buffer)
	// 重置文件读取位置，否则后续 PutObject 会从第512字节开始上传
	if _, err = file.Seek(0, 0); err != nil {
		glog.Errorf(ctx, "重置文件读取位置失败: %v", err)
		return
	}
	// 生成唯一的 MinIO 对象名（避免覆盖）
	fileExt := filepath.Ext(fileHeader.Filename)                        // 原文件后缀（如 .jpg）
	uniqueID := uuid.New().String()[:32]                                // 32位随机UUID
	timestamp := time.Now().Format("2006-01-02")                        // 日期目录（便于管理）
	objectName := fmt.Sprintf("/%s/%s%s", timestamp, uniqueID, fileExt) // 存储路径：20251209/abc12345.jpg
	// 设置存储桶公共读权限
	policy := `{"Version":"2012-10-17","Statement":[{"Effect":"Allow","Principal":{"AWS":["*"]},"Action":["s3:GetObject"],"Resource":["arn:aws:s3:::` + bucketName + `/*"]}]}`
	if err = minioClient.SetBucketPolicy(ctx, bucketName, policy); err != nil {
		glog.Errorf(ctx, "设置存储桶权限失败: %v", err)
		return
	}
	// 执行图片上传
	_, err = minioClient.PutObject(
		ctx,
		bucketName,
		objectName,
		file,
		fileHeader.Size,
		minio.PutObjectOptions{
			ContentType: contentType, // 关键：指定图片MIME类型，S3会根据此类型处理
			// 若需要图片可公开访问，添加如下配置（根据需求选择）
			//ACL: minio.ACLPublicRead,
		},
	)
	if err != nil {
		glog.Errorf(ctx, "上传图片失败: %v", err)
		return
	}
	return objectName, err
}

// GetFileAddressPrefix 拼接图片前缀地址
func GetFileAddressPrefix(ctx context.Context) (imageUrl string) {
	// 拼接图片前缀地址
	var url = "http://"
	if minioCfg.Secure {
		url = "https://"
	}
	imgAddressPrefix := url + minioCfg.FilePrefix + "/" + getBucketName(ctx)
	return imgAddressPrefix
}

func getBucketName(ctx context.Context) (bucketName string) {
	user, err := utils.GetUserInfo(ctx)
	if err != nil {
		glog.Errorf(ctx, "获取用户信息失败: %v", err)
		return
	}
	return "tenantid-" + gconv.String(user.TenantId)
}