using System.Linq;
using Infrastructure.Data;
using Infrastructure.Extensions;
using Infrastructure.Web.Mvc;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.EntityFrameworkCore;
using Application.Domain.Entities;
using Application.Models;
using Infrastructure.Application;

namespace IoTNode.Areas.Admin.Controllers
{
    [Authorize]
    [Area(nameof(Admin))]
    public class RoleController : CrudController<Role, PagedListModel<EditRoleModel>, EditRoleModel, EditRoleModel>
    {
        private readonly IRepository<Role> _roleRepo;
        private readonly AjaxController _ajax;

        public RoleController(IRepository<Role> roleRepo, AjaxController ajax) : base(roleRepo)
        {
            this._roleRepo = roleRepo;
            this._ajax = ajax;
        }

        public override IQueryable<Role> Include(IQueryable<Role> query)
        {
            return query.Include(o => o.RolePermissions).ThenInclude(o => o.Permission).ThenInclude(o => o.Category).ThenInclude(o => o.Parent);
        }

        public override void ToModel(Role entity, EditRoleModel model)
        {
            if (entity != null)
            {
                model.Permissions = entity.RolePermissions.Select(o => o.PermissionId).ToList();
            }
            this.ViewData.MultiSelectList(o => model.Permissions, () => this._ajax.GetPermissionMultiSelectList(model.Permissions));
        }

        public override void ToEntity(EditRoleModel model, Role entity)
        {
            foreach (var id in entity.RolePermissions.Select(o => o.PermissionId).ToList())
            {
                if (!model.Permissions.Any(o => o == id))
                {
                    entity.RolePermissions.RemoveAll(o => o.PermissionId == id);
                }
            }
            foreach (var id in model.Permissions)
            {
                if (!entity.RolePermissions.Any(o => o.PermissionId == id))
                {
                    entity.RolePermissions.Add(new RolePermission { PermissionId = id });
                }
            }
        }
    }
}