GreenTree.Nachtragsmanagement/GreenTree.Nachtragsmanagement.Web/Controllers/SiteController.cs

using GreenTree.Nachtragsmanagement.Core.Domain.Deviation;
using GreenTree.Nachtragsmanagement.Core.Domain.User;
using GreenTree.Nachtragsmanagement.Services.Appendix;
using GreenTree.Nachtragsmanagement.Services.Deviation;
using GreenTree.Nachtragsmanagement.Services.Site;
using GreenTree.Nachtragsmanagement.Services.User;
using GreenTree.Nachtragsmanagement.Web.Framework.Authorization;
using GreenTree.Nachtragsmanagement.Web.Models.Site;
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using System.Web.Mvc;

namespace GreenTree.Nachtragsmanagement.Web.Controllers
{
    public class SiteController : Controller
    {
        private readonly ISiteService _siteService;
        private readonly IDeviationService _deviationService;
        private readonly IAppendixService _appendixService;
        private readonly IUserService _userService;

        public SiteController(
            ISiteService siteService,
            IDeviationService deviationService,
            IAppendixService appendixService,
            IUserService userService)
        {
            _siteService = siteService;
            _deviationService = deviationService;
            _appendixService = appendixService;
            _userService = userService;

            ViewData["AllUsers"] =
                _userService.GetAllUsers()
                    .Select(u => new
                    {
                        Id = u.Id,
                        Description = String.Format("{0} - {1}", u.Lastname,
                            String.Join(", ", u.Roles
                                .Select(r => r.Description)))
                    })
                    .ToList();
        }

        #region Sites

        /// <summary>
        /// Basic site view function
        /// </summary>
        [FunctionAuthorize(true, "Site-Sites")]
        public ActionResult ViewSites()
        {
            var sites = _siteService.GetAllSites();
            var siteModels = sites
                .Select(u => SiteDataModel.FromSite(u, false))
                .ToList();

            return View("~/Views/Sites/View.cshtml", siteModels);
        }

        /// <summary>
        /// Get JSON data of specific site
        /// </summary>
        /// <param name="id">Site id.</param>
        public ActionResult GetSite(int id = -1)
        {
            var site = _siteService.GetSiteById(id);
            if (site == null)
                return new JsonResult
                {
                    Data = "notFound",
                    JsonRequestBehavior = JsonRequestBehavior.AllowGet
                };

            var siteModel = SiteDataModel.FromSite(site, false);
            var siteTreeModel = SiteTreeDataModel.TreeFromSite(site);

            siteModel.SiteTreeData = siteTreeModel;
            
            return new JsonResult
            {
                Data = JsonConvert.SerializeObject(siteModel),
                JsonRequestBehavior = JsonRequestBehavior.AllowGet
            };
        }

        /// <summary>
        /// Callback result for site grid
        /// </summary>
        public ActionResult PartialSites()
        {
            var sites = _siteService.GetAllSites();
            var siteModels = sites
                .Select(u => SiteDataModel.FromSite(u, false))
                .ToList();

            return PartialView("~/Views/Sites/_SiteGridPartial.cshtml", siteModels);
        }

        /// <summary>
        /// Callback result for site appendices
        /// </summary>
        /// <param name="siteId">Id of site.</param>
        public ActionResult PartialDeviationAppendices(int siteId)
        {
            var site = _siteService.GetSiteById(siteId);

            var siteModel = SiteDataModel.FromSite(site, false);
            var siteTreeModel = SiteTreeDataModel.TreeFromSite(site);

            siteModel.SiteTreeData = siteTreeModel;

            return PartialView("~/Views/Sites/_SiteEditTreePartial.cshtml", siteModel);
        }

        /// <summary>
        /// Partial edit for editing of existing or for new site
        /// </summary>
        /// <param name="id">Id for existing site, otherweise -1.</param>
        public ActionResult EditSite(int id = -1)
        {
            var site = _siteService.GetSiteById(id);
            var siteModel = SiteDataModel.FromSite(site, true);

            var siteTreeModel = SiteTreeDataModel.TreeFromSite(site);

            siteModel.SiteTreeData = siteTreeModel;

            return PartialView("~/Views/Sites/_SiteEditPartial.cshtml", siteModel);
        }

