From c9d7a957dc90742608361c595c506ebe9c137983 Mon Sep 17 00:00:00 2001
From: RaidMax <raidmax@live.com>
Date: Thu, 1 Jul 2021 13:12:19 -0500
Subject: [PATCH] add reset anticheat metric (!rsa) for issue #177

---
 .../Commands/ResetAnticheatMetricsCommand.cs  | 94 +++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 Plugins/Stats/Commands/ResetAnticheatMetricsCommand.cs

diff --git a/Plugins/Stats/Commands/ResetAnticheatMetricsCommand.cs b/Plugins/Stats/Commands/ResetAnticheatMetricsCommand.cs
new file mode 100644
index 000000000..23b455134
--- /dev/null
+++ b/Plugins/Stats/Commands/ResetAnticheatMetricsCommand.cs
@@ -0,0 +1,94 @@
+using System;
+using System.Linq;
+using System.Threading.Tasks;
+using Data.Abstractions;
+using Data.Models.Client;
+using Data.Models.Client.Stats;
+using IW4MAdmin.Plugins.Stats.Cheat;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging;
+using SharedLibraryCore;
+using SharedLibraryCore.Configuration;
+using SharedLibraryCore.Interfaces;
+using ILogger = Microsoft.Extensions.Logging.ILogger;
+
+namespace Stats.Commands
+{
+    public class ResetAnticheatMetricsCommand : Command
+    {
+        private readonly IDatabaseContextFactory _contextFactory;
+        private readonly ILogger _logger;
+
+        public ResetAnticheatMetricsCommand(ILogger<ResetAnticheatMetricsCommand> logger, CommandConfiguration config,
+            ITranslationLookup translationLookup,
+            IDatabaseContextFactory contextFactory) : base(config, translationLookup)
+        {
+            Name = "resetanticheat";
+            Description = translationLookup["PLUGINS_STATS_COMMANDS_RESETAC_DESC"];
+            Alias = "rsa";
+            Permission = EFClient.Permission.Owner;
+            RequiresTarget = true;
+
+            _contextFactory = contextFactory;
+            _logger = logger;
+        }
+
+        public override async Task ExecuteAsync(GameEvent gameEvent)
+        {
+            try
+            {
+                var clientDetection =
+                    gameEvent.Target.GetAdditionalProperty<Detection>(IW4MAdmin.Plugins.Stats.Helpers.StatManager
+                        .CLIENT_DETECTIONS_KEY);
+                var clientStats =
+                    gameEvent.Target.GetAdditionalProperty<EFClientStatistics>(IW4MAdmin.Plugins.Stats.Helpers
+                        .StatManager.CLIENT_STATS_KEY);
+
+                if (clientStats != null)
+                {
+                    clientStats.MaxStrain = 0;
+                    clientStats.AverageSnapValue = 0;
+                    clientStats.SnapHitCount = 0;
+                    clientStats.HitLocations.Clear();
+                }
+
+                clientDetection?.TrackedHits.Clear();
+
+                await using var context = _contextFactory.CreateContext();
+
+                var hitLocationCounts = await context.Set<EFHitLocationCount>()
+                    .Where(loc => loc.EFClientStatisticsClientId == gameEvent.Target.ClientId)
+                    .Select(loc => new EFHitLocationCount()
+                    {
+                        HitLocationCountId = loc.HitLocationCountId
+                    })
+                    .ToListAsync();
+
+                context.RemoveRange(hitLocationCounts);
+
+                await context.SaveChangesAsync();
+
+                var stats = await context.Set<EFClientStatistics>()
+                    .Where(stat => stat.ClientId == gameEvent.Target.ClientId)
+                    .ToListAsync();
+
+                foreach (var stat in stats)
+                {
+                    stat.MaxStrain = 0;
+                    stat.AverageSnapValue = 0;
+                    stat.SnapHitCount = 0;
+                }
+
+                context.UpdateRange(stats);
+                await context.SaveChangesAsync();
+
+                gameEvent.Origin.Tell(_translationLookup["PLUGINS_STATS_COMMANDS_RESETAC_SUCCESS"]);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "Could not reset anticheat metrics for {Target}", gameEvent.Target);
+                throw;
+            }
+        }
+    }
+}
\ No newline at end of file