EFB/Controllers/RouteController.cs

using System;
using System.Threading.Tasks;
using System.Threading;
using System.Collections.Generic;
using System.Text;
using System.Net.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Logging;
using Microsoft.AspNetCore.Http;
using Newtonsoft.Json;
using EFB.Models;
using EFB.Models.JSON;
using EFB.Sessions;
using EFB.Controllers.Form;
using EFB.Controllers.API;

namespace EFB.Controllers
{
    public class RouteController : Controller
    {
        private readonly ILogger<RouteController> _logger;

        public RouteController(ILogger<RouteController> logger)
        {
            _logger = logger;
        }

        public IActionResult Index()
        {
            //Check the user has a valid login
            UserModel user = HttpContext.Session.GetObject<UserModel>("User");
            if (user == null || user.UserToken.IsExpired())
            {
                return RedirectToAction("Index", "Home");
            }

            return View();
        }

        [ResponseCache(Duration = 0, Location = ResponseCacheLocation.None, NoStore = true)]
        public IActionResult Error()
        {
            return View("Error!");
        }

        public async Task<IActionResult> New(string departure, string arrival, string cruise)
        {
            UserModel user = HttpContext.Session.GetObject<UserModel>("User");
            if (!(user == null || user.UserToken.IsExpired()))
            {//If the user is still authenticated
                if (FormAuthenticator.ValidateICAOCode(departure) && FormAuthenticator.ValidateICAOCode(arrival))
                {//If the user has entered valid ICAOs

                    uint cruiseAlt;

                    if (uint.TryParse(cruise, out cruiseAlt) && FormAuthenticator.ValidateCruiseAlt(cruiseAlt))
                    {//If the cruise altitude if within limits.

                        //Submit route request...
                        APIInterface API = new APIInterface();

                        //Prepare data to be send off with request (route)
                        Dictionary<string, string> headerData = new Dictionary<string, string>();
                        headerData.Add("Authorization", $"Bearer {user.UserToken.TokenValue}");

                        RouteRequest routeRequest = new RouteRequest()
                        {
                            departure = departure,
                            destination = arrival,
                            preferredminlevel = cruiseAlt / 1000,
                            preferredmaxlevel = cruiseAlt / 1000,
                        };
                        StringContent content = new StringContent(JsonConvert.SerializeObject(routeRequest), Encoding.UTF8, "application/json");
                        
                        //Make initial Route Request
                        var requestRoute = API.Post<string>("https://api.autorouter.aero/v1.0/router", headerData, content);

                        ResponseModel<string> responseRoute = await requestRoute;

                        if (responseRoute.Error == null)
                        {//Update User session and add route ID
                            TokenModel routeToken = new TokenModel()
                            {
                                TokenValue = responseRoute.Result.ToString()
                            };

                            user.RouteToken = routeToken;
                            HttpContext.Session.SetObject("User", user);

                            return await Poll(departure, arrival, cruiseAlt);

                        }

                        TempData["Error"] = responseRoute.Error;
                        return RedirectToAction("Index", "Route");

                    }
                    TempData["Error"] = "Invalid Cruise Altitude";
                    TempData["Departure"] = departure;
                    TempData["Arrival"] = arrival;
                    return RedirectToAction("Index", "Route");

                }
                TempData["Error"] = "Invalid Departure or Arrival ICAO";
                return RedirectToAction("Index", "Route");
            }
            return RedirectToAction("Index", "Home");
        }


        public async Task<IActionResult> Poll(string departure, string arrival, uint cruise)
        {
            if (HttpContext.Session.GetString("User") != null)
            {//If the user is currently logged in
                UserModel user = HttpContext.Session.GetObject<UserModel>("User");

                if (user.RouteToken != null)
                {//If the user has a route object (e.g, they have been to the route page)

                    //Make calls to the server to fetch route
                    bool collected = false;
                    int pollCount = 0;
                    string routeString = "";

                    APIInterface API = new APIInterface();
                    Dictionary<string, string> headerData = new Dictionary<string, string>();

                    /*-----Chart Fetching Operations--------*/

                    //Prepare data to be send off with request(charts)
                    headerData = new Dictionary<string, string>();
                    headerData.Add("referer", "luke-else.co.uk");

                    Dictionary<string, string> formData = new Dictionary<string, string>();
                    formData.Add("token", "");
                    var body = new FormUrlEncodedContent(formData);

                    //make Charts request
                    var requestDepartureCharts = API.Post<ChartResponse>($"https://chartfox.org/api/charts/grouped/{departure}", headerData, body);
                    var requestArrivalCharts = API.Post<ChartResponse>($"https://chartfox.org/api/charts/grouped/{arrival}", headerData, body);

                    /*----------------------------------*/

                    //Run route Polling
                    headerData.Add("Authorization", $"Bearer {user.UserToken.TokenValue}");

                    while (collected == false && pollCount < 15)
                    {
                        //Make Polling Request
                        var pollingRequest = API.Put<List<PollResponse>>($"https://api.autorouter.aero/v1.0/router/{user.RouteToken.TokenValue}/longpoll", headerData, null);

                        ResponseModel<List<PollResponse>> responsePoll = await pollingRequest;

                        foreach (var item in responsePoll.Result)
                        {
                            if (item.Command == "fpl" || item.Command == "solution")
                            {
                                collected = true;
                                routeString = item.FlightPlan;
                                break;
                            }
                        }

                        Thread.Sleep(3000);
                        pollCount++;

                    }

                    if (collected)
                    {
                        //Get Response from Charts
                        ResponseModel<ChartResponse> responseDepartureCharts = await requestDepartureCharts;
                        user.DepartureCharts = new ChartModel(responseDepartureCharts.Result.Response);
                        
                        ResponseModel<ChartResponse> responseArrivalCharts = await requestArrivalCharts;
                        user.ArrivalCharts = new ChartModel(responseArrivalCharts.Result.Response);

                        //fill in route
                        string finalRoute = RouteModel.ParseRoute(routeString);
                        user.Route = finalRoute;
                        HttpContext.Session.SetObject("User", user);
                        
                        return RedirectToAction("Index", "Route");
                    }

                    TempData["Error"] = $"Unable to get route after {pollCount} Attempts!";
                    return RedirectToAction("Index", "Route");

                }
                else
                {
                    return RedirectToAction("Index", "Route");
                }

            }
            else
            {
                return RedirectToAction("Index", "Route");
            }
        }

        
    }
}