using System; using System.Collections.Generic; using System.Globalization; using System.Linq; using System.Net; using System.Threading.Tasks; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.Logging; using Nettle; using Wabbajack.Common; using Wabbajack.Server; using Wabbajack.Server.DataLayer; using Wabbajack.Server.DTOs; using WebSocketSharp; using LogLevel = Microsoft.Extensions.Logging.LogLevel; namespace Wabbajack.BuildServer.Controllers { [ApiController] [Route("/metrics")] public class MetricsController : ControllerBase { private SqlService _sql; private ILogger _logger; public MetricsController(ILogger logger, SqlService sql) { _sql = sql; _logger = logger; } [HttpGet] [Route("{subject}/{value}")] public async Task LogMetricAsync(string subject, string value) { var date = DateTime.UtcNow; await Log(date, subject, value, Request.Headers[Consts.MetricsKeyHeader].FirstOrDefault()); return new Result { Timestamp = date}; } [HttpGet] [Route("report/{subject}")] [ResponseCache(Duration = 60 * 60)] public async Task MetricsReport(string subject) { var metrics = (await _sql.MetricsReport(subject)).ToList(); var labels = metrics.GroupBy(m => m.Date) .OrderBy(m => m.Key) .Select(m => m.Key) .ToArray(); var labelStrings = labels.Select(l => l.ToString("MM-dd-yyy")).ToList(); var results = metrics .GroupBy(m => m.Subject) .Select(g => { var indexed = g.ToDictionary(m => m.Date, m => m.Count); return new MetricResult { SeriesName = g.Key, Labels = labelStrings, Values = labels.Select(l => indexed.TryGetValue(l, out var found) ? found : 0).ToList() }; }); return Ok(results.ToList()); } [HttpGet] [Route("badge/{name}/total_installs_badge.json")] public async Task TotalInstallsBadge(string name) { var results = await _sql.TotalInstalls(name); Response.ContentType = "application/json"; return Ok(results == 0 ? new Badge($"Modlist {name} not found!", "Error") {color = "red"} : new Badge("Installations: ", $"{results}") {color = "green"}); } [HttpGet] [Route("badge/{name}/unique_installs_badge.json")] public async Task UniqueInstallsBadge(string name) { var results = await _sql.UniqueInstalls(name); Response.ContentType = "application/json"; return Ok(results == 0 ? new Badge($"Modlist {name} not found!", "Error") {color = "red"} : new Badge("Installations: ", $"{results}"){color = "green"}) ; } private static readonly Func ReportTemplate = NettleEngine.GetCompiler().Compile(@"

Tar Report for {{$.key}}

Ban Status: {{$.status}}

"); [HttpGet] [Route("tarlog/{key}")] public async Task TarLog(string key) { var isTarKey = await _sql.IsTarKey(key); List<(DateTime, string, string)> report = new List<(DateTime, string, string)>(); if (isTarKey) report = await _sql.FullTarReport(key); var response = ReportTemplate(new { key = key, status = isTarKey ? "BANNED" : "NOT BANNED", log = report.Select(entry => new { Timestamp = entry.Item1, Path = entry.Item2, Key = entry.Item3 }).ToList() }); return new ContentResult { ContentType = "text/html", StatusCode = (int) HttpStatusCode.OK, Content = response }; } private async Task Log(DateTime timestamp, string action, string subject, string metricsKey = null) { //_logger.Log(LogLevel.Information, $"Log - {timestamp} {action} {subject} {metricsKey}"); await _sql.IngestMetric(new Metric { Timestamp = timestamp, Action = action, Subject = subject, MetricsKey = metricsKey }); } public class Result { public DateTime Timestamp { get; set; } } } }