Нет просмотра страниц

using System;
using System.Collections.Generic;
using System.Data.Entity;
using System.Linq;
using System.Web.Mvc;
using AVIALINIISait.Models;
using System.Web.Security;
 
namespace AVIALINIISait.Controllers
{
    public class AirlineController : Controller
    {
        private AirlineContext db = new AirlineContext();
 
        // Главная страница
        public ActionResult Index()
        {
            return View();
        }
 
        // Aircraft
        [RoleAuthorize("Admin", "Client")] // Allow both Admin and Client to view aircraft
        public ActionResult Aircrafts()
        {
            var aircrafts = db.Aircrafts.Include(a => a.Captain).ToList();
            return View(aircrafts);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")] // Only Admin can add aircraft
        public ActionResult AddAircraft()
        {
            ViewBag.CaptainID = new SelectList(db.Captains, "CaptainID", "FullName");
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddAircraft(Aircrafts aircraft)
        {
            if (ModelState.IsValid)
            {
                db.Aircrafts.Add(aircraft);
                db.SaveChanges();
                return RedirectToAction("Aircrafts");
            }
            ViewBag.CaptainID = new SelectList(db.Captains, "CaptainID", "FullName", aircraft.CaptainID);
            return View(aircraft);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")] // Only Admin can edit aircraft
        public ActionResult EditAircraft(int id)
        {
            var aircraft = db.Aircrafts.Find(id);
            if (aircraft == null)
            {
                return HttpNotFound();
            }
            ViewBag.CaptainID = new SelectList(db.Captains, "CaptainID", "FullName", aircraft.CaptainID);
            return View(aircraft);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditAircraft(Aircrafts aircraft)
        {
            if (ModelState.IsValid)
            {
                db.Entry(aircraft).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Aircrafts");
            }
            ViewBag.CaptainID = new SelectList(db.Captains, "CaptainID", "FullName", aircraft.CaptainID);
            return View(aircraft);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")] // Only Admin can delete aircraft
        public ActionResult DeleteAircraft(int id)
        {
            var aircraft = db.Aircrafts.Find(id);
            if (aircraft != null)
            {
                db.Aircrafts.Remove(aircraft);
                db.SaveChanges();
            }
            return RedirectToAction("Aircrafts");
        }
 
        // Captain
        [RoleAuthorize("Admin", "Client")]
        public ActionResult Captains()
        {
            var captains = db.Captains.ToList();
            return View(captains);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult AddCaptain()
        {
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddCaptain(Captain captain)
        {
            if (ModelState.IsValid)
            {
                db.Captains.Add(captain);
                db.SaveChanges();
                return RedirectToAction("Captains");
            }
            return View(captain);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult EditCaptain(int id)
        {
            var captain = db.Captains.Find(id);
            if (captain == null)
            {
                return HttpNotFound();
            }
            return View(captain);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditCaptain(Captain captain)
        {
            if (ModelState.IsValid)
            {
                db.Entry(captain).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Captains");
            }
            return View(captain);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult DeleteCaptain(int id)
        {
            var captain = db.Captains.Find(id);
            if (captain != null)
            {
                db.Captains.Remove(captain);
                db.SaveChanges();
            }
            return RedirectToAction("Captains");
        }
 
        // Flight
        [RoleAuthorize("Admin", "Client")]
        public ActionResult Flights()
        {
            var flights = db.Flights.Include(f => f.Route).ToList();
            return View(flights);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult AddFlight()
        {
            ViewBag.RouteID = new SelectList(db.Routes, "RouteID", "DepartureAirport");
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddFlight(Flight flight)
        {
            if (ModelState.IsValid)
            {
                db.Flights.Add(flight);
                db.SaveChanges();
                return RedirectToAction("Flights");
            }
            ViewBag.RouteID = new SelectList(db.Routes, "RouteID", "DepartureAirport", flight.RouteID);
            return View(flight);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult EditFlight(int id)
        {
            var flight = db.Flights.Find(id);
            if (flight == null)
            {
                return HttpNotFound();
            }
            ViewBag.RouteID = new SelectList(db.Routes, "RouteID", "DepartureAirport", flight.RouteID);
            return View(flight);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditFlight(Flight flight)
        {
            if (ModelState.IsValid)
            {
                db.Entry(flight).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Flights");
            }
            ViewBag.RouteID = new SelectList(db.Routes, "RouteID", "DepartureAirport", flight.RouteID);
            return View(flight);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult DeleteFlight(int id)
        {
            var flight = db.Flights.Find(id);
            if (flight != null)
            {
                db.Flights.Remove(flight);
                db.SaveChanges();
            }
            return RedirectToAction("Flights");
        }
 
        // Passenger
        [RoleAuthorize("Admin", "Client")]
        public ActionResult Passengers()
        {
            var passengers = db.Passengers.ToList();
            return View(passengers);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult AddPassenger()
        {
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddPassenger(Passenger passenger)
        {
            if (ModelState.IsValid)
            {
                db.Passengers.Add(passenger);
                db.SaveChanges();
                return RedirectToAction("Passengers");
            }
            return View(passenger);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult EditPassenger(string id)
        {
            var passenger = db.Passengers.Find(id);
            if (passenger == null)
            {
                return HttpNotFound();
            }
            return View(passenger);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditPassenger(Passenger passenger)
        {
            if (ModelState.IsValid)
            {
                db.Entry(passenger).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Passengers");
            }
            return View(passenger);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult DeletePassenger(string id)
        {
            var passenger = db.Passengers.Find(id);
            if (passenger != null)
            {
                db.Passengers.Remove(passenger);
                db.SaveChanges();
            }
            return RedirectToAction("Passengers");
        }
 
        // Route
        [RoleAuthorize("Admin", "Client")]
        public ActionResult Routes()
        {
            var routes = db.Routes.ToList();
            return View(routes);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult AddRoute()
        {
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddRoute(Route route)
        {
            if (ModelState.IsValid)
            {
                db.Routes.Add(route);
                db.SaveChanges();
                return RedirectToAction("Routes");
            }
            return View(route);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult EditRoute(int id)
        {
            var route = db.Routes.Find(id);
            if (route == null)
            {
                return HttpNotFound();
            }
            return View(route);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditRoute(Route route)
        {
            if (ModelState.IsValid)
            {
                db.Entry(route).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Routes");
            }
            return View(route);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult DeleteRoute(int id)
        {
            var route = db.Routes.Find(id);
            if (route != null)
            {
                db.Routes.Remove(route);
                db.SaveChanges();
            }
            return RedirectToAction("Routes");
        }
 
        // Ticket
        [RoleAuthorize("Admin", "Client")]
        public ActionResult Tickets()
        {
            var tickets = db.Tickets.Include(t => t.Flight).Include(t => t.Passenger).ToList();
            return View(tickets);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult AddTicket()
        {
            ViewBag.FlightID = new SelectList(db.Flights, "FlightID", "Route.DepartureAirport");
            return View();
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult AddTicket(Ticket ticket)
        {
            if (ModelState.IsValid)
            {
                db.Tickets.Add(ticket);
                db.SaveChanges();
                return RedirectToAction("Tickets");
            }
            ViewBag.FlightID = new SelectList(db.Flights, "FlightID", "Route.DepartureAirport", ticket.FlightID);
            return View(ticket);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult EditTicket(int id)
        {
            var ticket = db.Tickets.Find(id);
            if (ticket == null)
            {
                return HttpNotFound();
            }
            ViewBag.FlightID = new SelectList(db.Flights, "FlightID", "Route.DepartureAirport", ticket.FlightID);
            return View(ticket);
        }
 
        [HttpPost]
        [RoleAuthorize("Admin")]
        public ActionResult EditTicket(Ticket ticket)
        {
            if (ModelState.IsValid)
            {
                db.Entry(ticket).State = EntityState.Modified;
                db.SaveChanges();
                return RedirectToAction("Tickets");
            }
            ViewBag.FlightID = new SelectList(db.Flights, "FlightID", "Route.DepartureAirport", ticket.FlightID);
            return View(ticket);
        }
 
        [HttpGet]
        [RoleAuthorize("Admin")]
        public ActionResult DeleteTicket(int id)
        {
            var ticket = db.Tickets.Find(id);
            if (ticket != null)
            {
                db.Tickets.Remove(ticket);
                db.SaveChanges();
            }
            return RedirectToAction("Tickets");
        }
 
        // GET: Account/Login
        [HttpGet]
        public ActionResult Login()
        {
            return View();
        }
 
        // POST: Account/Login
        [HttpPost]
        [ValidateAntiForgeryToken]
        public ActionResult Login(string email, string password)
        {
            if (ModelState.IsValid)
            {
                var user = db.Users.FirstOrDefault(u => u.Email == email && u.Password == password);
                if (user != null)
                {
                    FormsAuthentication.SetAuthCookie(user.Email, false);
//                  Session["User Role"] = user.Role; // Сохраните роль пользователя в сессии
                    return RedirectToAction("Index", "Airline");
                }
                ModelState.AddModelError("", "Неверный логин или пароль.");
            }
            return View();
        }
 
 
        // GET: Account/Register
        [HttpGet]
        public ActionResult Register()
        {
            return View();
        }
 
        // POST: Account/Register
        [HttpPost]
        [ValidateAntiForgeryToken]
        public ActionResult Register(User user)
        {
            if (ModelState.IsValid)
            {
                user.Role = "Client"; // Роль по умолчанию - клиент
                db.Users.Add(user);
                db.SaveChanges();
                return RedirectToAction("Index", "Airline");
            }
            return View(user);
        }
 
        // Метод для выхода
        public ActionResult Logout()
        {
            FormsAuthentication.SignOut();
            return RedirectToAction("Index", "Airline");
        }
 
        // Освобождаем ресурсы
        protected override void Dispose(bool disposing)
        {
            if (disposing)
            {
                db.Dispose();
            }
            base.Dispose(disposing);
        }
    }
}