using Infrastructure.Domain;
using System.Collections.Generic;
using System.Linq;

namespace Infrastructure.Extensions
{
    public static class TreeEntityExtensions
    {
        public static List<T> ToTree<T>(this List<T> list) where T : BaseTreeEntity<T>
        {
            if (list != null)
            {
                foreach (var item in list.OrderBy(o=>o.Order))
                {
                    if (item.ParentId.HasValue)
                    {
                        item.Parent = list.FirstOrDefault(o => o.Id == item.ParentId.Value);
                        item.Parent?.Children.Add(item);
                    }
                }
            }
            return list;
        }
    }
}