lone-services/services/analysis/internal/logic/analysisuser/userStatusLogic.go

package analysisuserlogic

import (
	"context"

	"lone-services/pkg/utils"
	analysis "lone-services/rpc/analysis/pb"
	"lone-services/services/analysis/internal/logic/common"
	"lone-services/services/analysis/internal/svc"

	"github.com/zeromicro/go-zero/core/logx"
)

type UserStatusLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUserStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserStatusLogic {
	return &UserStatusLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *UserStatusLogic) UserStatus(in *analysis.UserStatusReq) (*analysis.Response, error) {
	bizId, _, fail := common.RequireAnalysisUserId(l.ctx)
	if fail != nil {
		return fail, nil
	}

	row, err := common.LoadAnalysisUser(bizId)
	if err != nil {
		l.Errorf("user status load: %v", err)
		return common.FailResponse(utils.Fail), nil
	}
	if row == nil {
		return common.FailResponse(utils.ErrorNotFund), nil
	}

	status := utils.NumberOne
	if row.IsRegistered == uint8(utils.StatusOk) {
		status = utils.NumberTwo
	}
	if row.IsFaceVerified == uint8(utils.StatusOk) {
		status = utils.NumberThree
	}
	if row.IsProfileCompleted == uint8(utils.StatusOk) {
		status = utils.NumberFour
	}

	return common.OkResponse(map[string]int{"status": status}), nil
}