package admin

import (
	"context"

	systemv1 "service.xpcool.com/api/admin/v1/system"
	"service.xpcool.com/internal/model/dto"
	"service.xpcool.com/internal/service"
)

// MenuTree returns the full menu tree (menus + button permissions).
func (c *Controller) MenuTree(ctx context.Context, req *systemv1.MenuTreeReq) (res *systemv1.MenuTreeRes, err error) {
	tree, err := service.MenuManage().Tree(ctx)
	if err != nil {
		return nil, err
	}
	out := make([]*systemv1.MenuItem, 0, len(tree))
	for _, n := range tree {
		out = append(out, menuNodeToV1(n))
	}
	return &systemv1.MenuTreeRes{Tree: out}, nil
}

// MenuCreate creates a menu or button node.
func (c *Controller) MenuCreate(ctx context.Context, req *systemv1.MenuCreateReq) (res *systemv1.MenuCreateRes, err error) {
	id, err := service.MenuManage().Create(ctx, dto.MenuCreateInput{
		ParentId: req.ParentId, Name: req.Name, Icon: req.Icon, Type: req.Type, Path: req.Path,
		Component: req.Component, Permission: req.Permission, Sort: req.Sort, Status: req.Status, Hidden: req.Hidden,
	})
	if err != nil {
		return nil, err
	}
	return &systemv1.MenuCreateRes{Id: id}, nil
}

// MenuUpdate updates a menu or button node.
func (c *Controller) MenuUpdate(ctx context.Context, req *systemv1.MenuUpdateReq) (res *systemv1.MenuUpdateRes, err error) {
	if err = service.MenuManage().Update(ctx, dto.MenuUpdateInput{
		Id: req.Id, ParentId: req.ParentId, Name: req.Name, Icon: req.Icon, Type: req.Type, Path: req.Path,
		Component: req.Component, Permission: req.Permission, Sort: req.Sort, Status: req.Status, Hidden: req.Hidden,
	}); err != nil {
		return nil, err
	}
	return &systemv1.MenuUpdateRes{}, nil
}

// MenuDelete deletes a menu node.
func (c *Controller) MenuDelete(ctx context.Context, req *systemv1.MenuDeleteReq) (res *systemv1.MenuDeleteRes, err error) {
	if err = service.MenuManage().Delete(ctx, req.Id); err != nil {
		return nil, err
	}
	return &systemv1.MenuDeleteRes{}, nil
}

func menuNodeToV1(n *dto.MenuNode) *systemv1.MenuItem {
	item := &systemv1.MenuItem{
		Id: n.Id, ParentId: n.ParentId, Name: n.Name, Icon: n.Icon, Type: n.Type, Path: n.Path,
		Component: n.Component, Permission: n.Permission, Sort: n.Sort, Status: n.Status, Hidden: n.Hidden,
	}
	for _, c := range n.Children {
		item.Children = append(item.Children, menuNodeToV1(c))
	}
	return item
}