everything/EveryThing/Pages/TransportLoadingOrder/Edit.cshtml.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Identity;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.RazorPages;
using Microsoft.AspNetCore.Mvc.Rendering;
using Microsoft.EntityFrameworkCore;
using EveryThing.Data;
using EveryThing.Models;
using EveryThing.Models.Transport;

namespace EveryThing.Pages.TransportLoadingOrder
{
    [Authorize(Roles = "Administrator,TransportThingUser")]
    public class EditModel : PageModel
    {
        private readonly ApplicationDbContext _context;
        private readonly UserManager<IdentityApplicationUser> _userManager;

        public EditModel(ApplicationDbContext context, UserManager<IdentityApplicationUser> userManager)
        {
            _context = context;
            _userManager = userManager;
        }

        [BindProperty]
        public Models.Transport.TransportLoadingOrder TransportLoadingOrder { get; set; }

        [BindProperty]
        public TransportLoadingOrderLoadUnload TransportLoadingOrderLoadUnload { get; set; }

        public IList<TransportLoadingOrderLoadUnload> TransportLoadingOrderLoadUnloadList { get; set; }

        public async Task<IActionResult> OnGetAsync(int? id)
        {
            var user = _userManager.GetUserAsync(User).Result;

            if (id == null)
            {
                return NotFound();
            }

            TransportLoadingOrder = await _context.TransportLoadingOrders
                .Include(t => t.Partner)
                .Include(t => t.Vehicle).FirstOrDefaultAsync(m => m.IdTransportLoadingOrder == id && m.IdCompanyFk == user.IdCompanyFk);

            TransportLoadingOrderLoadUnload = new TransportLoadingOrderLoadUnload();

            if (TransportLoadingOrder == null)
            {
                return NotFound();
            }

            ViewData["IdPartnerFk"] = new SelectList(_context.CodeTablePartners.Where(x => x.IdCompanyFk == user.IdCompanyFk), "IdPartner", "Title");
            ViewData["IdVehicleFk"] = new SelectList(_context.Vehicles.Where(x => x.IdCompanyFk == user.IdCompanyFk), "IdVehicle", "RegistrationNumber");
            ViewData["IdLoadingFk"] = new SelectList(_context.CodeTablePartners.Where(x => x.IdCompanyFk == user.IdCompanyFk), "IdPartner", "Title");
            ViewData["IdUnloadingFk"] = new SelectList(_context.CodeTablePartners.Where(x => x.IdCompanyFk == user.IdCompanyFk), "IdPartner", "Title");

            return Page();
        }

        public async Task<IActionResult> OnPostOrderAsync()
        {
            if (!ModelState.IsValid)
            {
                return Page();
            }

            TransportLoadingOrder.PostText = TransportLoadingOrder.PostText.Replace("<div><br></div>", "");
            TransportLoadingOrder.PreText = TransportLoadingOrder.PreText.Replace("<div><br></div>", "");

            _context.Attach(TransportLoadingOrder).State = EntityState.Modified;

            try
            {
                await _context.SaveChangesAsync();
            }
            catch (DbUpdateConcurrencyException)
            {
                if (!TransportLoadingOrderExists(TransportLoadingOrder.IdTransportLoadingOrder))
                {
                    return NotFound();
                }
                else
                {
                    throw;
                }
            }

            TransportLoadingOrderLoadUnload = new TransportLoadingOrderLoadUnload();

            return Page();
        }

        public async Task<IActionResult> OnPostPositionAddAsync()
        {
            TransportLoadingOrderLoadUnload.IdTransportLoadingOrderFk = TransportLoadingOrder.IdTransportLoadingOrder;
            TransportLoadingOrderLoadUnload.CargoDescription = TransportLoadingOrderLoadUnload.CargoDescription.Replace("<div><br></div>", "");

            _context.TransportLoadingOrderLoadUnloads.Add(TransportLoadingOrderLoadUnload);
            await _context.SaveChangesAsync();

            return Page();
        }

        public async Task<IActionResult> OnGetPositionRemoveAsync(int idTransportLoadingOrder, int id)
        {
            var user = _userManager.GetUserAsync(User).Result;

            TransportLoadingOrder = await _context.TransportLoadingOrders
                .Include(t => t.Partner)
                .Include(t => t.Vehicle).FirstOrDefaultAsync(m => m.IdTransportLoadingOrder == idTransportLoadingOrder && m.IdCompanyFk == user.IdCompanyFk);

            TransportLoadingOrderLoadUnload = new TransportLoadingOrderLoadUnload();

            if (TransportLoadingOrder != null)
            {
                var transportLoadingOrderLoadUnload = await _context.TransportLoadingOrderLoadUnloads.FirstOrDefaultAsync(x => x.IdTransportLoadingOrderFk == TransportLoadingOrder.IdTransportLoadingOrder && x.IdTransportLoadingOrderLoadUnload == id);

                if (transportLoadingOrderLoadUnload != null)
                {
                    _context.TransportLoadingOrderLoadUnloads.Remove(transportLoadingOrderLoadUnload);

                    await _context.SaveChangesAsync();
                }

                return Page();
            }

            return NotFound();
        }

        public async Task<IActionResult> OnGetIndexLoadUnloadAsync(int id)
        {
            var user = _userManager.GetUserAsync(User).Result;

            var transportLoadingOrder = await _context.TransportLoadingOrders.FirstOrDefaultAsync(x => x.IdTransportLoadingOrder == id && x.IdCompanyFk == user.IdCompanyFk);

            if (transportLoadingOrder != null)
            {
                var transportLoadingOrderLoadUnloadList = await _context.TransportLoadingOrderLoadUnloads
                    .Where(x => x.IdTransportLoadingOrderFk == transportLoadingOrder.IdTransportLoadingOrder)
                    .Include(x => x.Loading)
                    .Include(x => x.Unloading).ToListAsync();

                return Partial("IndexLoadUnload", transportLoadingOrderLoadUnloadList);
            }

            return NotFound();
        }

        private bool TransportLoadingOrderExists(int id)
        {
            return _context.TransportLoadingOrders.Any(e => e.IdTransportLoadingOrder == id);
        }
    }
}