// Code scaffolded by goctl. Safe to edit.
package menu

import (
	"context"
	"fmt"
	"github.com/zeromicro/go-zero/core/logx"
	"sundynix-micro-go/app/system/api/internal/svc"
	sysModel "sundynix-micro-go/app/system/model"
)

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

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

func (l *GetMenuListLogic) GetMenuList() (resp interface{}, err error) {
	var menus []sysModel.SundynixMenu
	if err := l.svcCtx.DB.Order("sort").Find(&menus).Error; err != nil {
		return nil, fmt.Errorf("查询菜单列表失败")
	}
	// 构建树形结构
	menuMap := make(map[string]*sysModel.SundynixMenu)
	for i := range menus {
		menus[i].Children = []*sysModel.SundynixMenu{}
		menuMap[menus[i].ID] = &menus[i]
	}
	var tree []*sysModel.SundynixMenu
	for _, m := range menuMap {
		if m.ParentID == "0" || m.ParentID == "" {
			tree = append(tree, m)
		} else if parent, ok := menuMap[m.ParentID]; ok {
			parent.Children = append(parent.Children, m)
		} else {
			tree = append(tree, m)
		}
	}
	return tree, nil
}