        /// <summary>
        /// Partial edit result if ModelState is valid, otherwise simple JSON result for success
        /// </summary>
        /// <param name="siteModel">Site model to be saved.</param>
        [HttpPost, ValidateInput(false)]
        public ActionResult EditSite(SiteDataModel siteModel)
        {
            if (!ModelState.IsValid)
            {
                foreach (var role in siteModel.UserValues)
                    siteModel.UserDescriptions.Add(
                        ((IList<User>)ViewData["AllUsers"])
                            .First(r => r.Id == role).Lastname);

                return PartialView("~/Views/Sites/_SiteEditPartial.cshtml", siteModel);
            }

            var selectedUsers = _userService.GetUsersByIds(siteModel.UserValues.ToArray());

            if (siteModel.Id == -1)
            {
                var site = siteModel.ToSite();

                site.SetUsers(selectedUsers);

                _siteService.InsertSite(site);
            }
            else
            {
                var site = _siteService.GetSiteById(siteModel.Id);

                site.CustomNumber = siteModel.CustomNumber;
                site.Description = siteModel.Description;
                site.Start = siteModel.Start;
                site.End = siteModel.End;
                site.Comment = siteModel.Comment;

                site.SetUsers(selectedUsers);

                _siteService.UpdateSite(site);
            }

            return new JsonResult
            {
                Data = "success"
            };
        }


        /// <summary>
        /// Partial edit result if ModelState is valid, otherwise simple JSON result for success
        /// </summary>
        /// <param name="siteModel">Site model to be saved.</param>
        [HttpPost, ValidateInput(false)]
        public ActionResult EditSiteForAppend(SiteDataModel siteModel)
        {
            if (!ModelState.IsValid)
            {
                foreach (var role in siteModel.UserValues)
                    siteModel.UserDescriptions.Add(
                        ((IList<User>)ViewData["AllUsers"])
                            .First(r => r.Id == role).Lastname);

                return PartialView("~/Views/Sites/_SiteEditPartial.cshtml", siteModel);
            }

            var selectedUsers = _userService.GetUsersByIds(siteModel.UserValues.ToArray());
            var siteId = siteModel.Id;

            if (siteModel.Id == -1)
            {
                var site = siteModel.ToSite();

                site.SetUsers(selectedUsers);

                _siteService.InsertSite(site);

                siteId = site.Id;
            }

            return EditSite(siteId);
        }

        /// <summary>
        /// Simple JSON result for deleting a specific site
        /// </summary>
        /// <param name="id">Site id.</param>
        [HttpPost]
        public ActionResult DeleteSite(int id)
        {
            var site = _siteService.GetSiteById(id);

            if (site != null)
                _siteService.DeleteSite(site);

            return new JsonResult
            {
                Data = "success"
            };
        }

        #endregion

        #region Assigning

        /// <summary>
        /// Assigns a deviation to an existing entity of type site or appendix
        /// </summary>
        /// <param name="siteId">The siteId in which the context works.</param>
        /// <param name="sourceKey">The sourceKey of the deviation.</param>
        /// <param name="targetKey">The targetKey of the site or appendix.</param>
        public ActionResult AssignDeviationToEntity(int siteId, string sourceKey, string targetKey)
        {
            if (!sourceKey.StartsWith("d"))
            {
                return new JsonResult
                {
                    Data = "invalidSourceKey"
                };
            }

            var deviationId = Convert.ToInt32(sourceKey.Replace("d_", String.Empty));
            var deviation = _deviationService.GetDeviationById(deviationId);

            if (targetKey == "a_0")
            {
                var site = _siteService.GetSiteById(siteId);

                deviation.Appendix = null;
                deviation.AppendixId = null;

                site.Deviations.Add(deviation);

                _siteService.UpdateSite(site);
                _deviationService.UpdateDeviation(deviation);
            }
            else
            {
                var appendixId = Convert.ToInt32(targetKey.Replace("a_", String.Empty));
                var appendix = _appendixService.GetAppendixById(appendixId);

                deviation.Site = null;
                deviation.SiteId = null;

                appendix.Deviations.Add(deviation);

                _appendixService.UpdateAppendix(appendix);
                _deviationService.UpdateDeviation(deviation);
            }

            return new JsonResult
            {
                Data = "success"
            };
        }

        #endregion
    }
}