Remotely/Server/API/AgentUpdateController.cs

using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Caching.Memory;
using Remotely.Server.Services;
using System;
using System.IO;
using System.Net;
using System.Threading.Tasks;

namespace Remotely.Server.API
{
    [Route("api/[controller]")]
    [ApiController]
    public class AgentUpdateController : ControllerBase
    {
        private static readonly MemoryCache downloadingAgents = new MemoryCache(new MemoryCacheOptions());


        public AgentUpdateController(IWebHostEnvironment hostingEnv, 
            DataService dataService,
            ApplicationConfig appConfig)
        {
            HostEnv = hostingEnv;
            DataService = dataService;
            AppConfig = appConfig;
        }

        private DataService DataService { get; }
        public ApplicationConfig AppConfig { get; }
        private IWebHostEnvironment HostEnv { get; }


        [HttpGet("[action]/{downloadId}")]
        public ActionResult ClearDownload(string downloadId)
        {
            downloadingAgents.Remove(downloadId);
            return Ok();
        }

        // TODO: Remove in the next few updates.
        [HttpGet("[action]")]
        public string CurrentVersion()
        {
            var filePath = Path.Combine(HostEnv.ContentRootPath, "CurrentVersion.txt");
            if (!System.IO.File.Exists(filePath))
            {
                return "0.0.0.0";
            }
            return System.IO.File.ReadAllText(filePath).Trim();
        }

        [HttpGet("[action]/{platform}/{downloadId}")]
        public async Task<ActionResult> DownloadPackage(string platform, string downloadId)
        {
            try
            {
                var startWait = DateTimeOffset.Now;
                while (downloadingAgents.Count > AppConfig.MaxConcurrentUpdates)
                {
                    await Task.Delay(new Random().Next(100, 10000));
                }
                var waitTime = DateTimeOffset.Now - startWait;
                DataService.WriteEvent($"Download started after wait time of {waitTime}.", Shared.Models.EventType.Debug, string.Empty);

                downloadingAgents.Set(downloadId, string.Empty, TimeSpan.FromMinutes(10));

                byte[] fileBytes;
                string filePath;

                switch (platform.ToLower())
                {
                    case "win-x64":
                        filePath = Path.Combine(HostEnv.WebRootPath, "Downloads", "Remotely-Win10-x64.zip");
                        fileBytes = await System.IO.File.ReadAllBytesAsync(filePath);
                        break;
                    case "win-x86":
                        filePath = Path.Combine(HostEnv.WebRootPath, "Downloads", "Remotely-Win10-x86.zip");
                        fileBytes = await System.IO.File.ReadAllBytesAsync(filePath);
                        break;
                    case "linux":
                        filePath = Path.Combine(HostEnv.WebRootPath, "Downloads", "Remotely-Linux.zip");
                        fileBytes = await System.IO.File.ReadAllBytesAsync(filePath);
                        break;
                    default:
                        return BadRequest();
                }

                return File(fileBytes, "application/octet-stream", "RemotelyUpdate.zip");
            }
            catch (Exception ex)
            {
                DataService.WriteEvent(ex, null);
                return StatusCode((int)HttpStatusCode.InternalServerError);
            }
        }

        // TODO: Remove in the next few updates.
        [HttpGet("[action]")]
        public int UpdateWindow()
        {
            return DataService.GetDeviceCount() * 10;
        }
    }
}