wabbajack/Wabbajack.Server/Controllers/ListDefinitions.cs
2021-09-27 06:42:46 -06:00

86 lines
2.9 KiB
C#

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 ILogger<ListDefinitions> _logger;
private SqlService _sql;
private DiscordWebHook _discord;
private readonly AppSettings _settings;
private readonly DTOSerializer _dtos;
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);
}
}
}