feat: analysis user auth and ws

2026-10-15 09:23:07 +08:00
parent 6fa185003d
commit a65b6d2803
102 changed files with 4966 additions and 786 deletions
@@ -0,0 +1,87 @@
+package analysisstorelogic
+
+import (
+	"context"
+
+	"lone-services/pkg/utils"
+	"lone-services/pkg/validate"
+	analysis "lone-services/rpc/analysis/pb"
+	userpb "lone-services/rpc/user/pb"
+	"lone-services/services/analysis/internal/dao"
+	"lone-services/services/analysis/internal/logic/common"
+	"lone-services/services/analysis/internal/svc"
+	"lone-services/services/analysis/validator"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type StoreAuthLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewStoreAuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StoreAuthLogic {
+	return &StoreAuthLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *StoreAuthLogic) StoreAuth(in *analysis.StoreAuthReq) (*analysis.Response, error) {
+	var req validator.StoreAuthValidator
+	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
+		return common.OutResponse(utils.ErrorParams, msg), nil
+	}
+
+	storeId, _, fail := common.RequireStoreId(l.ctx)
+	if fail != nil {
+		return fail, nil
+	}
+
+	if l.svcCtx.UserSvcName == utils.StringEmpty {
+		return common.FailResponse(utils.Fail), nil
+	}
+	cli, err := svc.GetRpcClient(l.svcCtx.UserSvcName)
+	if err != nil {
+		l.Errorf("store auth get user rpc: %v", err)
+		return common.FailResponse(utils.Fail), nil
+	}
+
+	verified, err := userpb.NewUserClient(cli.Conn()).InternalVerifyPassword(l.ctx, &userpb.InternalVerifyPasswordReq{
+		Username:    req.Username,
+		Password:    req.Password,
+		SubjectType: "user",
+		ClientCode:  dao.ClientCodeUserSkin,
+	})
+	if err != nil {
+		l.Errorf("store auth verify password: %v", err)
+		if st, ok := status.FromError(err); ok {
+			switch st.Code() {
+			case codes.NotFound:
+				return common.FailResponse(utils.ErrorNotFund), nil
+			case codes.InvalidArgument:
+				return common.FailResponse(utils.ErrorPwdError), nil
+			case codes.FailedPrecondition:
+				return common.OutResponse(utils.Fail, st.Message()), nil
+			}
+		}
+		return common.FailResponse(utils.Fail), nil
+	}
+	if verified == nil || verified.UserId < utils.NumberOne {
+		return common.FailResponse(utils.ErrorNotFund), nil
+	}
+	if verified.BizId < utils.NumberOne {
+		return common.OutResponse(utils.ErrorParams, "用户未开通"), nil
+	}
+
+	if err := common.BindStoreAuth(verified.UserId, verified.BizId, storeId, req.ValidityTime); err != nil {
+		l.Errorf("store auth bind: %v", err)
+		return common.FailResponse(utils.Fail), nil
+	}
+
+	return common.OkResponse(map[string]int64{"user_id": verified.UserId}), nil
+}
@@ -0,0 +1,126 @@
+package analysisstorelogic
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+
+	"lone-services/pkg/redis"
+	"lone-services/pkg/utils"
+	"lone-services/pkg/wechat/qrcode"
+	pkgws "lone-services/pkg/ws"
+	analysis "lone-services/rpc/analysis/pb"
+	chore "lone-services/rpc/chore/pb"
+	analysisconfig "lone-services/services/analysis/internal/config"
+	"lone-services/services/analysis/internal/logic/common"
+	"lone-services/services/analysis/internal/svc"
+
+	"github.com/google/uuid"
+	jsoniter "github.com/json-iterator/go"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type StoreAuthQrcodeLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewStoreAuthQrcodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StoreAuthQrcodeLogic {
+	return &StoreAuthQrcodeLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *StoreAuthQrcodeLogic) StoreAuthQrcode(in *analysis.StoreAuthQrcodeReq) (*analysis.Response, error) {
+	storeId, _, fail := common.RequireStoreId(l.ctx)
+	if fail != nil {
+		return fail, nil
+	}
+
+	wxCfg, err := analysisconfig.LoadWechatConfig()
+	if err != nil {
+		l.Errorf("store auth qrcode wechat config: %v", err)
+		return common.OutResponse(utils.Fail, "微信配置缺失"), nil
+	}
+
+	sessionID, err := l.createSession()
+	if err != nil {
+		l.Errorf("store auth qrcode session: %v", err)
+		return common.OutResponse(utils.Fail, "create qrcode session failed"), nil
+	}
+
+	storeKey := pkgws.RedisKeyQRSession + strconv.FormatInt(storeId, utils.NumberTen)
+	if err := redis.Client.Set(l.ctx, storeKey, sessionID, pkgws.SessionTTL).Err(); err != nil {
+		l.Errorf("store auth qrcode store mapping: %v", err)
+		return common.FailResponse(utils.Fail), nil
+	}
+
+	img, err := wxCfg.NewQrcodeClient().CreateWxaCode(l.ctx, &qrcode.WxaCodeRequest{
+		Path: qrcode.BuildPath(wxCfg.AuthPage, map[string]string{
+			"store_id":   strconv.FormatInt(storeId, utils.NumberTen),
+			"session_id": sessionID,
+		}),
+		Width:      wxCfg.Width,
+		EnvVersion: wxCfg.EnvVersion,
+		AutoColor:  false,
+		LineColor:  &qrcode.LineColor{R: 255, G: 255, B: 255},
+		IsHyaline:  true,
+	})
+	if err != nil {
+		l.Errorf("store auth qrcode wechat: %v", err)
+		return common.OutResponse(utils.Fail, err.Error()), nil
+	}
+
+	qrURL, err := l.uploadQrcode(img)
+	if err != nil {
+		l.Errorf("store auth qrcode upload: %v", err)
+		return common.OutResponse(utils.Fail, "upload qrcode failed"), nil
+	}
+
+	return common.OkResponse(map[string]string{
+		"qrcode":     qrURL,
+		"session_id": sessionID,
+	}), nil
+}
+
+func (l *StoreAuthQrcodeLogic) createSession() (string, error) {
+	if redis.Client == nil {
+		return "", fmt.Errorf("redis not initialized")
+	}
+	sessionID := uuid.New().String()
+	payload, err := jsoniter.MarshalToString(pkgws.Session{
+		Status: pkgws.StatusPending,
+	})
+	if err != nil {
+		return "", err
+	}
+	key := pkgws.RedisKeyQRSession + sessionID
+	if err := redis.Client.Set(l.ctx, key, payload, pkgws.SessionTTL).Err(); err != nil {
+		return "", err
+	}
+	return sessionID, nil
+}
+
+func (l *StoreAuthQrcodeLogic) uploadQrcode(content []byte) (string, error) {
+	if l.svcCtx.ChoreSvcName == utils.StringEmpty {
+		return "", fmt.Errorf("chore service name empty")
+	}
+	cli, err := svc.GetRpcClient(l.svcCtx.ChoreSvcName)
+	if err != nil {
+		return "", err
+	}
+	res, err := chore.NewChoreClient(cli.Conn()).InternalUpload(l.ctx, &chore.InternalUploadReq{
+		Filename: uuid.New().String() + ".png",
+		Content:  content,
+	})
+	if err != nil {
+		return "", err
+	}
+	if res == nil || res.Url == utils.StringEmpty {
+		return "", fmt.Errorf("chore upload empty url")
+	}
+	return res.Url, nil
+}