Aaru.Server/DiscImageChef.Server/Controllers/UploadStatsController.cs

// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename       : UploadStatsController.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : DiscImageChef Server.
//
// --[ Description ] ----------------------------------------------------------
//
//     Handles statistics uploads.
//
// --[ License ] --------------------------------------------------------------
//
//     This library is free software; you can redistribute it and/or modify
//     it under the terms of the GNU Lesser General Public License as
//     published by the Free Software Foundation; either version 2.1 of the
//     License, or (at your option) any later version.
//
//     This library is distributed in the hope that it will be useful, but
//     WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
//     Lesser General Public License for more details.
//
//     You should have received a copy of the GNU Lesser General Public
//     License along with this library; if not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2019 Natalia Portillo
// ****************************************************************************/

using System;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net;
using System.Threading;
using System.Threading.Tasks;
using System.Xml.Serialization;
using DiscImageChef.CommonTypes.Metadata;
using DiscImageChef.Server.Models;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Newtonsoft.Json;
using OperatingSystem = DiscImageChef.Server.Models.OperatingSystem;
using Version = DiscImageChef.Server.Models.Version;

namespace DiscImageChef.Server.Controllers
{
    public class UploadStatsController : Controller
    {
        private readonly DicServerContext _ctx;
        private IWebHostEnvironment _environment;

        public UploadStatsController(IWebHostEnvironment environment, DicServerContext ctx)
        {
            _environment = environment;
            _ctx = ctx;
        }

        /// <summary>
        ///     Receives statistics from DiscImageChef.Core, processes them and adds them to a server-side global statistics XML
        /// </summary>
        /// <returns>HTTP response</returns>
        [Route("api/uploadstats")]
        [HttpPost]
        public async Task<IActionResult> UploadStats()
        {
            var response = new ContentResult {StatusCode = (int) HttpStatusCode.OK, ContentType = "text/plain"};

            try
            {
                var newStats = new Stats();
                var request = HttpContext.Request;

                var xs = new XmlSerializer(newStats.GetType());
                newStats = (Stats) xs.Deserialize(
                    new StringReader(await new StreamReader(request.Body).ReadToEndAsync()));

                if (newStats == null)
                {
                    response.Content = "notstats";
                    return response;
                }

                StatsConverter.Convert(newStats);

                response.Content = "ok";
                return response;
            }
            catch (Exception ex)
            {
#if DEBUG
                if (Debugger.IsAttached) throw;
#endif
                response.Content = "error";
                return response;
            }
        }

        /// <summary>
        ///     Receives a report from DiscImageChef.Core, verifies it's in the correct format and stores it on the server
        /// </summary>
        /// <returns>HTTP response</returns>
        [Route("api/uploadstatsv2")]
        [HttpPost]
        public async Task<IActionResult> UploadStatsV2()
        {
            var response = new ContentResult {StatusCode = (int) HttpStatusCode.OK, ContentType = "text/plain"};

            try
            {
                var request = HttpContext.Request;

                var sr = new StreamReader(request.Body);
                var statsString = await sr.ReadToEndAsync();
                var newstats = JsonConvert.DeserializeObject<StatsDto>(statsString);

                if (newstats == null)
                {
                    response.Content = "notstats";
                    return response;
                }

                if (newstats.Commands != null)
                    foreach (var nvs in newstats.Commands)
                    {
                        var existing = _ctx.Commands.FirstOrDefault(c => c.Name == nvs.name);

                        if (existing == null) _ctx.Commands.Add(new Command {Name = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.Versions != null)
                    foreach (var nvs in newstats.Versions)
                    {
                        var existing = _ctx.Versions.FirstOrDefault(c => c.Value == nvs.name);

                        if (existing == null) _ctx.Versions.Add(new Version {Value = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.Filesystems != null)
                    foreach (var nvs in newstats.Filesystems)
                    {
                        var existing = _ctx.Filesystems.FirstOrDefault(c => c.Name == nvs.name);

                        if (existing == null) _ctx.Filesystems.Add(new Filesystem {Name = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.Partitions != null)
                    foreach (var nvs in newstats.Partitions)
                    {
                        var existing = _ctx.Partitions.FirstOrDefault(c => c.Name == nvs.name);

                        if (existing == null) _ctx.Partitions.Add(new Partition {Name = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.MediaFormats != null)
                    foreach (var nvs in newstats.MediaFormats)
                    {
                        var existing = _ctx.MediaFormats.FirstOrDefault(c => c.Name == nvs.name);

                        if (existing == null)
                            _ctx.MediaFormats.Add(new MediaFormat {Name = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.Filters != null)
                    foreach (var nvs in newstats.Filters)
                    {
                        var existing = _ctx.Filters.FirstOrDefault(c => c.Name == nvs.name);

                        if (existing == null) _ctx.Filters.Add(new Filter {Name = nvs.name, Count = nvs.Value});
                        else existing.Count += nvs.Value;
                    }

                if (newstats.OperatingSystems != null)
                    foreach (var operatingSystem in newstats.OperatingSystems)
                    {
                        var existing =
                            _ctx.OperatingSystems.FirstOrDefault(c => c.Name == operatingSystem.name &&
                                                                      c.Version == operatingSystem.version);

                        if (existing == null)
                            _ctx.OperatingSystems.Add(new OperatingSystem
                            {
                                Name = operatingSystem.name,
                                Version = operatingSystem.version,
                                Count = operatingSystem.Value
                            });
                        else existing.Count += operatingSystem.Value;
                    }

                if (newstats.Medias != null)
                    foreach (var media in newstats.Medias)
                    {
                        var existing = _ctx.Medias.FirstOrDefault(c => c.Type == media.type && c.Real == media.real);

                        if (existing == null)
                            _ctx.Medias.Add(new Media {Type = media.type, Real = media.real, Count = media.Value});
                        else existing.Count += media.Value;
                    }

                if (newstats.Devices != null)
                    foreach (var device in newstats.Devices)
                    {
                        var existing =
                            _ctx.DeviceStats.FirstOrDefault(c => c.Bus == device.Bus &&
                                                                 c.Manufacturer == device.Manufacturer &&
                                                                 c.Model == device.Model &&
                                                                 c.Revision == device.Revision);

                        if (existing == null)
                            _ctx.DeviceStats.Add(new DeviceStat
                            {
                                Bus = device.Bus,
                                Manufacturer = device.Manufacturer,
                                Model = device.Model,
                                Revision = device.Revision
                            });
                    }

                _ctx.SaveChanges();

                response.Content = "ok";
                return response;
            }
            // ReSharper disable once RedundantCatchClause
            catch
            {
#if DEBUG
                if (Debugger.IsAttached) throw;
#endif
                response.Content = "error";
                return response;
            }
        }

        private FileStream WaitForFile(string fullPath, FileMode mode, FileAccess access, FileShare share)
        {
            for (var numTries = 0; numTries < 100; numTries++)
            {
                FileStream fs = null;
                try
                {
                    fs = new FileStream(fullPath, mode, access, share);
                    return fs;
                }
                catch (IOException)
                {
                    if (fs != null) fs.Dispose();
                    Thread.Sleep(50);
                }
            }

            return null;
        }
    }
}