init: 1.0.0

internal/logic/public/user/queryUserAffiliateListLogic.go

@@ -0,0 +1,84 @@
+package user
+
+import (
+	"context"
+
+	"github.com/perfect-panel/ppanel-server/pkg/constant"
+
+	"github.com/perfect-panel/ppanel-server/internal/model/user"
+	"github.com/perfect-panel/ppanel-server/pkg/xerr"
+	"github.com/pkg/errors"
+	"gorm.io/gorm"
+
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/internal/types"
+	"github.com/perfect-panel/ppanel-server/pkg/logger"
+)
+
+type QueryUserAffiliateListLogic struct {
+	logger.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+// Query User Affiliate List
+func NewQueryUserAffiliateListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryUserAffiliateListLogic {
+	return &QueryUserAffiliateListLogic{
+		Logger: logger.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *QueryUserAffiliateListLogic) QueryUserAffiliateList(req *types.QueryUserAffiliateListRequest) (resp *types.QueryUserAffiliateListResponse, err error) {
+	u, ok := l.ctx.Value(constant.CtxKeyUser).(*user.User)
+	if !ok {
+		logger.Error("current user is not found in context")
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidAccess), "Invalid Access")
+	}
+	var data []*user.User
+	var total int64
+	err = l.svcCtx.UserModel.Transaction(l.ctx, func(db *gorm.DB) error {
+		return db.Model(&user.User{}).Order("id desc").Where("referer_id = ?", u.Id).Count(&total).Limit(req.Size).Offset((req.Page - 1) * req.Size).Find(&data).Error
+	})
+	if err != nil {
+		l.Errorw("Query User Affiliate List failed: %v", logger.Field("error", err.Error()))
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "Query User Affiliate List failed: %v", err.Error())
+	}
+
+	list := make([]types.UserAffiliate, 0)
+	for _, item := range data {
+		list = append(list, types.UserAffiliate{
+			Identifier:   GetAuthMethod(l, item).AuthIdentifier,
+			Avatar:       item.Avatar,
+			RegisteredAt: item.CreatedAt.UnixMilli(),
+			Enable:       *item.Enable,
+		})
+	}
+	return &types.QueryUserAffiliateListResponse{
+		Total: total,
+		List:  list,
+	}, nil
+}
+
+func GetAuthMethod(l *QueryUserAffiliateListLogic, item *user.User) user.AuthMethods {
+	authMethod := user.AuthMethods{}
+	authMethods, errs := l.svcCtx.UserModel.FindUserAuthMethods(l.ctx, item.Id)
+	if errs == nil && len(authMethods) > 0 {
+		for _, am := range authMethods {
+			if am.AuthType == "6" || am.AuthType == "7" {
+				authMethod = *am
+				break
+			}
+		}
+		if authMethod.AuthIdentifier == "" {
+			authMethod = *authMethods[0]
+		}
+
+		hideTextLength := len(authMethod.AuthIdentifier) / 3
+		if hideTextLength > 0 {
+			authMethod.AuthIdentifier = authMethod.AuthIdentifier[0:hideTextLength] + "***" + authMethod.AuthIdentifier[hideTextLength*2:]
+		}
+	}
+	return authMethod
+}