✨ encapsulate object storage service operations

app/auth/api/internal/logic/oauth/wechat_offiaccount_callback_logic.go

@@ -97,11 +97,11 @@ func (l *WechatOffiaccountCallbackLogic) WechatOffiaccountCallback(r *http.Reque
 
 // SendMessage 发送消息到客户端
 func (l *WechatOffiaccountCallbackLogic) SendMessage(openId string, clientId string) error {
-	encryptClientId, err := encrypt.Encrypt(clientId, l.svcCtx.Config.Encrypt.Key, l.svcCtx.Config.Encrypt.IV)
+	encryptClientId, err := encrypt.Encrypt(clientId, l.svcCtx.Config.Encrypt.Key)
 	if err != nil {
 		return err
 	}
-	encryptOpenId, err := encrypt.Encrypt(openId, l.svcCtx.Config.Encrypt.Key, l.svcCtx.Config.Encrypt.IV)
+	encryptOpenId, err := encrypt.Encrypt(openId, l.svcCtx.Config.Encrypt.Key)
 	if err != nil {
 		return err
 	}

app/auth/api/internal/logic/storage/set_storage_config_logic.go

@@ -0,0 +1,55 @@
+package storage
+
+import (
+	"context"
+	"errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"schisandra-album-cloud-microservices/app/auth/api/internal/svc"
+	"schisandra-album-cloud-microservices/app/auth/api/internal/types"
+	"schisandra-album-cloud-microservices/app/auth/model/mysql/model"
+	"schisandra-album-cloud-microservices/common/encrypt"
+)
+
+type SetStorageConfigLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewSetStorageConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SetStorageConfigLogic {
+	return &SetStorageConfigLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *SetStorageConfigLogic) SetStorageConfig(req *types.StorageConfigRequest) (resp string, err error) {
+
+	uid, ok := l.ctx.Value("user_id").(string)
+	if !ok {
+		return "", errors.New("user_id not found")
+	}
+	accessKey, err := encrypt.Encrypt(req.AccessKey, l.svcCtx.Config.Encrypt.Key)
+	if err != nil {
+		return "", err
+	}
+	secretKey, err := encrypt.Encrypt(req.SecretKey, l.svcCtx.Config.Encrypt.Key)
+	if err != nil {
+		return "", err
+	}
+	ossConfig := &model.ScaStorageConfig{
+		UserID:    uid,
+		Type:      req.Type,
+		Endpoint:  req.Endpoint,
+		Bucket:    req.Bucket,
+		AccessKey: accessKey,
+		SecretKey: secretKey,
+		Region:    req.Region,
+	}
+	err = l.svcCtx.DB.ScaStorageConfig.Create(ossConfig)
+	if err != nil {
+		return "", err
+	}
+	return "success", nil
+}

app/auth/api/internal/logic/storage/upload_file_logic.go

@@ -0,0 +1,75 @@
+package storage
+
+import (
+	"context"
+	"errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"net/http"
+	"schisandra-album-cloud-microservices/app/auth/api/internal/svc"
+	"schisandra-album-cloud-microservices/common/encrypt"
+	"schisandra-album-cloud-microservices/common/storage/config"
+)
+
+type UploadFileLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewUploadFileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadFileLogic {
+	return &UploadFileLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *UploadFileLogic) UploadFile(r *http.Request) (resp string, err error) {
+	uid, ok := l.ctx.Value("user_id").(string)
+	if !ok {
+		return "", errors.New("user_id not found")
+	}
+	file, header, err := r.FormFile("file")
+	if err != nil {
+		return "", errors.New("file not found")
+	}
+	defer file.Close()
+	//formValue := r.PostFormValue("result")
+	//
+	//var result types.File
+	//err = json.Unmarshal([]byte(formValue), &result)
+	//if err != nil {
+	//	return "", errors.New("invalid result")
+	//}
+	//fmt.Println(result)
+	ossConfig := l.svcCtx.DB.ScaStorageConfig
+	dbConfig, err := ossConfig.Where(ossConfig.UserID.Eq(uid)).First()
+	if err != nil {
+		return "", errors.New("oss config not found")
+	}
+	accessKey, err := encrypt.Decrypt(dbConfig.AccessKey, l.svcCtx.Config.Encrypt.Key)
+	if err != nil {
+		return "", errors.New("decrypt access key failed")
+	}
+	secretKey, err := encrypt.Decrypt(dbConfig.SecretKey, l.svcCtx.Config.Encrypt.Key)
+	if err != nil {
+		return "", errors.New("decrypt secret key failed")
+	}
+	storageConfig := &config.StorageConfig{
+		Provider:   dbConfig.Type,
+		Endpoint:   dbConfig.Endpoint,
+		AccessKey:  accessKey,
+		SecretKey:  secretKey,
+		BucketName: dbConfig.Bucket,
+		Region:     dbConfig.Region,
+	}
+	service, err := l.svcCtx.StorageManager.GetStorage(uid, storageConfig)
+	if err != nil {
+		return "", errors.New("get storage failed")
+	}
+	result, err := service.UploadFileSimple(l.ctx, dbConfig.Bucket, header.Filename, file, map[string]string{})
+	if err != nil {
+		return "", errors.New("upload file failed")
+	}
+	return *result.ContentMD5, nil
+}

app/auth/api/internal/logic/user/wechat_offiaccount_login_logic.go

@@ -36,7 +36,7 @@ func NewWechatOffiaccountLoginLogic(ctx context.Context, svcCtx *svc.ServiceCont
 }
 
 func (l *WechatOffiaccountLoginLogic) WechatOffiaccountLogin(r *http.Request, req *types.WechatOffiaccountLoginRequest) (resp *types.LoginResponse, err error) {
-	decryptedClientId, err := encrypt.Decrypt(req.ClientId, l.svcCtx.Config.Encrypt.Key, l.svcCtx.Config.Encrypt.IV)
+	decryptedClientId, err := encrypt.Decrypt(req.ClientId, l.svcCtx.Config.Encrypt.Key)
 	if err != nil {
 		return nil, err
 	}
@@ -44,7 +44,7 @@ func (l *WechatOffiaccountLoginLogic) WechatOffiaccountLogin(r *http.Request, re
 	if clientId == "" {
 		return nil, errors2.New(http.StatusUnauthorized, i18n.FormatText(l.ctx, "login.loginFailed"))
 	}
-	Openid, err := encrypt.Decrypt(req.Openid, l.svcCtx.Config.Encrypt.Key, l.svcCtx.Config.Encrypt.IV)
+	Openid, err := encrypt.Decrypt(req.Openid, l.svcCtx.Config.Encrypt.Key)
 	if err != nil {
 		return nil, err
 	}