iot/projects/IoT.Shared/Areas/IoTCenter/Controlls/SceneController.cs

using Infrastructure.Application;
using Infrastructure.Data;
using Infrastructure.Extensions;
using IoT.Shared.Application.Domain.Entities;
using IoT.Shared.Application.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Linq;

namespace IoT.Shared.Areas.IoTCenter.Controlls
{
    public class SceneController : SharedController<IoTScene, EditSceneModel>
    {
        private readonly IRepository<IoTGateway> _nodeRepo;
        private readonly AjaxBaseController _ajax;

        public SceneController(IRepository<IoTScene> repo, IRepository<IoTGateway> nodeRepo, AjaxBaseController ajax, IServiceProvider sp) : base(repo, sp)
        {
            this._nodeRepo = nodeRepo;
            this._ajax = ajax;
        }

        public override IQueryable<IoTScene> Include(IQueryable<IoTScene> query)
        {
            return query.Include(o => o.Node);
        }

        public override IQueryable<IoTScene> Query(PagedListModel<EditSceneModel> model, IQueryable<IoTScene> query)
        {
            ViewData.SelectList(o => model.Query.NodeId, () => this._ajax.GetNodeSelectList(model.Query.NodeId));
            return query.Where(o => o.NodeId != null)
                .WhereIf(model.Query.NodeId.HasValue, o => o.NodeId == model.Query.NodeId.Value)
                .WhereIf(!string.IsNullOrEmpty(model.Query.Name), o => o.Name.Contains(model.Query.Name))
                .WhereIf(model.Query.Hidden.HasValue, o => o.Hidden == model.Query.Hidden.Value)
                .OrderBy(o => o.DisplayOrder);
        }

        public override void ToDisplayModel(IoTScene entity, EditSceneModel model)
        {
            if (entity != null)
            {
                ViewData.Add(model.NodeId, entity.Node.Name);
            }
        }

        public override void ToEditModel(IoTScene entity, EditSceneModel model)
        {
            ViewData.SelectList(o => model.NodeId, () => this._ajax.GetNodeSelectList(model.NodeId));
        }

        public override string GetNodeNumber(EditSceneModel model)
        {
            return this._nodeRepo.ReadOnlyTable().Where(o => o.Id == model.NodeId).Select(o => o.Number).FirstOrDefault();
        }
    }
}