feat: minilogin改造

2026-05-24 23:04:09 +08:00
parent 076ed1509b
commit da02247794
36 changed files with 3523 additions and 11653 deletions
@@ -2,10 +2,13 @@ package complete

 import (
 	"context"
+	"time"
+
+	filePb "sundynix-micro-go/app/file/rpc/file"
+	"sundynix-micro-go/app/plant/api/internal/svc"
+	plantModel "sundynix-micro-go/app/plant/model"

 	"github.com/zeromicro/go-zero/core/logx"
-	"sundynix-micro-go/app/plant/api/internal/svc"
-	plantPb "sundynix-micro-go/app/plant/rpc/plant"
 )

 type GetBadgeConfigTreeLogic struct {
@@ -18,6 +21,142 @@ func NewGetBadgeConfigTreeLogic(ctx context.Context, svcCtx *svc.ServiceContext)
 	return &GetBadgeConfigTreeLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
 }

-func (l *GetBadgeConfigTreeLogic) GetBadgeConfigTree() (*plantPb.BadgeConfigTreeResp, error) {
-	return l.svcCtx.PlantRpc.GetBadgeConfigTree(l.ctx, &plantPb.IdReq{})
+func (l *GetBadgeConfigTreeLogic) GetBadgeConfigTree() (interface{}, error) {
+	var allBadges []plantModel.BadgeConfig
+	err := l.svcCtx.DB.Order("dimension desc, group_id asc, tier asc, sort asc").Find(&allBadges).Error
+	if err != nil {
+		return nil, err
+	}
+
+	iconIds := make([]string, 0)
+	iconIdMap := make(map[string]bool)
+	for _, b := range allBadges {
+		if b.IconID != "" && !iconIdMap[b.IconID] {
+			iconIdMap[b.IconID] = true
+			iconIds = append(iconIds, b.IconID)
+		}
+	}
+
+	fileMap := make(map[string]map[string]interface{})
+	if len(iconIds) > 0 {
+		resp, err := l.svcCtx.FileRpc.GetFilesByIds(l.ctx, &filePb.GetFilesByIdsReq{Ids: iconIds})
+		if err == nil && resp != nil {
+			for _, f := range resp.Files {
+				fileMap[f.Id] = map[string]interface{}{
+					"id":        f.Id,
+					"name":      f.Name,
+					"url":       f.Url,
+					"tag":       f.Tag,
+					"key":       f.Key,
+					"suffix":    f.Suffix,
+					"md5":       f.Md5,
+					"createdAt": time.Unix(f.CreatedAt, 0).Format(time.RFC3339),
+				}
+			}
+		}
+	}
+
+	dimLabelMap := map[string]string{
+		"PERSISTENCE": "勤勉成就",
+		"EXPERTISE":   "专家成就",
+		"JOURNAL":     "岁月记录",
+		"DISCOVERY":   "探索发现",
+	}
+
+	groupLabelMap := map[string]string{
+		"water_master":  "雨露均沾",
+		"alive_master":  "长情陪伴",
+		"fert_master":   "炼金术士",
+		"prune_master":  "园艺理发师",
+		"repot_master":  "乔迁之喜",
+		"doctor_master": "植物医生",
+		"photo_master":  "光影捕手",
+		"night_owl":     "守夜人",
+	}
+
+	dimOrder := []string{"PERSISTENCE", "EXPERTISE", "JOURNAL", "DISCOVERY"}
+
+	// Build dimension -> groupId -> badge list
+	tempMap := make(map[string]map[string][]map[string]interface{})
+	for _, b := range allBadges {
+		dim := b.Dimension
+		group := b.GroupID
+
+		badgeData := map[string]interface{}{
+			"id":             b.ID,
+			"name":           b.Name,
+			"description":    b.Description,
+			"iconId":         b.IconID,
+			"dimension":      b.Dimension,
+			"groupId":        b.GroupID,
+			"tier":           b.Tier,
+			"targetAction":   b.TargetAction,
+			"threshold":      b.Threshold,
+			"comparator":     b.Comparator,
+			"rewardSunlight": b.RewardSunlight,
+			"sort":           b.Sort,
+			"createdAt":      b.CreatedAt.Format("2006-01-02 15:04:05"),
+		}
+		if iconData, ok := fileMap[b.IconID]; ok {
+			badgeData["icon"] = iconData
+		} else {
+			badgeData["icon"] = nil
+		}
+
+		if tempMap[dim] == nil {
+			tempMap[dim] = make(map[string][]map[string]interface{})
+		}
+		tempMap[dim][group] = append(tempMap[dim][group], badgeData)
+	}
+
+	var tree []map[string]interface{}
+	for _, dimKey := range dimOrder {
+		if groupMap, exists := tempMap[dimKey]; exists {
+			var groupNodes []map[string]interface{}
+			for groupKey, badges := range groupMap {
+				gLabel := groupLabelMap[groupKey]
+				if gLabel == "" {
+					gLabel = groupKey
+				}
+				groupNodes = append(groupNodes, map[string]interface{}{
+					"groupId":    groupKey,
+					"groupLabel": gLabel,
+					"badges":     badges,
+				})
+			}
+			dLabel := dimLabelMap[dimKey]
+			if dLabel == "" {
+				dLabel = dimKey
+			}
+			tree = append(tree, map[string]interface{}{
+				"dimension": dimKey,
+				"label":     dLabel,
+				"groups":    groupNodes,
+			})
+			delete(tempMap, dimKey)
+		}
+	}
+
+	// Handle leftover dimensions
+	for dimKey, groupMap := range tempMap {
+		var groupNodes []map[string]interface{}
+		for groupKey, badges := range groupMap {
+			gLabel := groupLabelMap[groupKey]
+			if gLabel == "" {
+				gLabel = groupKey
+			}
+			groupNodes = append(groupNodes, map[string]interface{}{
+				"groupId":    groupKey,
+				"groupLabel": gLabel,
+				"badges":     badges,
+			})
+		}
+		tree = append(tree, map[string]interface{}{
+			"dimension": dimKey,
+			"label":     dimKey,
+			"groups":    groupNodes,
+		})
+	}
+
+	return tree, nil
 }