wabbajack/Wabbajack.Server/Controllers/ListDefinitions.cs

using System;
using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Logging;
using Wabbajack.Common;
using Wabbajack.DTOs;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.Paths.IO;
using Wabbajack.Server.DataLayer;
using Wabbajack.Server.Services;

namespace Wabbajack.BuildServer.Controllers;

[ApiController]
[Route("/list_definitions")]
public class ListDefinitions : ControllerBase
{
    private readonly DTOSerializer _dtos;
    private readonly AppSettings _settings;
    private DiscordWebHook _discord;
    private readonly ILogger<ListDefinitions> _logger;
    private SqlService _sql;

    public ListDefinitions(ILogger<ListDefinitions> logger, SqlService sql, DiscordWebHook discord,
        AppSettings settings,
        DTOSerializer dtos)
    {
        _logger = logger;
        _sql = sql;
        _discord = discord;
        _settings = settings;
        _dtos = dtos;
    }


    [Route("ingest")]
    [Authorize(Roles = "User")]
    [HttpPost]
    public async Task<IActionResult> PostIngest()
    {
        var user = Request.Headers[_settings.MetricsKeyHeader].First();
        var use_gzip = Request.Headers[_settings.CompressedBodyHeader].Any();
        _logger.Log(LogLevel.Information, $"Ingesting Modlist Definition for {user}");

        var modlistBytes = await Request.Body.ReadAllAsync();


        _logger.LogInformation("Spawning ingestion task");
        var tsk = Task.Run(async () =>
        {
            try
            {
                if (use_gzip)
                {
                    await using var os = new MemoryStream();
                    await using var gZipStream =
                        new GZipStream(new MemoryStream(modlistBytes), CompressionMode.Decompress);
                    await gZipStream.CopyToAsync(os);
                    modlistBytes = os.ToArray();
                }

                var modlist = _dtos.DeserializeAsync<ModList>(new MemoryStream(modlistBytes));

                var file = KnownFolders.EntryPoint.Combine("mod_list_definitions")
                    .Combine($"{user}_{DateTime.UtcNow.ToFileTimeUtc()}.json");
                file.Parent.CreateDirectory();
                await using var stream = file.Open(FileMode.Create, FileAccess.Write);
                await _dtos.Serialize(modlist, stream);
                _logger.Log(LogLevel.Information, $"Done Ingesting Modlist Definition for {user}");
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "Error ingesting uploaded modlist");
            }
        });

        return Accepted(0);
    }
}