using System; using System.IO; using System.IO.Compression; using System.Text; using System.Threading.Tasks; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.Options; using StardewModdingAPI.Web.Framework; using StardewModdingAPI.Web.Framework.Clients.Pastebin; using StardewModdingAPI.Web.Framework.ConfigModels; using StardewModdingAPI.Web.ViewModels; namespace StardewModdingAPI.Web.Controllers { /// Provides a web UI and API for parsing SMAPI log files. internal class LogParserController : Controller { /********* ** Properties *********/ /// The log parser config settings. private readonly ContextConfig Config; /// The underlying Pastebin client. private readonly IPastebinClient Pastebin; /// The first bytes in a valid zip file. /// See . private const uint GzipLeadBytes = 0x8b1f; /********* ** Public methods *********/ /*** ** Constructor ***/ /// Construct an instance. /// The context config settings. /// The Pastebin API client. public LogParserController(IOptions contextProvider, IPastebinClient pastebin) { this.Config = contextProvider.Value; this.Pastebin = pastebin; } /*** ** Web UI ***/ /// Render the log parser UI. /// The paste ID. [HttpGet] [Route("log")] [Route("log/{id}")] public ViewResult Index(string id = null) { return this.View("Index", new LogParserModel(this.Config.LogParserUrl, id)); } /*** ** JSON ***/ /// Fetch raw text from Pastebin. /// The Pastebin paste ID. [HttpGet, Produces("application/json")] [Route("log/fetch/{id}")] public async Task GetAsync(string id) { PasteInfo response = await this.Pastebin.GetAsync(id); response.Content = this.DecompressString(response.Content); return response; } /// Save raw log data. /// The log content to save. [HttpPost, Produces("application/json"), AllowLargePosts] [Route("log/save")] public async Task PostAsync([FromBody] string content) { content = this.CompressString(content); return await this.Pastebin.PostAsync(content); } /********* ** Private methods *********/ /// Compress a string. /// The text to compress. /// Derived from . private string CompressString(string text) { // get raw bytes byte[] buffer = Encoding.UTF8.GetBytes(text); // compressed byte[] compressedData; using (MemoryStream stream = new MemoryStream()) { using (GZipStream zipStream = new GZipStream(stream, CompressionLevel.Optimal, leaveOpen: true)) zipStream.Write(buffer, 0, buffer.Length); stream.Position = 0; compressedData = new byte[stream.Length]; stream.Read(compressedData, 0, compressedData.Length); } // prefix length var zipBuffer = new byte[compressedData.Length + 4]; Buffer.BlockCopy(compressedData, 0, zipBuffer, 4, compressedData.Length); Buffer.BlockCopy(BitConverter.GetBytes(buffer.Length), 0, zipBuffer, 0, 4); // return string representation return Convert.ToBase64String(zipBuffer); } /// Decompress a string. /// The compressed text. /// Derived from . private string DecompressString(string rawText) { // get raw bytes byte[] zipBuffer; try { zipBuffer = Convert.FromBase64String(rawText); } catch { return rawText; // not valid base64, wasn't compressed by the log parser } // skip if not gzip if (BitConverter.ToUInt16(zipBuffer, 4) != LogParserController.GzipLeadBytes) return rawText; // decompress using (MemoryStream memoryStream = new MemoryStream()) { // read length prefix int dataLength = BitConverter.ToInt32(zipBuffer, 0); memoryStream.Write(zipBuffer, 4, zipBuffer.Length - 4); // read data var buffer = new byte[dataLength]; memoryStream.Position = 0; using (GZipStream gZipStream = new GZipStream(memoryStream, CompressionMode.Decompress)) gZipStream.Read(buffer, 0, buffer.Length); // return original string return Encoding.UTF8.GetString(buffer); } } } }