package menu

import (
	"context"

	"muyu-apiserver/gateway/internal/svc"
	"muyu-apiserver/gateway/internal/types"
	"muyu-apiserver/rpc/system/pb"

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

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

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

func (l *ListMenuLogic) ListMenu() (resp *types.ListMenuResp, err error) {
	rpcResp, err := l.svcCtx.SystemRpc.ListMenu(l.ctx, &pb.ListMenuReq{})
	if err != nil {
		return nil, err
	}

	list := make([]types.MenuInfo, 0, len(rpcResp.List))
	for _, m := range rpcResp.List {
		if m != nil {
			list = append(list, *convertMenuInfo(m))
		}
	}

	return &types.ListMenuResp{
		List: list,
	}, nil
}